feat: Add max add/remove methods.

balancer · Jul 26, 2024 · a68c386 · a68c386
1 parent baa4dfc
commit a68c386
Show file tree

Hide file tree

Showing 4 changed files with 54 additions and 0 deletions.
diff --git a/typescript/src/constants.ts b/typescript/src/constants.ts
@@ -0,0 +1,2 @@
+export const MAX_UINT256 =
+    115792089237316195423570985008687907853269984665640564039457584007913129639935n;
diff --git a/typescript/src/stable/stablePool.ts b/typescript/src/stable/stablePool.ts
@@ -1,3 +1,4 @@
+import { MAX_UINT256 } from '../constants';
 import { MathSol } from '../utils/math';
 import {
     MaxSwapParams,
@@ -45,6 +46,27 @@ export class Stable implements PoolBase {
         );
     }
 
+    getMaxSingleTokenAddAmount(): bigint {
+        return MAX_UINT256;
+    }
+
+    getMaxSingleTokenExitAmount(
+        isExactIn: boolean,
+        totalSupply: bigint,
+        tokenOutBalance: bigint,
+        tokenOutScalingFactor: bigint,
+        tokenOutRate: bigint,
+    ): bigint {
+        return this.getMaxSwapAmount({
+            swapKind: isExactIn ? SwapKind.GivenIn : SwapKind.GivenOut,
+            balancesLiveScaled18: [totalSupply, tokenOutBalance],
+            tokenRates: [1000000000000000000n, tokenOutRate],
+            scalingFactors: [1000000000000000000n, tokenOutScalingFactor],
+            indexIn: 0,
+            indexOut: 1,
+        });
+    }
+
     onSwap(swapParams: SwapParams): bigint {
         const {
             swapKind,

diff --git a/typescript/src/vault/types.ts b/typescript/src/vault/types.ts
@@ -17,6 +17,14 @@ export enum SwapKind {
 
 export interface PoolBase {
     getMaxSwapAmount(maxSwapParams: MaxSwapParams): bigint;
+    getMaxSingleTokenExitAmount(
+        isExactIn: boolean,
+        totalSupply: bigint,
+        tokenOutBalance: bigint,
+        tokenOutScalingFactor: bigint,
+        tokenOutRate: bigint,
+    ): bigint;
+    getMaxSingleTokenAddAmount(): bigint;
     onSwap(swapParams: SwapParams): bigint;
     computeInvariant(balancesLiveScaled18: bigint[]): bigint;
     computeBalance(

diff --git a/typescript/src/weighted/weightedPool.ts b/typescript/src/weighted/weightedPool.ts
@@ -1,3 +1,4 @@
+import { MAX_UINT256 } from '../constants';
 import { MathSol } from '../utils/math';
 import {
     MaxSwapParams,
@@ -39,6 +40,27 @@ export class Weighted implements PoolBase {
         );
     }
 
+    getMaxSingleTokenAddAmount(): bigint {
+        return MAX_UINT256;
+    }
+
+    getMaxSingleTokenExitAmount(
+        isExactIn: boolean,
+        totalSupply: bigint,
+        tokenOutBalance: bigint,
+        tokenOutScalingFactor: bigint,
+        tokenOutRate: bigint,
+    ): bigint {
+        return this.getMaxSwapAmount({
+            swapKind: isExactIn ? SwapKind.GivenIn : SwapKind.GivenOut,
+            balancesLiveScaled18: [totalSupply, tokenOutBalance],
+            tokenRates: [1000000000000000000n, tokenOutRate],
+            scalingFactors: [1000000000000000000n, tokenOutScalingFactor],
+            indexIn: 0,
+            indexOut: 1,
+        });
+    }
+
     onSwap(swapParams: SwapParams): bigint {
         const {
             swapKind,