@paraswap/dex-lib 3.8.18 → 3.8.20-lite-psm

package/src/dex/lite-psm/lite-psm-e2e.test.ts

@@ -0,0 +1,79 @@
+import dotenv from 'dotenv';
+dotenv.config();
+
+import { testE2E } from '../../../tests/utils-e2e';
+import { Tokens, Holders } from '../../../tests/constants-e2e';
+import { Network, ContractMethod, SwapSide } from '../../constants';
+import { StaticJsonRpcProvider } from '@ethersproject/providers';
+import { generateConfig } from '../../config';
+
+describe('LitePsm E2E', () => {
+  const dexKey = 'LitePsm';
+
+  describe('LitePsm MAINNET_V6', () => {
+    const network = Network.MAINNET;
+    const tokens = Tokens[network];
+    const holders = Holders[network];
+    const provider = new StaticJsonRpcProvider(
+      generateConfig(network).privateHttpProvider,
+      network,
+    );
+
+    const tokenASymbol: string = 'USDC';
+    const tokenBSymbol: string = 'DAI';
+
+    const tokenAAmount: string = '10000000';
+    const tokenBAmount: string = '10000000000000000';
+
+    const sideToContractMethods = new Map([
+      [
+        SwapSide.SELL,
+        [
+          ContractMethod.swapExactAmountIn,
+          ContractMethod.swapExactAmountInOutOnMakerPSM,
+        ],
+      ],
+      [
+        SwapSide.BUY,
+        [
+          ContractMethod.swapExactAmountOut,
+          ContractMethod.swapExactAmountInOutOnMakerPSM,
+        ],
+      ],
+    ]);
+
+    sideToContractMethods.forEach((contractMethods, side) =>
+      contractMethods.forEach((contractMethod: ContractMethod) => {
+        describe(`${contractMethod}`, () => {
+          it(`${side}:${tokenASymbol} -> ${tokenBSymbol}`, async () => {
+            await testE2E(
+              tokens[tokenASymbol],
+              tokens[tokenBSymbol],
+              holders[tokenASymbol],
+              side === SwapSide.SELL ? tokenAAmount : tokenBAmount,
+              side,
+              dexKey,
+              contractMethod,
+              network,
+              provider,
+            );
+          });
+
+          it(`${side}:${tokenBSymbol} -> ${tokenASymbol}`, async () => {
+            await testE2E(
+              tokens[tokenBSymbol],
+              tokens[tokenASymbol],
+              holders[tokenBSymbol],
+              side === SwapSide.SELL ? tokenBAmount : tokenAAmount,
+              side,
+              dexKey,
+              contractMethod,
+              network,
+              provider,
+            );
+          });
+        });
+      }),
+    );
+  });
+});

package/src/dex/lite-psm/lite-psm-event-pool.ts

