fix(RedPacket): revalidate after claiming (#11453)

* fix(RedPacket): revalidate after claiming * fix: prettier --------- Co-authored-by: swkatmask <[email protected]> Co-authored-by: guanbinrui <[email protected]>
DimensionDev · Feb 27, 2024 · af09c45 · af09c45
1 parent 9a9dad8
commit af09c45
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/packages/plugins/RedPacket/src/SiteAdaptor/RedPacket/index.tsx b/packages/plugins/RedPacket/src/SiteAdaptor/RedPacket/index.tsx
@@ -19,6 +19,7 @@ import { useRedPacketContract } from '../hooks/useRedPacketContract.js'
 import { useRefundCallback } from '../hooks/useRefundCallback.js'
 import { OperationFooter } from './OperationFooter.js'
 import { useQuery } from '@tanstack/react-query'
+import { avalancheFuji } from 'viem/chains'
 
 const useStyles = makeStyles<{ outdated: boolean }>()((theme, { outdated }) => {
     return {
@@ -217,7 +218,7 @@ export const RedPacket = memo(function RedPacket({ payload }: RedPacketProps) {
         if (typeof hash === 'string') {
             revalidateAvailability()
         }
-    }, [canClaim, canRefund, claimCallback, checkResult, recheckClaimStatus])
+    }, [canClaim, canRefund, claimCallback, checkResult, recheckClaimStatus, revalidateAvailability])
 
     const handleShare = useCallback(() => {
         if (shareText) share?.(shareText, source ? source : undefined)
@@ -230,22 +231,23 @@ export const RedPacket = memo(function RedPacket({ payload }: RedPacketProps) {
 
     const { data } = useQuery({
         enabled: !!availability && !!payload.rpid && !!token?.symbol,
-        queryKey: ['red-packet', 'theme-id', payload.rpid],
+        queryKey: ['red-packet', 'theme-id', payload.rpid, availability?.balance, availability?.claimed],
         queryFn: async () => {
+            if (!token || !availability) return null
             const name = payload.sender.name
 
             return FireflyRedPacket.getCoverUrlByRpid(
                 payload.rpid,
-                token?.symbol,
-                token?.decimals,
+                token.symbol,
+                token.decimals,
                 payload.shares,
                 payload.total,
                 [isValidAddress, isValidDomain, (n: string) => n.startsWith('@')].some((f) => f(name)) ? name : (
                     `@${name}`
                 ),
                 payload.sender.message,
-                availability?.balance ?? payload.total,
-                toFixed(minus(payload.shares, availability?.claimed || 0)),
+                availability.balance ?? payload.total,
+                toFixed(minus(payload.shares, availability.claimed || 0)),
             )
         },
     })