use crate::generated::types::WriteType;
use borsh::BorshDeserialize;
use borsh::BorshSerialize;
pub struct MemoryWrite {
pub program_id: solana_program::pubkey::Pubkey,
pub system_program: solana_program::pubkey::Pubkey,
pub payer: solana_program::pubkey::Pubkey,
pub memory: solana_program::pubkey::Pubkey,
pub source_account: solana_program::pubkey::Pubkey,
}
impl MemoryWrite {
pub fn instruction(
&self,
args: MemoryWriteInstructionArgs,
) -> solana_program::instruction::Instruction {
self.instruction_with_remaining_accounts(args, &[])
}
#[allow(clippy::vec_init_then_push)]
pub fn instruction_with_remaining_accounts(
&self,
args: MemoryWriteInstructionArgs,
remaining_accounts: &[solana_program::instruction::AccountMeta],
) -> solana_program::instruction::Instruction {
let mut accounts = Vec::with_capacity(5 + remaining_accounts.len());
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
self.program_id,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
self.system_program,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new(
self.payer, true,
));
accounts.push(solana_program::instruction::AccountMeta::new(
self.memory,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
self.source_account,
false,
));
accounts.extend_from_slice(remaining_accounts);
let mut data = MemoryWriteInstructionData::new().try_to_vec().unwrap();
let mut args = args.try_to_vec().unwrap();
data.append(&mut args);
solana_program::instruction::Instruction {
program_id: crate::LIGHTHOUSE_ID,
accounts,
data,
}
}
}
#[derive(BorshDeserialize, BorshSerialize)]
struct MemoryWriteInstructionData {
discriminator: u8,
}
impl MemoryWriteInstructionData {
fn new() -> Self {
Self { discriminator: 0 }
}
}
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct MemoryWriteInstructionArgs {
pub memory_id: u8,
pub memory_bump: u8,
pub write_offset: u16,
pub write_type: WriteType,
}
#[derive(Default)]
pub struct MemoryWriteBuilder {
program_id: Option<solana_program::pubkey::Pubkey>,
system_program: Option<solana_program::pubkey::Pubkey>,
payer: Option<solana_program::pubkey::Pubkey>,
memory: Option<solana_program::pubkey::Pubkey>,
source_account: Option<solana_program::pubkey::Pubkey>,
memory_id: Option<u8>,
memory_bump: Option<u8>,
write_offset: Option<u16>,
write_type: Option<WriteType>,
__remaining_accounts: Vec<solana_program::instruction::AccountMeta>,
}
impl MemoryWriteBuilder {
pub fn new() -> Self {
Self::default()
}
#[inline(always)]
pub fn program_id(&mut self, program_id: solana_program::pubkey::Pubkey) -> &mut Self {
self.program_id = Some(program_id);
self
}
#[inline(always)]
pub fn system_program(&mut self, system_program: solana_program::pubkey::Pubkey) -> &mut Self {
self.system_program = Some(system_program);
self
}
#[inline(always)]
pub fn payer(&mut self, payer: solana_program::pubkey::Pubkey) -> &mut Self {
self.payer = Some(payer);
self
}
#[inline(always)]
pub fn memory(&mut self, memory: solana_program::pubkey::Pubkey) -> &mut Self {
self.memory = Some(memory);
self
}
#[inline(always)]
pub fn source_account(&mut self, source_account: solana_program::pubkey::Pubkey) -> &mut Self {
self.source_account = Some(source_account);
self
}
#[inline(always)]
pub fn memory_id(&mut self, memory_id: u8) -> &mut Self {
self.memory_id = Some(memory_id);
self
}
#[inline(always)]
pub fn memory_bump(&mut self, memory_bump: u8) -> &mut Self {
self.memory_bump = Some(memory_bump);
self
}
#[inline(always)]
pub fn write_offset(&mut self, write_offset: u16) -> &mut Self {
self.write_offset = Some(write_offset);
self
}
#[inline(always)]
pub fn write_type(&mut self, write_type: WriteType) -> &mut Self {
self.write_type = Some(write_type);
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 = MemoryWrite {
program_id: self.program_id.expect("program_id is not set"),
system_program: self
.system_program
.unwrap_or(solana_program::pubkey!("11111111111111111111111111111111")),
payer: self.payer.expect("payer is not set"),
memory: self.memory.expect("memory is not set"),
source_account: self.source_account.expect("source_account is not set"),
};
let args = MemoryWriteInstructionArgs {
memory_id: self.memory_id.clone().unwrap_or(0),
memory_bump: self.memory_bump.clone().expect("memory_bump is not set"),
write_offset: self.write_offset.clone().expect("write_offset is not set"),
write_type: self.write_type.clone().expect("write_type is not set"),
};
accounts.instruction_with_remaining_accounts(args, &self.__remaining_accounts)
}
}
pub struct MemoryWriteCpiAccounts<'a, 'b> {
pub program_id: &'b solana_program::account_info::AccountInfo<'a>,
pub system_program: &'b solana_program::account_info::AccountInfo<'a>,
pub payer: &'b solana_program::account_info::AccountInfo<'a>,
pub memory: &'b solana_program::account_info::AccountInfo<'a>,
pub source_account: &'b solana_program::account_info::AccountInfo<'a>,
}
pub struct MemoryWriteCpi<'a, 'b> {
pub __program: &'b solana_program::account_info::AccountInfo<'a>,
pub program_id: &'b solana_program::account_info::AccountInfo<'a>,
pub system_program: &'b solana_program::account_info::AccountInfo<'a>,
pub payer: &'b solana_program::account_info::AccountInfo<'a>,
pub memory: &'b solana_program::account_info::AccountInfo<'a>,
pub source_account: &'b solana_program::account_info::AccountInfo<'a>,
pub __args: MemoryWriteInstructionArgs,
}
impl<'a, 'b> MemoryWriteCpi<'a, 'b> {
pub fn new(
program: &'b solana_program::account_info::AccountInfo<'a>,
accounts: MemoryWriteCpiAccounts<'a, 'b>,
args: MemoryWriteInstructionArgs,
) -> Self {
Self {
__program: program,
program_id: accounts.program_id,
system_program: accounts.system_program,
payer: accounts.payer,
memory: accounts.memory,
source_account: accounts.source_account,
__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_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(5 + remaining_accounts.len());
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
*self.program_id.key,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
*self.system_program.key,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new(
*self.payer.key,
true,
));
accounts.push(solana_program::instruction::AccountMeta::new(
*self.memory.key,
false,
));
accounts.push(solana_program::instruction::AccountMeta::new_readonly(
*self.source_account.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 mut data = MemoryWriteInstructionData::new().try_to_vec().unwrap();
let mut args = self.__args.try_to_vec().unwrap();
data.append(&mut args);
let instruction = solana_program::instruction::Instruction {
program_id: crate::LIGHTHOUSE_ID,
accounts,
data,
};
let mut account_infos = Vec::with_capacity(5 + 1 + remaining_accounts.len());
account_infos.push(self.__program.clone());
account_infos.push(self.program_id.clone());
account_infos.push(self.system_program.clone());
account_infos.push(self.payer.clone());
account_infos.push(self.memory.clone());
account_infos.push(self.source_account.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 MemoryWriteCpiBuilder<'a, 'b> {
instruction: Box<MemoryWriteCpiBuilderInstruction<'a, 'b>>,
}
impl<'a, 'b> MemoryWriteCpiBuilder<'a, 'b> {
pub fn new(program: &'b solana_program::account_info::AccountInfo<'a>) -> Self {
let instruction = Box::new(MemoryWriteCpiBuilderInstruction {
__program: program,
program_id: None,
system_program: None,
payer: None,
memory: None,
source_account: None,
memory_id: None,
memory_bump: None,
write_offset: None,
write_type: None,
__remaining_accounts: Vec::new(),
});
Self { instruction }
}
#[inline(always)]
pub fn program_id(
&mut self,
program_id: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.program_id = Some(program_id);
self
}
#[inline(always)]
pub fn system_program(
&mut self,
system_program: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.system_program = Some(system_program);
self
}
#[inline(always)]
pub fn payer(&mut self, payer: &'b solana_program::account_info::AccountInfo<'a>) -> &mut Self {
self.instruction.payer = Some(payer);
self
}
#[inline(always)]
pub fn memory(
&mut self,
memory: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.memory = Some(memory);
self
}
#[inline(always)]
pub fn source_account(
&mut self,
source_account: &'b solana_program::account_info::AccountInfo<'a>,
) -> &mut Self {
self.instruction.source_account = Some(source_account);
self
}
#[inline(always)]
pub fn memory_id(&mut self, memory_id: u8) -> &mut Self {
self.instruction.memory_id = Some(memory_id);
self
}
#[inline(always)]
pub fn memory_bump(&mut self, memory_bump: u8) -> &mut Self {
self.instruction.memory_bump = Some(memory_bump);
self
}
#[inline(always)]
pub fn write_offset(&mut self, write_offset: u16) -> &mut Self {
self.instruction.write_offset = Some(write_offset);
self
}
#[inline(always)]
pub fn write_type(&mut self, write_type: WriteType) -> &mut Self {
self.instruction.write_type = Some(write_type);
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 args = MemoryWriteInstructionArgs {
memory_id: self.instruction.memory_id.clone().unwrap_or(0),
memory_bump: self
.instruction
.memory_bump
.clone()
.expect("memory_bump is not set"),
write_offset: self
.instruction
.write_offset
.clone()
.expect("write_offset is not set"),
write_type: self
.instruction
.write_type
.clone()
.expect("write_type is not set"),
};
let instruction = MemoryWriteCpi {
__program: self.instruction.__program,
program_id: self.instruction.program_id.expect("program_id is not set"),
system_program: self
.instruction
.system_program
.expect("system_program is not set"),
payer: self.instruction.payer.expect("payer is not set"),
memory: self.instruction.memory.expect("memory is not set"),
source_account: self
.instruction
.source_account
.expect("source_account is not set"),
__args: args,
};
instruction.invoke_signed_with_remaining_accounts(
signers_seeds,
&self.instruction.__remaining_accounts,
)
}
}
struct MemoryWriteCpiBuilderInstruction<'a, 'b> {
__program: &'b solana_program::account_info::AccountInfo<'a>,
program_id: Option<&'b solana_program::account_info::AccountInfo<'a>>,
system_program: Option<&'b solana_program::account_info::AccountInfo<'a>>,
payer: Option<&'b solana_program::account_info::AccountInfo<'a>>,
memory: Option<&'b solana_program::account_info::AccountInfo<'a>>,
source_account: Option<&'b solana_program::account_info::AccountInfo<'a>>,
memory_id: Option<u8>,
memory_bump: Option<u8>,
write_offset: Option<u16>,
write_type: Option<WriteType>,
__remaining_accounts: Vec<(
&'b solana_program::account_info::AccountInfo<'a>,
bool,
bool,
)>,
}