mirror of
https://github.com/d0zingcat/ore.git
synced 2026-05-16 23:16:47 +00:00
5x5
This commit is contained in:
@@ -4,22 +4,43 @@ use steel::*;
|
||||
/// Initializes the program.
|
||||
pub fn process_initialize(accounts: &[AccountInfo<'_>], _data: &[u8]) -> ProgramResult {
|
||||
// Load accounts.
|
||||
let [signer_info, config_info, market_info, mint_info, treasury_info, treasury_tokens_info, vault_info, system_program, token_program, associated_token_program] =
|
||||
let [signer_info, board_info, config_info, mint_info, treasury_info, vault_info, system_program, token_program, associated_token_program] =
|
||||
accounts
|
||||
else {
|
||||
return Err(ProgramError::NotEnoughAccountKeys);
|
||||
};
|
||||
signer_info.is_signer()?.has_address(&ADMIN_ADDRESS)?;
|
||||
board_info.has_seeds(&[BOARD], &ore_api::ID)?;
|
||||
config_info.has_seeds(&[CONFIG], &ore_api::ID)?;
|
||||
market_info.has_seeds(&[MARKET], &ore_api::ID)?;
|
||||
mint_info.has_address(&MINT_ADDRESS)?.as_mint()?;
|
||||
treasury_info.has_seeds(&[TREASURY], &ore_api::ID)?;
|
||||
treasury_tokens_info.is_writable()?;
|
||||
vault_info.has_address(&vault_address())?;
|
||||
system_program.is_program(&system_program::ID)?;
|
||||
token_program.is_program(&spl_token::ID)?;
|
||||
associated_token_program.is_program(&spl_associated_token_account::ID)?;
|
||||
|
||||
// Create board account.
|
||||
if board_info.data_is_empty() {
|
||||
create_program_account::<Board>(
|
||||
board_info,
|
||||
system_program,
|
||||
signer_info,
|
||||
&ore_api::ID,
|
||||
&[BOARD],
|
||||
)?;
|
||||
let board = board_info.as_account_mut::<Board>(&ore_api::ID)?;
|
||||
board.commits = [0; 25];
|
||||
board.id = 0;
|
||||
board.start_at = 0;
|
||||
board.start_slot = 0;
|
||||
board.end_slot = 0;
|
||||
board.slot_hash = [0; 32];
|
||||
board.total_commits = 0;
|
||||
board.total_burned = 0;
|
||||
} else {
|
||||
board_info.as_account::<Board>(&ore_api::ID)?;
|
||||
}
|
||||
|
||||
// Create config account.
|
||||
if config_info.data_is_empty() {
|
||||
create_program_account::<Config>(
|
||||
@@ -39,42 +60,6 @@ pub fn process_initialize(accounts: &[AccountInfo<'_>], _data: &[u8]) -> Program
|
||||
config_info.as_account::<Config>(&ore_api::ID)?;
|
||||
}
|
||||
|
||||
// Initialize market.
|
||||
if market_info.data_is_empty() {
|
||||
create_program_account::<Market>(
|
||||
market_info,
|
||||
system_program,
|
||||
signer_info,
|
||||
&ore_api::ID,
|
||||
&[MARKET],
|
||||
)?;
|
||||
let market = market_info.as_account_mut::<Market>(&ore_api::ID)?;
|
||||
market.base = TokenParams {
|
||||
mint: Pubkey::default(), // Virtual token
|
||||
balance: 0,
|
||||
balance_virtual: 0,
|
||||
};
|
||||
market.quote = TokenParams {
|
||||
mint: MINT_ADDRESS,
|
||||
balance: 0,
|
||||
balance_virtual: 0,
|
||||
};
|
||||
market.fee = FeeParams {
|
||||
rate: 0,
|
||||
uncollected: 0,
|
||||
cumulative: 0,
|
||||
};
|
||||
market.snapshot = Snapshot {
|
||||
enabled: 1,
|
||||
base_balance: 0,
|
||||
quote_balance: 0,
|
||||
slot: 0,
|
||||
};
|
||||
market.block_id = 0;
|
||||
} else {
|
||||
market_info.as_account::<Market>(&ore_api::ID)?;
|
||||
}
|
||||
|
||||
// Create treasury account.
|
||||
if treasury_info.data_is_empty() {
|
||||
create_program_account::<Treasury>(
|
||||
@@ -88,26 +73,11 @@ pub fn process_initialize(accounts: &[AccountInfo<'_>], _data: &[u8]) -> Program
|
||||
treasury_info.as_account::<Treasury>(&ore_api::ID)?;
|
||||
}
|
||||
|
||||
// Load treasury tokens.
|
||||
if treasury_tokens_info.data_is_empty() {
|
||||
create_associated_token_account(
|
||||
signer_info,
|
||||
treasury_info,
|
||||
treasury_tokens_info,
|
||||
mint_info,
|
||||
system_program,
|
||||
token_program,
|
||||
associated_token_program,
|
||||
)?;
|
||||
} else {
|
||||
treasury_tokens_info.as_associated_token_account(treasury_info.key, mint_info.key)?;
|
||||
}
|
||||
|
||||
// Initialize vault token account.
|
||||
if vault_info.data_is_empty() {
|
||||
create_associated_token_account(
|
||||
signer_info,
|
||||
market_info,
|
||||
board_info,
|
||||
vault_info,
|
||||
mint_info,
|
||||
system_program,
|
||||
@@ -115,7 +85,7 @@ pub fn process_initialize(accounts: &[AccountInfo<'_>], _data: &[u8]) -> Program
|
||||
associated_token_program,
|
||||
)?;
|
||||
} else {
|
||||
vault_info.as_associated_token_account(market_info.key, mint_info.key)?;
|
||||
vault_info.as_associated_token_account(board_info.key, mint_info.key)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
|
||||
Reference in New Issue
Block a user