/**
 * This code was AUTOGENERATED using the Codama library.
 * Please DO NOT EDIT THIS FILE, instead use visitors
 * to add features, then rerun Codama to update it.
 *
 * @see https://github.com/codama-idl/codama
 */

import {
  assertAccountExists,
  assertAccountsExist,
  combineCodec,
  decodeAccount,
  fetchEncodedAccount,
  fetchEncodedAccounts,
  fixDecoderSize,
  fixEncoderSize,
  getAddressDecoder,
  getAddressEncoder,
  getBytesDecoder,
  getBytesEncoder,
  getStructDecoder,
  getStructEncoder,
  getU128Decoder,
  getU128Encoder,
  getU64Decoder,
  getU64Encoder,
  getU8Decoder,
  getU8Encoder,
  transformEncoder,
  type Account,
  type Address,
  type EncodedAccount,
  type FetchAccountConfig,
  type FetchAccountsConfig,
  type FixedSizeCodec,
  type FixedSizeDecoder,
  type FixedSizeEncoder,
  type MaybeAccount,
  type MaybeEncodedAccount,
  type ReadonlyUint8Array,
} from '@solana/kit';
import { findStarbaseTreasuryPda, StarbaseTreasurySeeds } from '../pdas';

export const STARBASE_TREASURY_DISCRIMINATOR = new Uint8Array([
  3, 184, 34, 13, 195, 64, 187, 141,
]);

export function getStarbaseTreasuryDiscriminatorBytes() {
  return fixEncoderSize(getBytesEncoder(), 8).encode(
    STARBASE_TREASURY_DISCRIMINATOR,
  );
}

export type StarbaseTreasury = {
  discriminator: ReadonlyUint8Array;
  version: number;
  governance: Address;
  successorTreasury: Address;
  bump: number;
  atlasVault: bigint;
  rewardIndex: bigint;
  totalDistributed: bigint;
  undistributedRewards: bigint;
};

export type StarbaseTreasuryArgs = {
  version: number;
  governance: Address;
  successorTreasury: Address;
  bump: number;
  atlasVault: number | bigint;
  rewardIndex: number | bigint;
  totalDistributed: number | bigint;
  undistributedRewards: number | bigint;
};

/** Gets the encoder for {@link StarbaseTreasuryArgs} account data. */
export function getStarbaseTreasuryEncoder(): FixedSizeEncoder<StarbaseTreasuryArgs> {
  return transformEncoder(
    getStructEncoder([
      ['discriminator', fixEncoderSize(getBytesEncoder(), 8)],
      ['version', getU8Encoder()],
      ['governance', getAddressEncoder()],
      ['successorTreasury', getAddressEncoder()],
      ['bump', getU8Encoder()],
      ['atlasVault', getU64Encoder()],
      ['rewardIndex', getU128Encoder()],
      ['totalDistributed', getU64Encoder()],
      ['undistributedRewards', getU64Encoder()],
    ]),
    (value) => ({ ...value, discriminator: STARBASE_TREASURY_DISCRIMINATOR }),
  );
}

/** Gets the decoder for {@link StarbaseTreasury} account data. */
export function getStarbaseTreasuryDecoder(): FixedSizeDecoder<StarbaseTreasury> {
  return getStructDecoder([
    ['discriminator', fixDecoderSize(getBytesDecoder(), 8)],
    ['version', getU8Decoder()],
    ['governance', getAddressDecoder()],
    ['successorTreasury', getAddressDecoder()],
    ['bump', getU8Decoder()],
    ['atlasVault', getU64Decoder()],
    ['rewardIndex', getU128Decoder()],
    ['totalDistributed', getU64Decoder()],
    ['undistributedRewards', getU64Decoder()],
  ]);
}

/** Gets the codec for {@link StarbaseTreasury} account data. */
export function getStarbaseTreasuryCodec(): FixedSizeCodec<
  StarbaseTreasuryArgs,
  StarbaseTreasury
