reset step after signing

latticexyz · Apr 15, 2024 · 3115c64 · 3115c64
1 parent 97ed299
commit 3115c64
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx b/packages/account-kit/src/steps/app-account/AccountDelegationContent.tsx
@@ -3,9 +3,11 @@ import { AccountModalContent } from "../../AccountModalContent";
 import { AccountModalTitle } from "../../AccoutModalTitle";
 import { AppInfo } from "./AppInfo";
 import { useSignRegisterDelegation } from "./useSignRegisterDelegation";
+import { useOnboardingSteps } from "../../useOnboardingSteps";
 
 export function AccountDelegationContent() {
-  const { signRegisterDelegation, isPending, error } = useSignRegisterDelegation();
+  const { signRegisterDelegationAsync, isPending, error } = useSignRegisterDelegation();
+  const { resetStep } = useOnboardingSteps();
 
   return (
     <>
@@ -36,7 +38,14 @@ export function AccountDelegationContent() {
             for this app, and creating a signing key for a frictionless experience.
           </p>
 
-          <Button className="self-stretch" pending={isPending} onClick={() => signRegisterDelegation()}>
+          <Button
+            className="self-stretch"
+            pending={isPending}
+            onClick={async () => {
+              await signRegisterDelegationAsync();
+              resetStep();
+            }}
+          >
             Sign in
           </Button>
         </div>

diff --git a/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts b/packages/account-kit/src/steps/app-account/useSignRegisterDelegation.ts
@@ -63,6 +63,7 @@ export function useSignRegisterDelegation() {
     () => ({
       ...result,
       signRegisterDelegation: result.mutate,
+      signRegisterDelegationAsync: result.mutateAsync,
       registerDelegationSignature,
     }),
     [registerDelegationSignature, result],