/**
 * 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, fixDecoderSize, fixEncoderSize, getBooleanDecoder, getBooleanEncoder, getBytesDecoder, getBytesEncoder, getStructDecoder, getStructEncoder, getU128Decoder, getU128Encoder, getU64Decoder, getU64Encoder, transformEncoder, type AccountMeta, type AccountSignerMeta, type Address, type FixedSizeCodec, type FixedSizeDecoder, type FixedSizeEncoder, type Instruction, type InstructionWithAccounts, type InstructionWithData, type ReadonlyAccount, type ReadonlySignerAccount, type ReadonlyUint8Array, type TransactionSigner, type WritableAccount } from '@solana/kit';
import { AMM_V3_PROGRAM_ADDRESS } from '../programs';
import { getAccountMetaFactory, type ResolvedAccount } from '../shared';

export const SWAP_DISCRIMINATOR = new Uint8Array([248, 198, 158, 145, 225, 117, 135, 200]);

export function getSwapDiscriminatorBytes() { return fixEncoderSize(getBytesEncoder(), 8).encode(SWAP_DISCRIMINATOR); }

export type SwapInstruction<TProgram extends string = typeof AMM_V3_PROGRAM_ADDRESS, TAccountPayer extends string | AccountMeta<string> = string, TAccountAmmConfig extends string | AccountMeta<string> = string, TAccountPoolState extends string | AccountMeta<string> = string, TAccountInputTokenAccount extends string | AccountMeta<string> = string, TAccountOutputTokenAccount extends string | AccountMeta<string> = string, TAccountInputVault extends string | AccountMeta<string> = string, TAccountOutputVault extends string | AccountMeta<string> = string, TAccountObservationState extends string | AccountMeta<string> = string, TAccountTokenProgram extends string | AccountMeta<string> = "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA", TAccountTickArray extends string | AccountMeta<string> = string, TRemainingAccounts extends readonly AccountMeta<string>[] = []> =
Instruction<TProgram> & InstructionWithData<ReadonlyUint8Array> & InstructionWithAccounts<[TAccountPayer extends string ? ReadonlySignerAccount<TAccountPayer> & AccountSignerMeta<TAccountPayer> : TAccountPayer, TAccountAmmConfig extends string ? ReadonlyAccount<TAccountAmmConfig> : TAccountAmmConfig, TAccountPoolState extends string ? WritableAccount<TAccountPoolState> : TAccountPoolState, TAccountInputTokenAccount extends string ? WritableAccount<TAccountInputTokenAccount> : TAccountInputTokenAccount, TAccountOutputTokenAccount extends string ? WritableAccount<TAccountOutputTokenAccount> : TAccountOutputTokenAccount, TAccountInputVault extends string ? WritableAccount<TAccountInputVault> : TAccountInputVault, TAccountOutputVault extends string ? WritableAccount<TAccountOutputVault> : TAccountOutputVault, TAccountObservationState extends string ? WritableAccount<TAccountObservationState> : TAccountObservationState, TAccountTokenProgram extends string ? ReadonlyAccount<TAccountTokenProgram> : TAccountTokenProgram, TAccountTickArray extends string ? WritableAccount<TAccountTickArray> : TAccountTickArray, ...TRemainingAccounts]>;

export type SwapInstructionData = { discriminator: ReadonlyUint8Array; amount: bigint; otherAmountThreshold: bigint; sqrtPriceLimitX64: bigint; isBaseInput: boolean;  };

export type SwapInstructionDataArgs = { amount: number | bigint; otherAmountThreshold: number | bigint; sqrtPriceLimitX64: number | bigint; isBaseInput: boolean;  };

export function getSwapInstructionDataEncoder(): FixedSizeEncoder<SwapInstructionDataArgs> {
    return transformEncoder(getStructEncoder([['discriminator', fixEncoderSize(getBytesEncoder(), 8)], ['amount', getU64Encoder()], ['otherAmountThreshold', getU64Encoder()], ['sqrtPriceLimitX64', getU128Encoder()], ['isBaseInput', getBooleanEncoder()]]), (value) => ({ ...value, discriminator: SWAP_DISCRIMINATOR }));
}

export function getSwapInstructionDataDecoder(): FixedSizeDecoder<SwapInstructionData> {
    return getStructDecoder([['discriminator', fixDecoderSize(getBytesDecoder(), 8)], ['amount', getU64Decoder()], ['otherAmountThreshold', getU64Decoder()], ['sqrtPriceLimitX64', getU128Decoder()], ['isBaseInput', getBooleanDecoder()]]);
}

export function getSwapInstructionDataCodec(): FixedSizeCodec<SwapInstructionDataArgs, SwapInstructionData> {
    return combineCodec(getSwapInstructionDataEncoder(), getSwapInstructionDataDecoder());
}

export type SwapInput<TAccountPayer extends string = string, TAccountAmmConfig extends string = string, TAccountPoolState extends string = string, TAccountInputTokenAccount extends string = string, TAccountOutputTokenAccount extends string = string, TAccountInputVault extends string = string, TAccountOutputVault extends string = string, TAccountObservationState extends string = string, TAccountTokenProgram extends string = string, TAccountTickArray extends string = string> =  {
  payer: TransactionSigner<TAccountPayer>;
ammConfig: Address<TAccountAmmConfig>;
poolState: Address<TAccountPoolState>;
inputTokenAccount: Address<TAccountInputTokenAccount>;
outputTokenAccount: Address<TAccountOutputTokenAccount>;
inputVault: Address<TAccountInputVault>;
outputVault: Address<TAccountOutputVault>;
observationState: Address<TAccountObservationState>;
tokenProgram?: Address<TAccountTokenProgram>;
tickArray: Address<TAccountTickArray>;
amount: SwapInstructionDataArgs["amount"];
otherAmountThreshold: SwapInstructionDataArgs["otherAmountThreshold"];
sqrtPriceLimitX64: SwapInstructionDataArgs["sqrtPriceLimitX64"];
isBaseInput: SwapInstructionDataArgs["isBaseInput"];
}

export function getSwapInstruction<TAccountPayer extends string, TAccountAmmConfig extends string, TAccountPoolState extends string, TAccountInputTokenAccount extends string, TAccountOutputTokenAccount extends string, TAccountInputVault extends string, TAccountOutputVault extends string, TAccountObservationState extends string, TAccountTokenProgram extends string, TAccountTickArray extends string, TProgramAddress extends Address = typeof AMM_V3_PROGRAM_ADDRESS>(input: SwapInput<TAccountPayer, TAccountAmmConfig, TAccountPoolState, TAccountInputTokenAccount, TAccountOutputTokenAccount, TAccountInputVault, TAccountOutputVault, TAccountObservationState, TAccountTokenProgram, TAccountTickArray>, config?: { programAddress?: TProgramAddress } ): SwapInstruction<TProgramAddress, TAccountPayer, TAccountAmmConfig, TAccountPoolState, TAccountInputTokenAccount, TAccountOutputTokenAccount, TAccountInputVault, TAccountOutputVault, TAccountObservationState, TAccountTokenProgram, TAccountTickArray> {
  // Program address.
const programAddress = config?.programAddress ?? AMM_V3_PROGRAM_ADDRESS;

 // Original accounts.
const originalAccounts = { payer: { value: input.payer ?? null, isWritable: false }, ammConfig: { value: input.ammConfig ?? null, isWritable: false }, poolState: { value: input.poolState ?? null, isWritable: true }, inputTokenAccount: { value: input.inputTokenAccount ?? null, isWritable: true }, outputTokenAccount: { value: input.outputTokenAccount ?? null, isWritable: true }, inputVault: { value: input.inputVault ?? null, isWritable: true }, outputVault: { value: input.outputVault ?? null, isWritable: true }, observationState: { value: input.observationState ?? null, isWritable: true }, tokenProgram: { value: input.tokenProgram ?? null, isWritable: false }, tickArray: { value: input.tickArray ?? null, isWritable: true } }
const accounts = originalAccounts as Record<keyof typeof originalAccounts, ResolvedAccount>;


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


// Resolve default values.
if (!accounts.tokenProgram.value) {
accounts.tokenProgram.value = 'TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA' as Address<'TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA'>;
}

const getAccountMeta = getAccountMetaFactory(programAddress, 'programId');
return Object.freeze({ accounts: [getAccountMeta(accounts.payer), getAccountMeta(accounts.ammConfig), getAccountMeta(accounts.poolState), getAccountMeta(accounts.inputTokenAccount), getAccountMeta(accounts.outputTokenAccount), getAccountMeta(accounts.inputVault), getAccountMeta(accounts.outputVault), getAccountMeta(accounts.observationState), getAccountMeta(accounts.tokenProgram), getAccountMeta(accounts.tickArray)], data: getSwapInstructionDataEncoder().encode(args as SwapInstructionDataArgs), programAddress } as SwapInstruction<TProgramAddress, TAccountPayer, TAccountAmmConfig, TAccountPoolState, TAccountInputTokenAccount, TAccountOutputTokenAccount, TAccountInputVault, TAccountOutputVault, TAccountObservationState, TAccountTokenProgram, TAccountTickArray>);
}

export type ParsedSwapInstruction<TProgram extends string = typeof AMM_V3_PROGRAM_ADDRESS, TAccountMetas extends readonly AccountMeta[] = readonly AccountMeta[]> = { programAddress: Address<TProgram>;
accounts: {
payer: TAccountMetas[0];
ammConfig: TAccountMetas[1];
poolState: TAccountMetas[2];
inputTokenAccount: TAccountMetas[3];
outputTokenAccount: TAccountMetas[4];
inputVault: TAccountMetas[5];
outputVault: TAccountMetas[6];
observationState: TAccountMetas[7];
tokenProgram: TAccountMetas[8];
tickArray: TAccountMetas[9];
};
data: SwapInstructionData; };

export function parseSwapInstruction<TProgram extends string, TAccountMetas extends readonly AccountMeta[]>(instruction: Instruction<TProgram> & InstructionWithAccounts<TAccountMetas> & InstructionWithData<ReadonlyUint8Array>): ParsedSwapInstruction<TProgram, TAccountMetas> {
  if (instruction.accounts.length < 10) {
  // TODO: Coded error.
  throw new Error('Not enough accounts');
}
let accountIndex = 0;
const getNextAccount = () => {
  const accountMeta = (instruction.accounts as TAccountMetas)[accountIndex]!;
  accountIndex += 1;
  return accountMeta;
}
  return { programAddress: instruction.programAddress, accounts: { payer: getNextAccount(), ammConfig: getNextAccount(), poolState: getNextAccount(), inputTokenAccount: getNextAccount(), outputTokenAccount: getNextAccount(), inputVault: getNextAccount(), outputVault: getNextAccount(), observationState: getNextAccount(), tokenProgram: getNextAccount(), tickArray: getNextAccount() }, data: getSwapInstructionDataDecoder().decode(instruction.data) };
}