@@ -0,0 +1,213 @@
+import { Interface } from '@ethersproject/abi';
+import { DeepReadonly } from 'ts-essentials';
+import { Contract } from 'web3-eth-contract';
+import { Address, Log, Logger } from '../../types';
+import { StatefulEventSubscriber } from '../../stateful-event-subscriber';
+import { getBigIntPow } from '../../utils';
+import { IDexHelper } from '../../dex-helper/idex-helper';
+import { PoolState, PoolConfig } from './types';
+import { LitePsmConfig } from './config';
+import PsmABI from '../../abi/maker-psm/psm.json';
+import VatABI from '../../abi/maker-psm/vat.json';
+import { erc20Iface } from '../../lib/tokens/utils';
+
+const vatInterface = new Interface(VatABI);
+const psmInterface = new Interface(PsmABI);
+
+const bigIntify = (b: any) => BigInt(b.toString());
+
+export async function getOnChainState(
+  multiContract: Contract,
+  poolConfigs: PoolConfig[],
+  vatAddress: Address,
+  blockNumber: number | 'latest',
+  daiAddress: Address,
+): Promise<PoolState[]> {
+  const callData = poolConfigs
+    .map(c => [
+      {
+        target: c.psmAddress,
+        callData: psmInterface.encodeFunctionData('tin', []),
+      },
+      {
+        target: c.psmAddress,
+        callData: psmInterface.encodeFunctionData('tout', []),
+      },
+      {
+        target: vatAddress,
+        callData: vatInterface.encodeFunctionData('ilks', [c.identifier]),
+      },
+      // dai liquidity
+      // ceiling for `sellGem`
+      {
+        target: daiAddress,
+        callData: erc20Iface.encodeFunctionData('balanceOf', [c.psmAddress]),
+      },
+      // gem liquidity
+      // ceiling for `buyGem`
+      {
+        target: c.gem.address,
+        callData: erc20Iface.encodeFunctionData('balanceOf', [c.pocketAddress]),
+      },
+    ])
+    .flat();
+
+  const res = await multiContract.methods
+    .aggregate(callData)
+    .call({}, blockNumber);
+
+  let i = 0;
+  const result = poolConfigs.map(c => {
+    const tin = bigIntify(
+      psmInterface.decodeFunctionResult('tin', res.returnData[i++])[0],
+    );
+    const tout = bigIntify(
+      psmInterface.decodeFunctionResult('tout', res.returnData[i++])[0],
+    );
+    const ilks = vatInterface.decodeFunctionResult('ilks', res.returnData[i++]);
+    const rate = bigIntify(ilks.rate);
+    const daiBalance = bigIntify(
+      erc20Iface.decodeFunctionResult('balanceOf', res.returnData[i++])[0],
+    );
+    const gemBalance = bigIntify(
+      erc20Iface.decodeFunctionResult('balanceOf', res.returnData[i++])[0],
+    );
+    const response = {
+      tin,
+      tout,
+      daiBalance,
+      gemBalance,
+      rate,
+    };
+
+    return response;
+  });
+
+  return result;
+}
+
+export class LitePsmEventPool extends StatefulEventSubscriber<PoolState> {
+  handlers: {
+    [event: string]: (event: any, pool: PoolState, log: Log) => PoolState;
+  } = {};
+
+  logDecoder: (log: Log) => any;
+
+  to18ConversionFactor: bigint;
+  bytes32Tout =
+    '0x746f757400000000000000000000000000000000000000000000000000000000'; // bytes32('tout')
+  bytes32Tin =
+    '0x74696e0000000000000000000000000000000000000000000000000000000000'; // bytes32('tin')
+
+  daiAddress: Address;
+
+  constructor(
+    parentName: string,
+    protected network: number,
+    protected dexHelper: IDexHelper,
+    logger: Logger,
+    public poolConfig: PoolConfig,
+    protected vatAddress: Address,
+  ) {
+    super(parentName, poolConfig.identifier, dexHelper, logger);
+
+    this.daiAddress =
+      LitePsmConfig[this.parentName][this.network].dai.address.toLowerCase();
+    this.logDecoder = (log: Log) => {
+      const logAddress = log.address.toLowerCase();
+      if (
+        logAddress === this.daiAddress ||
+        logAddress === this.poolConfig.gem.address
+      ) {
+        return erc20Iface.parseLog(log);
+      } else {
+        return psmInterface.parseLog(log);
+      }
+    };
+    this.addressesSubscribed = [
+      poolConfig.psmAddress,
+      this.daiAddress,
+      this.poolConfig.gem.address,
+    ];
+    this.to18ConversionFactor = getBigIntPow(18 - poolConfig.gem.decimals);
+
+    // Add handlers
+    this.handlers['File'] = this.handleFile.bind(this);
+    this.handlers['Transfer'] = this.handleTransfer.bind(this);
+  }
+
+  handleFile(event: any, pool: PoolState, log: Log): PoolState {
+    if (event.args.what === this.bytes32Tin) {
+      pool.tin = bigIntify(event.args.data);
+    } else if (event.args.what === this.bytes32Tout) {
+      pool.tout = bigIntify(event.args.data);
+    }
+    return pool;
+  }
+
+  handleTransfer(event: any, pool: PoolState, log: Log): PoolState {
+    if (log.address.toLowerCase() === this.daiAddress) {
+      return this.handleTransferDAI(event, pool, log);
+    } else if (log.address.toLowerCase() === this.poolConfig.gem.address) {
+      return this.handleTransferGEM(event, pool, log);
+    }
+    return pool;
+  }
+
+  handleTransferDAI(event: any, pool: PoolState, log: Log): PoolState {
+    if (event.args?.from?.toLowerCase() === this.poolConfig.psmAddress) {
+      pool.daiBalance -= bigIntify(event.args.value);
+    } else if (event.args?.to?.toLowerCase() === this.poolConfig.psmAddress) {
+      pool.daiBalance += bigIntify(event.args.value);
+    }
+    return pool;
+  }
+
+  handleTransferGEM(event: any, pool: PoolState, log: Log): PoolState {
+    if (event.args?.from?.toLowerCase() === this.poolConfig.pocketAddress) {
+      pool.gemBalance -= bigIntify(event.args.value);
+    } else if (
+      event.args?.to?.toLowerCase() === this.poolConfig.pocketAddress
+    ) {
+      pool.gemBalance += bigIntify(event.args.value);
+    }
+    return pool;
+  }
+
+  getIdentifier(): string {
+    return `${this.parentName}_${this.poolConfig.psmAddress}`.toLowerCase();
+  }
+
+  protected processLog(
+    state: DeepReadonly<PoolState>,
+    log: Readonly<Log>,
+  ): DeepReadonly<PoolState> | null {
+    try {
+      const event = this.logDecoder(log);
+      if (event.name in this.handlers) {
+        return this.handlers[event.name](event, state, log);
+      }
+      return state;
+    } catch (e) {
+      this.logger.error(
+        `Error_${this.parentName}_processLog could not parse the log with topic ${log.topics}:`,
+        e,
+      );
+      return null;
+    }
+  }
+
+  async generateState(
+    blockNumber: number | 'latest',
+  ): Promise<Readonly<PoolState>> {
+    return (
+      await getOnChainState(
+        this.dexHelper.multiContract,
+        [this.poolConfig],
+        this.vatAddress,
+        blockNumber,
+        this.daiAddress,
+      )
+    )[0];
+  }
+}

