import { ComputeBudgetProgram, TransactionInstruction } from "@solana/web3.js";

import { ComputeBudgetConfig, InstructionType, TxVersion } from "../base";

export function addComputeBudget(config: ComputeBudgetConfig) {
  const ins: TransactionInstruction[] = []
  const insTypes: InstructionType[] = []
  if (config.microLamports) {
    ins.push(ComputeBudgetProgram.setComputeUnitPrice({microLamports: config.microLamports}))
    insTypes.push(InstructionType.setComputeUnitPrice)
  }
  if (config.units) {
    ins.push(ComputeBudgetProgram.setComputeUnitLimit({units: config.units}))
    insTypes.push(InstructionType.setComputeUnitPrice)
  }

  return {
    address: {},
    innerTransaction: {
      instructions: ins,
      signers: [],
      instructionTypes: insTypes,
      supportedVersion: [TxVersion.LEGACY, TxVersion.V0]
    }
  }
}