feat: add enable conditions to the usebalance and usereadcontract hooks

kleros · Oct 8, 2024 · 964eac7 · 964eac7
1 parent 6271ea6
commit 964eac7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 15 deletions.
diff --git a/web/src/pages/NewTransaction/NavigationButtons/DepositPaymentButton.tsx b/web/src/pages/NewTransaction/NavigationButtons/DepositPaymentButton.tsx
@@ -70,10 +70,12 @@ const DepositPaymentButton: React.FC = () => {
   const finalRecipientAddress = ensResult.data || sellerAddress;
 
   const { data: nativeBalance } = useBalance({
+    query: { enabled: isNativeTransaction },
     address: isNativeTransaction ? (address as `0x${string}`) : undefined,
   });
 
   const { data: tokenBalance } = useReadContract({
+    query: { enabled: !isNativeTransaction },
     address: !isNativeTransaction ? (sendingToken?.address as `0x${string}`) : undefined,
     abi: erc20Abi,
     functionName: "balanceOf",

diff --git a/...ction/Terms/Payment/GeneralTransaction/TokenAndAmount/TokenSelector/TokenItem/Balance.tsx b/...ction/Terms/Payment/GeneralTransaction/TokenAndAmount/TokenSelector/TokenItem/Balance.tsx
@@ -23,16 +23,18 @@ interface IBalance {
 
 const Balance: React.FC<IBalance> = ({ token }) => {
   const { address } = useAccount();
-  const isNativeTransaction = token?.address === 'native';
+  const isNativeTransaction = token?.address === "native";
 
   const { data: nativeBalance } = useBalance({
-    address: isNativeTransaction ? address as `0x${string}` : undefined,
+    query: { enabled: isNativeTransaction },
+    address: isNativeTransaction ? (address as `0x${string}`) : undefined,
   });
 
   const { data: tokenBalance } = useReadContract({
-    address: !isNativeTransaction ? token?.address as `0x${string}` : undefined,
+    query: { enabled: !isNativeTransaction },
+    address: !isNativeTransaction ? (token?.address as `0x${string}`) : undefined,
     abi: erc20Abi,
-    functionName: 'balanceOf',
+    functionName: "balanceOf",
     args: [address as `0x${string}`],
   });
 

diff --git a/web/src/pages/NewTransaction/Terms/Payment/GeneralTransaction/TokenAndAmount/index.tsx b/web/src/pages/NewTransaction/Terms/Payment/GeneralTransaction/TokenAndAmount/index.tsx
@@ -35,23 +35,26 @@ const TokenAndAmount: React.FC<ITokenAndAmount> = ({ quantity, setQuantity }) =>
   const { address } = useAccount();
   const { sendingToken, setHasSufficientNativeBalance } = useNewTransactionContext();
 
-  const isNativeTransaction = sendingToken?.address === 'native';
+  const isNativeTransaction = sendingToken?.address === "native";
 
   const { data: nativeBalance } = useBalance({
-    address: isNativeTransaction ? address as `0x${string}` : undefined,
+    query: { enabled: isNativeTransaction },
+    address: isNativeTransaction ? (address as `0x${string}`) : undefined,
   });
 
   const { data: tokenBalance } = useReadContract({
-    address: !isNativeTransaction ? sendingToken?.address as `0x${string}` : undefined,
+    query: { enabled: !isNativeTransaction },
+    address: !isNativeTransaction ? (sendingToken?.address as `0x${string}`) : undefined,
     abi: erc20Abi,
-    functionName: 'balanceOf',
+    functionName: "balanceOf",
     args: [address as `0x${string}`],
   });
 
   const { data: tokenDecimal } = useReadContract({
-    address: !isNativeTransaction ? sendingToken?.address as `0x${string}` : undefined,
+    query: { enabled: !isNativeTransaction },
+    address: !isNativeTransaction ? (sendingToken?.address as `0x${string}`) : undefined,
     abi: erc20Abi,
-    functionName: 'decimals',
+    functionName: "decimals",
   });
 
   const [error, setError] = useState("");
@@ -89,11 +92,7 @@ const TokenAndAmount: React.FC<ITokenAndAmount> = ({ quantity, setQuantity }) =>
       <AmountField quantity={quantity} setQuantity={setQuantity} error={error} />
       <TokenSelectorAndMaxBalance>
         <TokenSelector />
-        <MaxBalance
-          formattedBalance={formattedBalance}
-          rawBalance={balanceAmount}
-          setQuantity={setQuantity}
-        />
+        <MaxBalance rawBalance={balanceAmount} {...{ setQuantity, formattedBalance }} />
       </TokenSelectorAndMaxBalance>
     </Container>
   );