From 8a13b28c371e0eccf3aabe744f5fbbd815f84c1b Mon Sep 17 00:00:00 2001
From: ByeongSu Hong <hong@byeongsu.dev>
Date: Tue, 6 Aug 2024 22:12:28 +0900
Subject: [PATCH] feat: Add uniETH support for Mitosis & and some code cleanup
 (#11142)

* support uniETH

* pass test

* lesgoh

* fix scroll issue
---
 projects/mitosis/index.js | 151 ++++++++++++++++++++++++++++----------
 1 file changed, 111 insertions(+), 40 deletions(-)

diff --git a/projects/mitosis/index.js b/projects/mitosis/index.js
index e0b048e7703a..a2fcf67df27a 100644
--- a/projects/mitosis/index.js
+++ b/projects/mitosis/index.js
@@ -2,6 +2,25 @@ const ADDRESSES = require('../helper/coreAssets.json');
 const { sumTokens2 } = require('../helper/unwrapLPs');
 const sdk = require('@defillama/sdk');
 
+const UNIETH_ADDRESS = {
+  ethereum: { 
+    asset: "0xf1376bcef0f78459c0ed0ba5ddce976f1ddf51f4", 
+    vault: "0x02Ff1F648Ff443B5d88214341F0acE6ECFb94cF3",
+  },
+  arbitrum: { 
+    asset: "0x3d15fd46ce9e551498328b1c83071d9509e2c3a0", 
+    vault: "0x7E8cffBe165c6905a8AceC0f37B341c00353e8BA",
+  },
+  scroll: { 
+    asset: "0x15eefe5b297136b8712291b632404b66a8ef4d25", 
+    vault: "0xA0EeB418213f8472cba2c842378E1bB64e28bd28",
+  },
+  linea: { 
+    asset: "0x15eefe5b297136b8712291b632404b66a8ef4d25", 
+    vault: "0x56ceD49205e5D9b4d8D9B29f4aBfbe7bb8b08768",
+  },
+};
+
 const WEETH_ADDRESS = {
   ethereum: "0xCd5fE23C85820F7B72D0926FC9b05b43E359b7ee",
   arbitrum: "0x35751007a407ca6FEFfE80b3cB397736D2cf4dbe",
@@ -14,14 +33,18 @@ const WEETH_ADDRESS = {
 };
 
 const VAULT_weETHs = {
-  ethereum: "0x917ceE801a67f933F2e6b33fC0cD1ED2d5909D88",
+  ethereum: "0x917cee801a67f933f2e6b33fc0cd1ed2d5909d88",
 };
 
 const CAP_ADDRESS = {
-  scroll: "0x4a6219E25a41FD4165fbd158D89723a7175EA382",
   ethereum: "0x451d791b6e9a9b8c9237bb55e58a7757342b16f9",
+  arbitrum: "0xb883ee478d3b7fea8a5357a3c3e27e2d2292b1d2",
+  optimism: "0xb883ee478d3b7fea8a5357a3c3e27e2d2292b1d2",
+  mode: "0xb883ee478d3b7fea8a5357a3c3e27e2d2292b1d2",
+  manta: "0xb883ee478d3b7fea8a5357a3c3e27e2d2292b1d2",
   blast: "0x096430ef0a653c067df32e93ff77090e084169de",
   linea: "0xcd32876b9b483eb75e8ca74935e4b51725f33a91",
+  scroll: "0xcd32876b9B483eb75e8ca74935E4b51725F33A91",
 };
 
 const vaults = {
@@ -29,60 +52,108 @@ const vaults = {
   ethereum: '0xE4cf2D4eb9c01784798679F2FED4CF47cc59a3ec'
 };
 
-const ethCapTVL = async (api) => {
-  const ethcap = CAP_ADDRESS.ethereum;
-  const tvl = await api.call({ abi: "uint256:load", target: ethcap });
-  
-  const balance = {};
-  balance[WEETH_ADDRESS.ethereum] = tvl;
-
-  return balance;
-};
-
-const ethVaultTVL = async (api) => {
-  const ethvault = vaults.ethereum;
-  return sumTokens2({ api, owner: ethvault, tokens: [VAULT_weETHs.ethereum] });
-};
-
-const scrollVaultTVL = async (api) => {
-  const vault = vaults.scroll;
-  return sumTokens2({ api, owner: vault, tokens: [WEETH_ADDRESS.scroll] });
-};
-
-const capTVL = (chain) => async (api) => {
-  const cap = CAP_ADDRESS[chain] ?? "0xb883ee478d3b7fea8a5357a3c3e27e2d2292b1d2";
-  const tvl = await api.call({ abi: "uint256:load", target: cap });
-  api.add(WEETH_ADDRESS[chain], tvl);
-};
-
+/**
+ * @param {
+ *  (
+ *    {
+ *      type: 'cap';
+ *      data: {
+ *        asset: string;
+ *        cap: string;
+ *      }
+ *    } | {
+ *      type: 'vault'; 
+ *      data: {
+ *        asset: string; 
+ *        cap: string;
+ *      }
+ *    }
+ *  )[]} inputs
+ */
+const chainTVL = (chain, inputs) => async (api) => {
+  for (const { type, data } of inputs) {
+    if (type === 'cap') {
+      const tvl = await api.call({ abi: "uint256:load", target: data.cap });
+      console.log(chain, type, data.asset, tvl);
+      api.add(data.asset, tvl);
+    } else if (type === 'vault') {
+      const tvl = await sumTokens2({ api, owner: data.vault, tokens: [data.asset] });
+      const key = `${chain}:${data.asset}`;
+      // console.log(chain, type, data.asset, tvl, key, tvl[key]);
+      api.add(data.asset, tvl[key]);
+    }
+  }
+}
 
 module.exports = {
+  doublecounted: true,
   ethereum: {
-    tvl: sdk.util.sumChainTvls([ethCapTVL, ethVaultTVL])
+    tvl: chainTVL(
+      "ethereum",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.ethereum, cap: CAP_ADDRESS.ethereum }},
+        { type: 'vault', data: { asset: VAULT_weETHs.ethereum, vault: vaults.ethereum }},
+        { type: 'vault', data: UNIETH_ADDRESS.ethereum },
+      ],
+    ),
   },
   scroll: {
-    tvl: scrollVaultTVL,
+    tvl: chainTVL(
+      "scroll",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.scroll, cap: CAP_ADDRESS.scroll }},
+        { type: 'vault', data: UNIETH_ADDRESS.scroll },
+      ]
+    ),
   },
   arbitrum: {
-    tvl: capTVL("arbitrum"),
+    tvl: chainTVL(
+      "arbitrum",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.arbitrum, cap: CAP_ADDRESS.arbitrum }},
+        { type: 'vault', data: UNIETH_ADDRESS.arbitrum },
+      ]
+    ),
   },
   optimism: {
-    tvl: capTVL("optimism"),
+    tvl: chainTVL(
+      "optimism",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.optimism, cap: CAP_ADDRESS.optimism }},
+      ],
+    ),
   },
   mode: {
-    tvl: capTVL("mode"),
+    tvl: chainTVL(
+      "mode",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.mode, cap: CAP_ADDRESS.mode }},
+      ],
+    ),
   },
   manta: {
-    tvl: capTVL("manta"),
+    tvl: chainTVL(
+      "manta",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.manta, cap: CAP_ADDRESS.manta }},
+      ],
+    ),
   },
   blast: {
-    tvl: capTVL("blast"),
+    tvl: chainTVL(
+      "blast",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.blast, cap: CAP_ADDRESS.blast }},
+      ],
+    ),
   },
   linea: {
-    tvl: capTVL("linea"),
+    tvl: chainTVL(
+      "linea",
+      [
+        { type: 'cap', data: { asset: WEETH_ADDRESS.linea, cap: CAP_ADDRESS.linea }},
+        { type: 'vault', data: UNIETH_ADDRESS.linea },
+      ],
+    ),
   },
 };
-
-
-
-