Fix: Sushi trident using new bentobox helper (DefiLlama#15381)

Author: 0xpeluche

projects/sushiswap-bentobox/helper.js

@@ -1,5 +1,4 @@
 const sdk = require("@defillama/sdk");
-const { default: BigNumber } = require("bignumber.js");
 const { request, gql } = require("graphql-request");
 
 const CONFIG = {
@@ -214,4 +213,4 @@ async function fetchAllTokens(subgraph, query, block, type) {
   return allTokens;
 }
 
-module.exports = { CONFIG, getTokens };
+module.exports = { CONFIG, getTokens, fetchAllTokens };

projects/sushiswap-trident/index.js

@@ -1,4 +1,7 @@
-const { trident } = require("./trident.js");
+const sdk = require("@defillama/sdk");
+const { getTokens, fetchAllTokens } = require("../sushiswap-bentobox/helper.js");
+
+const polygonOldRouter = sdk.graph.modifyEndpoint('5LBvcUQthQDZTMe6cyJ7DbeokFkYeVpYYBZruHPUjMG5')
 
 const modulesToExport = {};
 const trident_chains = [
@@ -12,12 +15,43 @@ const trident_chains = [
   "avax",
 ];
 
+const tridentQuery = `
+  query get_tokens {
+    tokens(
+      first: 100
+      orderBy: liquidityUSD
+      orderDirection: desc
+      where: { liquidityUSD_gt: 0 }
+    ) {
+      id
+      symbol
+      liquidity
+    }
+  }
+`;
+
+async function fetchAndAddTridentTokens(api, block, subgraph) {
+  const tokens = subgraph
+    ? await fetchAllTokens(subgraph, tridentQuery, block, 'trident')
+    : await getTokens(api, block, 'trident');
+
+  for (const { id, liquidity } of tokens) {
+    api.add(id, liquidity);
+  }
+}
+
+const tvl = async (api) => {
+  const block = await api.getBlock();
+  await fetchAndAddTridentTokens(api, block);
+  if (api.chain === 'polygon')  await fetchAndAddTridentTokens(api, block, polygonOldRouter);
+}
+
 trident_chains.forEach((chain) => {
-  modulesToExport[chain] = {
-    tvl: trident(chain),
-  };
+  modulesToExport[chain] = { tvl }
 });
 
+module.exports.methodology = `TVL of Trident consist of tokens deployed into swapping pairs.`
+
 module.exports = {
     ...modulesToExport,
 };