From 27cc82fe5a67c26133857100eaae5df61da1a5e1 Mon Sep 17 00:00:00 2001
From: Nathan Richards <DNR500@users.noreply.github.com>
Date: Thu, 30 May 2024 14:48:10 +0200
Subject: [PATCH] chore: use find default token function in scripts

---
 examples/node/examples/bridge.ts                    | 2 +-
 examples/node/examples/constants.ts                 | 1 -
 examples/node/examples/polynomialDeposit.ts         | 9 ++++++---
 examples/node/examples/utils/checkTokenAllowance.ts | 3 ++-
 4 files changed, 9 insertions(+), 6 deletions(-)
 delete mode 100644 examples/node/examples/constants.ts

diff --git a/examples/node/examples/bridge.ts b/examples/node/examples/bridge.ts
index c5f3e546..a9dddd39 100644
--- a/examples/node/examples/bridge.ts
+++ b/examples/node/examples/bridge.ts
@@ -7,7 +7,7 @@ import {
   createConfig,
   EVM,
 } from '@lifi/sdk'
-import { Address, Chain, parseUnits } from 'viem'
+import type { Address, Chain } from 'viem'
 import { createWalletClient, http } from 'viem'
 import { privateKeyToAccount } from 'viem/accounts'
 import { mainnet, arbitrum, optimism, polygon } from 'viem/chains'
diff --git a/examples/node/examples/constants.ts b/examples/node/examples/constants.ts
deleted file mode 100644
index 970de9a8..00000000
--- a/examples/node/examples/constants.ts
+++ /dev/null
@@ -1 +0,0 @@
-export const AddressZero = '0x0000000000000000000000000000000000000000'
diff --git a/examples/node/examples/polynomialDeposit.ts b/examples/node/examples/polynomialDeposit.ts
index 359d2a51..f5290c18 100644
--- a/examples/node/examples/polynomialDeposit.ts
+++ b/examples/node/examples/polynomialDeposit.ts
@@ -1,6 +1,8 @@
-import { ContractCallsQuoteRequest, StatusResponse } from '@lifi/sdk'
+import * as lifiDataTypes from '@lifi/data-types'
+import type { ContractCallsQuoteRequest, StatusResponse } from '@lifi/sdk'
 import {
   ChainId,
+  CoinKey,
   createConfig,
   EVM,
   getContractCallsQuote,
@@ -19,10 +21,11 @@ import { privateKeyToAccount } from 'viem/accounts'
 import { mainnet, arbitrum, optimism, polygon } from 'viem/chains'
 import 'dotenv/config'
 import { promptConfirm } from '../helpers/promptConfirm'
-import { AddressZero } from './constants'
 import { checkTokenAllowance } from './utils/checkTokenAllowance'
 import { transformTxRequestToSendTxParams } from './utils/transformTxRequestToSendTxParams'
 
+const { findDefaultToken } = (lifiDataTypes as any).default
+
 const run = async () => {
   console.info('>> Starting Polynomial Demo: Deposit sETH on Optimism')
   console.info('>> Initialize LiFi SDK')
@@ -66,7 +69,7 @@ const run = async () => {
     // config for polynomial deposit run
     const config = {
       fromChain: ChainId.ETH,
-      fromToken: AddressZero,
+      fromToken: findDefaultToken(CoinKey.ETH, ChainId.ETH).address,
       amount: parseEther('0.00001').toString(),
       polynomialContractAddress: '0x2D46292cbB3C601c6e2c74C32df3A4FCe99b59C7', // Polynomial Ethereum Contract on Optimism
       polynomialContractToken: '0xE405de8F52ba7559f9df3C368500B6E6ae6Cee49', // sETH on Optimism
diff --git a/examples/node/examples/utils/checkTokenAllowance.ts b/examples/node/examples/utils/checkTokenAllowance.ts
index 9e1e41fb..0d9be229 100644
--- a/examples/node/examples/utils/checkTokenAllowance.ts
+++ b/examples/node/examples/utils/checkTokenAllowance.ts
@@ -1,7 +1,8 @@
 import type { LiFiStep } from '@lifi/sdk'
 import { getTokenAllowance, setTokenAllowance } from '@lifi/sdk'
 import type { PrivateKeyAccount, PublicClient, WalletClient } from 'viem'
-import { AddressZero } from '../constants'
+
+const AddressZero = '0x0000000000000000000000000000000000000000'
 
 export const checkTokenAllowance = async (
   contactCallsQuoteResponse: LiFiStep,