feat(assets): use yearn-assets for zapper tokens (#71)

yearn · Jul 15, 2021 · 6099cf9 · 6099cf9
1 parent 8551dbd
commit 6099cf9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/interfaces/simulation.ts b/src/interfaces/simulation.ts
@@ -10,7 +10,7 @@ import { EthAddress, WethAddress, ZeroAddress } from "../helpers";
 import { PickleJars } from "../services/partners/pickle";
 import { Address, Integer, SdkError, ZapApprovalTransactionOutput, ZapProtocol } from "../types";
 import { TransactionOutcome } from "../types/custom/simulation";
-import { PickleJarContract, VaultContract, YearnVaultContract } from "../vaultContract";
+import { PickleJarContract, VaultContract, YearnVaultContract } from "../vault";
 
 const baseUrl = "https://simulate.yearn.network";
 const latestBlockKey = -1;

diff --git a/src/interfaces/token.ts b/src/interfaces/token.ts
@@ -68,7 +68,12 @@ export class TokenInterface<C extends ChainId> extends ServiceInterface<C> {
   async supported(): Promise<Token[]> {
     if (this.chainId === 1 || this.chainId === 1337) {
       // only ETH Main is supported
-      return await this.yearn.services.zapper.supportedTokens();
+      const tokens = await this.yearn.services.zapper.supportedTokens();
+      const icons = this.yearn.services.asset.icon(tokens.map(token => token.address));
+      return tokens.map(token => {
+        const icon = icons[token.address];
+        return icon ? { ...token, icon } : token;
+      });
     }
     return [];
   }

diff --git a/src/vaultContract.ts → src/vault.ts b/src/vaultContract.ts → src/vault.ts