package/src/dex/lite-psm/lite-psm-events.test.ts

@@ -0,0 +1,61 @@
+import dotenv from 'dotenv';
+dotenv.config();
+
+import { LitePsmEventPool } from './lite-psm-event-pool';
+import { LitePsmConfig } from './config';
+import { Network } from '../../constants';
+import { DummyDexHelper } from '../../dex-helper/index';
+import { testEventSubscriber } from '../../../tests/utils-events';
+import { PoolState } from './types';
+
+jest.setTimeout(50 * 1000);
+const dexKey = 'LitePsm';
+const network = Network.MAINNET;
+const config = LitePsmConfig[dexKey][network];
+
+function fetchPoolState(
+  makerPsmPool: LitePsmEventPool,
+  blockNumber: number,
+): Promise<PoolState> {
+  return makerPsmPool.generateState(blockNumber);
+}
+
+describe('LitePsm Event', function () {
+  const blockNumbers: { [eventName: string]: number[] } = {
+    // File: [], // never happened
+    SellGem: [20559136, 20559876, 20561086, 20563099, 20563672, 20563887],
+    BuyGem: [
+      20554181, 20555354, 20555452, 20555880, 20556086, 20556431, 20558791,
+    ],
+  };
+
+  describe('LitePsmEventPool', function () {
+    Object.keys(blockNumbers).forEach((event: string) => {
+      blockNumbers[event].forEach((blockNumber: number) => {
+        it(`Should return the correct state after the ${blockNumber}:${event}`, async function () {
+          const dexHelper = new DummyDexHelper(network);
+          const logger = dexHelper.getLogger(dexKey);
+
+          const makerPsmPool = new LitePsmEventPool(
+            dexKey,
+            network,
+            dexHelper,
+            logger,
+            config.pools[0],
+            config.vatAddress,
+          );
+
+          await testEventSubscriber(
+            makerPsmPool,
+            makerPsmPool.addressesSubscribed,
+            (_blockNumber: number) =>
+              fetchPoolState(makerPsmPool, _blockNumber),
+            blockNumber,
+            `${dexKey}_${config.pools[0].psmAddress}`,
+            dexHelper.provider,
+          );
+        });
+      });
+    });
+  });
+});

