65 lines
2.8 KiB
Java
65 lines
2.8 KiB
Java
package net.gepafin.tendermanagement.config;
|
|
|
|
import java.io.IOException;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.security.core.Authentication;
|
|
import org.springframework.security.saml2.provider.service.authentication.Saml2AuthenticatedPrincipal;
|
|
import org.springframework.security.saml2.provider.service.authentication.Saml2Authentication;
|
|
import org.springframework.security.web.authentication.AuthenticationSuccessHandler;
|
|
import org.springframework.stereotype.Component;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
|
import jakarta.servlet.http.HttpServletRequest;
|
|
import jakarta.servlet.http.HttpServletResponse;
|
|
import net.gepafin.tendermanagement.entities.SamlResponseLogEntity;
|
|
import net.gepafin.tendermanagement.repositories.SamlResponseLogRepository;
|
|
|
|
@Component
|
|
public class SamlSuccessHandler implements AuthenticationSuccessHandler{
|
|
|
|
|
|
private final Logger logger = LoggerFactory.getLogger(SamlSuccessHandler.class);
|
|
|
|
@Autowired
|
|
private SamlResponseLogRepository samlResponseLogRepository;
|
|
|
|
@Override
|
|
public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response,
|
|
Authentication authentication) throws IOException {
|
|
try {
|
|
// Cast the authentication object to Saml2Authentication
|
|
Saml2Authentication samlAuth = (Saml2Authentication) authentication;
|
|
Saml2AuthenticatedPrincipal principal = (Saml2AuthenticatedPrincipal) samlAuth.getPrincipal();
|
|
|
|
// Extract the user attributes from the principal
|
|
Map<String, List<Object>> userAttributes = principal.getAttributes();
|
|
|
|
// Log the user attributes for debugging purposes
|
|
logger.info("SAML User Attributes: " + userAttributes);
|
|
|
|
// Save the authentication details in the database (Optional)
|
|
SamlResponseLogEntity samlResponseLogEntity = new SamlResponseLogEntity();
|
|
samlResponseLogEntity.setAuthenticationObject(authentication.toString());
|
|
|
|
// Convert user attributes to JSON and save in DB
|
|
ObjectMapper objectMapper = new ObjectMapper();
|
|
String userAttributesJson = objectMapper.writeValueAsString(userAttributes);
|
|
samlResponseLogEntity.setAuthenticationObject(userAttributesJson);
|
|
samlResponseLogRepository.save(samlResponseLogEntity);
|
|
|
|
// Successful login logic
|
|
logger.info("SAML login successful for user: " + principal.getName());
|
|
response.sendRedirect("http://gepafin-staging-fe.s3-website.eu-central-1.amazonaws.com/login");
|
|
} catch (Exception e) {
|
|
logger.error("Error processing SAML success handler", e);
|
|
}
|
|
}
|
|
|
|
}
|