diff --git a/consts/chains.ts b/consts/chains.ts index 5334d6ce..125e6d7f 100644 --- a/consts/chains.ts +++ b/consts/chains.ts @@ -30,3 +30,8 @@ export const SDK_LEGACY_SUPPORTED_CHAINS = [ export const SDK_SUPPORTED_MULTICHAIN_CHAINS = [ LIDO_MULTICHAIN_CHAINS.OptimismSepolia, ]; + +// TODO: move to @lidofinance/lido-ethereum-sdk package +export const isSDKSupportedL2Chain = (chainId) => { + return SDK_SUPPORTED_MULTICHAIN_CHAINS.indexOf(chainId) > -1; +}; diff --git a/shared/hooks/use-balance.ts b/shared/hooks/use-balance.ts index 7ddf9dee..53699215 100644 --- a/shared/hooks/use-balance.ts +++ b/shared/hooks/use-balance.ts @@ -17,8 +17,7 @@ import { import type { GetBalanceData } from 'wagmi/query'; import { config } from 'config'; - -import { useDappStatus } from './use-dapp-status'; +import { isSDKSupportedL2Chain } from 'consts/chains'; const nativeToBN = (data: bigint) => BigNumber.from(data.toString()); @@ -254,7 +253,6 @@ export const useStethBalance = ({ shouldSubscribeToUpdates = true, }: UseBalanceProps = {}) => { const { address } = useAccount(); - const { isDappActiveOnL2 } = useDappStatus(); const mergedAccount = account ?? address; const { steth, l2Steth, core } = useLidoSDK(); @@ -265,7 +263,9 @@ export const useStethBalance = ({ staleTime: Infinity, queryFn: async () => - isDappActiveOnL2 ? l2Steth.getContract() : steth.getContract(), + isSDKSupportedL2Chain(core.chainId) + ? l2Steth.getContract() + : steth.getContract(), }); const balanceData = useTokenBalance( @@ -282,7 +282,6 @@ export const useWstethBalance = ({ shouldSubscribeToUpdates = true, }: UseBalanceProps = {}) => { const { address } = useAccount(); - const { isDappActiveOnL2 } = useDappStatus(); const mergedAccount = account ?? address; const { wsteth, l2Wsteth, core } = useLidoSDK(); @@ -292,7 +291,9 @@ export const useWstethBalance = ({ enabled: !!mergedAccount, staleTime: Infinity, queryFn: async () => - isDappActiveOnL2 ? l2Wsteth.getContract() : wsteth.getContract(), + isSDKSupportedL2Chain(core.chainId) + ? l2Wsteth.getContract() + : wsteth.getContract(), }); const balanceData = useTokenBalance( diff --git a/shared/hooks/use-dapp-status.ts b/shared/hooks/use-dapp-status.ts index cf639c1f..7d651713 100644 --- a/shared/hooks/use-dapp-status.ts +++ b/shared/hooks/use-dapp-status.ts @@ -1,10 +1,7 @@ import { useMemo } from 'react'; import { useAccount } from 'wagmi'; -import { - LIDO_MULTICHAIN_CHAINS, - SDK_SUPPORTED_MULTICHAIN_CHAINS, -} from 'consts/chains'; +import { isSDKSupportedL2Chain, LIDO_MULTICHAIN_CHAINS } from 'consts/chains'; import { useIsSupportedChain } from './use-is-supported-chain'; import { useConfig } from 'config'; @@ -32,7 +29,7 @@ export const useDappStatus = () => { const isDappActiveOnL2 = useMemo(() => { if (!chainId) return false; - return SDK_SUPPORTED_MULTICHAIN_CHAINS.indexOf(chainId) > -1; + return isSDKSupportedL2Chain(chainId); }, [chainId]); return {