package/src/dex/lite-psm/lite-psm-integration.test.ts

@@ -0,0 +1,96 @@
+import dotenv from 'dotenv';
+dotenv.config();
+
+import { DummyDexHelper } from '../../dex-helper/index';
+import { Network, SwapSide } from '../../constants';
+import { LitePsm } from './lite-psm';
+import { checkPoolPrices, checkPoolsLiquidity } from '../../../tests/utils';
+import { Tokens } from '../../../tests/constants-e2e';
+import { BI_POWS } from '../../bigint-constants';
+
+const network = Network.MAINNET;
+const TokenASymbol = 'USDC';
+const TokenA = Tokens[network][TokenASymbol];
+
+const TokenBSymbol = 'DAI';
+const TokenB = Tokens[network][TokenBSymbol];
+
+const tokenAAmounts = [0n, BI_POWS[8], 200000000n];
+
+const tokenBAmounts = [0n, BI_POWS[18], 2000000000000000000n];
+
+const dexKey = 'LitePsm';
+
+describe('LitePsm', function () {
+  it('getPoolIdentifiers and getPricesVolume SELL', async function () {
+    const dexHelper = new DummyDexHelper(network);
+    const blocknumber = await dexHelper.web3Provider.eth.getBlockNumber();
+    const litePsm = new LitePsm(network, dexKey, dexHelper);
+
+    await litePsm.initializePricing(blocknumber);
+
+    const pools = await litePsm.getPoolIdentifiers(
+      TokenA,
+      TokenB,
+      SwapSide.SELL,
+      blocknumber,
+    );
+    console.log(`${TokenASymbol} <> ${TokenBSymbol} Pool Identifiers: `, pools);
+
+    expect(pools.length).toBeGreaterThan(0);
+
+    const poolPrices = await litePsm.getPricesVolume(
+      TokenA,
+      TokenB,
+      tokenAAmounts,
+      SwapSide.SELL,
+      blocknumber,
+      pools,
+    );
+    console.log(`${TokenASymbol} <> ${TokenBSymbol} Pool Prices: `, poolPrices);
+
+    expect(poolPrices).not.toBeNull();
+    checkPoolPrices(poolPrices!, tokenAAmounts, SwapSide.SELL, dexKey);
+  });
+
+  it('getPoolIdentifiers and getPricesVolume BUY', async function () {
+    const dexHelper = new DummyDexHelper(network);
+    const blocknumber = await dexHelper.web3Provider.eth.getBlockNumber();
+    const litePsm = new LitePsm(network, dexKey, dexHelper);
+
+    await litePsm.initializePricing(blocknumber);
+
+    const pools = await litePsm.getPoolIdentifiers(
+      TokenA,
+      TokenB,
+      SwapSide.BUY,
+      blocknumber,
+    );
+    console.log(`${TokenASymbol} <> ${TokenBSymbol} Pool Identifiers: `, pools);
+
+    expect(pools.length).toBeGreaterThan(0);
+
+    const poolPrices = await litePsm.getPricesVolume(
+      TokenA,
+      TokenB,
+      tokenBAmounts,
+      SwapSide.BUY,
+      blocknumber,
+      pools,
+    );
+    console.log(`${TokenASymbol} <> ${TokenBSymbol} Pool Prices: `, poolPrices);
+
+    expect(poolPrices).not.toBeNull();
+    checkPoolPrices(poolPrices!, tokenBAmounts, SwapSide.BUY, dexKey);
+  });
+
+  it('getTopPoolsForToken', async function () {
+    const dexHelper = new DummyDexHelper(network);
+    const litePsm = new LitePsm(network, dexKey, dexHelper);
+
+    const poolLiquidity = await litePsm.getTopPoolsForToken(TokenA.address, 10);
+    console.log(`${TokenASymbol} Top Pools:`, poolLiquidity);
+
+    checkPoolsLiquidity(poolLiquidity, TokenA.address, dexKey);
+  });
+});