fix hugo callback

my-blog/static/js/ask-ai.js

@@ -284,8 +284,52 @@ function setupAskAIEventListeners() {
     });
 }
 
+// OAuth Callback handling
+function handleOAuthCallback() {
+    // Check if we're on the callback page
+    if (window.location.pathname === '/oauth/callback') {
+        const urlParams = new URLSearchParams(window.location.search);
+        const code = urlParams.get('code');
+        const error = urlParams.get('error');
+        
+        if (error) {
+            console.error('OAuth error:', error);
+            // Redirect to home page with error
+            setTimeout(() => {
+                window.location.href = '/?oauth_error=' + encodeURIComponent(error);
+            }, 1000);
+            return;
+        }
+        
+        if (code) {
+            console.log('OAuth callback successful, code received');
+            
+            // Get the original page from localStorage or use home page
+            const originalPage = localStorage.getItem('oauth_original_page') || '/';
+            localStorage.removeItem('oauth_original_page');
+            
+            // Wait a bit for OAuth app to process the callback
+            setTimeout(() => {
+                console.log('Redirecting back to:', originalPage);
+                window.location.href = originalPage;
+            }, 2000);
+            
+            return;
+        }
+    }
+    
+    // Store current page before OAuth if we're not on callback page
+    if (window.location.pathname !== '/oauth/callback') {
+        localStorage.setItem('oauth_original_page', window.location.href);
+    }
+}
+
 // Initialize Ask AI when DOM is loaded
 document.addEventListener('DOMContentLoaded', function() {
+    // Handle OAuth callback first
+    handleOAuthCallback();
+    
+    // Then initialize Ask AI
     setupAskAIEventListeners();
     console.log('Ask AI initialized successfully');
 });