venus: track ethereum

Author: g1nt0ki

projects/venus-isolated-pools/index.js

@@ -4,6 +4,11 @@ const { compoundExports2 } = require('../helper/compound')
 const config = {
   bsc: {
     endpoint: 'https://api.thegraph.com/subgraphs/name/venusprotocol/venus-isolated-pools',
+    corePools: ['0xfd36e2c2a6789db23113685031d7f16329158384'],
+  },
+  ethereum: {
+    endpoint: 'https://api.thegraph.com/subgraphs/name/venusprotocol/venus-isolated-pools-ethereum',
+    corePools: ['0x67aA3eCc5831a65A5Ba7be76BED3B5dc7DB60796'],
   }
 }
 
@@ -12,9 +17,10 @@ Object.keys(config).forEach(chain => {
 })
 
 async function getPools(api) {
-  const { endpoint } = config[api.chain]
-  const { pools } = await cachedGraphQuery('venus-v4', endpoint, `{ pools { id  }}`)
-  return pools.map(i => i.id)
+  const { endpoint, corePools } = config[api.chain]
+  const poolSet = new Set(corePools.map(i=>i.toLowerCase()))
+  const { pools } = await cachedGraphQuery('venus-v4/'+api.chain, endpoint, `{ pools { id  }}`)
+  return pools.map(i => i.id).filter(i => !poolSet.has(i.toLowerCase()))
 }
 
 async function tvl(...args) {
@@ -29,4 +35,4 @@ async function borrowed(...args) {
   const pools = await getPools(api)
   const tvls = pools.map(i => compoundExports2({ comptroller: i, fetchBalances: true, }))
   return sdk.util.sumChainTvls(tvls.map(i => i.borrowed))(...args)
-}
+}

projects/venus.js

@@ -1,5 +1,5 @@
 const ADDRESSES = require('./helper/coreAssets.json')
-const {fullCoumpoundExports} = require('./helper/compound');
+const {fullCoumpoundExports, compoundExports2} = require('./helper/compound');
 
 const replace = {
   [ADDRESSES.bsc.BETH]: ADDRESSES.ethereum.WETH, // beth->weth
@@ -9,3 +9,5 @@ const replace = {
 module.exports = fullCoumpoundExports("0xfd36e2c2a6789db23113685031d7f16329158384", "bsc", "0xA07c5b74C9B40447a954e1466938b865b6BBea36", ADDRESSES.bsc.WBNB, addr=>{
   return replace[addr.toLowerCase()] || `bsc:${addr}`
 })
+
+module.exports.ethereum = compoundExports2({ comptroller: '0x67aA3eCc5831a65A5Ba7be76BED3B5dc7DB60796'})