Some changes for handle events

masa-finance · Feb 15, 2023 · d36318b · d36318b
1 parent a417717
commit d36318b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/provider/modules/session/session.ts b/src/provider/modules/session/session.ts
@@ -14,8 +14,8 @@ export const useSession = (
   isLoading: boolean;
   error: unknown;
 } => {
-  const queryKey: string = useMemo(() => {
-    return `session-${walletAddress}`;
+  const queryKey: any[] = useMemo(() => {
+    return ["session", walletAddress];
   }, [walletAddress]);
 
   const {
@@ -28,23 +28,23 @@ export const useSession = (
   });
 
   useEffect(() => {
-    if (loggedIn && walletAddress) {
-      void queryClient.invalidateQueries(queryKey);
+    if (loggedIn && !walletAddress) {
+      void queryClient.invalidateQueries('session');
     }
   }, [walletAddress, loggedIn, queryKey]);
 
   const login = useCallback(async () => {
     const logged = await masa?.session.login();
     if (logged) {
-      await queryClient.invalidateQueries(queryKey);
+      await queryClient.invalidateQueries('session');
       await queryClient.refetchQueries();
     }
   }, [masa, queryKey]);
 
   const logout = useCallback(
     async (callback?: () => void) => {
       await masa?.session.logout();
-      await queryClient.invalidateQueries(queryKey);
+      await queryClient.invalidateQueries('session');
       await queryClient.refetchQueries();
 
       if (callback) {

diff --git a/src/provider/use-metamask.ts b/src/provider/use-metamask.ts
@@ -83,6 +83,7 @@ export const useMetamask = ({
             await handleLogout?.();
             await disconnect();
             await queryClient.invalidateQueries('wallet');
+            await queryClient.invalidateQueries('session');
           }
         }
       );