Enhance OAuth2SuccessHandler with logging and error handling

Added logging for OAuth2 success handling and improved error handling.

Author: LJHCS

src/main/java/com/blockcloud/exception/handler/OAuth2SuccessHandler.java

@@ -11,6 +11,7 @@
 import java.net.URLEncoder;
 import java.nio.charset.StandardCharsets;
 import lombok.AllArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.http.HttpStatus;
 import org.springframework.security.core.Authentication;
 import org.springframework.security.web.authentication.AuthenticationSuccessHandler;
@@ -19,6 +20,7 @@
 import java.util.Map;
 
 @AllArgsConstructor
+@Slf4j
 public class OAuth2SuccessHandler implements AuthenticationSuccessHandler {
 
     private final JWTUtil jwtUtil;
@@ -29,39 +31,47 @@ public void onAuthenticationSuccess(HttpServletRequest request, HttpServletRespo
 
         CustomOAuth2User customOAuth2User = (CustomOAuth2User) authentication.getPrincipal();
         User user = customOAuth2User.getUser();
-        ObjectMapper objectMapper = new ObjectMapper();
+        ObjectMapper objectMapper = new ObjectMapper(); 
 
-		try {
-				String accessToken = jwtUtil.createJwt("access", user.getEmail(), String.valueOf(user.getRole()), 30 * 60 * 1000L);
-				String refreshToken = jwtUtil.createJwt("refresh", user.getEmail(), String.valueOf(user.getRole()), 24 * 60 * 60 * 1000L);
+        try {
+            log.info("Starting OAuth2 Success Handling for user: {}", user.getEmail());
+
+            String accessToken = jwtUtil.createJwt("access", user.getEmail(), String.valueOf(user.getRole()), 30 * 60 * 1000L);
+            String refreshToken = jwtUtil.createJwt("refresh", user.getEmail(), String.valueOf(user.getRole()), 24 * 60 * 60 * 1000L);
 
             Cookie refreshCookie = cookieService.createCookie("refresh", refreshToken, 24 * 60 * 60 * 1000L);
             response.addCookie(refreshCookie);
+            log.debug("Access Token created. Refresh Token cookie added.");
 
             response.setStatus(HttpStatus.OK.value());
             response.setContentType("application/json;charset=UTF-8");
 
             String userJson = objectMapper.writeValueAsString(
-                Map.of(
-                    "message", "Login successful",
-                    "email", user.getEmail(),
-                    "imgUrl", user.getImgUrl(),
-                    "userName", user.getUsername(),
-                    "role", user.getRole()
-                )
+                    Map.of(
+                            "message", "Login successful",
+                            "email", user.getEmail(),
+                            "imgUrl", user.getImgUrl(),
+                            "userName", user.getUsername(),
+                            "role", user.getRole()
+                    )
             );
-			String redirectUrl =  "https://blockcloud.dev/oauth2/callback"
-				+ "?access=" + URLEncoder.encode(accessToken, StandardCharsets.UTF_8)
-				+ "&user=" + URLEncoder.encode(userJson, StandardCharsets.UTF_8);
 
-			response.sendRedirect(redirectUrl);
-        } catch (IOException e) {
-            e.printStackTrace();
+            String redirectUrl =  "https://blockcloud.dev/oauth2/callback"
+                    + "?access=" + URLEncoder.encode(accessToken, StandardCharsets.UTF_8)
+                    + "&user=" + URLEncoder.encode(userJson, StandardCharsets.UTF_8);
+
+            log.info("Redirecting successfully to: {}", redirectUrl);
+            response.sendRedirect(redirectUrl);
+
+        } catch (Exception e) {
+
+            log.error("Critical error occurred in OAuth2SuccessHandler while generating tokens or redirecting.", e);
+
             response.setStatus(HttpStatus.INTERNAL_SERVER_ERROR.value());
             response.setContentType("application/json;charset=UTF-8");
 
             try (PrintWriter writer = response.getWriter()) {
-                writer.write("{\"message\": \"An error occurred during authentication\"}");
+                writer.write("{\"message\": \"An error occurred during authentication success handling\"}");
                 writer.flush();
             }
         }