From ccbed18aaa70f986e7c82cccc8f66c1d61b068d7 Mon Sep 17 00:00:00 2001
From: Peter <peter@thx.network>
Date: Wed, 26 Jun 2024 09:10:08 +0200
Subject: [PATCH] Fetch pool safely

---
 apps/api/src/app/services/BalancerService.ts | 41 ++++++++++++--------
 1 file changed, 24 insertions(+), 17 deletions(-)

diff --git a/apps/api/src/app/services/BalancerService.ts b/apps/api/src/app/services/BalancerService.ts
index 4e9616823..0190a9862 100644
--- a/apps/api/src/app/services/BalancerService.ts
+++ b/apps/api/src/app/services/BalancerService.ts
@@ -6,7 +6,6 @@ import { logger } from '../util/logger';
 import { WalletDocument } from '../models';
 import { ChainId } from '@thxnetwork/common/enums';
 import { contractArtifacts, contractNetworks } from '@thxnetwork/api/hardhat';
-import { formatUnits } from 'ethers/lib/utils';
 import { BigNumber } from 'alchemy-sdk';
 
 class BalancerService {
@@ -98,27 +97,35 @@ class BalancerService {
         }
     }
 
+    async fetchPool() {
+        try {
+            return await this.balancer.pools.find(BALANCER_POOL_ID);
+        } catch (error) {
+            logger.error(error);
+            return;
+        }
+    }
+
     async updatePricesJob() {
-        // const pool = await this.balancer.pools.find(BALANCER_POOL_ID);
-        // const [usdc, thx] = pool.tokens as unknown as {
-        //     symbol: string;
-        //     balance: number;
-        //     token: { latestUSDPrice: number };
-        // }[];
-        // const totalShares = pool.totalShares as unknown as number;
-        // const thxValue = thx.balance * thx.token.latestUSDPrice;
-        // const usdcValue = usdc.balance * usdc.token.latestUSDPrice;
-        // const btpPrice = (thxValue + usdcValue) / totalShares;
+        const pool = await this.fetchPool();
+        if (!pool) return;
+
+        const [usdc, thx] = pool.tokens as unknown as {
+            symbol: string;
+            balance: number;
+            token: { latestUSDPrice: number };
+        }[];
+        const totalShares = pool.totalShares as unknown as number;
+        const thxValue = thx.balance * thx.token.latestUSDPrice;
+        const usdcValue = usdc.balance * usdc.token.latestUSDPrice;
+        const btpPrice = (thxValue + usdcValue) / totalShares;
         const balPrice = await this.fetchPrice('BAL', 'USDC');
 
         this.pricing = {
             'BAL': Number(balPrice),
-            '20USDC-80THX': 0,
-            'USDC': 0,
-            'THX': 0,
-            // '20USDC-80THX': btpPrice,
-            // 'USDC': Number(usdc.token.latestUSDPrice),
-            // 'THX': Number(thx.token.latestUSDPrice),
+            '20USDC-80THX': btpPrice,
+            'USDC': Number(usdc.token.latestUSDPrice),
+            'THX': Number(thx.token.latestUSDPrice),
         };
     }