feat: support token list logoUri

dinngo · Jun 6, 2024 · 40c1d04 · 40c1d04
1 parent edcbdcb
commit 40c1d04
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 18 deletions.
diff --git a/.changeset/four-insects-destroy.md b/.changeset/four-insects-destroy.md
@@ -0,0 +1,5 @@
+---
+'@protocolink/logics': patch
+---
+
+support token list logoUri
diff --git a/src/logics/balancer-v2/logic.flash-loan.ts b/src/logics/balancer-v2/logic.flash-loan.ts
@@ -39,9 +39,9 @@ export class FlashLoanLogic extends core.Logic implements core.LogicTokenListInt
 
     const tmp: Record<string, boolean> = {};
     const tokenList: FlashLoanLogicTokenList = [];
-    for (const { chainId, address, decimals, symbol, name } of data.tokens) {
+    for (const { chainId, address, decimals, symbol, name, logoURI } of data.tokens) {
       if (tmp[address] || chainId !== this.chainId || !name || !symbol || !decimals) continue;
-      tokenList.push(new common.Token(chainId, address, decimals, symbol, name));
+      tokenList.push(new common.Token(chainId, address, decimals, symbol, name, logoURI));
       tmp[address] = true;
     }
 

diff --git a/src/logics/openocean-v2/logic.swap-token.ts b/src/logics/openocean-v2/logic.swap-token.ts
@@ -32,8 +32,8 @@ export class SwapTokenLogic
     const resp = await axios.get(url + '/tokenList');
     const tokens = resp.data.data;
     const tokenList: SwapTokenLogicTokenList = [];
-    for (const { address, decimals, symbol, name } of tokens) {
-      tokenList.push(new common.Token(this.chainId, address, decimals, symbol, name));
+    for (const { address, decimals, symbol, name, icon } of tokens) {
+      tokenList.push(new common.Token(this.chainId, address, decimals, symbol, name, icon));
     }
     return tokenList;
   }

diff --git a/src/logics/uniswap-v3/logic.swap-token.ts b/src/logics/uniswap-v3/logic.swap-token.ts
@@ -31,9 +31,9 @@ export class SwapTokenLogic
 
     const tmp: Record<string, boolean> = { [this.nativeToken.address]: true };
     const tokenList: SwapTokenLogicTokenList = [this.nativeToken];
-    for (const { chainId, address, decimals, symbol, name } of data.tokens) {
+    for (const { chainId, address, decimals, symbol, name, logoURI } of data.tokens) {
       if (tmp[address] || chainId !== this.chainId) continue;
-      tokenList.push(new common.Token(chainId, address, decimals, symbol, name));
+      tokenList.push(new common.Token(chainId, address, decimals, symbol, name, logoURI));
       tmp[address] = true;
     }
 

diff --git a/src/utils/tokens.ts b/src/utils/tokens.ts
@@ -4,28 +4,28 @@ import * as common from '@protocolink/common';
 import { ethers } from 'ethers';
 
 export async function get1InchTokens(chainId: number) {
-  const { data } = await axios.get<Record<string, { symbol: string; name: string; decimals: number; address: string }>>(
-    `https://tokens.1inch.io/v1.2/${chainId}`
-  );
+  const { data } = await axios.get<
+    Record<string, { symbol: string; name: string; decimals: number; address: string; logoURI: string }>
+  >(`https://tokens.1inch.io/v1.2/${chainId}`);
 
   const nativeToken = common.getNativeToken(chainId);
   const elasticAddress = common.ELASTIC_ADDRESS.toLowerCase();
-  const tokens = Object.values(data).map(({ address, decimals, symbol, name }) =>
-    address === elasticAddress ? nativeToken : new common.Token(chainId, address, decimals, symbol, name)
+  const tokens = Object.values(data).map(({ address, decimals, symbol, name, logoURI }) =>
+    address === elasticAddress ? nativeToken : new common.Token(chainId, address, decimals, symbol, name, logoURI)
   );
 
   return tokens;
 }
 
 export async function getMetisTokens() {
   const chainId = common.ChainId.metis;
-  const { data } = await axios.get<{ tokens: { symbol: string; name: string; decimals: number; address: string }[] }>(
-    `https://tokens.coingecko.com/metis-andromeda/all.json`
-  );
+  const { data } = await axios.get<{
+    tokens: { symbol: string; name: string; decimals: number; address: string; logoURI: string }[];
+  }>(`https://tokens.coingecko.com/metis-andromeda/all.json`);
 
   const tokens = [common.getNativeToken(chainId)];
-  for (const { address, decimals, symbol, name } of data.tokens) {
-    tokens.push(new common.Token(chainId, address, decimals, symbol, name));
+  for (const { address, decimals, symbol, name, logoURI } of data.tokens) {
+    tokens.push(new common.Token(chainId, address, decimals, symbol, name, logoURI));
   }
 
   return tokens;
@@ -93,7 +93,7 @@ export async function getTokenList(
   const tmp: Record<string, boolean> = defaultTokenList.reduce((acc, token) => ({ [token.address]: true }), {});
   const tokenList: common.Token[] = [...defaultTokenList];
   for (const { tokens } of tokenLists) {
-    for (const { chainId, address, decimals, symbol, name } of tokens) {
+    for (const { chainId, address, decimals, symbol, name, logoURI } of tokens) {
       const lowerCaseAddress = address.toLowerCase();
 
       if (
@@ -105,7 +105,7 @@ export async function getTokenList(
         !ethers.utils.isAddress(address)
       )
         continue;
-      tokenList.push(new common.Token(chainId, address, decimals, symbol, name));
+      tokenList.push(new common.Token(chainId, address, decimals, symbol, name, logoURI));
       tmp[lowerCaseAddress] = true;
     }
   }