/* eslint-disable @typescript-eslint/no-unused-vars */
import {
  Address,
  isSome,
  AccountMeta,
  AccountSignerMeta,
  Instruction,
  Option,
  TransactionSigner,
} from "@solana/kit"
/* eslint-enable @typescript-eslint/no-unused-vars */
import BN from "bn.js" // eslint-disable-line @typescript-eslint/no-unused-vars
import * as borsh from "@coral-xyz/borsh" // eslint-disable-line @typescript-eslint/no-unused-vars
import { borshAddress } from "../utils" // eslint-disable-line @typescript-eslint/no-unused-vars
import * as types from "../types" // eslint-disable-line @typescript-eslint/no-unused-vars
import { PROGRAM_ID } from "../programId"

export const DISCRIMINATOR = Buffer.from([195, 136, 1, 79, 10, 123, 121, 100])

export interface UpdateReserveAllocationV2Args {
  weight: BN
  cap: BN
  ctokenAllocationCap: BN
}

export interface UpdateReserveAllocationV2Accounts {
  signer: TransactionSigner
  vaultState: Address
  baseVaultAuthority: Address
  reserveCollateralMint: Address
  reserve: Address
  ctokenVault: Address
  reserveWhitelistEntry: Option<Address>
  reserveCollateralTokenProgram: Address
  systemProgram: Address
  rent: Address
}

export const layout = borsh.struct<UpdateReserveAllocationV2Args>([
  borsh.u64("weight"),
  borsh.u64("cap"),
  borsh.u64("ctokenAllocationCap"),
])

export function updateReserveAllocationV2(
  args: UpdateReserveAllocationV2Args,
  accounts: UpdateReserveAllocationV2Accounts,
  remainingAccounts: Array<AccountMeta | AccountSignerMeta> = [],
  programAddress: Address = PROGRAM_ID
) {
  const keys: Array<AccountMeta | AccountSignerMeta> = [
    { address: accounts.signer.address, role: 3, signer: accounts.signer },
    { address: accounts.vaultState, role: 1 },
    { address: accounts.baseVaultAuthority, role: 0 },
    { address: accounts.reserveCollateralMint, role: 1 },
    { address: accounts.reserve, role: 0 },
    { address: accounts.ctokenVault, role: 1 },
    isSome(accounts.reserveWhitelistEntry)
      ? { address: accounts.reserveWhitelistEntry.value, role: 0 }
      : { address: programAddress, role: 0 },
    { address: accounts.reserveCollateralTokenProgram, role: 0 },
    { address: accounts.systemProgram, role: 0 },
    { address: accounts.rent, role: 0 },
    ...remainingAccounts,
  ]
  const buffer = Buffer.alloc(1000)
  const len = layout.encode(
    {
      weight: args.weight,
      cap: args.cap,
      ctokenAllocationCap: args.ctokenAllocationCap,
    },
    buffer
  )
  const data = Buffer.concat([DISCRIMINATOR, buffer]).slice(0, 8 + len)
  const ix: Instruction = { accounts: keys, programAddress, data }
  return ix
}