Feat boosted pool add and remove

src/abi/erc4626.ts

@@ -0,0 +1,177 @@
+export const erc4626Abi = [
+    {
+        inputs: [
+            {
+                internalType: 'address',
+                name: 'asset',
+                type: 'address',
+            },
+        ],
+        stateMutability: 'nonpayable',
+        type: 'constructor',
+    },
+    {
+        inputs: [
+            {
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+        ],
+        name: 'deposit',
+        outputs: [
+            {
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        stateMutability: 'nonpayable',
+        type: 'function',
+    },
+    {
+        inputs: [
+            {
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        name: 'withdraw',
+        outputs: [
+            {
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+        ],
+        stateMutability: 'nonpayable',
+        type: 'function',
+    },
+    {
+        inputs: [],
+        name: 'asset',
+        outputs: [
+            {
+                internalType: 'address',
+                name: '',
+                type: 'address',
+            },
+        ],
+        stateMutability: 'view',
+        type: 'function',
+    },
+    {
+        inputs: [
+            {
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+        ],
+        name: 'convertToShares',
+        outputs: [
+            {
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        stateMutability: 'view',
+        type: 'function',
+    },
+    {
+        inputs: [
+            {
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        name: 'convertToAssets',
+        outputs: [
+            {
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+        ],
+        stateMutability: 'view',
+        type: 'function',
+    },
+    {
+        inputs: [],
+        name: 'totalAssets',
+        outputs: [
+            {
+                internalType: 'uint256',
+                name: '',
+                type: 'uint256',
+            },
+        ],
+        stateMutability: 'view',
+        type: 'function',
+    },
+    {
+        anonymous: false,
+        inputs: [
+            {
+                indexed: true,
+                internalType: 'address',
+                name: 'caller',
+                type: 'address',
+            },
+            {
+                indexed: true,
+                internalType: 'address',
+                name: 'owner',
+                type: 'address',
+            },
+            {
+                indexed: false,
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+            {
+                indexed: false,
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        name: 'Deposit',
+        type: 'event',
+    },
+    {
+        anonymous: false,
+        inputs: [
+            {
+                indexed: true,
+                internalType: 'address',
+                name: 'caller',
+                type: 'address',
+            },
+            {
+                indexed: true,
+                internalType: 'address',
+                name: 'receiver',
+                type: 'address',
+            },
+            {
+                indexed: false,
+                internalType: 'uint256',
+                name: 'assets',
+                type: 'uint256',
+            },
+            {
+                indexed: false,
+                internalType: 'uint256',
+                name: 'shares',
+                type: 'uint256',
+            },
+        ],
+        name: 'Withdraw',
+        type: 'event',
+    },
+];

src/abi/index.ts

@@ -18,3 +18,4 @@ export * from './weightedPool.V3';
 export * from './vaultAdmin.V3';
 export * from './stablePoolFactory.V3';
 export * from './balancerCompositeLiquidityRouter';
+export * from './erc4626';

src/data/providers/balancer-api/modules/pool-state/index.ts

@@ -1,5 +1,9 @@
 import { BalancerApiClient } from '../../client';
-import { PoolState, PoolStateWithBalances } from '../../../../../entities';
+import {
+    PoolState,
+    PoolStateWithBalances,
+    PoolStateWithUnderlyings,
+} from '../../../../../entities';
 import { mapPoolType } from '../../../../../utils/poolTypeMapper';
 
 import { API_CHAIN_NAMES } from '../../../../../utils/constants';
@@ -39,6 +43,29 @@ export class Pools {
       }
     }`;
 
+    readonly boostedPoolStateWithUnderlying = `
+    query GetPool($id: String!, $chain: GqlChain!) {
+      poolGetPool(id:$id, chain:$chain) {
+        id
+        address
+        type
+        protocolVersion
+        poolTokens {
+          index
+          address
+          decimals
+          balance
+          underlyingToken {
+            address
+            decimals
+          }
+        }
+        dynamicData {
+          totalShares
+        }
+      }
+    }`;
+
     constructor(private readonly balancerApiClient: BalancerApiClient) {}
 
     async fetchPoolState(id: string): Promise<PoolState> {
@@ -76,4 +103,24 @@ export class Pools {
         };
         return poolStateWithBalances;
     }
+
+    async fetchPoolStateWithUnderlyingTokens(
+        id: string,
+    ): Promise<PoolStateWithUnderlyings> {
+        const { data } = await this.balancerApiClient.fetch({
+            query: this.boostedPoolStateWithUnderlying,
+            variables: {
+                id: id.toLowerCase(),
+                chain: API_CHAIN_NAMES[this.balancerApiClient.chainId],
+            },
+        });
+        const poolStateWithUnderlyings: PoolStateWithUnderlyings = {
+            ...data.poolGetPool,
+            tokens: data.poolGetPool.poolTokens,
+            type: mapPoolType(data.poolGetPool.type),
+            //underlyings: data.poolGetPool.poolTokens.map((token) => {token}),
+            totalShares: data.poolGetPool.dynamicData.totalShares,
+        };
+        return poolStateWithUnderlyings;
+    }
 }

src/data/types.ts

@@ -11,3 +11,7 @@ export interface MinimalToken {
 export interface PoolTokenWithBalance extends MinimalToken {
     balance: HumanAmount;
 }
+
+export interface PoolTokenWithUnderlying extends MinimalToken {
+    underlying: Address;
+}

src/entities/addLiquidityBoosted/doAddLiquidityPropotionalQuery.ts

@@ -0,0 +1,31 @@
+import { createPublicClient, http, getAddress } from 'viem';
+
+import { BALANCER_COMPOSITE_LIQUIDITY_ROUTER, CHAINS } from '@/utils';
+
+import { AddLiquidityProportionalInput } from '../addLiquidity/types';
+import { Address } from '@/types';
+
+import { balancerCompositeLiquidityRouterAbi } from '@/abi';
+
+export const doAddLiquidityProportionalQuery = async (
+    { rpcUrl, chainId, referenceAmount }: AddLiquidityProportionalInput,
+    poolAddress: Address,
+): Promise<bigint[]> => {
+    const client = createPublicClient({
+        transport: http(rpcUrl),
+        chain: CHAINS[chainId],
+    });
+
+    const { result: exactAmountsIn } = await client.simulateContract({
+        address: BALANCER_COMPOSITE_LIQUIDITY_ROUTER[chainId],
+        abi: balancerCompositeLiquidityRouterAbi,
+        functionName: 'queryAddLiquidityProportionalToERC4626Pool',
+        args: [
+            poolAddress,
+            referenceAmount.rawAmount,
+            getAddress('0xa5cc3c03994db5b0d9a5eEdD10Cabab0813678ac'),
+            '0x',
+        ],
+    });
+    return [...exactAmountsIn];
+};

src/entities/addLiquidityBoosted/doAddLiquidityUnbalancedQuery.ts

@@ -0,0 +1,31 @@
+import { createPublicClient, http, getAddress } from 'viem';
+
+import { BALANCER_COMPOSITE_LIQUIDITY_ROUTER, CHAINS } from '@/utils';
+
+import { AddLiquidityUnbalancedInput } from '../addLiquidity/types';
+import { Address } from '@/types';
+
+import { balancerCompositeLiquidityRouterAbi } from '@/abi';
+
+export const doAddLiquidityUnbalancedQuery = async (
+    { rpcUrl, chainId, amountsIn }: AddLiquidityUnbalancedInput,
+    poolAddress: Address,
+): Promise<bigint> => {
+    const client = createPublicClient({
+        transport: http(rpcUrl),
+        chain: CHAINS[chainId],
+    });
+
+    const { result: bptAmountOut } = await client.simulateContract({
+        address: BALANCER_COMPOSITE_LIQUIDITY_ROUTER[chainId],
+        abi: balancerCompositeLiquidityRouterAbi,
+        functionName: 'queryAddLiquidityUnbalancedToERC4626Pool',
+        args: [
+            poolAddress,
+            amountsIn.map((input) => input.rawAmount),
+            getAddress('0xa5cc3c03994db5b0d9a5eEdD10Cabab0813678ac'),
+            '0x',
+        ],
+    });
+    return bptAmountOut;
+};