fix: stabilize zone authorization state

Author: juandolealt

src/lib/useZoneAuthorization.ts

@@ -1,6 +1,6 @@
 'use client'
 
-import { useMutation, useQuery } from '@tanstack/react-query'
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query'
 import type { Hex } from 'viem'
 import { Storage as ZoneStorage } from 'viem/tempo'
 
@@ -29,6 +29,7 @@ export function useZoneAuthorization(parameters: {
   zoneClient: ZoneAuthClientLike | undefined
 }) {
   const { address, chainId, queryKey, zoneClient } = parameters
+  const queryClient = useQueryClient()
 
   const statusQuery = useQuery({
     enabled: Boolean(address && zoneClient),
@@ -85,8 +86,15 @@ export function useZoneAuthorization(parameters: {
 
       return zoneClient.zone.signAuthorizationToken()
     },
-    onSuccess: async () => {
-      await statusQuery.refetch()
+    onSuccess: async (result) => {
+      if (address) {
+        queryClient.setQueryData(queryKey, {
+          account: address,
+          expiresAt: BigInt(result.authentication.expiresAt),
+        })
+      }
+
+      await queryClient.invalidateQueries({ queryKey })
     },
   })
 
@@ -109,7 +117,10 @@ function withTimeout<T>(promise: Promise<T>, timeoutMs: number) {
         reject(error)
       }, timeoutMs)
 
-      promise.finally(() => clearTimeout(timeout))
+      promise.then(
+        () => clearTimeout(timeout),
+        () => clearTimeout(timeout),
+      )
     }),
   ])
 }