import { AnyNumber } from "@aptos-labs/ts-sdk";

interface PositionToRemove {
  positionAddr: string;
  mintedShare: AnyNumber;
  minAmount0: AnyNumber;
  minAmount1: AnyNumber;
}

export interface RemoveSingleAMMLiquidityParams extends PositionToRemove {
  poolId: string;
}

export interface RemoveMultipleAMMLiquidityParams {
  poolId: string;
  positions: PositionToRemove[];
}

export interface RemoveSingleCLMMLiquidityParams extends PositionToRemove {
  poolId: string;
}

export interface RemoveMultipleCLMMLiquidityParams {
  poolId: string;
  positions: PositionToRemove[];
}

export interface RemoveSingleStableLiquidityParams {
  poolId: string;
  liquidityType: number;
  position: {
    positionAddr: string;
    mintedShare: AnyNumber;
    amounts: AnyNumber[];
  };
}

export interface RemoveMultipleStableLiquidityParams {
  poolId: string;
  liquidityType: number;
  positions: {
    positionAddr: string;
    mintedShare: AnyNumber;
    amounts: AnyNumber[];
  }[];
}

export interface AddAMMLiquidityParams {
  poolId: string;
  amountA: number;
  amountB: number;
}

export interface AddCLMMLiquidityParams {
  poolId: string;
  minPrice: number;
  maxPrice: number;
  amountA: number;
  amountB: number;
  fee: number;
  isMaxAmountB: boolean;
}

export interface AddStableLiquidityParams {
  poolId: string;
  amounts: number[];
}

export interface CreateAMMPoolAndAddLiquidityParams {
  tokenAddress: string[];
  fee: number;
  amounts: number[];
}

export interface CreateCLMMPoolAndAddLiquidityParams {
  tokenAddress: string[];
  fee: number;
  amounts: number[];
  initialPrice: number;
  minPrice: number;
  maxPrice: number;
  isMaxAmountB: boolean;
}

export interface CreateStablePoolAndAddLiquidityParams {
  tokenAddress: string[];
  fee: number;
  amounts: number[];
  amplificationFactor: number;
  offpeg_fee_multiplier?: number;
}

export interface CollectFeeParams {
  poolId: string;
  positionAddr: string;
}