import { PublicKey, SYSVAR_INSTRUCTIONS_PUBKEY } from '@solana/web3.js';
import * as anchor from '@coral-xyz/anchor';
import { KaminoReserve } from '../classes';
import {
  FlashBorrowReserveLiquidityArgs,
  FlashBorrowReserveLiquidityAccounts,
  flashBorrowReserveLiquidity,
  FlashRepayReserveLiquidityArgs,
  FlashRepayReserveLiquidityAccounts,
  flashRepayReserveLiquidity,
} from '../lib';
import Decimal from 'decimal.js';

export const getFlashLoanInstructions = (args: {
  borrowIxIndex: number;
  walletPublicKey: PublicKey;
  lendingMarketAuthority: PublicKey;
  lendingMarketAddress: PublicKey;
  reserve: KaminoReserve;
  amountLamports: Decimal;
  destinationAta: PublicKey;
  referrerAccount: PublicKey;
  referrerTokenState: PublicKey;
  programId: PublicKey;
}) => {
  const flashBorrowIx = getBorrowFlashLoanInstruction({
    walletPublicKey: args.walletPublicKey,
    lendingMarketAuthority: args.lendingMarketAuthority,
    lendingMarketAddress: args.lendingMarketAddress,
    reserve: args.reserve,
    amountLamports: args.amountLamports,
    destinationAta: args.destinationAta,
    referrerAccount: args.referrerAccount,
    referrerTokenState: args.referrerTokenState,
    programId: args.programId,
  });
  const flashRepayIx = getRepayFlashLoanInstruction({
    borrowIxIndex: args.borrowIxIndex,
    walletPublicKey: args.walletPublicKey,
    lendingMarketAuthority: args.lendingMarketAuthority,
    lendingMarketAddress: args.lendingMarketAddress,
    reserve: args.reserve,
    amountLamports: args.amountLamports,
    userSourceLiquidity: args.destinationAta,
    referrerAccount: args.referrerAccount,
    referrerTokenState: args.referrerTokenState,
    programId: args.programId,
  });

  return { flashBorrowIx, flashRepayIx };
};

export const getBorrowFlashLoanInstruction = ({
  walletPublicKey,
  lendingMarketAuthority,
  lendingMarketAddress,
  reserve,
  amountLamports,
  destinationAta,
  referrerAccount,
  referrerTokenState,
  programId,
}: {
  walletPublicKey: PublicKey;
  lendingMarketAuthority: PublicKey;
  lendingMarketAddress: PublicKey;
  reserve: KaminoReserve;
  amountLamports: Decimal;
  destinationAta: PublicKey;
  referrerAccount: PublicKey;
  referrerTokenState: PublicKey;
  programId: PublicKey;
}) => {
  const args: FlashBorrowReserveLiquidityArgs = {
    liquidityAmount: new anchor.BN(amountLamports.floor().toString()),
  };
  const accounts: FlashBorrowReserveLiquidityAccounts = {
    userTransferAuthority: walletPublicKey,
    lendingMarketAuthority: lendingMarketAuthority,
    lendingMarket: lendingMarketAddress,
    reserve: reserve.address,
    reserveLiquidityMint: reserve.getLiquidityMint(),
    reserveSourceLiquidity: reserve.state.liquidity.supplyVault,
    userDestinationLiquidity: destinationAta,
    referrerAccount: referrerAccount,
    referrerTokenState: referrerTokenState,
    reserveLiquidityFeeReceiver: reserve.state.liquidity.feeVault,
    sysvarInfo: SYSVAR_INSTRUCTIONS_PUBKEY,
    tokenProgram: reserve.getLiquidityTokenProgram(),
  };

  return flashBorrowReserveLiquidity(args, accounts, programId);
};

export const getRepayFlashLoanInstruction = ({
  borrowIxIndex,
  walletPublicKey,
  lendingMarketAuthority,
  lendingMarketAddress,
  reserve,
  amountLamports,
  userSourceLiquidity,
  referrerAccount,
  referrerTokenState,
  programId,
}: {
  borrowIxIndex: number;
  walletPublicKey: PublicKey;
  lendingMarketAuthority: PublicKey;
  lendingMarketAddress: PublicKey;
  reserve: KaminoReserve;
  amountLamports: Decimal;
  userSourceLiquidity: PublicKey;
  referrerAccount: PublicKey;
  referrerTokenState: PublicKey;
  programId: PublicKey;
}) => {
  const args: FlashRepayReserveLiquidityArgs = {
    borrowInstructionIndex: borrowIxIndex,
    liquidityAmount: new anchor.BN(amountLamports.floor().toString()),
  };

  const accounts: FlashRepayReserveLiquidityAccounts = {
    userTransferAuthority: walletPublicKey,
    lendingMarketAuthority: lendingMarketAuthority,
    lendingMarket: lendingMarketAddress,
    reserve: reserve.address,
    reserveLiquidityMint: reserve.getLiquidityMint(),
    reserveDestinationLiquidity: reserve.state.liquidity.supplyVault,
    userSourceLiquidity: userSourceLiquidity,
    referrerAccount: referrerAccount,
    referrerTokenState: referrerTokenState,
    reserveLiquidityFeeReceiver: reserve.state.liquidity.feeVault,
    sysvarInfo: SYSVAR_INSTRUCTIONS_PUBKEY,
    tokenProgram: reserve.getLiquidityTokenProgram(),
  };

  return flashRepayReserveLiquidity(args, accounts, programId);
};