import { Transaction } from "@mysten/sui/transactions";
import { AggregatorConfig } from "../config";

export async function makeCETUSPTB(
  txb: Transaction,
  poolId: string,
  byAmountIn: boolean,
  coinA: any,
  amount: any,
  a2b: boolean,
  typeArguments: any
) {
  let coinTypeA = typeArguments[0];
  let coinTypeB = typeArguments[1];

  const sqrtPriceLimit = BigInt(
    a2b ? "4295048016" : "79226673515401279992447579055"
  );

  const [cetusReceiveA, cetusReceiveB, cetusflashReceipt] = txb.moveCall({
    target: `${AggregatorConfig.cetusPackageId}::pool::flash_swap`,
    arguments: [
      txb.object(AggregatorConfig.cetusConfigId),
      txb.object(poolId),
      txb.pure.bool(a2b),
      txb.pure.bool(byAmountIn),
      amount,
      txb.pure.u128(sqrtPriceLimit),
      txb.object(AggregatorConfig.clockAddress),
    ],
    typeArguments: [coinTypeA, coinTypeB],
  });
  txb.moveCall({
    target: `${AggregatorConfig.cetusPackageId}::pool::swap_pay_amount`,
    arguments: [cetusflashReceipt],
    typeArguments: [coinTypeA, coinTypeB],
  });

  if (a2b) {
    const pay_coin_b = txb.moveCall({
      target: "0x2::balance::zero",
      typeArguments: [coinTypeB],
    });

    txb.moveCall({
      target: `${AggregatorConfig.cetusPackageId}::pool::repay_flash_swap`,
      arguments: [
        txb.object(AggregatorConfig.cetusConfigId),
        txb.object(poolId),
        coinA,
        pay_coin_b,
        cetusflashReceipt,
      ],
      typeArguments: [coinTypeA, coinTypeB],
    });

    const coin_a = txb.moveCall({
      target: `0x2::coin::from_balance`,
      arguments: [cetusReceiveA],
      typeArguments: [coinTypeA],
    });
    const receive_coin_b = txb.moveCall({
      target: `0x2::coin::from_balance`,
      arguments: [cetusReceiveB],
      typeArguments: [coinTypeB],
    });

    return { receiveCoin: receive_coin_b, leftCoin: coin_a };
  }

  const [pay_coin_a] = txb.moveCall({
    target: "0x2::balance::zero",
    typeArguments: [coinTypeA],
  });

  txb.moveCall({
    target: `${AggregatorConfig.cetusPackageId}::pool::repay_flash_swap`,
    arguments: [
      txb.object(AggregatorConfig.cetusConfigId),
      txb.object(poolId),
      pay_coin_a,
      coinA,
      cetusflashReceipt,
    ],
    typeArguments: [coinTypeA, coinTypeB],
  });

  const leftCoin: any = txb.moveCall({
    target: `0x2::coin::from_balance`,
    arguments: [cetusReceiveB],
    typeArguments: [coinTypeB],
  });
  const receiveCoin: any = txb.moveCall({
    target: `0x2::coin::from_balance`,
    arguments: [cetusReceiveA],
    typeArguments: [coinTypeA],
  });

  return { receiveCoin, leftCoin };
}