check for stRIF balance

RootstockCollective · Nov 12, 2024 · ce6e0a5 · ce6e0a5
1 parent 6f14485
commit ce6e0a5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/app/user/Delegation/DelegationSection.tsx b/src/app/user/Delegation/DelegationSection.tsx
@@ -13,13 +13,16 @@ import { useAccount } from 'wagmi'
 import { RenderTotalBalance } from '../Balances/RenderTotalBalance'
 import { BalancesProvider } from '../Balances/context/BalancesContext'
 import { useGetDelegates } from './hooks/useGetDelegates'
-import { ZeroAddress } from 'ethers'
 import { ReclaimCell } from './ReclaimCell'
 import { DelegationAction } from './type'
+import { useGetAddressBalances } from '../Balances/hooks/useGetAddressBalances'
 
 export const DelegationSection = () => {
   const { address } = useAccount()
   const { delegateeAddress } = useGetDelegates(address)
+  const {
+    stRIF: { balance: stRIFBalance },
+  } = useGetAddressBalances()
 
   const [isDelegateModalOpened, setIsDelegateModalOpened] = useState(false)
   const [hash, setHash] = useState<Hash | undefined>(undefined)
@@ -43,7 +46,7 @@ export const DelegationSection = () => {
     }
   }, [action, isError, isSuccess, setGlobalMessage])
 
-  const isValidDelegatee = delegateeAddress !== address && delegateeAddress !== ZeroAddress
+  const isValidDelegatee = delegateeAddress !== address && Number(stRIFBalance) > 0
   const delegatee = {
     'Voting Power Delegated': isValidDelegatee ? <HolderColumn address={delegateeAddress || ''} /> : '-',
     Amount: isValidDelegatee ? <RenderTotalBalance symbol="stRIF" /> : '-',