Merge pull request #489 from balancer/488-refactor-nested-pool-tokens

Refactor nestedPoolState logic to fetch mainTokens from api
balancer · Nov 14, 2024 · 0117f7b · 0117f7b
2 parents c892033 + f105cbb
commit 0117f7b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 18 deletions.
diff --git a/.changeset/fuzzy-coins-sin.md b/.changeset/fuzzy-coins-sin.md
@@ -0,0 +1,5 @@
+---
+"@balancer/sdk": patch
+---
+
+Refactor nestedPoolState logic to fetch mainTokens from api
diff --git a/src/data/providers/balancer-api/modules/nested-pool-state/index.ts b/src/data/providers/balancer-api/modules/nested-pool-state/index.ts
@@ -14,11 +14,6 @@ export type PoolGetPool = {
     protocolVersion: 1 | 2 | 3;
     address: Address;
     type: string;
-    allTokens: {
-        address: Address;
-        decimals: number;
-        isMainToken: boolean;
-    }[];
     poolTokens: Token[];
 };
 
@@ -48,11 +43,6 @@ export class NestedPools {
       protocolVersion
       address
       type
-      allTokens {
-        address
-        decimals
-        isMainToken
-      }
       poolTokens {
         index
         address
@@ -258,14 +248,10 @@ export function mapPoolToNestedPoolStateV2(pool: PoolGetPool): NestedPoolState {
         });
     });
 
-    const mainTokens = pool.allTokens
-        .filter((t) => t.isMainToken)
-        .map((t) => {
-            return {
-                address: t.address,
-                decimals: t.decimals,
-            };
-        });
+    // mainTokens are pool tokens filtering out nested pools and phantomBPTs
+    const mainTokens = pools
+        .flatMap((p) => p.tokens)
+        .filter((t) => !pools.find((p) => p.address === t.address));
 
     return {
         protocolVersion: 2,