Refactored saml code

src/main/java/net/gepafin/tendermanagement/config/SamlSuccessHandler.java

@@ -0,0 +1,64 @@
+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);
+         }
+     }
+
+}