fix. URI 사용하여 path 추출

CodeSquad-Airbnb-Team07 · Jun 5, 2024 · 208fade · 208fade
1 parent afb03cf
commit 208fade
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/BE/src/main/java/team07/airbnb/config/security/CustomSuccessHandler.java b/BE/src/main/java/team07/airbnb/config/security/CustomSuccessHandler.java
@@ -12,25 +12,36 @@
 
 
 import java.io.IOException;
+import java.net.URI;
+import java.net.URISyntaxException;
 
 @Component
 public class CustomSuccessHandler extends SavedRequestAwareAuthenticationSuccessHandler {
 
     private final RequestCache requestCache = new HttpSessionRequestCache();
 
-    @Override
-    public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) throws ServletException, IOException, ServletException {
+    public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) throws ServletException, IOException {
         SavedRequest savedRequest = requestCache.getRequest(request, response);
 
-
         if (savedRequest == null) {
             super.onAuthenticationSuccess(request, response, authentication);
             return;
         }
 
-        String targetUrl = savedRequest.getRedirectUrl().replace("http:/squadbnb.site", "");
-        targetUrl = "/api" + targetUrl;
+        String targetUrl = savedRequest.getRedirectUrl();
+
+        try {
+            URI targetUri = new URI(targetUrl);
+            String path = targetUri.getPath();
+            String query = targetUri.getQuery();
+            String newTargetUrl = request.getContextPath() + "/api" + path;
+            if (query != null) {
+                newTargetUrl += "?" + query;
+            }
 
-        getRedirectStrategy().sendRedirect(request, response, targetUrl);
+            getRedirectStrategy().sendRedirect(request, response, newTargetUrl);
+        } catch (URISyntaxException e) {
+            super.onAuthenticationSuccess(request, response, authentication);
+        }
     }
 }