use borsh::BorshDeserialize;
use borsh::BorshSerialize;
#[derive(Debug)]
pub struct Swap {
pub payer: solana_pubkey::Pubkey,
pub amm_config: solana_pubkey::Pubkey,
pub pool_state: solana_pubkey::Pubkey,
pub input_token_account: solana_pubkey::Pubkey,
pub output_token_account: solana_pubkey::Pubkey,
pub input_vault: solana_pubkey::Pubkey,
pub output_vault: solana_pubkey::Pubkey,
pub observation_state: solana_pubkey::Pubkey,
pub token_program: solana_pubkey::Pubkey,
pub tick_array: solana_pubkey::Pubkey,
}
impl Swap {
pub fn instruction(&self, args: SwapInstructionArgs) -> solana_instruction::Instruction {
self.instruction_with_remaining_accounts(args, &[])
}
#[allow(clippy::arithmetic_side_effects)]
#[allow(clippy::vec_init_then_push)]
pub fn instruction_with_remaining_accounts(
&self,
args: SwapInstructionArgs,
remaining_accounts: &[solana_instruction::AccountMeta],
) -> solana_instruction::Instruction {
let mut accounts = Vec::with_capacity(10 + remaining_accounts.len());
accounts.push(solana_instruction::AccountMeta::new_readonly(
self.payer, true,
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
self.amm_config,
false,
));
accounts.push(solana_instruction::AccountMeta::new(self.pool_state, false));
accounts.push(solana_instruction::AccountMeta::new(
self.input_token_account,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
self.output_token_account,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
self.input_vault,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
self.output_vault,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
self.observation_state,
false,
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
self.token_program,
false,
));
accounts.push(solana_instruction::AccountMeta::new(self.tick_array, false));
accounts.extend_from_slice(remaining_accounts);
let mut data = borsh::to_vec(&SwapInstructionData::new()).unwrap();
let mut args = borsh::to_vec(&args).unwrap();
data.append(&mut args);
solana_instruction::Instruction {
program_id: crate::AMM_V3_ID,
accounts,
data,
}
}
}
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SwapInstructionData {
discriminator: [u8; 8],
}
impl SwapInstructionData {
pub fn new() -> Self {
Self {
discriminator: [248, 198, 158, 145, 225, 117, 135, 200],
}
}
}
impl Default for SwapInstructionData {
fn default() -> Self {
Self::new()
}
}
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SwapInstructionArgs {
pub amount: u64,
pub other_amount_threshold: u64,
pub sqrt_price_limit_x64: u128,
pub is_base_input: bool,
}
#[derive(Clone, Debug, Default)]
pub struct SwapBuilder {
payer: Option<solana_pubkey::Pubkey>,
amm_config: Option<solana_pubkey::Pubkey>,
pool_state: Option<solana_pubkey::Pubkey>,
input_token_account: Option<solana_pubkey::Pubkey>,
output_token_account: Option<solana_pubkey::Pubkey>,
input_vault: Option<solana_pubkey::Pubkey>,
output_vault: Option<solana_pubkey::Pubkey>,
observation_state: Option<solana_pubkey::Pubkey>,
token_program: Option<solana_pubkey::Pubkey>,
tick_array: Option<solana_pubkey::Pubkey>,
amount: Option<u64>,
other_amount_threshold: Option<u64>,
sqrt_price_limit_x64: Option<u128>,
is_base_input: Option<bool>,
__remaining_accounts: Vec<solana_instruction::AccountMeta>,
}
impl SwapBuilder {
pub fn new() -> Self {
Self::default()
}
#[inline(always)]
pub fn payer(&mut self, payer: solana_pubkey::Pubkey) -> &mut Self {
self.payer = Some(payer);
self
}
#[inline(always)]
pub fn amm_config(&mut self, amm_config: solana_pubkey::Pubkey) -> &mut Self {
self.amm_config = Some(amm_config);
self
}
#[inline(always)]
pub fn pool_state(&mut self, pool_state: solana_pubkey::Pubkey) -> &mut Self {
self.pool_state = Some(pool_state);
self
}
#[inline(always)]
pub fn input_token_account(&mut self, input_token_account: solana_pubkey::Pubkey) -> &mut Self {
self.input_token_account = Some(input_token_account);
self
}
#[inline(always)]
pub fn output_token_account(
&mut self,
output_token_account: solana_pubkey::Pubkey,
) -> &mut Self {
self.output_token_account = Some(output_token_account);
self
}
#[inline(always)]
pub fn input_vault(&mut self, input_vault: solana_pubkey::Pubkey) -> &mut Self {
self.input_vault = Some(input_vault);
self
}
#[inline(always)]
pub fn output_vault(&mut self, output_vault: solana_pubkey::Pubkey) -> &mut Self {
self.output_vault = Some(output_vault);
self
}
#[inline(always)]
pub fn observation_state(&mut self, observation_state: solana_pubkey::Pubkey) -> &mut Self {
self.observation_state = Some(observation_state);
self
}
#[inline(always)]
pub fn token_program(&mut self, token_program: solana_pubkey::Pubkey) -> &mut Self {
self.token_program = Some(token_program);
self
}
#[inline(always)]
pub fn tick_array(&mut self, tick_array: solana_pubkey::Pubkey) -> &mut Self {
self.tick_array = Some(tick_array);
self
}
#[inline(always)]
pub fn amount(&mut self, amount: u64) -> &mut Self {
self.amount = Some(amount);
self
}
#[inline(always)]
pub fn other_amount_threshold(&mut self, other_amount_threshold: u64) -> &mut Self {
self.other_amount_threshold = Some(other_amount_threshold);
self
}
#[inline(always)]
pub fn sqrt_price_limit_x64(&mut self, sqrt_price_limit_x64: u128) -> &mut Self {
self.sqrt_price_limit_x64 = Some(sqrt_price_limit_x64);
self
}
#[inline(always)]
pub fn is_base_input(&mut self, is_base_input: bool) -> &mut Self {
self.is_base_input = Some(is_base_input);
self
}
#[inline(always)]
pub fn add_remaining_account(&mut self, account: solana_instruction::AccountMeta) -> &mut Self {
self.__remaining_accounts.push(account);
self
}
#[inline(always)]
pub fn add_remaining_accounts(
&mut self,
accounts: &[solana_instruction::AccountMeta],
) -> &mut Self {
self.__remaining_accounts.extend_from_slice(accounts);
self
}
#[allow(clippy::clone_on_copy)]
pub fn instruction(&self) -> solana_instruction::Instruction {
let accounts = Swap {
payer: self.payer.expect("payer is not set"),
amm_config: self.amm_config.expect("amm_config is not set"),
pool_state: self.pool_state.expect("pool_state is not set"),
input_token_account: self
.input_token_account
.expect("input_token_account is not set"),
output_token_account: self
.output_token_account
.expect("output_token_account is not set"),
input_vault: self.input_vault.expect("input_vault is not set"),
output_vault: self.output_vault.expect("output_vault is not set"),
observation_state: self
.observation_state
.expect("observation_state is not set"),
token_program: self.token_program.unwrap_or(solana_pubkey::pubkey!(
"TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA"
)),
tick_array: self.tick_array.expect("tick_array is not set"),
};
let args = SwapInstructionArgs {
amount: self.amount.clone().expect("amount is not set"),
other_amount_threshold: self
.other_amount_threshold
.clone()
.expect("other_amount_threshold is not set"),
sqrt_price_limit_x64: self
.sqrt_price_limit_x64
.clone()
.expect("sqrt_price_limit_x64 is not set"),
is_base_input: self
.is_base_input
.clone()
.expect("is_base_input is not set"),
};
accounts.instruction_with_remaining_accounts(args, &self.__remaining_accounts)
}
}
pub struct SwapCpiAccounts<'a, 'b> {
pub payer: &'b solana_account_info::AccountInfo<'a>,
pub amm_config: &'b solana_account_info::AccountInfo<'a>,
pub pool_state: &'b solana_account_info::AccountInfo<'a>,
pub input_token_account: &'b solana_account_info::AccountInfo<'a>,
pub output_token_account: &'b solana_account_info::AccountInfo<'a>,
pub input_vault: &'b solana_account_info::AccountInfo<'a>,
pub output_vault: &'b solana_account_info::AccountInfo<'a>,
pub observation_state: &'b solana_account_info::AccountInfo<'a>,
pub token_program: &'b solana_account_info::AccountInfo<'a>,
pub tick_array: &'b solana_account_info::AccountInfo<'a>,
}
pub struct SwapCpi<'a, 'b> {
pub __program: &'b solana_account_info::AccountInfo<'a>,
pub payer: &'b solana_account_info::AccountInfo<'a>,
pub amm_config: &'b solana_account_info::AccountInfo<'a>,
pub pool_state: &'b solana_account_info::AccountInfo<'a>,
pub input_token_account: &'b solana_account_info::AccountInfo<'a>,
pub output_token_account: &'b solana_account_info::AccountInfo<'a>,
pub input_vault: &'b solana_account_info::AccountInfo<'a>,
pub output_vault: &'b solana_account_info::AccountInfo<'a>,
pub observation_state: &'b solana_account_info::AccountInfo<'a>,
pub token_program: &'b solana_account_info::AccountInfo<'a>,
pub tick_array: &'b solana_account_info::AccountInfo<'a>,
pub __args: SwapInstructionArgs,
}
impl<'a, 'b> SwapCpi<'a, 'b> {
pub fn new(
program: &'b solana_account_info::AccountInfo<'a>,
accounts: SwapCpiAccounts<'a, 'b>,
args: SwapInstructionArgs,
) -> Self {
Self {
__program: program,
payer: accounts.payer,
amm_config: accounts.amm_config,
pool_state: accounts.pool_state,
input_token_account: accounts.input_token_account,
output_token_account: accounts.output_token_account,
input_vault: accounts.input_vault,
output_vault: accounts.output_vault,
observation_state: accounts.observation_state,
token_program: accounts.token_program,
tick_array: accounts.tick_array,
__args: args,
}
}
#[inline(always)]
pub fn invoke(&self) -> solana_program_entrypoint::ProgramResult {
self.invoke_signed_with_remaining_accounts(&[], &[])
}
#[inline(always)]
pub fn invoke_with_remaining_accounts(
&self,
remaining_accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
) -> solana_program_entrypoint::ProgramResult {
self.invoke_signed_with_remaining_accounts(&[], remaining_accounts)
}
#[inline(always)]
pub fn invoke_signed(
&self,
signers_seeds: &[&[&[u8]]],
) -> solana_program_entrypoint::ProgramResult {
self.invoke_signed_with_remaining_accounts(signers_seeds, &[])
}
#[allow(clippy::arithmetic_side_effects)]
#[allow(clippy::clone_on_copy)]
#[allow(clippy::vec_init_then_push)]
pub fn invoke_signed_with_remaining_accounts(
&self,
signers_seeds: &[&[&[u8]]],
remaining_accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
) -> solana_program_entrypoint::ProgramResult {
let mut accounts = Vec::with_capacity(10 + remaining_accounts.len());
accounts.push(solana_instruction::AccountMeta::new_readonly(
*self.payer.key,
true,
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
*self.amm_config.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.pool_state.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.input_token_account.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.output_token_account.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.input_vault.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.output_vault.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.observation_state.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
*self.token_program.key,
false,
));
accounts.push(solana_instruction::AccountMeta::new(
*self.tick_array.key,
false,
));
remaining_accounts.iter().for_each(|remaining_account| {
accounts.push(solana_instruction::AccountMeta {
pubkey: *remaining_account.0.key,
is_signer: remaining_account.1,
is_writable: remaining_account.2,
})
});
let mut data = borsh::to_vec(&SwapInstructionData::new()).unwrap();
let mut args = borsh::to_vec(&self.__args).unwrap();
data.append(&mut args);
let instruction = solana_instruction::Instruction {
program_id: crate::AMM_V3_ID,
accounts,
data,
};
let mut account_infos = Vec::with_capacity(11 + remaining_accounts.len());
account_infos.push(self.__program.clone());
account_infos.push(self.payer.clone());
account_infos.push(self.amm_config.clone());
account_infos.push(self.pool_state.clone());
account_infos.push(self.input_token_account.clone());
account_infos.push(self.output_token_account.clone());
account_infos.push(self.input_vault.clone());
account_infos.push(self.output_vault.clone());
account_infos.push(self.observation_state.clone());
account_infos.push(self.token_program.clone());
account_infos.push(self.tick_array.clone());
remaining_accounts
.iter()
.for_each(|remaining_account| account_infos.push(remaining_account.0.clone()));
if signers_seeds.is_empty() {
solana_cpi::invoke(&instruction, &account_infos)
} else {
solana_cpi::invoke_signed(&instruction, &account_infos, signers_seeds)
}
}
}
#[derive(Clone, Debug)]
pub struct SwapCpiBuilder<'a, 'b> {
instruction: Box<SwapCpiBuilderInstruction<'a, 'b>>,
}
impl<'a, 'b> SwapCpiBuilder<'a, 'b> {
pub fn new(program: &'b solana_account_info::AccountInfo<'a>) -> Self {
let instruction = Box::new(SwapCpiBuilderInstruction {
__program: program,
payer: None,
amm_config: None,
pool_state: None,
input_token_account: None,
output_token_account: None,
input_vault: None,
output_vault: None,
observation_state: None,
token_program: None,
tick_array: None,
amount: None,
other_amount_threshold: None,
sqrt_price_limit_x64: None,
is_base_input: None,
__remaining_accounts: Vec::new(),
});
Self { instruction }
}
#[inline(always)]
pub fn payer(&mut self, payer: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.payer = Some(payer);
self
}
#[inline(always)]
pub fn amm_config(
&mut self,
amm_config: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.amm_config = Some(amm_config);
self
}
#[inline(always)]
pub fn pool_state(
&mut self,
pool_state: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.pool_state = Some(pool_state);
self
}
#[inline(always)]
pub fn input_token_account(
&mut self,
input_token_account: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.input_token_account = Some(input_token_account);
self
}
#[inline(always)]
pub fn output_token_account(
&mut self,
output_token_account: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.output_token_account = Some(output_token_account);
self
}
#[inline(always)]
pub fn input_vault(
&mut self,
input_vault: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.input_vault = Some(input_vault);
self
}
#[inline(always)]
pub fn output_vault(
&mut self,
output_vault: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.output_vault = Some(output_vault);
self
}
#[inline(always)]
pub fn observation_state(
&mut self,
observation_state: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.observation_state = Some(observation_state);
self
}
#[inline(always)]
pub fn token_program(
&mut self,
token_program: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.token_program = Some(token_program);
self
}
#[inline(always)]
pub fn tick_array(
&mut self,
tick_array: &'b solana_account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.tick_array = Some(tick_array);
self
}
#[inline(always)]
pub fn amount(&mut self, amount: u64) -> &mut Self {
self.instruction.amount = Some(amount);
self
}
#[inline(always)]
pub fn other_amount_threshold(&mut self, other_amount_threshold: u64) -> &mut Self {
self.instruction.other_amount_threshold = Some(other_amount_threshold);
self
}
#[inline(always)]
pub fn sqrt_price_limit_x64(&mut self, sqrt_price_limit_x64: u128) -> &mut Self {
self.instruction.sqrt_price_limit_x64 = Some(sqrt_price_limit_x64);
self
}
#[inline(always)]
pub fn is_base_input(&mut self, is_base_input: bool) -> &mut Self {
self.instruction.is_base_input = Some(is_base_input);
self
}
#[inline(always)]
pub fn add_remaining_account(
&mut self,
account: &'b solana_account_info::AccountInfo<'a>,
is_writable: bool,
is_signer: bool,
) -> &mut Self {
self.instruction
.__remaining_accounts
.push((account, is_writable, is_signer));
self
}
#[inline(always)]
pub fn add_remaining_accounts(
&mut self,
accounts: &[(&'b solana_account_info::AccountInfo<'a>, bool, bool)],
) -> &mut Self {
self.instruction
.__remaining_accounts
.extend_from_slice(accounts);
self
}
#[inline(always)]
pub fn invoke(&self) -> solana_program_entrypoint::ProgramResult {
self.invoke_signed(&[])
}
#[allow(clippy::clone_on_copy)]
#[allow(clippy::vec_init_then_push)]
pub fn invoke_signed(
&self,
signers_seeds: &[&[&[u8]]],
) -> solana_program_entrypoint::ProgramResult {
let args = SwapInstructionArgs {
amount: self.instruction.amount.clone().expect("amount is not set"),
other_amount_threshold: self
.instruction
.other_amount_threshold
.clone()
.expect("other_amount_threshold is not set"),
sqrt_price_limit_x64: self
.instruction
.sqrt_price_limit_x64
.clone()
.expect("sqrt_price_limit_x64 is not set"),
is_base_input: self
.instruction
.is_base_input
.clone()
.expect("is_base_input is not set"),
};
let instruction = SwapCpi {
__program: self.instruction.__program,
payer: self.instruction.payer.expect("payer is not set"),
amm_config: self.instruction.amm_config.expect("amm_config is not set"),
pool_state: self.instruction.pool_state.expect("pool_state is not set"),
input_token_account: self
.instruction
.input_token_account
.expect("input_token_account is not set"),
output_token_account: self
.instruction
.output_token_account
.expect("output_token_account is not set"),
input_vault: self
.instruction
.input_vault
.expect("input_vault is not set"),
output_vault: self
.instruction
.output_vault
.expect("output_vault is not set"),
observation_state: self
.instruction
.observation_state
.expect("observation_state is not set"),
token_program: self
.instruction
.token_program
.expect("token_program is not set"),
tick_array: self.instruction.tick_array.expect("tick_array is not set"),
__args: args,
};
instruction.invoke_signed_with_remaining_accounts(
signers_seeds,
&self.instruction.__remaining_accounts,
)
}
}
#[derive(Clone, Debug)]
struct SwapCpiBuilderInstruction<'a, 'b> {
__program: &'b solana_account_info::AccountInfo<'a>,
payer: Option<&'b solana_account_info::AccountInfo<'a>>,
amm_config: Option<&'b solana_account_info::AccountInfo<'a>>,
pool_state: Option<&'b solana_account_info::AccountInfo<'a>>,
input_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
output_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
input_vault: Option<&'b solana_account_info::AccountInfo<'a>>,
output_vault: Option<&'b solana_account_info::AccountInfo<'a>>,
observation_state: Option<&'b solana_account_info::AccountInfo<'a>>,
token_program: Option<&'b solana_account_info::AccountInfo<'a>>,
tick_array: Option<&'b solana_account_info::AccountInfo<'a>>,
amount: Option<u64>,
other_amount_threshold: Option<u64>,
sqrt_price_limit_x64: Option<u128>,
is_base_input: Option<bool>,
__remaining_accounts: Vec<(&'b solana_account_info::AccountInfo<'a>, bool, bool)>,
}