fix: Circular deps.

balancer · Sep 26, 2023 · 1d54c4c · 1d54c4c
1 parent e1efe6b
commit 1d54c4c
Show file tree

Hide file tree

Showing 40 changed files with 214 additions and 185 deletions.
diff --git a/src/data/enrichers/onChainPoolDataEnricher.ts b/src/data/enrichers/onChainPoolDataEnricher.ts
@@ -6,6 +6,7 @@ import {
     RawPool,
     RawPoolTokenWithRate,
     RawWeightedPoolToken,
+    HumanAmount,
 } from '../types';
 
 import {
@@ -16,7 +17,7 @@ import {
     poolHasVirtualSupply,
     poolIsLinearPool,
 } from '../../utils';
-import { HumanAmount, SwapOptions } from '../../types';
+import { SwapOptions } from '../../types';
 
 interface OnChainPoolData {
     id: string;

diff --git a/src/data/types.ts b/src/data/types.ts
@@ -1,5 +1,6 @@
 import { Address, Hex } from 'viem';
-import { HumanAmount } from '../types';
+
+export type HumanAmount = `${number}`;
 
 // These are only the known pool types, additional pool types can be added via
 // extension through custom PoolFactories and PoolDataProviders

diff --git a/src/entities/exit/weighted/validateInputs.ts b/src/entities/exit/weighted/validateInputs.ts
@@ -1,4 +1,4 @@
-import { ExitInput, ExitKind } from '..';
+import { ExitInput, ExitKind } from '../types';
 import { PoolStateInput } from '../../types';
 import { areTokensInArray } from '../../utils/areTokensInArray';
 

diff --git a/src/entities/join/weighted/validateInputs.ts b/src/entities/join/weighted/validateInputs.ts
@@ -1,4 +1,4 @@
-import { JoinInput, JoinKind } from '..';
+import { JoinInput, JoinKind } from '../types';
 import { PoolStateInput } from '../../types';
 import { areTokensInArray } from '../../utils/areTokensInArray';
 

diff --git a/src/entities/join/weighted/weightedJoin.ts b/src/entities/join/weighted/weightedJoin.ts
@@ -1,5 +1,7 @@
 import { encodeFunctionData } from 'viem';
-import { Token, TokenAmount, WeightedEncoder } from '../../..';
+import { Token } from '../../../entities/token';
+import { TokenAmount } from '../../../entities/tokenAmount';
+import { WeightedEncoder } from '../../../entities/encoders/weighted';
 import { Address } from '../../../types';
 import { BALANCER_VAULT, MAX_UINT256, ZERO_ADDRESS } from '../../../utils';
 import { vaultAbi } from '../../../abi';
@@ -10,7 +12,7 @@ import {
     JoinInput,
     JoinKind,
     JoinQueryResult,
-} from '..';
+} from '../types';
 import { AmountsJoin, PoolState } from '../../types';
 import { doQueryJoin, getAmounts, parseJoinArgs } from '../../utils';
 

diff --git a/src/entities/path.ts b/src/entities/path.ts
@@ -1,5 +1,6 @@
 import { BasePool } from './pools/';
-import { Token, TokenAmount } from './';
+import { Token } from './token';
+import { TokenAmount } from './tokenAmount';
 import { SwapKind } from '../types';
 
 export class Path {

diff --git a/src/entities/pools/fx/fxFactory.ts b/src/entities/pools/fx/fxFactory.ts
@@ -1,5 +1,5 @@
 import { BasePool, BasePoolFactory } from '../';
-import { FxPool } from './';
+import { FxPool } from './fxPool';
 import { RawFxPool, RawPool } from '../../../data/types';
 
 export class FxPoolFactory implements BasePoolFactory {

diff --git a/src/entities/pools/fx/fxMath.ts b/src/entities/pools/fx/fxMath.ts
@@ -1,6 +1,6 @@
 import { parseUnits } from 'viem';
-import { RAY } from '../../../utils';
-import { FxPoolPairData } from './fxPool';
+import { RAY } from '../../../utils/math';
+import { FxPoolPairData } from './types';
 import { SwapKind } from '../../../types';
 
 export const CURVEMATH_MAX_DIFF = parseUnits('-0.000001000000000000024', 36);

diff --git a/src/entities/pools/fx/fxPool.ts b/src/entities/pools/fx/fxPool.ts
@@ -1,11 +1,14 @@
 import { Hex, parseEther, parseUnits } from 'viem';
-import { HumanAmount, PoolType, SwapKind } from '../../../types';
-import { BigintIsh, Token, TokenAmount } from '../../';
+import { PoolType, SwapKind } from '../../../types';
+import { Token } from '../../token';
+import { TokenAmount } from '../../tokenAmount';
 import { BasePool } from '../../pools';
-import { RAY, WAD, getPoolAddress } from '../../../utils';
+import { RAY, getPoolAddress } from '../../../utils';
 import { _calcInGivenOut, _calcOutGivenIn } from './fxMath';
 import { RawFxPool } from '../../../data/types';
 import { MathFx, parseFixedCurveParam } from './helpers';
+import { FxPoolPairData } from './types';
+import { FxPoolToken } from './fxPoolToken';
 
 const isUSDC = (address: string): boolean => {
     return (
@@ -15,104 +18,6 @@ const isUSDC = (address: string): boolean => {
     );
 };
 
-export type FxPoolPairData = {
-    tIn: FxPoolToken;
-    tOut: FxPoolToken;
-    alpha: bigint;
-    beta: bigint;
-    delta: bigint;
-    lambda: bigint;
-    _oGLiq: bigint;
-    _nGLiq: bigint;
-    _oBals: bigint[];
-    _nBals: bigint[];
-    givenToken: FxPoolToken;
-    swapKind: SwapKind;
-};
-
-export class FxPoolToken extends TokenAmount {
-    public readonly index: number;
-    public readonly latestFXPrice: HumanAmount;
-    public readonly fxOracleDecimals: number;
-    public numeraire: bigint; // in 36 decimals
-    private readonly scalar36 = this.scalar * WAD;
-
-    public constructor(
-        token: Token,
-        amount: BigintIsh,
-        latestFXPrice: HumanAmount,
-        fxOracleDecimals: number,
-        index: number,
-    ) {
-        super(token, amount);
-        this.latestFXPrice = latestFXPrice;
-        this.fxOracleDecimals = fxOracleDecimals;
-        const truncatedNumeraire = MathFx.mulDownFixed(
-            this.amount,
-            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
-            this.fxOracleDecimals,
-        );
-        this.numeraire = truncatedNumeraire * this.scalar36;
-        this.index = index;
-    }
-
-    public increase(amount: bigint): TokenAmount {
-        this.amount = this.amount + amount;
-        this.scale18 = this.amount * this.scalar;
-        const truncatedNumeraire = MathFx.mulDownFixed(
-            this.amount,
-            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
-            this.fxOracleDecimals,
-        );
-        this.numeraire = truncatedNumeraire * this.scalar36;
-        return this;
-    }
-
-    public decrease(amount: bigint): TokenAmount {
-        this.amount = this.amount - amount;
-        this.scale18 = this.amount * this.scalar;
-        const truncatedNumeraire = MathFx.mulDownFixed(
-            this.amount,
-            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
-            this.fxOracleDecimals,
-        );
-        this.numeraire = truncatedNumeraire * this.scalar36;
-        return this;
-    }
-
-    public static fromNumeraire(
-        poolToken: FxPoolToken,
-        numeraire: BigintIsh,
-        divUp?: boolean,
-    ): FxPoolToken {
-        const truncatedNumeraire = BigInt(numeraire) / poolToken.scalar36; // loss of precision required to match SC implementation
-        const amount = divUp
-            ? MathFx.divUpFixed(
-                  BigInt(truncatedNumeraire),
-                  parseUnits(
-                      poolToken.latestFXPrice,
-                      poolToken.fxOracleDecimals,
-                  ),
-                  poolToken.fxOracleDecimals,
-              )
-            : MathFx.divDownFixed(
-                  BigInt(truncatedNumeraire),
-                  parseUnits(
-                      poolToken.latestFXPrice,
-                      poolToken.fxOracleDecimals,
-                  ),
-                  poolToken.fxOracleDecimals,
-              );
-        return new FxPoolToken(
-            poolToken.token,
-            amount,
-            poolToken.latestFXPrice,
-            poolToken.fxOracleDecimals,
-            poolToken.index,
-        );
-    }
-}
-
 export class FxPool implements BasePool {
     public readonly chainId: number;
     public readonly id: Hex;

diff --git a/src/entities/pools/fx/fxPoolToken.ts b/src/entities/pools/fx/fxPoolToken.ts
@@ -0,0 +1,90 @@
+import { parseUnits } from 'viem';
+import { Token } from '../../token';
+import { TokenAmount, BigintIsh } from '../../tokenAmount';
+import { WAD } from '../../../utils/math';
+import { _calcInGivenOut, _calcOutGivenIn } from './fxMath';
+import { HumanAmount } from '../../../data/types';
+import { MathFx } from './helpers';
+
+export class FxPoolToken extends TokenAmount {
+    public readonly index: number;
+    public readonly latestFXPrice: HumanAmount;
+    public readonly fxOracleDecimals: number;
+    public numeraire: bigint; // in 36 decimals
+    private readonly scalar36 = this.scalar * WAD;
+
+    public constructor(
+        token: Token,
+        amount: BigintIsh,
+        latestFXPrice: HumanAmount,
+        fxOracleDecimals: number,
+        index: number,
+    ) {
+        super(token, amount);
+        this.latestFXPrice = latestFXPrice;
+        this.fxOracleDecimals = fxOracleDecimals;
+        const truncatedNumeraire = MathFx.mulDownFixed(
+            this.amount,
+            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
+            this.fxOracleDecimals,
+        );
+        this.numeraire = truncatedNumeraire * this.scalar36;
+        this.index = index;
+    }
+
+    public increase(amount: bigint): TokenAmount {
+        this.amount = this.amount + amount;
+        this.scale18 = this.amount * this.scalar;
+        const truncatedNumeraire = MathFx.mulDownFixed(
+            this.amount,
+            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
+            this.fxOracleDecimals,
+        );
+        this.numeraire = truncatedNumeraire * this.scalar36;
+        return this;
+    }
+
+    public decrease(amount: bigint): TokenAmount {
+        this.amount = this.amount - amount;
+        this.scale18 = this.amount * this.scalar;
+        const truncatedNumeraire = MathFx.mulDownFixed(
+            this.amount,
+            parseUnits(this.latestFXPrice, this.fxOracleDecimals),
+            this.fxOracleDecimals,
+        );
+        this.numeraire = truncatedNumeraire * this.scalar36;
+        return this;
+    }
+
+    public static fromNumeraire(
+        poolToken: FxPoolToken,
+        numeraire: BigintIsh,
+        divUp?: boolean,
+    ): FxPoolToken {
+        const truncatedNumeraire = BigInt(numeraire) / poolToken.scalar36; // loss of precision required to match SC implementation
+        const amount = divUp
+            ? MathFx.divUpFixed(
+                  BigInt(truncatedNumeraire),
+                  parseUnits(
+                      poolToken.latestFXPrice,
+                      poolToken.fxOracleDecimals,
+                  ),
+                  poolToken.fxOracleDecimals,
+              )
+            : MathFx.divDownFixed(
+                  BigInt(truncatedNumeraire),
+                  parseUnits(
+                      poolToken.latestFXPrice,
+                      poolToken.fxOracleDecimals,
+                  ),
+                  poolToken.fxOracleDecimals,
+              );
+        return new FxPoolToken(
+            poolToken.token,
+            amount,
+            poolToken.latestFXPrice,
+            poolToken.fxOracleDecimals,
+            poolToken.index,
+        );
+    }
+}
diff --git a/src/entities/pools/fx/helpers.ts b/src/entities/pools/fx/helpers.ts
@@ -1,5 +1,5 @@
 import { parseUnits } from 'viem';
-import { HumanAmount } from '../../../types';
+import { HumanAmount } from '../../../data/types';
 
 export class MathFx {
     static mulDownFixed(a: bigint, b: bigint, decimals = 36): bigint {

diff --git a/src/entities/pools/fx/index.ts b/src/entities/pools/fx/index.ts
@@ -1,3 +1,5 @@
 export * from './fxFactory';
 export * from './fxPool';
 export * from './fxMath';
+export * from './types';
+export * from './fxPoolToken';
diff --git a/src/entities/pools/fx/types.ts b/src/entities/pools/fx/types.ts
@@ -0,0 +1,17 @@
+import { SwapKind } from '../../../types';
+import { FxPoolToken } from './fxPoolToken';
+
+export type FxPoolPairData = {
+    tIn: FxPoolToken;
+    tOut: FxPoolToken;
+    alpha: bigint;
+    beta: bigint;
+    delta: bigint;
+    lambda: bigint;
+    _oGLiq: bigint;
+    _nGLiq: bigint;
+    _oBals: bigint[];
+    _nBals: bigint[];
+    givenToken: FxPoolToken;
+    swapKind: SwapKind;
+};
diff --git a/src/entities/pools/gyro2/gyro2Factory.ts b/src/entities/pools/gyro2/gyro2Factory.ts
@@ -1,5 +1,5 @@
-import { Gyro2Pool } from '.';
-import { BasePool, BasePoolFactory } from '..';
+import { Gyro2Pool } from './gyro2Pool';
+import { BasePool, BasePoolFactory } from '../index';
 import { RawGyro2Pool, RawPool } from '../../../data/types';
 
 export class Gyro2PoolFactory implements BasePoolFactory {

diff --git a/src/entities/pools/gyro2/gyro2Pool.ts b/src/entities/pools/gyro2/gyro2Pool.ts
@@ -7,7 +7,8 @@ import {
     _findVirtualParams,
 } from './gyro2Math';
 import { BasePool } from '..';
-import { BigintIsh, Token, TokenAmount } from '../..';
+import { Token } from '../../token';
+import { TokenAmount, BigintIsh } from '../../tokenAmount';
 import { RawGyro2Pool } from '../../../data/types';
 import { PoolType, SwapKind } from '../../../types';
 import { getPoolAddress, MathSol, WAD } from '../../../utils';

diff --git a/src/entities/pools/gyro3/gyro3Factory.ts b/src/entities/pools/gyro3/gyro3Factory.ts
@@ -1,5 +1,5 @@
-import { Gyro3Pool } from '.';
-import { BasePool, BasePoolFactory } from '..';
+import { Gyro3Pool } from './gyro3Pool';
+import { BasePool, BasePoolFactory } from '../index';
 import { RawGyro3Pool, RawPool } from '../../../data/types';
 
 export class Gyro3PoolFactory implements BasePoolFactory {

diff --git a/src/entities/pools/gyro3/gyro3Pool.ts b/src/entities/pools/gyro3/gyro3Pool.ts
@@ -6,7 +6,8 @@ import {
     _calculateInvariant,
 } from './gyro3Math';
 import { BasePool } from '..';
-import { BigintIsh, Token, TokenAmount } from '../..';
+import { Token } from '../../token';
+import { TokenAmount, BigintIsh } from '../../tokenAmount';
 import { RawGyro3Pool } from '../../../data/types';
 import { PoolType, SwapKind } from '../../../types';
 import { getPoolAddress, MathSol, WAD } from '../../../utils';

diff --git a/src/entities/pools/gyroE/gyroEFactory.ts b/src/entities/pools/gyroE/gyroEFactory.ts
@@ -1,5 +1,5 @@
-import { GyroEPool } from '.';
-import { BasePool, BasePoolFactory } from '..';
+import { GyroEPool } from './gyroEPool';
+import { BasePool, BasePoolFactory } from '../index';
 import { RawGyroEPool, RawPool } from '../../../data/types';
 
 export class GyroEPoolFactory implements BasePoolFactory {

diff --git a/src/entities/pools/gyroE/gyroEMath.ts b/src/entities/pools/gyroE/gyroEMath.ts
@@ -11,7 +11,7 @@ import {
     normalizedLiquidityXIn,
     normalizedLiquidityYIn,
 } from './gyroEMathFunctions';
-import { DerivedGyroEParams, GyroEParams, Vector2 } from './gyroEPool';
+import { DerivedGyroEParams, GyroEParams, Vector2 } from './types';
 import { MathGyro, ONE_XP, SMALL } from '../../../utils/gyroHelpers/math';
 
 export function calculateNormalizedLiquidity(

diff --git a/src/entities/pools/gyroE/gyroEMathFunctions.ts b/src/entities/pools/gyroE/gyroEMathFunctions.ts
@@ -1,7 +1,7 @@
 import { WAD } from '../../../utils';
 import { MathGyro } from '../../../utils/gyroHelpers/math';
 import { virtualOffset0, virtualOffset1 } from './gyroEMathHelpers';
-import { DerivedGyroEParams, GyroEParams, Vector2 } from './gyroEPool';
+import { DerivedGyroEParams, GyroEParams, Vector2 } from './types';
 
 /////////
 /// SPOT PRICE DERIVATIVE CALCULATIONS

diff --git a/src/entities/pools/gyroE/gyroEMathHelpers.ts b/src/entities/pools/gyroE/gyroEMathHelpers.ts
@@ -1,5 +1,5 @@
 import { MAX_BALANCES } from './constants';
-import { DerivedGyroEParams, GyroEParams, Vector2 } from './gyroEPool';
+import { DerivedGyroEParams, GyroEParams, Vector2 } from './types';
 import { MathGyro, ONE_XP } from '../../../utils/gyroHelpers/math';
 
 /////////