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

export type WithdrawFundParams = {
  accounts?: {
    mint: PublicKey;
  },
  inputs: {
    amount: BN;
  };
};

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

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

  return ix;
}