import * as anchor from "@project-serum/anchor";
import { ComputeBudgetProgram } from "@solana/web3.js";

/**
 * This action lets users request additional compute units
 * for their transactions.
 * @param unitsRequested Number of units to be requested
 * @param additionalFee Additional fee to prioritize tx
 * @returns A transaction instruction to be attached in the transaction
 * object for increasing the compute budget.
 */
export const requestComputeUnits = (
    provider: anchor.AnchorProvider,
    unitsRequested: number,
    additionalFee: number
) => {
  const additionalComputeBudgetInstruction = ComputeBudgetProgram.requestUnits({
    units: unitsRequested,
    additionalFee: additionalFee,
  });

  return additionalComputeBudgetInstruction;
};