From dbb72c6ee4fb98cb91998079db728ba90ae1ea4e Mon Sep 17 00:00:00 2001
From: Taylan Pince <taylanpince@gmail.com>
Date: Fri, 7 Jun 2024 17:51:41 +0200
Subject: [PATCH] Fix issue with ERC20 balance check always hitting Polygon
 (#22)

---
 src/components/views/SendERC20View.tsx | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/components/views/SendERC20View.tsx b/src/components/views/SendERC20View.tsx
index 1254121..7039efc 100644
--- a/src/components/views/SendERC20View.tsx
+++ b/src/components/views/SendERC20View.tsx
@@ -1,7 +1,7 @@
 import { useEffect, useState } from 'react'
 import { Box, Text, Button, TextInput, Spinner, Select } from '@0xsequence/design-system'
 import { ethers } from 'ethers'
-import { node, sequence } from '../../main'
+import { sequence } from '../../main'
 import { FeeOption, isSentTransactionResponse, Network, erc20 } from '@0xsequence/waas'
 import { checkTransactionFeeOptions, TransactionFeeOptions } from './TransactionFeeOptions.tsx'
 
@@ -46,6 +46,7 @@ export function SendERC20View(props: { network?: Network }) {
 
     setTokenBalance('...')
 
+    const node = new ethers.providers.JsonRpcProvider(`https://nodes.sequence.app/${props.network?.name}`)
     const contract = new ethers.Contract(
       tokenAddress,
       [
@@ -56,14 +57,18 @@ export function SendERC20View(props: { network?: Network }) {
       node
     )
 
-    const [balance, decimals, symbol] = await Promise.all([
-      contract.balanceOf(sequence.getAddress()),
-      contract.decimals(),
-      contract.symbol()
-    ])
-
-    setDecimals(decimals)
-    setTokenBalance(`${ethers.utils.formatUnits(balance, decimals)} ${symbol}`)
+    try {
+      const [balance, decimals, symbol] = await Promise.all([
+        contract.balanceOf(sequence.getAddress()),
+        contract.decimals(),
+        contract.symbol()
+      ])
+
+      setDecimals(decimals)
+      setTokenBalance(`${ethers.utils.formatUnits(balance, decimals)} ${symbol}`)
+    } catch (e) {
+      setTokenBalance('---')
+    }
   }
 
   const checkFeeOptions = async () => {