fix(vault): refetch all balances

thematters · Dec 18, 2024 · 1a037a5 · 1a037a5
1 parent d81ab24
commit 1a037a5
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 18 deletions.
diff --git a/src/common/enums/events.ts b/src/common/enums/events.ts
@@ -52,6 +52,9 @@ export const SUPPORT_SUCCESS = 'supportSuccess'
 export const SUPPORT_SUCCESS_USDT_VISITOR = 'supportSuccessUSDTVisitor'
 export const SUPPORT_SUCCESS_ANIMATION = 'supportSuccessAnimation'
 
+// Wallet
+export const REFETCH_BALANCE_USDT = 'refetchBalanceUSDT'
+
 // Universal auth
 export enum UNIVERSAL_AUTH_TRIGGER {
   appreciation = 'appreciation',

diff --git a/src/components/Dialogs/WithdrawVaultUSDTDialog/Confirming.tsx b/src/components/Dialogs/WithdrawVaultUSDTDialog/Confirming.tsx
@@ -3,16 +3,13 @@ import gql from 'graphql-tag'
 import { useContext, useEffect, useState } from 'react'
 import { FormattedMessage } from 'react-intl'
 
-import { OPEN_WITHDRAW_VAULT_USDT_DIALOG, PATHS } from '~/common/enums'
-import { formatAmount, truncate } from '~/common/utils'
 import {
-  Dialog,
-  Spinner,
-  toast,
-  useRoute,
-  useVaultBalanceUSDT,
-  ViewerContext,
-} from '~/components'
+  OPEN_WITHDRAW_VAULT_USDT_DIALOG,
+  PATHS,
+  REFETCH_BALANCE_USDT,
+} from '~/common/enums'
+import { formatAmount, truncate } from '~/common/utils'
+import { Dialog, Spinner, toast, useRoute, ViewerContext } from '~/components'
 import {
   TransactionState,
   WithdrawVaultUsdtMutation,
@@ -57,7 +54,6 @@ const Confirming: React.FC<ConfirmingProps> = ({ amount, closeDialog }) => {
   const { router } = useRoute()
   const viewer = useContext(ViewerContext)
   const address = viewer.info.ethAddress!
-  const { refetch: refetchVaultBalanceUSDT } = useVaultBalanceUSDT()
 
   const [withdraw] = useMutation<WithdrawVaultUsdtMutation>(WITHDRAW_VAULT_USDT)
   const [txId, setTxId] = useState<string | null>(null)
@@ -111,7 +107,7 @@ const Confirming: React.FC<ConfirmingProps> = ({ amount, closeDialog }) => {
       ],
     })
 
-    refetchVaultBalanceUSDT()
+    window.dispatchEvent(new CustomEvent(REFETCH_BALANCE_USDT, {}))
 
     closeDialog()
   }

diff --git a/src/components/Forms/PaymentForm/PayTo/CurrencyChoice/Tips/index.tsx b/src/components/Forms/PaymentForm/PayTo/CurrencyChoice/Tips/index.tsx
@@ -15,7 +15,7 @@ const Tips = () => {
   const viewer = useContext(ViewerContext)
   const { lang } = useContext(LanguageContext)
   const { address } = useAccount()
-  const { data: balanceUSDTData } = useBalanceUSDT({})
+  const { data: balanceUSDTData } = useBalanceUSDT()
 
   const isZeroBalance = balanceUSDTData?.value && balanceUSDTData?.value < 0n
   const hasEthAddress = !!viewer.info.ethAddress

diff --git a/src/components/Forms/PaymentForm/PayTo/CurrencyChoice/USDTChoice.tsx b/src/components/Forms/PaymentForm/PayTo/CurrencyChoice/USDTChoice.tsx
@@ -45,7 +45,7 @@ const USDTChoice: React.FC<FormProps> = ({
   const { address } = useAccount()
 
   const { data: balanceUSDTData, isLoading: balanceUSDTLoading } =
-    useBalanceUSDT({})
+    useBalanceUSDT()
   const balanceUSDT = parseFloat(balanceUSDTData?.formatted || '0')
 
   const curatorAddress = viewer.info.ethAddress

diff --git a/src/components/Hook/useERC20.ts b/src/components/Hook/useERC20.ts
@@ -38,7 +38,7 @@ export const useBalanceUSDT = ({
   address: addr,
 }: {
   address?: string | null
-}) => {
+} = {}) => {
   const viewer = useContext(ViewerContext)
   const viewerEthAddress = viewer.info.ethAddress
   const targetNetwork = featureSupportedChains.curation[0]

diff --git a/src/views/Me/Wallet/Balance/USDT.tsx b/src/views/Me/Wallet/Balance/USDT.tsx
@@ -9,6 +9,7 @@ import {
   contract,
   OPEN_WITHDRAW_VAULT_USDT_DIALOG,
   PATHS,
+  REFETCH_BALANCE_USDT,
 } from '~/common/enums'
 import { formatAmount } from '~/common/utils'
 import {
@@ -19,6 +20,7 @@ import {
   TextIcon,
   Translate,
   useBalanceUSDT,
+  useEventListener,
   useVaultBalanceUSDT,
   ViewerContext,
 } from '~/components'
@@ -34,10 +36,16 @@ interface USDTBalanceProps {
 export const USDTBalance = ({ currency, exchangeRate }: USDTBalanceProps) => {
   const viewer = useContext(ViewerContext)
   const address = viewer.info.ethAddress
-  const { data: balanceUSDTData, isLoading: balanceUSDTLoading } =
-    useBalanceUSDT({})
-  const { data: vaultBalanceUSDTData, isLoading: vaultBalanceUSDTLoading } =
-    useVaultBalanceUSDT()
+  const {
+    data: balanceUSDTData,
+    refetch: refetchBalanceUSDT,
+    isLoading: balanceUSDTLoading,
+  } = useBalanceUSDT({})
+  const {
+    data: vaultBalanceUSDTData,
+    refetch: refetchVaultBalanceUSDT,
+    isLoading: vaultBalanceUSDTLoading,
+  } = useVaultBalanceUSDT()
   const balanceUSDT = parseFloat(balanceUSDTData?.formatted || '0')
   const vaultBalanceUSDT = parseFloat(
     formatUnits(
@@ -59,6 +67,11 @@ export const USDTBalance = ({ currency, exchangeRate }: USDTBalanceProps) => {
     window.dispatchEvent(new CustomEvent(OPEN_WITHDRAW_VAULT_USDT_DIALOG, {}))
   }
 
+  useEventListener(REFETCH_BALANCE_USDT, () => {
+    refetchBalanceUSDT()
+    refetchVaultBalanceUSDT()
+  })
+
   if (!address && !vaultBalanceUSDT) {
     return (
       <section className={classes}>