feat(stake page): useBalance for OP Sepolia

lidofinance · Sep 19, 2024 · a36d0b2 · a36d0b2
1 parent 79ccaeb
commit a36d0b2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/consts/staking-router.ts b/consts/staking-router.ts
@@ -5,16 +5,17 @@ import invariant from 'tiny-invariant';
 import { PartialStakingRouterAbi__factory } from 'generated/factories/PartialStakingRouterAbi__factory';
 
 export const STAKING_ROUTER_BY_NETWORK: {
-  [key in CHAINS]?: string;
+  [key in CHAINS]?: string | null;
 } = {
   [CHAINS.Mainnet]: '0xFdDf38947aFB03C621C71b06C9C70bce73f12999',
   [CHAINS.Holesky]: '0xd6EbF043D30A7fe46D1Db32BA90a0A51207FE229',
   [CHAINS.Sepolia]: '0x4F36aAEb18Ab56A4e380241bea6ebF215b9cb12c',
+  [CHAINS.OptimismSepolia]: null,
 };
 
-export const getStakingRouterAddress = (chainId: CHAINS): string => {
+export const getStakingRouterAddress = (chainId: CHAINS): string | null => {
   const address = STAKING_ROUTER_BY_NETWORK[chainId];
-  invariant(address, 'chain is not supported');
+  invariant(address !== undefined, 'chain is not supported');
   return address;
 };
 

diff --git a/shared/hooks/use-balance.ts b/shared/hooks/use-balance.ts
@@ -17,6 +17,7 @@ import type { AbstractLidoSDKErc20 } from '@lidofinance/lido-ethereum-sdk/erc20'
 import type { GetBalanceData } from 'wagmi/query';
 import type { Address, WatchContractEventOnLogsFn } from 'viem';
 import { config } from 'config';
+import { CHAINS } from 'consts/chains';
 
 const nativeToBN = (data: bigint) => BigNumber.from(data.toString());
 
@@ -251,17 +252,19 @@ export const useStethBalance = ({
   account,
   shouldSubscribeToUpdates = true,
 }: UseBalanceProps = {}) => {
-  const { address } = useAccount();
+  const { address, chainId } = useAccount();
   const mergedAccount = account ?? address;
 
-  const { steth, core } = useLidoSDK();
+  const { steth, core, l2 } = useLidoSDK();
+
+  const isL2 = chainId === CHAINS.OPSepoliaTestnet;
 
   const { data: contract, isLoading } = useQuery({
     queryKey: ['steth-contract', core.chainId],
     enabled: !!mergedAccount,
 
     staleTime: Infinity,
-    queryFn: async () => steth.getContract(),
+    queryFn: async () => (isL2 ? l2.steth.getContract() : steth.getContract()),
   });
 
   const balanceData = useTokenBalance(
@@ -277,16 +280,19 @@ export const useWstethBalance = ({
   account,
   shouldSubscribeToUpdates = true,
 }: UseBalanceProps = {}) => {
-  const { address } = useAccount();
+  const { address, chainId } = useAccount();
   const mergedAccount = account ?? address;
 
-  const { wsteth, core } = useLidoSDK();
+  const { wsteth, core, l2 } = useLidoSDK();
+
+  const isL2 = chainId === CHAINS.OPSepoliaTestnet;
 
   const { data: contract, isLoading } = useQuery({
     queryKey: ['wsteth-contract', core.chainId],
     enabled: !!mergedAccount,
     staleTime: Infinity,
-    queryFn: async () => wsteth.getContract(),
+    queryFn: async () =>
+      isL2 ? l2.wsteth.getContract() : wsteth.getContract(),
   });
 
   const balanceData = useTokenBalance(