use borsh::BorshSerialize;
use borsh::BorshDeserialize;
#[derive(Debug)]
pub struct Withdraw {
pub authority: solana_pubkey::Pubkey,
pub treasury: solana_pubkey::Pubkey,
pub position: solana_pubkey::Pubkey,
pub treasury_staked_token_account: solana_pubkey::Pubkey,
pub treasury_reward_token_account: solana_pubkey::Pubkey,
pub authority_staked_token_account: solana_pubkey::Pubkey,
pub authority_reward_token_account: solana_pubkey::Pubkey,
pub token_program: solana_pubkey::Pubkey,
}
impl Withdraw {
pub fn instruction(&self) -> solana_instruction::Instruction {
self.instruction_with_remaining_accounts(&[])
}
#[allow(clippy::arithmetic_side_effects)]
#[allow(clippy::vec_init_then_push)]
pub fn instruction_with_remaining_accounts(&self, remaining_accounts: &[solana_instruction::AccountMeta]) -> solana_instruction::Instruction {
let mut accounts = Vec::with_capacity(8+ remaining_accounts.len());
accounts.push(solana_instruction::AccountMeta::new(
self.authority,
true
));
accounts.push(solana_instruction::AccountMeta::new(
self.treasury,
false
));
accounts.push(solana_instruction::AccountMeta::new(
self.position,
false
));
accounts.push(solana_instruction::AccountMeta::new(
self.treasury_staked_token_account,
false
));
accounts.push(solana_instruction::AccountMeta::new(
self.treasury_reward_token_account,
false
));
accounts.push(solana_instruction::AccountMeta::new(
self.authority_staked_token_account,
false
));
accounts.push(solana_instruction::AccountMeta::new(
self.authority_reward_token_account,
false
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
self.token_program,
false
));
accounts.extend_from_slice(remaining_accounts);
let data = borsh::to_vec(&WithdrawInstructionData::new()).unwrap();
solana_instruction::Instruction {
program_id: crate::TUNA_STAKING_ID,
accounts,
data,
}
}
}
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct WithdrawInstructionData {
discriminator: [u8; 8],
}
impl WithdrawInstructionData {
pub fn new() -> Self {
Self {
discriminator: [183, 18, 70, 156, 148, 109, 161, 34],
}
}
}
impl Default for WithdrawInstructionData {
fn default() -> Self {
Self::new()
}
}
#[derive(Clone, Debug, Default)]
pub struct WithdrawBuilder {
authority: Option<solana_pubkey::Pubkey>,
treasury: Option<solana_pubkey::Pubkey>,
position: Option<solana_pubkey::Pubkey>,
treasury_staked_token_account: Option<solana_pubkey::Pubkey>,
treasury_reward_token_account: Option<solana_pubkey::Pubkey>,
authority_staked_token_account: Option<solana_pubkey::Pubkey>,
authority_reward_token_account: Option<solana_pubkey::Pubkey>,
token_program: Option<solana_pubkey::Pubkey>,
__remaining_accounts: Vec<solana_instruction::AccountMeta>,
}
impl WithdrawBuilder {
pub fn new() -> Self {
Self::default()
}
#[inline(always)]
pub fn authority(&mut self, authority: solana_pubkey::Pubkey) -> &mut Self {
self.authority = Some(authority);
self
}
#[inline(always)]
pub fn treasury(&mut self, treasury: solana_pubkey::Pubkey) -> &mut Self {
self.treasury = Some(treasury);
self
}
#[inline(always)]
pub fn position(&mut self, position: solana_pubkey::Pubkey) -> &mut Self {
self.position = Some(position);
self
}
#[inline(always)]
pub fn treasury_staked_token_account(&mut self, treasury_staked_token_account: solana_pubkey::Pubkey) -> &mut Self {
self.treasury_staked_token_account = Some(treasury_staked_token_account);
self
}
#[inline(always)]
pub fn treasury_reward_token_account(&mut self, treasury_reward_token_account: solana_pubkey::Pubkey) -> &mut Self {
self.treasury_reward_token_account = Some(treasury_reward_token_account);
self
}
#[inline(always)]
pub fn authority_staked_token_account(&mut self, authority_staked_token_account: solana_pubkey::Pubkey) -> &mut Self {
self.authority_staked_token_account = Some(authority_staked_token_account);
self
}
#[inline(always)]
pub fn authority_reward_token_account(&mut self, authority_reward_token_account: solana_pubkey::Pubkey) -> &mut Self {
self.authority_reward_token_account = Some(authority_reward_token_account);
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 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 = Withdraw {
authority: self.authority.expect("authority is not set"),
treasury: self.treasury.expect("treasury is not set"),
position: self.position.expect("position is not set"),
treasury_staked_token_account: self.treasury_staked_token_account.expect("treasury_staked_token_account is not set"),
treasury_reward_token_account: self.treasury_reward_token_account.expect("treasury_reward_token_account is not set"),
authority_staked_token_account: self.authority_staked_token_account.expect("authority_staked_token_account is not set"),
authority_reward_token_account: self.authority_reward_token_account.expect("authority_reward_token_account is not set"),
token_program: self.token_program.unwrap_or(solana_pubkey::pubkey!("TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA")),
};
accounts.instruction_with_remaining_accounts(&self.__remaining_accounts)
}
}
pub struct WithdrawCpiAccounts<'a, 'b> {
pub authority: &'b solana_account_info::AccountInfo<'a>,
pub treasury: &'b solana_account_info::AccountInfo<'a>,
pub position: &'b solana_account_info::AccountInfo<'a>,
pub treasury_staked_token_account: &'b solana_account_info::AccountInfo<'a>,
pub treasury_reward_token_account: &'b solana_account_info::AccountInfo<'a>,
pub authority_staked_token_account: &'b solana_account_info::AccountInfo<'a>,
pub authority_reward_token_account: &'b solana_account_info::AccountInfo<'a>,
pub token_program: &'b solana_account_info::AccountInfo<'a>,
}
pub struct WithdrawCpi<'a, 'b> {
pub __program: &'b solana_account_info::AccountInfo<'a>,
pub authority: &'b solana_account_info::AccountInfo<'a>,
pub treasury: &'b solana_account_info::AccountInfo<'a>,
pub position: &'b solana_account_info::AccountInfo<'a>,
pub treasury_staked_token_account: &'b solana_account_info::AccountInfo<'a>,
pub treasury_reward_token_account: &'b solana_account_info::AccountInfo<'a>,
pub authority_staked_token_account: &'b solana_account_info::AccountInfo<'a>,
pub authority_reward_token_account: &'b solana_account_info::AccountInfo<'a>,
pub token_program: &'b solana_account_info::AccountInfo<'a>,
}
impl<'a, 'b> WithdrawCpi<'a, 'b> {
pub fn new(
program: &'b solana_account_info::AccountInfo<'a>,
accounts: WithdrawCpiAccounts<'a, 'b>,
) -> Self {
Self {
__program: program,
authority: accounts.authority,
treasury: accounts.treasury,
position: accounts.position,
treasury_staked_token_account: accounts.treasury_staked_token_account,
treasury_reward_token_account: accounts.treasury_reward_token_account,
authority_staked_token_account: accounts.authority_staked_token_account,
authority_reward_token_account: accounts.authority_reward_token_account,
token_program: accounts.token_program,
}
}
#[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(8+ remaining_accounts.len());
accounts.push(solana_instruction::AccountMeta::new(
*self.authority.key,
true
));
accounts.push(solana_instruction::AccountMeta::new(
*self.treasury.key,
false
));
accounts.push(solana_instruction::AccountMeta::new(
*self.position.key,
false
));
accounts.push(solana_instruction::AccountMeta::new(
*self.treasury_staked_token_account.key,
false
));
accounts.push(solana_instruction::AccountMeta::new(
*self.treasury_reward_token_account.key,
false
));
accounts.push(solana_instruction::AccountMeta::new(
*self.authority_staked_token_account.key,
false
));
accounts.push(solana_instruction::AccountMeta::new(
*self.authority_reward_token_account.key,
false
));
accounts.push(solana_instruction::AccountMeta::new_readonly(
*self.token_program.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 data = borsh::to_vec(&WithdrawInstructionData::new()).unwrap();
let instruction = solana_instruction::Instruction {
program_id: crate::TUNA_STAKING_ID,
accounts,
data,
};
let mut account_infos = Vec::with_capacity(9 + remaining_accounts.len());
account_infos.push(self.__program.clone());
account_infos.push(self.authority.clone());
account_infos.push(self.treasury.clone());
account_infos.push(self.position.clone());
account_infos.push(self.treasury_staked_token_account.clone());
account_infos.push(self.treasury_reward_token_account.clone());
account_infos.push(self.authority_staked_token_account.clone());
account_infos.push(self.authority_reward_token_account.clone());
account_infos.push(self.token_program.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 WithdrawCpiBuilder<'a, 'b> {
instruction: Box<WithdrawCpiBuilderInstruction<'a, 'b>>,
}
impl<'a, 'b> WithdrawCpiBuilder<'a, 'b> {
pub fn new(program: &'b solana_account_info::AccountInfo<'a>) -> Self {
let instruction = Box::new(WithdrawCpiBuilderInstruction {
__program: program,
authority: None,
treasury: None,
position: None,
treasury_staked_token_account: None,
treasury_reward_token_account: None,
authority_staked_token_account: None,
authority_reward_token_account: None,
token_program: None,
__remaining_accounts: Vec::new(),
});
Self { instruction }
}
#[inline(always)]
pub fn authority(&mut self, authority: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.authority = Some(authority);
self
}
#[inline(always)]
pub fn treasury(&mut self, treasury: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.treasury = Some(treasury);
self
}
#[inline(always)]
pub fn position(&mut self, position: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.position = Some(position);
self
}
#[inline(always)]
pub fn treasury_staked_token_account(&mut self, treasury_staked_token_account: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.treasury_staked_token_account = Some(treasury_staked_token_account);
self
}
#[inline(always)]
pub fn treasury_reward_token_account(&mut self, treasury_reward_token_account: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.treasury_reward_token_account = Some(treasury_reward_token_account);
self
}
#[inline(always)]
pub fn authority_staked_token_account(&mut self, authority_staked_token_account: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.authority_staked_token_account = Some(authority_staked_token_account);
self
}
#[inline(always)]
pub fn authority_reward_token_account(&mut self, authority_reward_token_account: &'b solana_account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.authority_reward_token_account = Some(authority_reward_token_account);
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 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 instruction = WithdrawCpi {
__program: self.instruction.__program,
authority: self.instruction.authority.expect("authority is not set"),
treasury: self.instruction.treasury.expect("treasury is not set"),
position: self.instruction.position.expect("position is not set"),
treasury_staked_token_account: self.instruction.treasury_staked_token_account.expect("treasury_staked_token_account is not set"),
treasury_reward_token_account: self.instruction.treasury_reward_token_account.expect("treasury_reward_token_account is not set"),
authority_staked_token_account: self.instruction.authority_staked_token_account.expect("authority_staked_token_account is not set"),
authority_reward_token_account: self.instruction.authority_reward_token_account.expect("authority_reward_token_account is not set"),
token_program: self.instruction.token_program.expect("token_program is not set"),
};
instruction.invoke_signed_with_remaining_accounts(signers_seeds, &self.instruction.__remaining_accounts)
}
}
#[derive(Clone, Debug)]
struct WithdrawCpiBuilderInstruction<'a, 'b> {
__program: &'b solana_account_info::AccountInfo<'a>,
authority: Option<&'b solana_account_info::AccountInfo<'a>>,
treasury: Option<&'b solana_account_info::AccountInfo<'a>>,
position: Option<&'b solana_account_info::AccountInfo<'a>>,
treasury_staked_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
treasury_reward_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
authority_staked_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
authority_reward_token_account: Option<&'b solana_account_info::AccountInfo<'a>>,
token_program: Option<&'b solana_account_info::AccountInfo<'a>>,
__remaining_accounts: Vec<(&'b solana_account_info::AccountInfo<'a>, bool, bool)>,
}