/**
 * 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 {
    combineCodec,
    getStructDecoder,
    getStructEncoder,
    getU32Decoder,
    getU32Encoder,
    SOLANA_ERROR__PROGRAM_CLIENTS__INSUFFICIENT_ACCOUNT_METAS,
    SolanaError,
    transformEncoder,
    type AccountMeta,
    type AccountSignerMeta,
    type Address,
    type FixedSizeCodec,
    type FixedSizeDecoder,
    type FixedSizeEncoder,
    type Instruction,
    type InstructionWithAccounts,
    type InstructionWithData,
    type ReadonlyAccount,
    type ReadonlyUint8Array,
    type TransactionSigner,
    type WritableAccount,
    type WritableSignerAccount,
} from '@solana/kit';
import { getAccountMetaFactory, type ResolvedInstructionAccount } from '@solana/kit/program-client-core';
import { LOADER_V3_PROGRAM_ADDRESS } from '../programs';

export const EXTEND_PROGRAM_DISCRIMINATOR = 6;

export function getExtendProgramDiscriminatorBytes(): ReadonlyUint8Array {
    return getU32Encoder().encode(EXTEND_PROGRAM_DISCRIMINATOR);
}

export type ExtendProgramInstruction<
    TProgram extends string = typeof LOADER_V3_PROGRAM_ADDRESS,
    TAccountProgramDataAccount extends string | AccountMeta<string> = string,
    TAccountProgramAccount extends string | AccountMeta<string> = string,
    TAccountSystemProgram extends string | AccountMeta<string> = string,
    TAccountPayer extends string | AccountMeta<string> = string,
    TRemainingAccounts extends readonly AccountMeta<string>[] = [],
> = Instruction<TProgram> &
    InstructionWithData<ReadonlyUint8Array> &
    InstructionWithAccounts<
        [
            TAccountProgramDataAccount extends string
                ? WritableAccount<TAccountProgramDataAccount>
                : TAccountProgramDataAccount,
            TAccountProgramAccount extends string ? WritableAccount<TAccountProgramAccount> : TAccountProgramAccount,
            TAccountSystemProgram extends string ? ReadonlyAccount<TAccountSystemProgram> : TAccountSystemProgram,
            TAccountPayer extends string
                ? WritableSignerAccount<TAccountPayer> & AccountSignerMeta<TAccountPayer>
                : TAccountPayer,
            ...TRemainingAccounts,
        ]
    >;

export type ExtendProgramInstructionData = { discriminator: number; additionalBytes: number };

export type ExtendProgramInstructionDataArgs = { additionalBytes: number };

export function getExtendProgramInstructionDataEncoder(): FixedSizeEncoder<ExtendProgramInstructionDataArgs> {
    return transformEncoder(
        getStructEncoder([
            ['discriminator', getU32Encoder()],
            ['additionalBytes', getU32Encoder()],
        ]),
        value => ({ ...value, discriminator: EXTEND_PROGRAM_DISCRIMINATOR }),
    );
}

export function getExtendProgramInstructionDataDecoder(): FixedSizeDecoder<ExtendProgramInstructionData> {
    return getStructDecoder([
        ['discriminator', getU32Decoder()],
        ['additionalBytes', getU32Decoder()],
    ]);
}

export function getExtendProgramInstructionDataCodec(): FixedSizeCodec<
    ExtendProgramInstructionDataArgs,
    ExtendProgramInstructionData
> {
    return combineCodec(getExtendProgramInstructionDataEncoder(), getExtendProgramInstructionDataDecoder());
}

export type ExtendProgramInput<
    TAccountProgramDataAccount extends string = string,
    TAccountProgramAccount extends string = string,
    TAccountSystemProgram extends string = string,
    TAccountPayer extends string = string,
> = {
    /** ProgramData account. */
    programDataAccount: Address<TAccountProgramDataAccount>;
    /** Program account. */
    programAccount: Address<TAccountProgramAccount>;
    /** System program (optional). */
    systemProgram?: Address<TAccountSystemProgram>;
    /** Payer. */
    payer?: TransactionSigner<TAccountPayer>;
    additionalBytes: ExtendProgramInstructionDataArgs['additionalBytes'];
};

