From 58117f932fd10704f94ce1868d8aae62963ce240 Mon Sep 17 00:00:00 2001
From: Jeremy <git@jrmyphlmn.com>
Date: Fri, 8 Nov 2024 17:59:06 +0300
Subject: [PATCH] Persist auth spinner (#503)

---
 app/(auth)/login/page.tsx           |  4 +++-
 app/(auth)/register/page.tsx        |  5 ++++-
 components/custom/submit-button.tsx | 18 +++++++++++++-----
 3 files changed, 20 insertions(+), 7 deletions(-)

diff --git a/app/(auth)/login/page.tsx b/app/(auth)/login/page.tsx
index 97f9cdc..19a39b3 100644
--- a/app/(auth)/login/page.tsx
+++ b/app/(auth)/login/page.tsx
@@ -14,6 +14,7 @@ export default function Page() {
   const router = useRouter();
 
   const [email, setEmail] = useState('');
+  const [isSuccessful, setIsSuccessful] = useState(false);
 
   const [state, formAction] = useActionState<LoginActionState, FormData>(
     login,
@@ -28,6 +29,7 @@ export default function Page() {
     } else if (state.status === 'invalid_data') {
       toast.error('Failed validating your submission!');
     } else if (state.status === 'success') {
+      setIsSuccessful(true);
       router.refresh();
     }
   }, [state.status, router]);
@@ -47,7 +49,7 @@ export default function Page() {
           </p>
         </div>
         <AuthForm action={handleSubmit} defaultEmail={email}>
-          <SubmitButton>Sign in</SubmitButton>
+          <SubmitButton isSuccessful={isSuccessful}>Sign in</SubmitButton>
           <p className="text-center text-sm text-gray-600 mt-4 dark:text-zinc-400">
             {"Don't have an account? "}
             <Link
diff --git a/app/(auth)/register/page.tsx b/app/(auth)/register/page.tsx
index cd51f86..7c6f54c 100644
--- a/app/(auth)/register/page.tsx
+++ b/app/(auth)/register/page.tsx
@@ -14,6 +14,8 @@ export default function Page() {
   const router = useRouter();
 
   const [email, setEmail] = useState('');
+  const [isSuccessful, setIsSuccessful] = useState(false);
+
   const [state, formAction] = useActionState<RegisterActionState, FormData>(
     register,
     {
@@ -30,6 +32,7 @@ export default function Page() {
       toast.error('Failed validating your submission!');
     } else if (state.status === 'success') {
       toast.success('Account created successfully');
+      setIsSuccessful(true);
       router.refresh();
     }
   }, [state, router]);
@@ -49,7 +52,7 @@ export default function Page() {
           </p>
         </div>
         <AuthForm action={handleSubmit} defaultEmail={email}>
-          <SubmitButton>Sign Up</SubmitButton>
+          <SubmitButton isSuccessful={isSuccessful}>Sign Up</SubmitButton>
           <p className="text-center text-sm text-gray-600 mt-4 dark:text-zinc-400">
             {'Already have an account? '}
             <Link
diff --git a/components/custom/submit-button.tsx b/components/custom/submit-button.tsx
index 903330d..34e4f17 100644
--- a/components/custom/submit-button.tsx
+++ b/components/custom/submit-button.tsx
@@ -6,24 +6,32 @@ import { LoaderIcon } from '@/components/custom/icons';
 
 import { Button } from '../ui/button';
 
-export function SubmitButton({ children }: { children: React.ReactNode }) {
+export function SubmitButton({
+  children,
+  isSuccessful,
+}: {
+  children: React.ReactNode;
+  isSuccessful: boolean;
+}) {
   const { pending } = useFormStatus();
 
   return (
     <Button
       type={pending ? 'button' : 'submit'}
-      aria-disabled={pending}
-      disabled={pending}
+      aria-disabled={pending || isSuccessful}
+      disabled={pending || isSuccessful}
       className="relative"
     >
       {children}
-      {pending && (
+
+      {(pending || isSuccessful) && (
         <span className="animate-spin absolute right-4">
           <LoaderIcon />
         </span>
       )}
+
       <span aria-live="polite" className="sr-only" role="status">
-        {pending ? 'Loading' : 'Submit form'}
+        {pending || isSuccessful ? 'Loading' : 'Submit form'}
       </span>
     </Button>
   );