update project equilibria adapter (DefiLlama#13018)

Author: yulodl

projects/equilibria/index.js

@@ -1,114 +1,113 @@
-const sdk = require("@defillama/sdk");
-const { sumTokensExport } = require("../helper/unwrapLPs");
-
-const contracts = {
+const config = {
   ethereum: {
-    boosterAddress: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
-    staker: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
-    eqbAddress: "0xfE80D611c6403f70e5B1b9B722D2B3510B740B2B",
-    eqbRewardsAddress: "0xd8967B2B15b3CDF96039b7407813B4037f73ec27",
-    pendleAddress: "0x808507121b80c02388fad14726482e061b8da827",
+    pendleBooster: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
+    pendleProxy: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
+    eqb: "0xfE80D611c6403f70e5B1b9B722D2B3510B740B2B",
+    vlEqb: "0xd8967B2B15b3CDF96039b7407813B4037f73ec27",
+    pendle: "0x808507121b80c02388fad14726482e061b8da827",
+    ePendle: "0x22Fc5A29bd3d6CCe19a06f844019fd506fCe4455",
+    ePendleReward: "0x357F55b46821A6C6e476CC32EBB2674cD125e849",
   },
   arbitrum: {
-    boosterAddress: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
-    staker: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
-    eqbRewardsAddress: "0x70f61901658aAFB7aE57dA0C30695cE4417e72b9",
-    pendleAddress: "0x0c880f6761F1af8d9Aa9C466984b80DAb9a8c9e8",
+    pendleBooster: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
+    pendleProxy: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
+    eqb: "0xBfbCFe8873fE28Dfa25f1099282b088D52bbAD9C",
+    vlEqb: "0x70f61901658aAFB7aE57dA0C30695cE4417e72b9",
+    pendle: "0x0c880f6761F1af8d9Aa9C466984b80DAb9a8c9e8",
+    ePendle: "0x3EaBE18eAE267D1B57f917aBa085bb5906114600",
+    ePendleReward: "0x9739d1E515C5291faA26D92a5D02761b6BbB4D6F",
   },
   bsc: {
-    boosterAddress: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
-    staker: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
-    eqbRewardsAddress: "0x0140dE476f49B6B42f7b73612b6dc317aB91D3BC",
-    pendleAddress: "0xb3Ed0A426155B79B898849803E3B36552f7ED507",
+    pendleBooster: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
+    pendleProxy: "0x64627901dAdb46eD7f275fD4FC87d086cfF1e6E3",
+    eqb: "0x374Ca32fd7934c5d43240E1e73fa9B2283468609",
+    vlEqb: "0x0140dE476f49B6B42f7b73612b6dc317aB91D3BC",
+    pendle: "0xb3Ed0A426155B79B898849803E3B36552f7ED507",
+    ePendle: "0x898CA9B3ef8b6a30dA5fc7202f70E7992b3602B3",
+    ePendleReward: "0xE2dB20ce7D845f99338BbA4bdFF00e733801Dde7",
   },
   optimism: {
-    boosterAddress: "0x18C61629E6CBAdB85c29ba7993f251b3EbE2B356",
-    staker: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
-    eqbRewardsAddress: "0x22Fc5A29bd3d6CCe19a06f844019fd506fCe4455",
-    pendleAddress: "0xBC7B1Ff1c6989f006a1185318eD4E7b5796e66E1",
+    pendleBooster: "0x18C61629E6CBAdB85c29ba7993f251b3EbE2B356",
+    pendleProxy: "0x4D32C8Ff2fACC771eC7Efc70d6A8468bC30C26bF",
+    eqb: "0xaf3A6f67Af1624d3878A8d30b09FAe7915DcA2a0",
+    vlEqb: "0x22Fc5A29bd3d6CCe19a06f844019fd506fCe4455",
+    pendle: "0xBC7B1Ff1c6989f006a1185318eD4E7b5796e66E1",
+    ePendle: "0x86a20111fEae36f3511A30c0640d2099b3A818C5",
+    ePendleReward: "0x898CA9B3ef8b6a30dA5fc7202f70E7992b3602B3",
   },
   mantle: {
-    boosterAddress: "0x920873E5b302A619C54c908aDFB77a1C4256A3B8",
-    staker: "0x479603DE0a8B6D2f4D4eaA1058Eea0d7Ac9E218d",
-    eqbRewardsAddress: "0x71e0ce200a10f0bBFB9F924fE466ACf0B7401EbF",
-    pendleAddress: "0xd27B18915e7acc8FD6Ac75DB6766a80f8D2f5729",
+    pendleBooster: "0x920873E5b302A619C54c908aDFB77a1C4256A3B8",
+    pendleProxy: "0x479603DE0a8B6D2f4D4eaA1058Eea0d7Ac9E218d",
+    eqb: "0x3e7eF8f50246f725885102E8238CBba33F276747",
+    vlEqb: "0x71e0ce200a10f0bBFB9F924fE466ACf0B7401EbF",
+    pendle: "0xd27B18915e7acc8FD6Ac75DB6766a80f8D2f5729",
+    ePendle: "0x6EE066b813cd8C2586D86F79eD0025e81801b923",
+    ePendleReward: "0x741620136cf08a782c1Df1Fc9E3cAA760Cc4Fecc",
+  },
+  base: {
+    pendleBooster: "0x2583A2538272f31e9A15dD12A432B8C96Ab4821d",
+    pendleProxy: "0x920873E5b302A619C54c908aDFB77a1C4256A3B8",
+    eqb: "0x010cd9b9be7E416E07793dc6Ce2F45868A80a50a",
+    vlEqb: "0x70f61901658aAFB7aE57dA0C30695cE4417e72b9",
+    pendle: "0xA99F6e6785Da0F5d6fB42495Fe424BCE029Eeb3E",
+    ePendle: "0x741620136cf08a782c1Df1Fc9E3cAA760Cc4Fecc",
+    ePendleReward: "0xB2D167EB1Fff17EbD727B0a21e7f5F50424264c8",
   }
 };
 
-async function tvl(chain, block) {
-  const balances = {};
+const abi = {
+    "poolLength": "uint256:poolLength",
+    "poolInfo": "function poolInfo(uint256) view returns (address market, address token, address rewardPool, bool shutdown)"
+};
+
+const { staking } = require('../helper/staking')
+
+async function tvl(api) {
+  const { pendleBooster, vlEqb, pendleProxy, pendle } = config[api.chain];
 
-  if (chain === "ethereum") {
-    const pendleLocked = await sdk.api2.abi.call({
+  if (api.chain === "ethereum") {
+    const pendleLocked = await api.call({
       target: "0x4f30A9D41B80ecC5B94306AB4364951AE3170210",
-      params: contracts[chain].staker,
+      params: pendleProxy,
       abi: "erc20:balanceOf",
-      block,
     });
-    sdk.util.sumSingleBalance(
-      balances,
-      contracts[chain].pendleAddress,
+    api.add(
+      pendle,
       pendleLocked
     );
   }
 
-  const poolLength = await sdk.api2.abi.call({
-    chain,
-    block,
-    target: contracts[chain].boosterAddress,
-    abi: "function poolLength() view returns (uint256)",
+  const poolInfos = await api.fetchList({
+    lengthAbi: abi.poolLength,
+    itemAbi: abi.poolInfo,
+    target: pendleBooster,
   });
-
-  const poolInfos = await Promise.all(
-    Array.from(Array(Number(poolLength)).keys()).map((params) =>
-      sdk.api2.abi.call({
-        chain,
-        block,
-        target: contracts[chain].boosterAddress,
-        params,
-        abi: "function poolInfo(uint256) view returns (address market, address token, address rewardPool, bool shutdown)",
-      })
-    )
-  ); // multicall fails here
-
-  const { output: gaugeBalances } = await sdk.api.abi.multiCall({
-    abi: "erc20:balanceOf",
-    calls: Array.from(new Set(poolInfos.map((p) => p.market))).map((i) => ({
-      target: i,
-      params: contracts[chain].staker,
-    })),
-    chain,
-    block,
-  });
-
-  gaugeBalances.forEach(({ output, input }, i) =>
-    sdk.util.sumSingleBalance(
-      balances,
-      chain +
-        ":" +
-        poolInfos.find(
-          (p) => p.market.toLowerCase() === input.target.toLowerCase()
-        ).market,
-      output
-    )
-  );
-
-  return balances;
+  const poolTokens = poolInfos.map(pool => pool.market);
+  const blacklistedTokens = [];
+  if (vlEqb) blacklistedTokens.push(vlEqb);
+  return api.sumTokens({ tokens: poolTokens, owner: pendleProxy, blacklistedTokens });
 }
 
-const chains = ["ethereum", "arbitrum", 'bsc', 'optimism', 'mantle'];
-
-module.exports = {
-  doublecounted: true,
-};
+Object.keys(config).forEach((chain) => {
+  const { eqb, vlEqb, ePendle, ePendleReward } = config[chain];
 
-chains.forEach((chain) => {
   module.exports[chain] = {
-    tvl: async (_, _b, { [chain]: block }) => tvl(chain, block),
+    tvl,
   };
-});
 
-module.exports.ethereum.staking = sumTokensExport({
-  owner: contracts.ethereum.eqbRewardsAddress,
-  tokens: [contracts.ethereum.eqbAddress],
+  const stakingContracts = [];
+  const stakingTokens= [];
+  if (eqb && vlEqb) {
+    stakingContracts.push(vlEqb);
+    stakingTokens.push(eqb);
+  }
+
+  if (ePendle && ePendleReward) {
+    stakingContracts.push(ePendleReward);
+    stakingTokens.push(ePendle);
+  }
+
+  if (stakingContracts.length) module.exports[chain].staking = staking(stakingContracts, stakingTokens);
 });
+
+module.exports.doublecounted = true;