From 93a288a5fa08005767570597b34592db7cdc3949 Mon Sep 17 00:00:00 2001
From: Anton Shalimov <a.shalimof@yandex.ru>
Date: Fri, 20 Sep 2024 13:00:34 +0300
Subject: [PATCH] revert: make getTokenAddress as wrapper

---
 packages/constants/src/tokens.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/packages/constants/src/tokens.ts b/packages/constants/src/tokens.ts
index bc4362a..f600be5 100644
--- a/packages/constants/src/tokens.ts
+++ b/packages/constants/src/tokens.ts
@@ -49,7 +49,7 @@ export const L2_TOKENS_BY_NETWORK: {
   },
 };
 
-export const getTokenAddress = (chainId: CHAINS, token: TOKENS): string => {
+export const getL1TokenAddress = (chainId: CHAINS, token: TOKENS): string => {
   const tokens = TOKENS_BY_NETWORK[chainId];
   invariant(tokens, 'Chain is not supported');
 
@@ -68,3 +68,15 @@ export const getL2TokenAddress = (chainId: CHAINS, token: TOKENS): string => {
 
   return address;
 };
+
+export const getTokenAddress = (chainId: CHAINS, token: TOKENS): string => {
+  if (token === TOKENS.LDO) {
+    const _chainId =
+      chainId === CHAINS.OptimismSepolia ? CHAINS.Sepolia : chainId;
+    return getL1TokenAddress(_chainId, token);
+  } else {
+    return chainId === CHAINS.OptimismSepolia
+      ? getL2TokenAddress(chainId, token)
+      : getL1TokenAddress(chainId, token);
+  }
+};