feat: add balance polling to two other pages (#138)

src/pages/Home/components/Portfolio/Assets.tsx

@@ -28,12 +28,16 @@ import { isBitcoinNetwork } from '@src/background/services/network/utils/isBitco
 import { useAnalyticsContext } from '@src/contexts/AnalyticsProvider';
 import { useTokenPriceMissing } from '@src/hooks/useTokenPriceIsMissing';
 import { PAndL } from '@src/components/common/ProfitAndLoss';
+import { useLiveBalance } from '@src/hooks/useLiveBalance';
+import { TokenType } from '@avalabs/vm-module-types';
 
 enum AssetsTabs {
   TOKENS,
   ACTIVITY,
 }
 
+const POLLED_BALANCES = [TokenType.NATIVE, TokenType.ERC20];
+
 export function Assets() {
   const { t } = useTranslation();
   const { network } = useNetworkContext();
@@ -70,6 +74,8 @@ export function Assets() {
     return isPriceMissingFromNetwork(network?.chainId);
   }, [isPriceMissingFromNetwork, network]);
 
+  useLiveBalance(POLLED_BALANCES);
+
   return (
     <Stack sx={{ flexGrow: 1 }}>
       <Stack

src/pages/Wallet/TokenFlow.tsx

@@ -38,6 +38,10 @@ import { isXchainNetwork } from '@src/background/services/network/utils/isAvalan
 import { getUnconfirmedBalanceInCurrency } from '@src/background/services/balances/models';
 import { isTokenMalicious } from '@src/utils/isTokenMalicious';
 import { MaliciousTokenWarningBox } from '@src/components/common/MaliciousTokenWarning';
+import { useLiveBalance } from '@src/hooks/useLiveBalance';
+import { TokenType } from '@avalabs/vm-module-types';
+
+const POLLED_BALANCES = [TokenType.NATIVE, TokenType.ERC20];
 
 export function TokenFlow() {
   const { t } = useTranslation();
@@ -97,6 +101,8 @@ export function TokenFlow() {
     }
   }, [activeAccount, hasAccessToActiveNetwork, tokensWithBalances]);
 
+  useLiveBalance(POLLED_BALANCES);
+
   if (!hasAccessToActiveNetwork) {
     return (
       <NotSupportedByWallet