Refactor useErc20 and useLock hooks to use erc20IQConfig

EveripediaNetwork · May 14, 2024 · f5e15f9 · f5e15f9
1 parent 558fccb
commit f5e15f9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 35 deletions.
diff --git a/src/hooks/useErc20.ts b/src/hooks/useErc20.ts
@@ -3,27 +3,15 @@ import { useAccount, useBalance } from 'wagmi'
 
 export const useErc20 = () => {
   const { address } = useAccount()
-  const { data: erc20Balance, refetch: refetchErc20Balance } =
-    config.alchemyChain === 'iqChain'
-      ? useBalance({
-          address: address,
-          token: config.wiqAddress as `0x${string}`,
-        })
-      : useBalance({
-          address: address,
-          token: config.iqAddress as `0x${string}`,
-        })
+  const { data: erc20Balance, refetch: refetchErc20Balance } = useBalance({
+    address: address,
+    token: config.erc20IQConfig.address,
+  })
 
-  const { data: totalValueLocked } =
-    config.alchemyChain === 'iqChain'
-      ? useBalance({
-          address: config.hiiqAddress as `0x${string}`,
-          token: config.wiqAddress as `0x${string}`,
-        })
-      : useBalance({
-          address: config.hiiqAddress as `0x${string}`,
-          token: config.iqAddress as `0x${string}`,
-        })
+  const { data: totalValueLocked } = useBalance({
+    address: config.hiiqAddress as `0x${string}`,
+    token: config.erc20IQConfig.address,
+  })
 
   const getUserBalance = () => {
     if (address) {

diff --git a/src/hooks/useLock.ts b/src/hooks/useLock.ts
@@ -2,25 +2,12 @@ import hiIQABI from '@/abis/hiIQABI.abi'
 import config from '@/config'
 import { useAccount, useContractRead, useContractWrite } from 'wagmi'
 import { waitForTransaction } from 'wagmi/actions'
-import erc20Abi from '@/abis/erc20.abi'
 
 const hiiqContractConfig = {
   address: config.hiiqAddress as `0x${string}`,
   abi: hiIQABI,
 }
 
-//use WIQ for dev
-const erc20ContractConfig =
-  config.alchemyChain === 'iqChain'
-    ? {
-        address: config.wiqAddress as `0x${string}`,
-        abi: erc20Abi,
-      }
-    : {
-        address: config.iqAddress as `0x${string}`,
-        abi: erc20Abi,
-      }
-
 export const useLock = () => {
   const { address } = useAccount()
 
@@ -46,13 +33,13 @@ export const useLock = () => {
 
   const { data: allowanceToken, refetch: refetchedAllowanceToken } =
     useContractRead({
-      ...erc20ContractConfig,
+      ...config.erc20IQConfig,
       functionName: 'allowance',
       args: [address as `0x${string}`, config.hiiqAddress as `0x${string}`],
     })
 
   const { writeAsync: approve } = useContractWrite({
-    ...erc20ContractConfig,
+    ...config.erc20IQConfig,
     functionName: 'approve',
   })