80 lines
4.2 KiB
Java
80 lines
4.2 KiB
Java
package eu.eudat.controllers;
|
|
|
|
import eu.eudat.logic.proxy.config.configloaders.ConfigLoader;
|
|
import eu.eudat.logic.security.customproviders.ConfigurableProvider.entities.saml2.Saml2ConfigurableProvider;
|
|
import eu.eudat.logic.security.validators.configurableProvider.Saml2SSOUtils;
|
|
import eu.eudat.logic.services.ApiContext;
|
|
import eu.eudat.models.data.helpers.responses.ResponseItem;
|
|
import eu.eudat.models.data.saml2.AuthnRequestModel;
|
|
import eu.eudat.types.ApiMessageCode;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.http.HttpHeaders;
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
@RestController
|
|
@CrossOrigin
|
|
@RequestMapping(value = {"/api/saml2/"})
|
|
public class Saml2MetadataController extends BaseController {
|
|
|
|
private final ConfigLoader configLoader;
|
|
|
|
@Autowired
|
|
public Saml2MetadataController(ApiContext apiContext, ConfigLoader configLoader) {
|
|
super(apiContext);
|
|
this.configLoader = configLoader;
|
|
}
|
|
|
|
@RequestMapping(method = RequestMethod.GET, value = {"metadata/{configurableProviderId}"})
|
|
public @ResponseBody
|
|
ResponseEntity getMetadata(@PathVariable String configurableProviderId) {
|
|
Saml2ConfigurableProvider saml2ConfigurableProvider = (Saml2ConfigurableProvider) this.configLoader.getConfigurableProviders().getProviders().stream()
|
|
.filter(prov -> prov.getConfigurableLoginId().equals(configurableProviderId))
|
|
.findFirst().orElse(null);
|
|
if (saml2ConfigurableProvider != null) {
|
|
try {
|
|
String metadataXml = Saml2SSOUtils.getMetadata(saml2ConfigurableProvider);
|
|
HttpHeaders responseHeaders = new HttpHeaders();
|
|
responseHeaders.setContentLength(metadataXml.length());
|
|
responseHeaders.setContentType(MediaType.APPLICATION_OCTET_STREAM);
|
|
responseHeaders.set("Content-Disposition", "attachment;filename=" + configurableProviderId + ".xml");
|
|
responseHeaders.set("Access-Control-Expose-Headers", "Content-Disposition");
|
|
responseHeaders.get("Access-Control-Expose-Headers").add("Content-Type");
|
|
return new ResponseEntity<>(metadataXml.getBytes(StandardCharsets.UTF_8),
|
|
responseHeaders,
|
|
HttpStatus.OK);
|
|
} catch (Exception e) {
|
|
return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(new ResponseItem<String>().status(ApiMessageCode.ERROR_MESSAGE).message("Failed to fetch metadata."));
|
|
}
|
|
}
|
|
else {
|
|
return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(new ResponseItem<String>().status(ApiMessageCode.ERROR_MESSAGE).message("Failed to fetch metadata."));
|
|
}
|
|
}
|
|
@RequestMapping(method = RequestMethod.GET, value = {"authnRequest/{configurableProviderId}"})
|
|
public @ResponseBody
|
|
ResponseEntity getAuthnRequest(@PathVariable String configurableProviderId) {
|
|
Saml2ConfigurableProvider saml2ConfigurableProvider = (Saml2ConfigurableProvider) this.configLoader.getConfigurableProviders().getProviders().stream()
|
|
.filter(prov -> prov.getConfigurableLoginId().equals(configurableProviderId))
|
|
.findFirst().orElse(null);
|
|
if (saml2ConfigurableProvider != null) {
|
|
try {
|
|
AuthnRequestModel authnRequest = Saml2SSOUtils.getAuthnRequest(saml2ConfigurableProvider);
|
|
return ResponseEntity.status(HttpStatus.OK).body(new ResponseItem<AuthnRequestModel>().status(ApiMessageCode.SUCCESS_MESSAGE).message("Created").payload(authnRequest));
|
|
}
|
|
catch (Exception e) {
|
|
return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(new ResponseItem<String>().status(ApiMessageCode.ERROR_MESSAGE).message("Failed to create authentication request."));
|
|
}
|
|
|
|
}
|
|
else {
|
|
return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(new ResponseItem<String>().status(ApiMessageCode.ERROR_MESSAGE).message("Unknown provider."));
|
|
}
|
|
}
|
|
|
|
}
|