code refactor

Author: g1nt0ki

projects/blueshift/index.js

@@ -1,6 +1,6 @@
 const sdk = require('@defillama/sdk');
 const { default: BigNumber } = require('bignumber.js');
-const { transformMilkomedaAddress } = require('../helper/portedTokens');
+const { getChainTransform } = require('../helper/portedTokens');
 
 const abi = require('./abi.json');
 
@@ -13,7 +13,7 @@ const BLUESHIFT_INDEX_PORTFOLIO = '0xB2A76Ce2D5eD32aD7F8B93a1098C1Fee473e27bA';
 
 async function staking(timestamp, block, chainBlocks) {
   const balances = {};
-  const transform = await transformMilkomedaAddress();
+  const transform = await getChainTransform('milkomeda');
 
   const value = (await sdk.api.abi.call({
     abi: abi.BlueshiftEarning.getAccDeposit,
@@ -62,7 +62,7 @@ async function staking(timestamp, block, chainBlocks) {
 
 async function tvl(timestamp, block, chainBlocks) {
   const balances = {};
-  const transform = await transformMilkomedaAddress();
+  const transform = await getChainTransform('milkomeda');
 
   const portfolios = (await sdk.api.abi.call({
     abi: abi.BlueshiftRegistry.getPortfolios,

projects/earnifty/index.js

@@ -1,13 +1,13 @@
 const sdk = require('@defillama/sdk');
 const vaultContractABI = require("./abis/vault.json");
-const { transformVelasAddress } = require('../helper/portedTokens');
+const { getChainTransform } = require('../helper/portedTokens');
 const { unwrapUniswapLPs } = require('../helper/unwrapLPs');
 const vaults_info = require('./vaults_info.json');
 
 
 async function tvl(timestamp, block, chainBlocks) {
     const balances = {};
-    const transform = await transformVelasAddress();
+    const transform = await getChainTransform('velas');
 
     for (const info of vaults_info) {
         const balance = ((await sdk.api.abi.call({

projects/elephantdex/index.js

@@ -1,27 +1,5 @@
-const { getBlock } = require("../helper/getBlock");
-const { transformHarmonyAddress } = require("../helper/portedTokens");
-const { calculateUniTvl } = require("../helper/calculateUniTvl");
 const { getUniTVL } = require("../helper/unknownTokens");
 
-const wONE = "bsc:0xdE976f3344cd9F06E451aF3A94a324afC3E154F4";
-async function tvl(timestamp, block, chainBlocks) {
-  block = await getBlock(timestamp, "harmony", chainBlocks);
-  const transform = await transformHarmonyAddress();
-
-  let balances = await calculateUniTvl(
-    transform,
-    block,
-    "harmony",
-    "0x0Dea90EC11032615E027664D2708BC292Bbd976B",
-    18105518,
-    true
-  );
-
-  balances["harmony"] = balances[wONE] / 10 ** 18;
-  delete balances[wONE];
-  return balances;
-}
-
 module.exports = {
   harmony: {
     tvl: getUniTVL({

projects/helper/portedTokens.js

@@ -1,45 +1,9 @@
 const { aaveChainTvl } = require("../helper/aave");
-const { getFixBalances } = require("../helper/portedTokens");
-
-async function transformKlaytnAddress() {
-  const mapping = {
-    "0x5388ce775de8f7a69d17fd5caa9f7dbfee65dfce":
-      "0x4576E6825B462b6916D2a41E187626E9090A92c6", // Donkey
-    "0x9eaefb09fe4aabfbe6b1ca316a3c36afc83a393f": "ripple", // XRP
-    "0x02cbe46fb8a1f579254a9b485788f2d86cad51aa":
-      "0x26fb86579e371c7aedc461b2ddef0a8628c93d3b", // bora
-    "0x078dB7827a5531359f6CB63f62CFA20183c4F10c":
-      "0x6b175474e89094c44da98b954eedeac495271d0f", // dai
-    "0x6270B58BE569a7c0b8f47594F191631Ae5b2C86C":
-      "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48", // USDC
-    "0x0268dbed3832b87582b1fa508acf5958cbb1cd74":
-      "bsc:0xf258f061ae2d68d023ea6e7cceef97962785c6c1", // IJM
-    "0xd6dAb4CfF47dF175349e6e7eE2BF7c40Bb8C05A3":
-      "0xdac17f958d2ee523a2206206994597c13d831ec7", // USDT
-    "0x168439b5eebe8c83db9eef44a0d76c6f54767ae4":
-      "0x6b175474e89094c44da98b954eedeac495271d0f", // pUSD
-    "0x4fa62f1f404188ce860c8f0041d6ac3765a72e67":
-      "0x6b175474e89094c44da98b954eedeac495271d0f", // KSD
-    "0xce40569d65106c32550626822b91565643c07823":
-      "0x6b175474e89094c44da98b954eedeac495271d0f", // KASH
-    "0x210bc03f49052169d5588a52c317f71cf2078b85":
-      "bsc:0xe9e7cea3dedca5984780bafc599bd69add087d56", // kBUSD
-    "0xDCbacF3f7a069922E677912998c8d57423C37dfA":
-      "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599", // WBTC
-    "0xCD6f29dC9Ca217d0973d3D21bF58eDd3CA871a86":
-      "0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2", // WETH
-    "0xe4f05A66Ec68B54A58B17c22107b02e0232cC817": "0xe4f05a66ec68b54a58b17c22107b02e0232cc817" // Klaytn
-  };
-  return addr => {
-    const mappingResult = mapping[addr]
-    addr = addr.toLowerCase();
-    return mappingResult || `klaytn:${addr}`;
-  };
-}
+const { getFixBalances, getChainTransform } = require("../helper/portedTokens");
 
 function lending(borrowed) {
   return async (timestamp, ethBlock, chainBlocks) => {
-    const transform = await transformKlaytnAddress();
+    const transform = await getChainTransform('klaytn');
     const balances = await aaveChainTvl(
       "klaytn",
       "0x969E4A05c2F3F3029048e7943274eC2E762497AB",

projects/helper/tokenMapping.js

@@ -1,7 +1,7 @@
 const { tombTvl } = require("./helper/tomb");
 const sdk = require("@defillama/sdk");
-const { transformCronosAddress } = require("./helper/portedTokens");
-const { unwrapUniswapLPs } = require('./helper/unwrapLPs');
+const { getChainTransform } = require("./helper/portedTokens");
+const { unwrapUniswapLPs, unwrapLPsAuto } = require('./helper/unwrapLPs');
 
 const PES = "0x8efbaa6080412d7832025b03b9239d0be1e2aa3b";
 const SPES = "0xBBd4650EeA85f9DBd83d6Fb2a6E8B3d8f32FE1C5";
@@ -38,7 +38,7 @@ function addToExports(chain, key, fn) {
 
 async function tvl(timestamp, block, chainBlocks) {
     const balances = {};
-    const transform = await transformCronosAddress();
+    const transform = await getChainTransform('cronos');
     const result = await sdk.api.abi.multiCall({
         calls: genesisTokens.map((t) => ({
           target: t,
@@ -64,7 +64,7 @@ async function tvl(timestamp, block, chainBlocks) {
         );
 
     delete balances[`cronos:${PES}`];
-
+    await unwrapLPsAuto({ balances, block: chainBlocks.cronos, chain: 'cronos',  })
     return balances; 
 };
 

projects/klap/index.js

@@ -1,7 +1,7 @@
 const { sumTokens, } = require('../helper/unwrapLPs')
 const axios = require('axios')
 const { staking } = require('../helper/staking');
-const { transformBscAddress, transformEthereumAddress, transformPolygonAddress, transformArbitrumAddress } = require("../helper/portedTokens");
+const { getChainTransform } = require("../helper/portedTokens");
 
 // token list
 const tokenListsApiEndpoint = "https://token-list.solv.finance/vouchers-prod.json"
@@ -13,22 +13,22 @@ const solvBscTokenAddress = '0xC073c4eD65622A9423b5e5BDe2BFC8B81EBC471c'
 const solvBscPoolAddress = '0xE5742912EDb4599779ACC1CE2acB6a06E01f1089'
 
 const ethereumTVL = async (timestamp, block, chainBlocks) => {
-    const transform = await transformEthereumAddress();
+    const transform = await getChainTransform('ethereum');
     return tvl(timestamp, block, chainBlocks, "ethereum", 1, transform);
 }
 
 const polygonTVL = async (timestamp, block, chainBlocks) => {
-    const transform = await transformPolygonAddress();
+    const transform = await getChainTransform('polygon');
     return tvl(timestamp, block, chainBlocks, "polygon", 137, transform);
 }
 
 const arbitrumTVL = async (timestamp, block, chainBlocks) => {
-    const transform = await transformArbitrumAddress();
+    const transform = await getChainTransform('arbitrum');
     return tvl(timestamp, block, chainBlocks, "arbitrum", 42161, transform);
 };
 
 const bscTVL = async (timestamp, block, chainBlocks) => {
-    const transform = await transformBscAddress();
+    const transform = await getChainTransform('bsc');
     return tvl(timestamp, block, chainBlocks, "bsc", 56, transform);
 };
 

projects/pegasusdao.js

@@ -1,16 +1,10 @@
 const sdk = require('@defillama/sdk');
 const {
-  transformBscAddress,
-  transformAvaxAddress,
-  transformPolygonAddress,
-  transformBobaAddress,
-  transformAuroraAddress,
-  transformTelosAddress
+  getChainTransform,
 } = require('../helper/portedTokens');
 
 const getTokenAbi = require("./abi/getToken.json");
 const getTokenBalanceAbi = require("./abi/getTokenBalance.json");
-const BigNumber = require("bignumber.js");
 
 const config = {
   chains: [
@@ -84,23 +78,7 @@ const config = {
 }
 
 async function getTransform(chainName) {
-  let transform = (address) => {
-    return address
-  }
-  if (chainName === 'bsc') {
-    transform = await transformBscAddress();
-  } else if (chainName === 'avax') {
-    transform = await transformAvaxAddress();
-  } else if (chainName === 'polygon') {
-    transform = await transformPolygonAddress();
-  } else if (chainName === 'boba') {
-    transform = await transformBobaAddress();
-  } else if (chainName === 'aurora') {
-    transform = await transformAuroraAddress();
-  } else if (chainName === 'telos') {
-    transform = await transformTelosAddress();
-  }
-  return transform
+  return getChainTransform(chainName)
 }
 
 async function tvl(chainName, timestamp, block, chainBlocks) {

projects/solv-protocol/index.js

@@ -1,6 +1,6 @@
 const { sumSingleBalance } = require('@defillama/sdk/build/generalUtil')
 const { get } = require('../helper/http')
-const { wavesMapping } = require('../helper/portedTokens')
+const { fixBalancesTokens: { waves: wavesMapping } } = require('../helper/tokenMapping')
 
 
 const endpoint = "https://waves.exchange/api/v1/liquidity_pools/stats"