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 UpdateFlexibleVaultParams = {
  accounts?: {
    mint: PublicKey;
  }
  inputs: {
    rewardRate: BN;
    withdrawalDelayTime: BN;
  };
};

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

  const ix = await program.methods
    .updateFlexibleVault(
      inputs.rewardRate,
      inputs.withdrawalDelayTime,
    )
    .accounts({
      ...accounts,
    })
    .instruction();

  return ix;
}