> {
  return combineCodec(
    getStarbaseTreasuryEncoder(),
    getStarbaseTreasuryDecoder(),
  );
}

export function decodeStarbaseTreasury<TAddress extends string = string>(
  encodedAccount: EncodedAccount<TAddress>,
): Account<StarbaseTreasury, TAddress>;
export function decodeStarbaseTreasury<TAddress extends string = string>(
  encodedAccount: MaybeEncodedAccount<TAddress>,
): MaybeAccount<StarbaseTreasury, TAddress>;
export function decodeStarbaseTreasury<TAddress extends string = string>(
  encodedAccount: EncodedAccount<TAddress> | MaybeEncodedAccount<TAddress>,
):
  | Account<StarbaseTreasury, TAddress>
  | MaybeAccount<StarbaseTreasury, TAddress> {
  return decodeAccount(
    encodedAccount as MaybeEncodedAccount<TAddress>,
    getStarbaseTreasuryDecoder(),
  );
}

export async function fetchStarbaseTreasury<TAddress extends string = string>(
  rpc: Parameters<typeof fetchEncodedAccount>[0],
  address: Address<TAddress>,
  config?: FetchAccountConfig,
): Promise<Account<StarbaseTreasury, TAddress>> {
  const maybeAccount = await fetchMaybeStarbaseTreasury(rpc, address, config);
  assertAccountExists(maybeAccount);
  return maybeAccount;
}

export async function fetchMaybeStarbaseTreasury<
  TAddress extends string = string,
>(
  rpc: Parameters<typeof fetchEncodedAccount>[0],
  address: Address<TAddress>,
  config?: FetchAccountConfig,
): Promise<MaybeAccount<StarbaseTreasury, TAddress>> {
  const maybeAccount = await fetchEncodedAccount(rpc, address, config);
  return decodeStarbaseTreasury(maybeAccount);
}

export async function fetchAllStarbaseTreasury(
  rpc: Parameters<typeof fetchEncodedAccounts>[0],
  addresses: Array<Address>,
  config?: FetchAccountsConfig,
): Promise<Account<StarbaseTreasury>[]> {
  const maybeAccounts = await fetchAllMaybeStarbaseTreasury(
    rpc,
    addresses,
    config,
  );
  assertAccountsExist(maybeAccounts);
  return maybeAccounts;
}

export async function fetchAllMaybeStarbaseTreasury(
  rpc: Parameters<typeof fetchEncodedAccounts>[0],
  addresses: Array<Address>,
  config?: FetchAccountsConfig,
): Promise<MaybeAccount<StarbaseTreasury>[]> {
  const maybeAccounts = await fetchEncodedAccounts(rpc, addresses, config);
  return maybeAccounts.map((maybeAccount) =>
    decodeStarbaseTreasury(maybeAccount),
  );
}

export async function fetchStarbaseTreasuryFromSeeds(
  rpc: Parameters<typeof fetchEncodedAccount>[0],
  seeds: StarbaseTreasurySeeds,
  config: FetchAccountConfig & { programAddress?: Address } = {},
): Promise<Account<StarbaseTreasury>> {
  const maybeAccount = await fetchMaybeStarbaseTreasuryFromSeeds(
    rpc,
    seeds,
    config,
  );
  assertAccountExists(maybeAccount);
  return maybeAccount;
}

export async function fetchMaybeStarbaseTreasuryFromSeeds(
  rpc: Parameters<typeof fetchEncodedAccount>[0],
  seeds: StarbaseTreasurySeeds,
  config: FetchAccountConfig & { programAddress?: Address } = {},
): Promise<MaybeAccount<StarbaseTreasury>> {
  const { programAddress, ...fetchConfig } = config;
  const [address] = await findStarbaseTreasuryPda(seeds, { programAddress });
  return await fetchMaybeStarbaseTreasury(rpc, address, fetchConfig);
}