Fix stale execution context (#778)

cartridge-gg · Sep 25, 2024 · 3446f22 · 3446f22
1 parent 1c5244e
commit 3446f22
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/packages/keychain/src/components/ExecutionContainer.tsx b/packages/keychain/src/components/ExecutionContainer.tsx
@@ -98,6 +98,7 @@ export function ExecutionContainer({
       setCtrlError(error);
     } finally {
       setIsLoading(false);
+      setMaxFee(null);
     }
   };
 

diff --git a/packages/keychain/src/hooks/connection.ts b/packages/keychain/src/hooks/connection.ts
@@ -67,11 +67,12 @@ export function useConnectionValue() {
         code: ResponseCodes.CANCELED,
         message: "User aborted",
       });
+      setContext(undefined); // clears context
       await parent.close();
     } catch (e) {
       // Always fails for some reason
     }
-  }, [context, parent]);
+  }, [context, parent, setContext]);
 
   const openModal = useCallback(async () => {
     if (!parent) return;

diff --git a/packages/keychain/src/utils/connection/types.ts b/packages/keychain/src/utils/connection/types.ts
@@ -21,7 +21,8 @@ export type ConnectionCtx =
   | LogoutCtx
   | ExecuteCtx
   | SignMessageCtx
-  | OpenSettingsCtx;
+  | OpenSettingsCtx
+  | undefined;
 
 export type ConnectCtx = {
   origin: string;