From 4274faa8bbc1195d846cdf98c9c85fabdfe73c52 Mon Sep 17 00:00:00 2001
From: Slava <slava@interlay.io>
Date: Mon, 2 Dec 2024 23:43:36 +0300
Subject: [PATCH] chore: use read contracts

---
 apps/evm/src/hooks/useBalances.ts | 59 ++++++++++++++-----------------
 1 file changed, 27 insertions(+), 32 deletions(-)

diff --git a/apps/evm/src/hooks/useBalances.ts b/apps/evm/src/hooks/useBalances.ts
index 3083b8b2..3d31a51b 100644
--- a/apps/evm/src/hooks/useBalances.ts
+++ b/apps/evm/src/hooks/useBalances.ts
@@ -1,15 +1,14 @@
 import { ChainId } from '@gobob/chains';
 import { CurrencyAmount, ERC20Token, Ether } from '@gobob/currency';
 import { chain } from '@react-aria/utils';
-import { useQuery } from '@tanstack/react-query';
 import { useCallback, useEffect, useMemo, useRef } from 'react';
 import { erc20Abi } from 'viem';
-import { useAccount, useBalance, usePublicClient } from 'wagmi';
+import { useAccount, useBalance, usePublicClient, useReadContracts } from 'wagmi';
 import { watchContractEvent } from '@wagmi/core';
 
 import { useTokens } from './useTokens';
 
-import { INTERVAL, isProd } from '@/constants';
+import { isProd } from '@/constants';
 import { getConfig } from '@/lib/wagmi';
 
 type Balances = Record<string, CurrencyAmount<ERC20Token | Ether>>;
@@ -22,40 +21,36 @@ const useBalances = (chainId: ChainId) => {
 
   const { data: tokens } = useTokens(chainId);
 
-  // TODO: useReadContracts instead https://wagmi.sh/react/guides/migrate-from-v1-to-v2#deprecated-usebalance-token-parameter
   const {
     data: erc20Balances,
     refetch: refetchErc20,
     ...queryResult
-  } = useQuery({
-    queryKey: ['balances', chainId, address],
-    enabled: Boolean(address && publicClient && tokens),
-    queryFn: async () => {
-      if (!tokens || !publicClient) return;
-
-      const erc20List = tokens.filter((token) => token.currency.isToken);
-
-      const balancesMulticallResult = await publicClient.multicall({
-        contracts: erc20List.map((token) => ({
-          abi: erc20Abi,
-          address: token.raw.address,
-          functionName: 'balanceOf',
-          args: [address]
-        }))
-      });
-
-      return erc20List.reduce<Balances>(
-        (result, token, index) => ({
-          ...result,
-          [token.raw.symbol]: CurrencyAmount.fromRawAmount(
-            token.currency,
-            (balancesMulticallResult[index]?.result as bigint) || 0n
-          )
-        }),
-        {} as Balances
-      );
+  } = useReadContracts({
+    allowFailure: false,
+    query: {
+      enabled: Boolean(address && publicClient && tokens),
+      select(data: (string | number | bigint)[]): Balances {
+        if (!tokens) return {} as Balances;
+
+        return tokens.reduce<Balances>((result, token) => {
+          if (token.currency.isToken) {
+            const idx = Object.keys(result).length;
+
+            result[token.raw.symbol] = CurrencyAmount.fromRawAmount(token.currency, data[idx]!);
+          }
+
+          return result;
+        }, {} as Balances);
+      }
     },
-    refetchInterval: INTERVAL.MINUTE
+    contracts: tokens
+      ?.filter((token) => token.currency.isToken)
+      .map((token) => ({
+        abi: erc20Abi,
+        address: token.raw.address,
+        functionName: 'balanceOf',
+        args: [address]
+      }))
   });
 
   const shouldRefetchRef = useRef(false);