Update TrueFi adapter to include new portfolios

projects/truefi/index.js

@@ -1,5 +1,8 @@
 const sdk = require('../../sdk');
 const abi = require('./abi.json');
+const ethers = require('ethers')
+const managedPortfolioFactoryAbi = require('./managedPortfolioFactoryAbi.json')
+const managedPortfolioAbi = require('./managedPortfolioAbi.json')
 
 const tfBUSD = '0x1Ed460D149D48FA7d91703bf4890F97220C09437';
 const tfTUSD = '0xa1e72267084192db7387c8cc1328fade470e4149';
@@ -11,6 +14,49 @@ const BUSD = '0x4Fabb145d64652a948d72533023f6E7A623C7C53';
 const TUSD = '0x0000000000085d4780B73119b644AE5ecd22b376';
 const USDC = '0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48';
 const USDT = '0xdAC17F958D2ee523a2206206994597C13D831ec7';
+const managedPortfolioFactory = '0x17b7b75FD4288197cFd99D20e13B0dD9da1FF3E7';
+
+async function getTvlForManagedPortfolios(block) {
+  if (block < 14043986) {
+    return {}
+  }
+  let portfolios = []
+  try {
+    portfolios = (await sdk.api.abi.call({
+      target: managedPortfolioFactory,
+      abi: managedPortfolioFactoryAbi['getPortfolios'],
+      block
+    })).output;
+  } catch (e) { return {} }
+  const portfoliosBalances = {};
+  for (let i = 0; i < portfolios.length; i++) {
+    let value;
+    let underlyingToken;
+    try {
+      value = (await sdk.api.abi.call({
+        target: portfolios[i],
+        abi: managedPortfolioAbi['value'],
+        block
+      })).output;
+      underlyingToken = (await sdk.api.abi.call({
+        target: portfolios[i],
+        abi: managedPortfolioAbi['underlyingToken'],
+        block
+      })).output;
+    } catch (e) { break; }
+
+    if (!portfoliosBalances[underlyingToken]) {
+      portfoliosBalances[underlyingToken] = []
+    }
+    portfoliosBalances[underlyingToken].push(value);
+  }
+  let balances = {}
+  const underlyingTokens = Object.keys(portfoliosBalances);
+  for (let i = 0; i < underlyingTokens.length; i++) {
+    balances[underlyingTokens[i]] = portfoliosBalances[underlyingTokens[i]].reduce((acc, curr) => acc.add(curr), ethers.BigNumber.from(0))
+  }
+  return balances;
+}
 
 async function tvl(timestamp, block) {
   let balances = {};
@@ -25,43 +71,46 @@ async function tvl(timestamp, block) {
       abi: abi['poolValue'],
       block: block
     });
-  }catch (e){tfbusdTVL = {output: '0'}}
+  } catch (e) { tfbusdTVL = { output: '0' } }
   try {
     tftusdTVL = await sdk.api.abi.call({
       target: tfTUSD,
       abi: abi['poolValue'],
       block: block
     });
-  }catch (e){tftusdTVL = {output: '0'}}
+  } catch (e) { tftusdTVL = { output: '0' } }
   try {
     tfusdcTVL = await sdk.api.abi.call({
       target: tfUSDC,
       abi: abi['poolValue'],
       block: block
     });
-  }catch (e){tfusdcTVL = {output: '0'}}
+  } catch (e) { tfusdcTVL = { output: '0' } }
   try {
     tfusdtTVL = await sdk.api.abi.call({
       target: tfUSDT,
       abi: abi['poolValue'],
       block: block
     });
-  }catch (e){tfusdtTVL = {output: '0'}}
-  try{
+  } catch (e) { tfusdtTVL = { output: '0' } }
+  try {
     truTVL = await sdk.api.abi.call({
       target: stkTRU,
       abi: abi['stakeSupply'],
       block: block
     });
-  }catch (e) {truTVL = {output: '0'}}
+  } catch (e) { truTVL = { output: '0' } }
 
   balances[BUSD] = tfbusdTVL.output;
   balances[TUSD] = tftusdTVL.output;
   balances[USDC] = tfusdcTVL.output;
   balances[USDT] = tfusdtTVL.output;
   balances[TRU] = truTVL.output;
 
-    return balances;
+  const managedPortfoliosBalances = await getTvlForManagedPortfolios(block)
+  Object.keys(managedPortfoliosBalances).map(token => balances[token] = managedPortfoliosBalances[token].add(balances[token]).toString())
+
+  return balances;
 }
 
 module.exports = {

projects/truefi/managedPortfolioAbi.json

@@ -0,0 +1,28 @@
+{
+  "value": {
+    "inputs": [],
+    "name": "value",
+    "outputs": [
+      {
+        "internalType": "uint256",
+        "name": "",
+        "type": "uint256"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  },
+  "underlyingToken": {
+    "inputs": [],
+    "name": "underlyingToken",
+    "outputs": [
+      {
+        "internalType": "contract IERC20WithDecimals",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  }
+}

projects/truefi/managedPortfolioFactoryAbi.json

@@ -0,0 +1,15 @@
+{
+  "getPortfolios": {
+    "inputs": [],
+    "name": "getPortfolios",
+    "outputs": [
+      {
+        "internalType": "contract IManagedPortfolio[]",
+        "name": "",
+        "type": "address[]"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  }
+}