d0zingcat/ore
1
0
Fork 0
mirror of https://github.com/d0zingcat/ore.git synced 2026-05-18 15:10:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

admin functions

Browse Source
This commit is contained in:
Hardhat Chad
2024-02-14 07:22:01 +00:00
parent 418738127c
commit f917e49b3f
11 changed files with 287 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files

121
src/processor/initialize.rs
View File

@@ -3,11 +3,8 @@ use std::mem::size_of;
use solana_program::program_pack::Pack;
use solana_program::{self, sysvar};
use solana_program::{
account_info::{next_account_info, AccountInfo},
entrypoint::ProgramResult,
program_error::ProgramError,
pubkey::Pubkey,
system_program,
account_info::AccountInfo, entrypoint::ProgramResult, program_error::ProgramError,
pubkey::Pubkey, system_program,
};
use spl_token::state::Mint;
@@ -24,87 +21,43 @@ pub fn process_initialize<'a, 'info>(
accounts: &'a [AccountInfo<'info>],
data: &[u8],
) -> ProgramResult {
let accounts_iter = &mut accounts.iter();
// Parse args
let args = bytemuck::try_from_bytes::<InitializeArgs>(data)
.or(Err(ProgramError::InvalidInstructionData))?;
// Account 1: Signer
let signer = load_signer(next_account_info(accounts_iter)?)?;
// Accounts 2-9: Busses
let busses = vec![
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[0], &[args.bus_0_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[1], &[args.bus_1_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[2], &[args.bus_2_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[3], &[args.bus_3_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[4], &[args.bus_4_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[5], &[args.bus_5_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[6], &[args.bus_6_bump]],
)?,
load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[BUS, &[7], &[args.bus_7_bump]],
)?,
];
// Account 10: Mint
let mint = load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[MINT, &[args.mint_bump]],
)?;
if !mint.key.eq(&MINT_ADDRESS) {
// Validate accounts
let [signer, bus_0_info, bus_1_info, bus_2_info, bus_3_info, bus_4_info, bus_5_info, bus_6_info, bus_7_info, mint_info, treasury_info, treasury_tokens_info, system_program, token_program, associated_token_program, rent_sysvar] = accounts else {
return Err(ProgramError::NotEnoughAccountKeys);
};
load_signer(signer)?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(bus_0_info, &[BUS, &[0], &[args.bus_0_bump]])?;
load_uninitialized_pda(mint_info, &[MINT, &[args.mint_bump]])?;
if !mint_info.key.eq(&MINT_ADDRESS) {
return Err(ProgramError::InvalidAccountData);
}
// Account 11: Treasury
let treasury_info = load_uninitialized_pda(
next_account_info(accounts_iter)?,
&[TREASURY, &[args.treasury_bump]],
)?;
load_uninitialized_pda(treasury_info, &[TREASURY, &[args.treasury_bump]])?;
if !treasury_info.key.eq(&TREASURY_ADDRESS) {
return Err(ProgramError::InvalidSeeds);
}
// Account 12: Treasury tokens
let treasury_tokens = load_uninitialized_account(next_account_info(accounts_iter)?)?;
// Account 13: System program
let system_program = load_account(next_account_info(accounts_iter)?, system_program::id())?;
// Account 14: Token program
let token_program = load_account(next_account_info(accounts_iter)?, spl_token::id())?;
// Account 15: Associated token program
let associated_token_program = load_account(
next_account_info(accounts_iter)?,
spl_associated_token_account::id(),
)?;
// Account 16: Rent sysvar
let rent_sysvar = load_account(next_account_info(accounts_iter)?, sysvar::rent::id())?;
load_uninitialized_account(treasury_tokens_info)?;
load_account(system_program, system_program::id())?;
load_account(token_program, spl_token::id())?;
load_account(associated_token_program, spl_associated_token_account::id())?;
load_account(rent_sysvar, sysvar::rent::id())?;
// Initialize bus accounts
let bus_bumps = vec![
let bus_infos = [
bus_0_info, bus_1_info, bus_2_info, bus_3_info, bus_4_info, bus_5_info, bus_6_info,
bus_7_info,
];
let bus_bumps = [
args.bus_0_bump,
args.bus_1_bump,
args.bus_2_bump,
@@ -116,14 +69,14 @@ pub fn process_initialize<'a, 'info>(
];
for i in 0..BUS_COUNT {
create_pda(
busses[i],
bus_infos[i],
&crate::id(),
size_of::<Bus>(),
&[BUS, &[i as u8], &[bus_bumps[i]]],
system_program,
signer,
)?;
busses[i].try_borrow_mut_data()?.copy_from_slice(
bus_infos[i].try_borrow_mut_data()?.copy_from_slice(
Bus {
bump: bus_bumps[i] as u32,
id: i as u32,
@@ -154,7 +107,7 @@ pub fn process_initialize<'a, 'info>(
// Initialize mint
create_pda(
mint,
mint_info,
&spl_token::id(),
Mint::LEN,
&[MINT, &[args.mint_bump]],
@@ -164,14 +117,14 @@ pub fn process_initialize<'a, 'info>(
solana_program::program::invoke_signed(
&spl_token::instruction::initialize_mint(
&spl_token::id(),
mint.key,
mint_info.key,
treasury_info.key,
None,
TOKEN_DECIMALS,
)?,
&[
token_program.clone(),
mint.clone(),
mint_info.clone(),
treasury_info.clone(),
rent_sysvar.clone(),
],
@@ -183,15 +136,15 @@ pub fn process_initialize<'a, 'info>(
&spl_associated_token_account::instruction::create_associated_token_account(
signer.key,
treasury_info.key,
mint.key,
mint_info.key,
&spl_token::id(),
),
&[
associated_token_program.clone(),
signer.clone(),
treasury_tokens.clone(),
treasury_tokens_info.clone(),
treasury_info.clone(),
mint.clone(),
mint_info.clone(),
system_program.clone(),
token_program.clone(),
],