From c550534d6ce10d4616de3cb4b7249e25723c52eb Mon Sep 17 00:00:00 2001
From: Joey Yakimowich-Payne <jyapayne@pm.me>
Date: Fri, 16 Jan 2026 10:39:58 -0700
Subject: [PATCH] Fix sign in redirect

---
 components/AuthButton.tsx     | 5 ++++-
 components/Landing.tsx        | 5 ++++-
 components/SharedQuizView.tsx | 5 ++++-
 index.tsx                     | 9 ++++++++-
 4 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/components/AuthButton.tsx b/components/AuthButton.tsx
index 330dd47..e12927c 100644
--- a/components/AuthButton.tsx
+++ b/components/AuthButton.tsx
@@ -90,7 +90,10 @@ export const AuthButton: React.FC<AuthButtonProps> = ({ onAccountSettingsClick }
 
   return (
     <button
-      onClick={() => auth.signinRedirect()}
+      onClick={() => {
+        sessionStorage.setItem('kaboot_auth_return_url', window.location.pathname + window.location.search);
+        auth.signinRedirect();
+      }}
       className="flex items-center gap-2 bg-white/10 px-4 py-2 rounded-xl hover:bg-white/20 transition font-bold"
     >
       <LogIn size={20} />
diff --git a/components/Landing.tsx b/components/Landing.tsx
index c2801b8..f932bb3 100644
--- a/components/Landing.tsx
+++ b/components/Landing.tsx
@@ -348,7 +348,10 @@ export const Landing: React.FC<LandingProps> = ({ onGenerate, onCreateManual, on
                 <Lock size={32} className="mx-auto mb-3 text-gray-400" />
                 <p className="text-gray-500 font-bold mb-4">Sign in to host quizzes</p>
                 <button
-                  onClick={() => auth.signinRedirect()}
+                  onClick={() => {
+                    sessionStorage.setItem('kaboot_auth_return_url', window.location.pathname + window.location.search);
+                    auth.signinRedirect();
+                  }}
                   className="bg-theme-primary text-white py-3 px-6 rounded-2xl font-black hover:opacity-90 transition-all"
                 >
                   Sign In
diff --git a/components/SharedQuizView.tsx b/components/SharedQuizView.tsx
index f9d5107..41d944a 100644
--- a/components/SharedQuizView.tsx
+++ b/components/SharedQuizView.tsx
@@ -190,7 +190,10 @@ export const SharedQuizView: React.FC<SharedQuizViewProps> = ({ onHostQuiz, shar
             </button>
           ) : (
             <button
-              onClick={() => auth.signinRedirect()}
+              onClick={() => {
+                sessionStorage.setItem('kaboot_auth_return_url', window.location.pathname);
+                auth.signinRedirect();
+              }}
               className="w-full bg-gray-100 text-gray-600 py-3 rounded-2xl text-lg font-black hover:bg-gray-200 shadow-[0_4px_0_#d1d5db] active:shadow-none active:translate-y-[4px] transition-all flex items-center justify-center gap-2"
             >
               <LogIn size={20} /> Sign in to Save
diff --git a/index.tsx b/index.tsx
index c8fd3e4..237a65f 100644
--- a/index.tsx
+++ b/index.tsx
@@ -34,7 +34,14 @@ if (!rootElement) {
 const shouldSkipSigninCallback = !window.location.pathname.startsWith('/callback');
 
 const onSigninCallback = () => {
-  window.history.replaceState({}, document.title, '/');
+  const returnUrl = sessionStorage.getItem('kaboot_auth_return_url');
+  sessionStorage.removeItem('kaboot_auth_return_url');
+  
+  if (returnUrl && returnUrl !== '/') {
+    window.location.replace(returnUrl);
+  } else {
+    window.history.replaceState({}, document.title, '/');
+  }
 };
 
 const root = ReactDOM.createRoot(rootElement);