import { PublicKey, TransactionInstruction } from "@solana/web3.js";
import { BN, Program } from "@coral-xyz/anchor";
import { Staking } from "../artifacts/staking";
import { StakingSpl } from "../artifacts/staking_spl";

export type WithdrawParams = {
  accounts: {
    user: PublicKey;
    withdrawPosition?: PublicKey;
    mint?: PublicKey;
  };
  inputs: {
    withdrawPositionBump: number;
    index: number;
    amount: BN;
  };
};

export async function withdraw(
  program: Program<Staking> | Program<StakingSpl>,
  params: WithdrawParams,
): Promise<TransactionInstruction> {
  const { accounts, inputs } = params;

  const ix = await program.methods
    .withdraw(inputs.withdrawPositionBump, inputs.index, inputs.amount)
    .accounts({
      ...accounts,
    })
    .instruction();

  return ix;
}