From b98bc6b6e05d64cae1ee9dd36570262ec1214c21 Mon Sep 17 00:00:00 2001
From: Junwei Zhao <wahyd4@gmail.com>
Date: Tue, 5 Aug 2025 17:31:38 +1000
Subject: [PATCH] Update ui

---
 web/index.html | 55 +++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 50 insertions(+), 5 deletions(-)

diff --git a/web/index.html b/web/index.html
index e8e40f8..d7d26a5 100644
--- a/web/index.html
+++ b/web/index.html
@@ -256,6 +256,28 @@
             return errorMessage;
         }
 
+        // Redirect handling for nginx auth_request
+        function getRedirectUrl() {
+            const urlParams = new URLSearchParams(window.location.search);
+            return urlParams.get('redirect') || urlParams.get('rd');
+        }
+
+        function redirectAfterAuth() {
+            const redirectUrl = getRedirectUrl();
+            if (redirectUrl) {
+                try {
+                    // Decode the URL if it's encoded
+                    const decodedUrl = decodeURIComponent(redirectUrl);
+                    console.log('Redirecting to:', decodedUrl);
+                    window.location.href = decodedUrl;
+                    return true;
+                } catch (error) {
+                    console.error('Error decoding redirect URL:', error);
+                }
+            }
+            return false;
+        }
+
         function arrayBufferToBase64url(buffer) {
             const bytes = new Uint8Array(buffer);
             let binary = '';
@@ -673,8 +695,11 @@
 
             showAlert('Account created successfully! Welcome to the platform.');
 
-            // Check auth status to show welcome panel
-            setTimeout(checkAuthStatus, 100);
+            // Check for redirect first, then fall back to checking auth status
+            if (!redirectAfterAuth()) {
+                // Check auth status to show welcome panel
+                setTimeout(checkAuthStatus, 100);
+            }
         }
 
         // Sign In Handler
@@ -727,8 +752,11 @@
             const result = await finishResponse.json();
             showAlert(`Welcome back, ${result.user.display_name}!`);
 
-            // Check auth status to show welcome panel
-            setTimeout(checkAuthStatus, 100);
+            // Check for redirect first, then fall back to checking auth status
+            if (!redirectAfterAuth()) {
+                // Check auth status to show welcome panel
+                setTimeout(checkAuthStatus, 100);
+            }
         }
 
         // Users management
@@ -840,8 +868,25 @@
             // Load theme first
             loadTheme();
 
+            // Check if there's a redirect URL and update header
+            const redirectUrl = getRedirectUrl();
+            if (redirectUrl) {
+                try {
+                    const decodedUrl = decodeURIComponent(redirectUrl);
+                    const targetDomain = new URL(decodedUrl).hostname;
+                    document.getElementById('headerSubtitle').textContent = `Authenticating for ${targetDomain}`;
+                } catch (error) {
+                    document.getElementById('headerSubtitle').textContent = 'Authenticating for protected resource';
+                }
+            }
+
             // Check authentication status to determine initial UI state
-            await checkAuthStatus();
+            const user = await checkAuthStatus();
+
+            // If user is already authenticated and there's a redirect parameter, redirect immediately
+            if (user && redirectUrl) {
+                setTimeout(() => redirectAfterAuth(), 1000); // Give user a moment to see the message
+            }
         });
     </script>
 </body>