assets page: fetch native eth balance on l2

renegade-fi · Oct 18, 2024 · 56c8988 · 56c8988
1 parent 88be534
commit 56c8988
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/app/api/tokens/get-combined-balances/route.ts b/app/api/tokens/get-combined-balances/route.ts
@@ -47,9 +47,13 @@ export async function GET(req: Request) {
   )
 
   // Add native ETH balance to WETH
-  const ethBalance = await readEthBalance(process.env.RPC_URL_MAINNET!, address)
+  const ethBalanceL1 = await readEthBalance(
+    process.env.RPC_URL_MAINNET!,
+    address,
+  )
+  const ethBalanceL2 = await readEthBalance(process.env.RPC_URL!, address)
   const currentWethBalance = combinedBalances.get("WETH") || BigInt(0)
-  combinedBalances.set("WETH", currentWethBalance + ethBalance)
+  combinedBalances.set("WETH", currentWethBalance + ethBalanceL1 + ethBalanceL2)
 
   // Add USDC.e balance to USDC
   const usdceBalance = await readErc20BalanceOf(

diff --git a/hooks/use-combined-balances.ts b/hooks/use-combined-balances.ts
@@ -1,9 +1,12 @@
 import { Token } from "@renegade-fi/react"
 import { useQuery } from "@tanstack/react-query"
 
-async function fetchCombinedBalances(address: string) {
+async function fetchCombinedBalances(address: `0x${string}`) {
   const response = await fetch(
     `/api/tokens/get-combined-balances?address=${address}`,
+    {
+      cache: "no-store",
+    },
   )
   if (!response.ok) {
     throw new Error("Failed to fetch combined balances")