FluffyLabs
diff --git a/‎demo/src/App.tsx‎
Lines changed: 19 additions & 3 deletions b/‎demo/src/App.tsx‎
Lines changed: 19 additions & 3 deletions
@@ -6,6 +6,7 @@ import { initializeTheme } from "../../lib/components/DarkMode";
 import { SupabaseProvider } from "../../lib/supabase/SupabaseProvider";
 import { UserMenu } from "../../lib/supabase/UserMenu";
 import { AuthFlow } from "../../lib/supabase/AuthFlow";
+import { AuthCallback } from "../../lib/supabase/AuthCallback";
 import { Settings } from "../../lib/supabase/Settings";
 import { SubscriptionStatus } from "../../lib/supabase/SubscriptionStatus";
 import { PricingCard } from "../../lib/supabase/PricingCard";
@@ -22,7 +23,7 @@ const SUPABASE_ANON_KEY = import.meta.env.VITE_SUPABASE_ANON_KEY || "";
 
 initializeTheme();
 
-type Page = "home" | "login" | "settings" | "pricing" | "gated";
+type Page = "home" | "login" | "settings" | "pricing" | "gated" | "auth-callback";
 
 export function App() {
   if (!SUPABASE_URL || !SUPABASE_ANON_KEY) {
@@ -46,8 +47,14 @@ export function App() {
   );
 }
 
+const AUTH_CALLBACK_PATH = new URL("auth/callback", document.baseURI).pathname;
+const HOME_PATH = new URL(".", document.baseURI).pathname;
+
 function DemoApp() {
-  const [page, setPage] = useState<Page>("home");
+  const [page, setPage] = useState<Page>(() => {
+    if (window.location.pathname === AUTH_CALLBACK_PATH) return "auth-callback";
+    return "home";
+  });
 
   return (
     <div className="flex min-h-screen flex-col">
@@ -77,6 +84,15 @@ function PageContent({ page, setPage }: { page: Page; setPage: (p: Page) => void
       return <PricingPage />;
     case "gated":
       return <GatedFeaturePage />;
+    case "auth-callback":
+      return (
+        <AuthCallback
+          onSuccess={() => {
+            window.history.replaceState({}, "", HOME_PATH);
+            setPage("home");
+          }}
+        />
+      );
     default:
       return <HomePage setPage={setPage} />;
   }
@@ -126,7 +142,7 @@ function HomePage({ setPage }: { setPage: (p: Page) => void }) {
 function LoginPage({ onSuccess }: { onSuccess: () => void }) {
   return (
     <div className="mx-auto max-w-sm pt-8">
-      <AuthFlow onSuccess={onSuccess} />
+      <AuthFlow onSuccess={onSuccess} redirectTo={new URL("auth/callback", document.baseURI).toString()} />
     </div>
   );
 }