Merge pull request #6 from valorem-labs-inc/fix-siwe-sign-out

fix: SIWE sign out
valorem-labs-inc · Dec 6, 2023 · 39d5eab · 39d5eab
2 parents 0957cb1 + 77a8c27
commit 39d5eab
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 8 deletions.
diff --git a/.changeset/stale-coats-retire.md b/.changeset/stale-coats-retire.md
@@ -0,0 +1,5 @@
+---
+'@valorem-labs-inc/react-hooks': patch
+---
+
+fix: SIWE provider sign out
diff --git a/src/context/SIWEProvider.tsx b/src/context/SIWEProvider.tsx
@@ -20,6 +20,9 @@ const siweQueryProps = {
   refetchOnWindowFocus: false,
   refetchOnMount: false,
   refetchOnReconnect: false,
+  keepPreviousData: false,
+  cacheTime: 0,
+  staleTime: 1,
 };
 
 export function SIWEProvider({ onSignIn, onSignOut, children }: SIWEProps) {

diff --git a/src/utils/siwe.ts b/src/utils/siwe.ts
@@ -67,19 +67,47 @@ export const getSIWEConfig = ({
     },
     async signOut() {
       logger.debug('Signing out...');
-      await signOutQuery.refetch();
-      await nonceQuery.refetch();
+      try {
+        await signOutQuery.refetch();
+        queryClient.setQueryData(['valorem.trade.v1.Auth', 'Nonce'], undefined);
+        queryClient.setQueryData(
+          ['valorem.trade.v1.Auth', 'Session'],
+          undefined,
+        );
+        queryClient.setQueryData(
+          ['valorem.trade.v1.Auth', 'Authenticate'],
+          undefined,
+        );
+        queryClient.setQueryData(['valorem.trade.v1.Auth', 'signed-out'], true);
+        logger.info('Signed out');
+        return true;
+      } catch (error) {
+        logger.error('Error signing out');
+        return false;
+      }
+    },
+    async getSession() {
+      logger.debug('Getting session...');
+      await new Promise((resolve) => {
+        setTimeout(resolve, 750);
+      });
+      if (
+        queryClient.getQueryData(['valorem.trade.v1.Auth', 'signed-out']) ===
+        true
+      ) {
+        logger.debug('User is signed out');
+        queryClient.setQueryData(
+          ['valorem.trade.v1.Auth', 'signed-out'],
+          false,
+        );
+        return null;
+      }
       queryClient.setQueryData(['valorem.trade.v1.Auth', 'Nonce'], undefined);
       queryClient.setQueryData(['valorem.trade.v1.Auth', 'Session'], undefined);
       queryClient.setQueryData(
         ['valorem.trade.v1.Auth', 'Authenticate'],
         undefined,
       );
-      logger.info('Signed out');
-      return true;
-    },
-    async getSession() {
-      logger.debug('Getting session...');
 
       // check auth endpoint to ensure session is valid
       const { data: authData } = await authenticateQuery.refetch({});
@@ -90,6 +118,7 @@ export const getSIWEConfig = ({
       const authorizedAddress = fromH160ToAddress(authData);
       if (authorizedAddress.toLowerCase() !== address?.toLowerCase()) {
         logger.error('Authorized address does not match connected address');
+        return null;
       }
       // get session data
       const { data: sessionData } = await sessionQuery.refetch();
@@ -98,8 +127,12 @@ export const getSIWEConfig = ({
         return null;
       }
       const sessionAddress = fromH160ToAddress(sessionData.address);
-      if (sessionAddress.toLowerCase() === address?.toLowerCase()) {
+      if (sessionAddress.toLowerCase() === address.toLowerCase()) {
         logger.debug('Session is valid');
+        queryClient.setQueryData(
+          ['valorem.trade.v1.Auth', 'signed-out'],
+          false,
+        );
         return {
           address: sessionAddress,
           chainId: Number(fromH256(sessionData.chainId).toString()),