From 0527f8b92fdaf61d171573ef02c95f103684876a Mon Sep 17 00:00:00 2001
From: Anukul Pandey <anukul.030601@gmail.com>
Date: Thu, 5 Sep 2024 13:21:08 +0530
Subject: [PATCH] feat: listening balance changes

---
 src/pages/dashboard/TokenBalances.tsx | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/src/pages/dashboard/TokenBalances.tsx b/src/pages/dashboard/TokenBalances.tsx
index cd7d7ac..cdb2e34 100644
--- a/src/pages/dashboard/TokenBalances.tsx
+++ b/src/pages/dashboard/TokenBalances.tsx
@@ -1,6 +1,6 @@
-import { Token,Components } from '@reef-chain/react-lib';
+import { Token,Components, hooks } from '@reef-chain/react-lib';
 import Uik from '@reef-chain/ui-kit';
-import React, { useContext } from 'react';
+import React, { useContext, useEffect, useState } from 'react';
 import { faPlus } from '@fortawesome/free-solid-svg-icons';
 import { Link } from 'react-router-dom';
 import BigNumber from 'bignumber.js';
@@ -40,7 +40,8 @@ const balanceValue = (token: Token, price = 0): number => (new BigNumber(token.b
 
 export const TokenBalances = ({ tokens }: TokenBalances): JSX.Element => {
   const tokenPrices = useContext(TokenPricesContext);
-  const { selectedSigner, network,accounts,provider } = useContext(ReefSigners);
+  const [isReefBalanceZero,setIsReefBalanceZero] = useState(true);
+  const { selectedSigner,network,accounts,provider,reefState } = useContext(ReefSigners);
   const pools = useContext(PoolContext);
   const hidebalance = useContext(HideBalance)
   const {selExtensionName} = useConnectedWallet();
@@ -49,6 +50,9 @@ export const TokenBalances = ({ tokens }: TokenBalances): JSX.Element => {
 
   const isWalletConnect = selExtensionName == walletSelectorOptions[reefExt.REEF_WALLET_CONNECT_IDENT].name
 
+  const signers = hooks.useObservableState((reefState as any).accounts$);
+  const selectedAddress = hooks.useObservableState((reefState as any).selectedAddress$);
+
   const handleWalletConnectModal = (hasStarted:boolean)=>{
       setWcPreloader({
 value:hasStarted,
@@ -56,7 +60,19 @@ message:"waiting for transaction approval"
       })
   }
 
-  const isReefBalanceZero = selectedSigner?.balance._hex === '0x00';
+  useEffect(()=>{
+    if(selectedAddress){
+      (signers as any).forEach((sgnr)=>{
+        if(sgnr.address == selectedAddress){
+          if(!sgnr.balance){
+            setIsReefBalanceZero(true);
+          }else{
+            setIsReefBalanceZero(false);
+          }
+        }
+      })
+    }
+  },[signers,selectedAddress])
 
   const getUrl = (): string => {
     if (network.name === 'mainnet') return BUY_URL;