fix: fix migrated tokens showing up in legacy balance (race condition)

src/hooks/JBV3Token/contractReader/useTotalLegacyTokenBalance.ts

@@ -13,15 +13,21 @@ export function useTotalLegacyTokenBalance({
   const v1UnclaimedBalance = useV1UnclaimedBalanceForV3Token()
   const v1ClaimedBalance = v1TotalBalance?.sub(v1UnclaimedBalance)
 
-  const { data: v2TotalBalance } = useV2TotalBalance({ projectId })
-  const { data: v2UnclaimedBalance } = useV2UnclaimedBalanceForV3Token({
-    projectId,
-  })
-  const v2ClaimedBalance =
-    v2TotalBalance?.sub(v2UnclaimedBalance ?? 0) ?? BigNumber.from(0)
+  const { data: v2TotalBalance, loading: v2BalanceLoading } = useV2TotalBalance(
+    { projectId },
+  )
+  const { data: v2UnclaimedBalance, loading: unclaimedBalanceLoading } =
+    useV2UnclaimedBalanceForV3Token({
+      projectId,
+    })
+
+  if (v2BalanceLoading || unclaimedBalanceLoading) return BigNumber.from(0)
 
   const totalLegacyTokenBalance = v1TotalBalance?.add(v2TotalBalance ?? 0)
 
+  const v2ClaimedBalance =
+    v2TotalBalance?.sub(v2UnclaimedBalance ?? 0) ?? BigNumber.from(0)
+
   return {
     v1TotalBalance,
     v1UnclaimedBalance,

src/hooks/JBV3Token/contractReader/useV2UnclaimedBalanceForV3Token.ts

@@ -1,5 +1,6 @@
+import { BigNumber } from 'ethers'
 import { useWallet } from 'hooks/Wallet'
-import { useUnclaimedTokenBalance } from 'hooks/v2v3/contractReader/useUnclaimedTokenBalance'
+import useV2ContractReader from 'hooks/v2v3/contractReader/useV2ContractReader'
 import { useJBTokenStoreForV3Token } from '../contracts/useJBTokenStoreForV3Token'
 
 export function useV2UnclaimedBalanceForV3Token({
@@ -10,9 +11,12 @@ export function useV2UnclaimedBalanceForV3Token({
   const { userAddress } = useWallet()
   const v2TokenStoreContract = useJBTokenStoreForV3Token()
 
-  return useUnclaimedTokenBalance({
-    userAddress,
-    projectId,
-    JBTokenStore: v2TokenStoreContract,
+  return useV2ContractReader<BigNumber>({
+    contract: v2TokenStoreContract,
+    functionName: 'unclaimedBalanceOf',
+    args:
+      userAddress && projectId && v2TokenStoreContract
+        ? [userAddress, projectId]
+        : null,
   })
 }

src/hooks/v2v3/contractReader/useUnclaimedTokenBalance.ts

@@ -1,27 +1,21 @@
 import { BigNumber } from 'ethers'
-import { bigNumbersDiff } from 'utils/bigNumbers'
-
-import { Contract } from 'ethers'
 import { V2V3ContractName } from 'models/v2v3/contracts'
 import useContractReader from './useV2ContractReader'
 
 /** Returns unclaimed balance of user with `userAddress`. */
 export function useUnclaimedTokenBalance({
   projectId,
   userAddress,
-  JBTokenStore,
 }: {
   projectId: number | undefined
   userAddress: string | undefined
-  JBTokenStore?: Contract
 }) {
   return useContractReader<BigNumber>({
-    contract: JBTokenStore ?? V2V3ContractName.JBTokenStore,
+    contract: V2V3ContractName.JBTokenStore,
     functionName: 'unclaimedBalanceOf',
     args:
       userAddress && projectId
         ? [userAddress, BigNumber.from(projectId).toHexString()]
         : null,
-    valueDidChange: bigNumbersDiff,
   })
 }