#[cfg(feature = "anchor")]
use anchor_lang::prelude::{AnchorDeserialize, AnchorSerialize};
#[cfg(not(feature = "anchor"))]
use borsh::{BorshDeserialize, BorshSerialize};
pub struct Collect {
pub recipient1: solana_program::pubkey::Pubkey,
pub recipient2: solana_program::pubkey::Pubkey,
}
impl Collect {
pub fn instruction(&self) -> solana_program::instruction::Instruction {
self.instruction_with_remaining_accounts(&[])
}
#[allow(clippy::vec_init_then_push)]
pub fn instruction_with_remaining_accounts(
&self,
remaining_accounts: &[solana_program::instruction::AccountMeta],
) -> solana_program::instruction::Instruction {
let mut accounts = Vec::with_capacity(2 + remaining_accounts.len());
accounts.push(solana_program::instruction::AccountMeta::new(
self.recipient1,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new(
self.recipient2,
false,
));
accounts.extend_from_slice(remaining_accounts);
let data = CollectInstructionData::new().try_to_vec().unwrap();
solana_program::instruction::Instruction {
program_id: crate::MPL_CORE_ID,
accounts,
data,
}
}
}
#[cfg_attr(not(feature = "anchor"), derive(BorshSerialize, BorshDeserialize))]
#[cfg_attr(feature = "anchor", derive(AnchorSerialize, AnchorDeserialize))]
pub struct CollectInstructionData {
discriminator: u8,
}
impl CollectInstructionData {
pub fn new() -> Self {
Self { discriminator: 19 }
}
}
#[derive(Default)]
pub struct CollectBuilder {
recipient1: Option<solana_program::pubkey::Pubkey>,
recipient2: Option<solana_program::pubkey::Pubkey>,
__remaining_accounts: Vec<solana_program::instruction::AccountMeta>,
}
impl CollectBuilder {
pub fn new() -> Self {
Self::default()
}
#[inline(always)]
pub fn recipient1(&mut self, recipient1: solana_program::pubkey::Pubkey) -> &mut Self {
self.recipient1 = Some(recipient1);
self
}
#[inline(always)]
pub fn recipient2(&mut self, recipient2: solana_program::pubkey::Pubkey) -> &mut Self {
self.recipient2 = Some(recipient2);
self
}
#[inline(always)]
pub fn add_remaining_account(
&mut self,
account: solana_program::instruction::AccountMeta,
) -> &mut Self {
self.__remaining_accounts.push(account);
self
}
#[inline(always)]
pub fn add_remaining_accounts(
&mut self,
accounts: &[solana_program::instruction::AccountMeta],
) -> &mut Self {
self.__remaining_accounts.extend_from_slice(accounts);
self
}
#[allow(clippy::clone_on_copy)]
pub fn instruction(&self) -> solana_program::instruction::Instruction {
let accounts = Collect {
recipient1: self.recipient1.unwrap_or(solana_program::pubkey!(
"8AT6o8Qk5T9QnZvPThMrF9bcCQLTGkyGvVZZzHgCw11v"
)),
recipient2: self.recipient2.unwrap_or(solana_program::pubkey!(
"MmHsqX4LxTfifxoH8BVRLUKrwDn1LPCac6YcCZTHhwt"
)),
};
accounts.instruction_with_remaining_accounts(&self.__remaining_accounts)
}
}
pub struct CollectCpiAccounts<'a, 'b> {
pub recipient1: &'b solana_program::account_info::AccountInfo<'a>,
pub recipient2: &'b solana_program::account_info::AccountInfo<'a>,
}
pub struct CollectCpi<'a, 'b> {
pub __program: &'b solana_program::account_info::AccountInfo<'a>,
pub recipient1: &'b solana_program::account_info::AccountInfo<'a>,
pub recipient2: &'b solana_program::account_info::AccountInfo<'a>,
}
impl<'a, 'b> CollectCpi<'a, 'b> {
pub fn new(
program: &'b solana_program::account_info::AccountInfo<'a>,
accounts: CollectCpiAccounts<'a, 'b>,
) -> Self {
Self {
__program: program,
recipient1: accounts.recipient1,
recipient2: accounts.recipient2,
}
}
#[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_program::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::clone_on_copy)]
#[allow(clippy::vec_init_then_push)]
pub fn invoke_signed_with_remaining_accounts(
&self,
signers_seeds: &[&[&[u8]]],
remaining_accounts: &[(
&'b solana_program::account_info::AccountInfo<'a>,
bool,
bool,
)],
) -> solana_program::entrypoint::ProgramResult {
let mut accounts = Vec::with_capacity(2 + remaining_accounts.len());
accounts.push(solana_program::instruction::AccountMeta::new(
*self.recipient1.key,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new(
*self.recipient2.key,
false,
));
remaining_accounts.iter().for_each(|remaining_account| {
accounts.push(solana_program::instruction::AccountMeta {
pubkey: *remaining_account.0.key,
is_signer: remaining_account.1,
is_writable: remaining_account.2,
})
});
let data = CollectInstructionData::new().try_to_vec().unwrap();
let instruction = solana_program::instruction::Instruction {
program_id: crate::MPL_CORE_ID,
accounts,
data,
};
let mut account_infos = Vec::with_capacity(2 + 1 + remaining_accounts.len());
account_infos.push(self.__program.clone());
account_infos.push(self.recipient1.clone());
account_infos.push(self.recipient2.clone());
remaining_accounts
.iter()
.for_each(|remaining_account| account_infos.push(remaining_account.0.clone()));
if signers_seeds.is_empty() {
solana_program::program::invoke(&instruction, &account_infos)
} else {
solana_program::program::invoke_signed(&instruction, &account_infos, signers_seeds)
}
}
}
pub struct CollectCpiBuilder<'a, 'b> {
instruction: Box<CollectCpiBuilderInstruction<'a, 'b>>,
}
impl<'a, 'b> CollectCpiBuilder<'a, 'b> {
pub fn new(program: &'b solana_program::account_info::AccountInfo<'a>) -> Self {
let instruction = Box::new(CollectCpiBuilderInstruction {
__program: program,
recipient1: None,
recipient2: None,
__remaining_accounts: Vec::new(),
});
Self { instruction }
}
#[inline(always)]
pub fn recipient1(
&mut self,
recipient1: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.recipient1 = Some(recipient1);
self
}
#[inline(always)]
pub fn recipient2(
&mut self,
recipient2: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.recipient2 = Some(recipient2);
self
}
#[inline(always)]
pub fn add_remaining_account(
&mut self,
account: &'b solana_program::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_program::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 = CollectCpi {
__program: self.instruction.__program,
recipient1: self.instruction.recipient1.expect("recipient1 is not set"),
recipient2: self.instruction.recipient2.expect("recipient2 is not set"),
};
instruction.invoke_signed_with_remaining_accounts(
signers_seeds,
&self.instruction.__remaining_accounts,
)
}
}
struct CollectCpiBuilderInstruction<'a, 'b> {
__program: &'b solana_program::account_info::AccountInfo<'a>,
recipient1: Option<&'b solana_program::account_info::AccountInfo<'a>>,
recipient2: Option<&'b solana_program::account_info::AccountInfo<'a>>,
__remaining_accounts: Vec<(
&'b solana_program::account_info::AccountInfo<'a>,
bool,
bool,
)>,
}