mirror of
https://github.com/d0zingcat/ore.git
synced 2026-05-14 15:10:13 +00:00
230 lines
8.0 KiB
Rust
230 lines
8.0 KiB
Rust
use std::mem::size_of;
|
|
|
|
use solana_program::{
|
|
account_info::AccountInfo,
|
|
entrypoint::ProgramResult,
|
|
program_error::ProgramError,
|
|
program_pack::Pack,
|
|
pubkey::Pubkey,
|
|
system_program, {self, sysvar},
|
|
};
|
|
use spl_token::state::Mint;
|
|
|
|
use crate::{
|
|
instruction::*,
|
|
loaders::*,
|
|
state::{Bus, Config, Treasury},
|
|
utils::create_pda,
|
|
utils::AccountDeserialize,
|
|
utils::Discriminator,
|
|
BUS, BUS_COUNT, CONFIG, INITIAL_BASE_REWARD_RATE, INITIAL_TOLERANCE, METADATA, MINT,
|
|
MINT_ADDRESS, MINT_NOISE, TOKEN_DECIMALS, TREASURY,
|
|
};
|
|
|
|
/// Initialize sets up the Ore program. Its responsibilities include:
|
|
/// 1. Initialize the 8 bus accounts.
|
|
/// 2. Initialize the treasury account.
|
|
/// 3. Initialize the Ore mint account.
|
|
/// 4. Initialize the mint metadata account.
|
|
/// 5. Initialize the treasury token account.
|
|
/// 6. Set the signer as the program admin.
|
|
///
|
|
/// Safety requirements:
|
|
/// - Initialize is a permissionless instruction and can be called by anyone.
|
|
/// - Can only succeed once for the entire lifetime of the program.
|
|
/// - Can only succeed if all provided PDAs match their expected values.
|
|
/// - Can only succeed if provided system program, token program,
|
|
/// associated token program, metadata program, and rent sysvar are valid.
|
|
///
|
|
/// Discussion
|
|
/// - The signer of this instruction is set as the program admin and the
|
|
/// upgrade authority of the mint metadata account.
|
|
pub fn process_initialize<'a, 'info>(
|
|
_program_id: &Pubkey,
|
|
accounts: &'a [AccountInfo<'info>],
|
|
data: &[u8],
|
|
) -> ProgramResult {
|
|
// Parse args
|
|
let args = InitializeArgs::try_from_bytes(data)?;
|
|
|
|
// Load 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, config_info, metadata_info, mint_info, treasury_info, treasury_tokens_info, system_program, token_program, associated_token_program, metadata_program, rent_sysvar] =
|
|
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, config_info, metadata_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, &crate::id())?;
|
|
load_uninitialized_pda(bus_1_info, &[BUS, &[1]], args.bus_1_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_2_info, &[BUS, &[2]], args.bus_2_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_3_info, &[BUS, &[3]], args.bus_3_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_4_info, &[BUS, &[4]], args.bus_4_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_5_info, &[BUS, &[5]], args.bus_5_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_6_info, &[BUS, &[6]], args.bus_6_bump, &crate::id())?;
|
|
load_uninitialized_pda(bus_7_info, &[BUS, &[7]], args.bus_7_bump, &crate::id())?;
|
|
load_uninitialized_pda(config_info, &[CONFIG], args.config_bump, &crate::id())?;
|
|
load_uninitialized_pda(
|
|
metadata_info,
|
|
&[
|
|
METADATA,
|
|
mpl_token_metadata::ID.as_ref(),
|
|
MINT_ADDRESS.as_ref(),
|
|
],
|
|
args.metadata_bump,
|
|
&mpl_token_metadata::ID,
|
|
)?;
|
|
load_uninitialized_pda(
|
|
mint_info,
|
|
&[MINT, MINT_NOISE.as_slice()],
|
|
args.mint_bump,
|
|
&crate::id(),
|
|
)?;
|
|
load_uninitialized_pda(treasury_info, &[TREASURY], args.treasury_bump, &crate::id())?;
|
|
load_uninitialized_account(treasury_tokens_info)?;
|
|
load_program(system_program, system_program::id())?;
|
|
load_program(token_program, spl_token::id())?;
|
|
load_program(associated_token_program, spl_associated_token_account::id())?;
|
|
// load_program(metadata_program, mpl_token_metadata::ID)?;
|
|
load_sysvar(rent_sysvar, sysvar::rent::id())?;
|
|
|
|
// Initialize bus accounts
|
|
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,
|
|
args.bus_3_bump,
|
|
args.bus_4_bump,
|
|
args.bus_5_bump,
|
|
args.bus_6_bump,
|
|
args.bus_7_bump,
|
|
];
|
|
for i in 0..BUS_COUNT {
|
|
create_pda(
|
|
bus_infos[i],
|
|
&crate::id(),
|
|
8 + size_of::<Bus>(),
|
|
&[BUS, &[i as u8], &[bus_bumps[i]]],
|
|
system_program,
|
|
signer,
|
|
)?;
|
|
let mut bus_data = bus_infos[i].try_borrow_mut_data()?;
|
|
bus_data[0] = Bus::discriminator() as u8;
|
|
let bus = Bus::try_from_bytes_mut(&mut bus_data)?;
|
|
bus.id = i as u64;
|
|
bus.rewards = 0;
|
|
}
|
|
|
|
// Initialize config
|
|
create_pda(
|
|
config_info,
|
|
&crate::id(),
|
|
8 + size_of::<Config>(),
|
|
&[CONFIG, &[args.config_bump]],
|
|
system_program,
|
|
signer,
|
|
)?;
|
|
let mut config_data = config_info.data.borrow_mut();
|
|
config_data[0] = Config::discriminator() as u8;
|
|
let config = Config::try_from_bytes_mut(&mut config_data)?;
|
|
config.admin = *signer.key;
|
|
config.base_reward_rate = INITIAL_BASE_REWARD_RATE;
|
|
config.last_reset_at = 0;
|
|
config.paused = 0;
|
|
config.tolerance_liveness = INITIAL_TOLERANCE;
|
|
config.tolerance_spam = INITIAL_TOLERANCE;
|
|
|
|
// Initialize treasury
|
|
create_pda(
|
|
treasury_info,
|
|
&crate::id(),
|
|
8 + size_of::<Treasury>(),
|
|
&[TREASURY, &[args.treasury_bump]],
|
|
system_program,
|
|
signer,
|
|
)?;
|
|
let mut treasury_data = treasury_info.data.borrow_mut();
|
|
treasury_data[0] = Treasury::discriminator() as u8;
|
|
let treasury = Treasury::try_from_bytes_mut(&mut treasury_data)?;
|
|
treasury.bump = args.treasury_bump as u64;
|
|
drop(treasury_data);
|
|
|
|
// Initialize mint
|
|
create_pda(
|
|
mint_info,
|
|
&spl_token::id(),
|
|
Mint::LEN,
|
|
&[MINT, MINT_NOISE.as_slice(), &[args.mint_bump]],
|
|
system_program,
|
|
signer,
|
|
)?;
|
|
solana_program::program::invoke_signed(
|
|
&spl_token::instruction::initialize_mint(
|
|
&spl_token::id(),
|
|
mint_info.key,
|
|
treasury_info.key,
|
|
None,
|
|
TOKEN_DECIMALS,
|
|
)?,
|
|
&[
|
|
token_program.clone(),
|
|
mint_info.clone(),
|
|
treasury_info.clone(),
|
|
rent_sysvar.clone(),
|
|
],
|
|
&[&[MINT, MINT_NOISE.as_slice(), &[args.mint_bump]]],
|
|
)?;
|
|
|
|
// TODO Fix metadata initialization
|
|
// Initialize mint metadata
|
|
// mpl_token_metadata::instructions::CreateMetadataAccountV3Cpi {
|
|
// __program: metadata_program,
|
|
// metadata: metadata_info,
|
|
// mint: mint_info,
|
|
// mint_authority: treasury_info,
|
|
// payer: signer,
|
|
// update_authority: (signer, true),
|
|
// system_program,
|
|
// rent: Some(rent_sysvar),
|
|
// __args: mpl_token_metadata::instructions::CreateMetadataAccountV3InstructionArgs {
|
|
// data: mpl_token_metadata::types::DataV2 {
|
|
// name: METADATA_NAME.to_string(),
|
|
// symbol: METADATA_SYMBOL.to_string(),
|
|
// uri: METADATA_URI.to_string(),
|
|
// seller_fee_basis_points: 0,
|
|
// creators: None,
|
|
// collection: None,
|
|
// uses: None,
|
|
// },
|
|
// is_mutable: true,
|
|
// collection_details: None,
|
|
// },
|
|
// }
|
|
// .invoke_signed(&[&[TREASURY, &[args.treasury_bump]]])?;
|
|
|
|
// Initialize treasury token account
|
|
solana_program::program::invoke(
|
|
&spl_associated_token_account::instruction::create_associated_token_account(
|
|
signer.key,
|
|
treasury_info.key,
|
|
mint_info.key,
|
|
&spl_token::id(),
|
|
),
|
|
&[
|
|
associated_token_program.clone(),
|
|
signer.clone(),
|
|
treasury_tokens_info.clone(),
|
|
treasury_info.clone(),
|
|
mint_info.clone(),
|
|
system_program.clone(),
|
|
token_program.clone(),
|
|
],
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|