Update internal/auth/handler/auth_handler.go

internal/auth/handler/auth_handler.go

@@ -34,49 +34,50 @@ func (rw *LoggingResponseWriter) WriteHeader(statusCode int) {
 
 func Login(w http.ResponseWriter, r *http.Request) {
     log.Println("Login handler triggered")
-	state, _ := utils.SecureRandom(32)
-	codeVerifier, _ := utils.SecureRandom(64)
-	codeChallenge := crypto.SHA256Base64URL(codeVerifier)
-
-	http.SetCookie(w, &http.Cookie{
-		Name:     "oauth_state",
-		Value:    state,
-		HttpOnly: true,
-		Secure:   true,
-		SameSite: http.SameSiteNoneMode,
-		Path:     "/",
-		MaxAge:   300,
-	})
     
-	http.SetCookie(w, &http.Cookie{
-		Name:     "pkce_verifier",
-		Value:    codeVerifier,
-		HttpOnly: true,
-		Secure:   true,
-		SameSite: http.SameSiteNoneMode,
-		Path:     "/",
-		MaxAge:   300,
-	})
-	authURL := fmt.Sprintf("%s/realms/%s/protocol/openid-connect/auth?client_id=%s&response_type=code&scope=openid profile email&redirect_uri=%s&state=%s&code_challenge=%s&code_challenge_method=S256",
-		os.Getenv("KEYCLOAK_URL"),
-		os.Getenv("KEYCLOAK_REALM"),
-		os.Getenv("KEYCLOAK_CLIENT_ID"),
-		url.QueryEscape(os.Getenv("KEYCLOAK_REDIRECT_URL")),
-		state,
-		codeChallenge,
-	)
-
-	http.Redirect(w, r, authURL, http.StatusTemporaryRedirect)
-
-        log.Printf("Redirecting to URL: %s", authURL)
+    // Generate secure random values
+    state, _ := utils.SecureRandom(32)
+    codeVerifier, _ := utils.SecureRandom(64)
+    codeChallenge := crypto.SHA256Base64URL(codeVerifier)
+
+    // Set cookies
+    http.SetCookie(w, &http.Cookie{
+        Name:     "oauth_state",
+        Value:    state,
+        HttpOnly: true,
+        Secure:   true,
+        SameSite: http.SameSiteNoneMode,
+        Path:     "/",
+        MaxAge:   300,
+    })
+    log.Println("Cookies set 1")
+    
+    http.SetCookie(w, &http.Cookie{
+        Name:     "pkce_verifier",
+        Value:    codeVerifier,
+        HttpOnly: true,
+        Secure:   true,
+        SameSite: http.SameSiteNoneMode,
+        Path:     "/",
+        MaxAge:   300,
+    })
+    log.Println("Cookies set 2")
+
+    // Construct the authorization URL for Keycloak
+    authURL := fmt.Sprintf("%s/realms/%s/protocol/openid-connect/auth?client_id=%s&response_type=code&scope=openid profile email&redirect_uri=%s&state=%s&code_challenge=%s&code_challenge_method=S256",
+        os.Getenv("KEYCLOAK_URL"),
+        os.Getenv("KEYCLOAK_REALM"),
+        os.Getenv("KEYCLOAK_CLIENT_ID"),
+        url.QueryEscape(os.Getenv("KEYCLOAK_REDIRECT_URL")),
+        state,
+        codeChallenge,
+    )
 
-    // Perform the redirect
-    http.Redirect(loggingRW, r, authURL, http.StatusTemporaryRedirect)
+    // Log the URL being redirected to
+    log.Printf("Redirecting to URL: %s", authURL)
 
-    // Check if redirect status code was 3xx (redirection)
-    if loggingRW.StatusCode != http.StatusTemporaryRedirect {
-        log.Printf("Redirection failed with status code: %d", loggingRW.StatusCode)
-    }
+    // Perform the actual redirect
+    http.Redirect(w, r, authURL, http.StatusTemporaryRedirect)
 }
 
 func Callback(w http.ResponseWriter, r *http.Request) {