add Merkl rewards to existing pools

Author: mastilver

src/adaptors/euler-v2/index.js

@@ -131,7 +131,7 @@ const getApys = async () => {
       }
     }
   }
-  return result;
+  return utils.addMerklRewardApy(result, 'euler');
 };
 
 module.exports = {

src/adaptors/gamma/index.js

@@ -388,11 +388,13 @@ const getApy = async () => {
     '0x431f6e577a431d9ee87a535fde2db830e352e33c',
     '0xed17209ab7f9224e29cc9894fa14a011f37b6115',
   ];
-  return pools.flat().map((i) => ({
+  const data = pools.flat().map((i) => ({
     ...i,
     apyReward: x.includes(i.pool) || i.chain === 'Manta' ? null : i.apyReward,
     rewardTokens: x.includes(i.pool) || i.chain === 'Manta' ? null : i.rewardTokens,
   })).filter(p => p.chain != 'Binance');
+
+  return utils.addMerklRewardApy(data, 'gamma', (p) => p.pool.split('-')[0]);
 };
 
 module.exports = {

src/adaptors/merkl/index.js

@@ -9,6 +9,10 @@ const networks = {
   1101: 'polygon_zkevm',
   8453: 'base',
   60808: 'bob',
+  146: 'sonic',
+  43114: 'avax',
+  80094: 'berachain',
+  56: 'bsc',
 };
 
 // Protocols that should not be listed under Merkl
@@ -18,6 +22,7 @@ const protocolsBlacklist = [
   'crosscurve',
   'aerodrome',
   'gamma',
+  'uniswap',
 ];
 
 // Allow specific pools from blacklisted protocols
@@ -50,6 +55,7 @@ const main = async () => {
         data = await utils.getData(`https://api.merkl.xyz/v4/opportunities?chainId=${chainId}&status=LIVE&items=100&page=${pageI}`);
       } catch (err) {
         console.log('failed to fetch Merkl data on chain ' + chain);
+        break;
       }
 
       if (data.length === 0) {

src/adaptors/uniswap-v2/index.js

@@ -126,7 +126,10 @@ const main = async (timestamp = null) => {
     }
   }
 
-  return data.filter((p) => utils.keepFinite(p));
+  return utils.addMerklRewardApy(
+    data.filter((p) => utils.keepFinite(p)),
+    'uniswap'
+  );
 };
 
 module.exports = {

src/adaptors/uniswap-v3/index.js

@@ -352,7 +352,8 @@ const main = async (timestamp = null) => {
     );
   }
   data.push(...(await getOnchainPools()))
-  return data
+  return utils.addMerklRewardApy(
+    data
     .flat()
     .filter(
       (p) =>
@@ -361,7 +362,9 @@ const main = async (timestamp = null) => {
           '0x0c6d9d0f82ed2e0b86c4d3e9a9febf95415d1b76',
           '0xc809d13e9ea08f296d3b32d4c69d46ff90f73fd8',
         ].includes(p.pool)
-    );
+    ),
+    'uniswap'
+  );
 };
 
 module.exports = {

src/adaptors/utils.js

@@ -513,3 +513,83 @@ exports.getTotalSupply = async (tokenMintAddress) => {
 
   return supplyInTokens;
 };
+
+const merklNetworks = {
+  1: 'ethereum',
+  137: 'polygon',
+  10: 'optimism',
+  42161: 'arbitrum',
+  1101: 'polygon_zkevm',
+  8453: 'base',
+  60808: 'bob',
+  146: 'sonic',
+  43114: 'avax',
+  80094: 'berachain',
+  56: 'bsc',
+}
+
+exports.addMerklRewardApy = async (
+  pools,
+  protocolId,
+  poolAddressGetter,
+) => {
+  try {
+    let merklPools = [];
+    let pageI = 0;
+
+    while(true) {
+      let data;
+      try {
+        data = await exports.getData(`https://api.merkl.xyz/v4/opportunities?mainProtocolId=${protocolId}&status=LIVE&items=100&page=${pageI}`);
+      } catch (err) {
+        console.log(`failed to fetch Merkl data for ${protocolId}: ${err}`);
+        break;
+      }
+
+      if (data.length === 0) {
+        break;
+      }
+
+      merklPools.push(...data);
+      pageI++;
+    }
+
+
+    const merklPoolsMap = Object.fromEntries(Object.keys(merklNetworks).map(id => [merklNetworks[id], {}]));
+    merklPools.forEach(pool => {
+      if (!merklNetworks[pool.chainId]) {
+        return;
+      }
+
+      merklPoolsMap[merklNetworks[pool.chainId]][pool.identifier.toLowerCase()] = {
+        apyReward: pool.apr,
+        rewardTokens: [...new Set(pool.rewardsRecord?.breakdowns.map(x => x.token.address) || [])]
+      }
+    });
+
+    return pools.map(pool => {
+      const poolAddress = poolAddressGetter ? poolAddressGetter(pool) : pool.pool;
+      const merklRewards = merklPoolsMap[pool.chain.toLowerCase()][poolAddress.toLowerCase()];
+
+      if (!merklRewards) {
+        return pool;
+      }
+
+      // if the data is already present, don't overwrite it
+      if (pool.apyReward || (pool.rewardTokens && pool.rewardTokens.length !== 0)) {
+        console.log('pool already has apyReward or rewardTokens', pool.pool);
+        return pool;
+      }
+
+      return {
+        ...pool,
+        ...merklRewards,
+      }
+    });
+  } catch (err) {
+    console.log(`Failed to add Merkl reward apy to ${protocolId}: ${err}`);
+
+    // If we fail to fetch Merkl data, just return the original pools
+    return pools;
+  }
+};