APR Services - Add some yield tokens APR to linear pools

modules/network/mainnet.ts

@@ -17,6 +17,8 @@ import { GithubContentService } from '../content/github-content.service';
 import { gaugeSubgraphService } from '../subgraphs/gauge-subgraph/gauge-subgraph.service';
 import { coingeckoService } from '../coingecko/coingecko.service';
 import { CoingeckoPriceHandlerService } from '../token/lib/token-price-handlers/coingecko-price-handler.service';
+import { IbTokensAprService } from "../pool/lib/apr-data-sources/ib-tokens-apr.service";
+import { networkContext } from "./network-context.service";
 
 const mainnetNetworkData: NetworkData = {
     chain: {
@@ -157,6 +159,7 @@ export const mainnetNetworkConfig: NetworkConfig = {
     contentService: new GithubContentService(),
     provider: new ethers.providers.JsonRpcProvider(mainnetNetworkData.rpcUrl),
     poolAprServices: [
+        new IbTokensAprService(),
         new WstethAprService(tokenService, mainnetNetworkData.lido!.wstEthContract),
         new ReaperCryptAprService(
             mainnetNetworkData.reaper.linearPoolFactories,

modules/pool/lib/apr-data-sources/apr-types.ts

@@ -1,3 +1,5 @@
+import { Dictionary } from 'lodash'
+
 export interface YearnVault {
     inception: number;
     address: string;

modules/pool/lib/apr-data-sources/ib-tokens-apr.service.ts

@@ -0,0 +1,59 @@
+import { PoolAprService } from "../../pool-types";
+import { PrismaPoolWithExpandedNesting } from "../../../../prisma/prisma-types";
+import { prisma } from "../../../../prisma/prisma-client";
+import { networkContext } from "../../../network/network-context.service";
+import { prismaBulkExecuteOperations } from "../../../../prisma/prisma-util";
+import { fetchAllAprs } from "../../../token/lib/token-apr-handler/fetch-all";
+import { TokenApr } from "../../../token/lib/token-apr-handler/types";
+import { PrismaPoolAprItemGroup } from "@prisma/client";
+
+export class IbTokensAprService implements PoolAprService {
+
+  getAprServiceName(): string {
+    return "IbTokensAprService";
+  }
+
+  public async updateAprForPools(pools: PrismaPoolWithExpandedNesting[]): Promise<void> {
+    const operations: any[] = [];
+    const aprs = await this.fetchYieldTokensApr();
+    const tokenYieldPools = pools.filter((pool) => {
+        return pool.tokens.find((token) => {
+          return Array.from(aprs.keys()).map((key) => key.toLowerCase()).includes(token.address.toLowerCase());
+        })
+      }
+    );
+    for (const pool of tokenYieldPools) {
+      for (const token of pool.tokens) {
+        if ((aprs.get(token.address) !== undefined)) {
+          const tokenSymbol = token.token.symbol ?? (<TokenApr>aprs.get(token.address)).name
+          const itemId = `${ pool.id }-${ tokenSymbol }-yield-apr`
+
+          operations.push(prisma.prismaPoolAprItem.upsert({
+            where: { id_chain: { id: itemId, chain: networkContext.chain } },
+            create: {
+              id: itemId,
+              chain: networkContext.chain,
+              poolId: pool.id,
+              title: `${ tokenSymbol} APR`,
+              apr: aprs.get(token.address)?.val ?? 0,
+              group: (aprs.get(token.address)?.group as PrismaPoolAprItemGroup) ?? null,
+              type: pool.type === 'LINEAR' ? 'LINEAR_BOOSTED' : 'IB_YIELD',
+            },
+            update: {
+              title: `${ tokenSymbol } APR`,
+              apr: aprs.get(token.address)?.val
+            },
+          }));
+        }
+      }
+    }
+
+    await prismaBulkExecuteOperations(operations);
+  }
+
+  private async fetchYieldTokensApr(): Promise<Map<string, TokenApr>> {
+    const data = await fetchAllAprs()
+    return new Map<string, TokenApr>(data.filter((apr)=>!isNaN(apr.val)).map((apr) => [apr.address, apr]));
+  }
+
+}

modules/pool/pool.prisma

@@ -299,6 +299,14 @@ enum PrismaPoolAprItemGroup {
     OVERNIGHT
     REAPER
     YEARN
+    IDLE
+    TRANCHESS
+    GEARBOX
+    AAVE
+    ANKR
+    TESSERA
+    TETU
+    OVIX
 }
 
 model PrismaPoolCategory {

modules/token/lib/token-apr-handler/fetch-all.ts

@@ -0,0 +1,16 @@
+import { tokens } from "./tokens";
+import { TokenApr } from "./types";
+
+export const fetchAllAprs = async (): Promise<TokenApr[]> => {
+  const res = await Promise.all(
+    tokens.map(async ({ name, group, fetchFn }) => {
+      const fetchedResponse: {[key:string]:number} = await fetchFn()
+      return Object.entries(fetchedResponse).map(([address, aprValue]) => ({
+        val: aprValue * 0.0001 /*Values come in BPS 10000=>100% */,
+        group,
+        name,
+        address
+      }))
+    }));
+  return res.flat();
+}