From f07360d05c6ea84122c2f44883406d6aad779feb Mon Sep 17 00:00:00 2001
From: "Yeyang (Justin) Sun" <i@jsun.lol>
Date: Sun, 18 Feb 2024 23:55:15 +1030
Subject: [PATCH] Revert "Revert "fix: Header buttons""

---
 src/app/(account)/signin/page.tsx |  4 ++++
 src/components/Header.tsx         | 26 ++++++++++++++++----------
 2 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/src/app/(account)/signin/page.tsx b/src/app/(account)/signin/page.tsx
index 332bb7d7..5690b6ed 100644
--- a/src/app/(account)/signin/page.tsx
+++ b/src/app/(account)/signin/page.tsx
@@ -6,6 +6,7 @@ import FancyRectangle from '@/components/FancyRectangle';
 import { useSignIn } from '@clerk/clerk-react';
 import { zodResolver } from '@hookform/resolvers/zod';
 import Link from 'next/link';
+import { useRouter } from 'next/navigation';
 import { useForm } from 'react-hook-form';
 import { FcGoogle } from 'react-icons/fc';
 import { z } from 'zod';
@@ -25,6 +26,8 @@ export default function SignInPage() {
         resolver: zodResolver(signInSchema),
     });
 
+    const router = useRouter();
+
     const handleSignIn = form.handleSubmit(async ({ email, password }) => {
         if (!isLoaded) return;
         try {
@@ -35,6 +38,7 @@ export default function SignInPage() {
 
             if (result.status === 'complete') {
                 await setActive({ session: result.createdSessionId });
+                router.push('/');
             } else {
                 console.log(result);
             }
diff --git a/src/components/Header.tsx b/src/components/Header.tsx
index 9972d5a0..ad7ffcd5 100644
--- a/src/components/Header.tsx
+++ b/src/components/Header.tsx
@@ -28,7 +28,12 @@ export default function Header() {
     });
 
     const checkUserPaid = useSWR<{ paid: boolean }>(['payment'], fetcher.get.query, {
-        isPaused: () => clerkUser.isLoaded && !clerkUser.isSignedIn,
+        isPaused: () => {
+            if (clerkUser.isLoaded && !clerkUser.isSignedIn) {
+                return true;
+            }
+            return !checkUserExists.data?.exists;
+        },
     });
 
     const [isScrolled, setIsScrolled] = useState(false);
@@ -156,15 +161,16 @@ export default function Header() {
                                                 Continue Signing Up
                                             </Button>
                                         )}
-                                        {!checkUserPaid.data?.paid && (
-                                            <Button
-                                                colour="orange"
-                                                href="/settings"
-                                                onClick={closeMenu}
-                                            >
-                                                Continue to payment
-                                            </Button>
-                                        )}
+                                        {checkUserExists.data?.exists &&
+                                            !checkUserPaid.data?.paid && (
+                                                <Button
+                                                    colour="orange"
+                                                    href="/settings"
+                                                    onClick={closeMenu}
+                                                >
+                                                    Continue to payment
+                                                </Button>
+                                            )}
                                         <UserButton
                                             userExists={Boolean(checkUserExists.data?.exists)}
                                             userPaid={Boolean(checkUserPaid.data?.paid)}