export function getExtendProgramInstruction<
    TAccountProgramDataAccount extends string,
    TAccountProgramAccount extends string,
    TAccountSystemProgram extends string,
    TAccountPayer extends string,
    TProgramAddress extends Address = typeof LOADER_V3_PROGRAM_ADDRESS,
>(
    input: ExtendProgramInput<TAccountProgramDataAccount, TAccountProgramAccount, TAccountSystemProgram, TAccountPayer>,
    config?: { programAddress?: TProgramAddress },
): ExtendProgramInstruction<
    TProgramAddress,
    TAccountProgramDataAccount,
    TAccountProgramAccount,
    TAccountSystemProgram,
    TAccountPayer
> {
    // Program address.
    const programAddress = config?.programAddress ?? LOADER_V3_PROGRAM_ADDRESS;

    // Original accounts.
    const originalAccounts = {
        programDataAccount: { value: input.programDataAccount ?? null, isWritable: true },
        programAccount: { value: input.programAccount ?? null, isWritable: true },
        systemProgram: { value: input.systemProgram ?? null, isWritable: false },
        payer: { value: input.payer ?? null, isWritable: true },
    };
    const accounts = originalAccounts as Record<keyof typeof originalAccounts, ResolvedInstructionAccount>;

    // Original args.
    const args = { ...input };

    const getAccountMeta = getAccountMetaFactory(programAddress, 'programId');
    return Object.freeze({
        accounts: [
            getAccountMeta('programDataAccount', accounts.programDataAccount),
            getAccountMeta('programAccount', accounts.programAccount),
            getAccountMeta('systemProgram', accounts.systemProgram),
            getAccountMeta('payer', accounts.payer),
        ],
        data: getExtendProgramInstructionDataEncoder().encode(args as ExtendProgramInstructionDataArgs),
        programAddress,
    } as ExtendProgramInstruction<
        TProgramAddress,
        TAccountProgramDataAccount,
        TAccountProgramAccount,
        TAccountSystemProgram,
        TAccountPayer
    >);
}

export type ParsedExtendProgramInstruction<
    TProgram extends string = typeof LOADER_V3_PROGRAM_ADDRESS,
    TAccountMetas extends readonly AccountMeta[] = readonly AccountMeta[],
> = {
    programAddress: Address<TProgram>;
    accounts: {
        /** ProgramData account. */
        programDataAccount: TAccountMetas[0];
        /** Program account. */
        programAccount: TAccountMetas[1];
        /** System program (optional). */
        systemProgram?: TAccountMetas[2] | undefined;
        /** Payer. */
        payer?: TAccountMetas[3] | undefined;
    };
    data: ExtendProgramInstructionData;
};

export function parseExtendProgramInstruction<TProgram extends string, TAccountMetas extends readonly AccountMeta[]>(
    instruction: Instruction<TProgram> &
        InstructionWithAccounts<TAccountMetas> &
        InstructionWithData<ReadonlyUint8Array>,
): ParsedExtendProgramInstruction<TProgram, TAccountMetas> {
    if (instruction.accounts.length < 4) {
        throw new SolanaError(SOLANA_ERROR__PROGRAM_CLIENTS__INSUFFICIENT_ACCOUNT_METAS, {
            actualAccountMetas: instruction.accounts.length,
            expectedAccountMetas: 4,
        });
    }
    let accountIndex = 0;
    const getNextAccount = () => {
        const accountMeta = (instruction.accounts as TAccountMetas)[accountIndex]!;
        accountIndex += 1;
        return accountMeta;
    };
    const getNextOptionalAccount = () => {
        const accountMeta = getNextAccount();
        return accountMeta.address === LOADER_V3_PROGRAM_ADDRESS ? undefined : accountMeta;
    };
    return {
        programAddress: instruction.programAddress,
        accounts: {
            programDataAccount: getNextAccount(),
            programAccount: getNextAccount(),
            systemProgram: getNextOptionalAccount(),
            payer: getNextOptionalAccount(),
        },
        data: getExtendProgramInstructionDataDecoder().decode(instruction.data),
    };
}