boost

program/src/boost.rs

@@ -0,0 +1,39 @@
+use ore_api::prelude::*;
+use steel::*;
+
+/// Sets the admin.
+pub fn process_boost(accounts: &[AccountInfo<'_>], _data: &[u8]) -> ProgramResult {
+    // Load accounts.
+    let clock = Clock::get()?;
+    let [signer_info, config_info, mint_info, reserve_tokens_info, treasury_info, system_program, token_program] =
+        accounts
+    else {
+        return Err(ProgramError::NotEnoughAccountKeys);
+    };
+    signer_info.is_signer()?;
+    let config = config_info
+        .as_account_mut::<Config>(&ore_api::ID)?
+        .assert_mut(|c| clock.unix_timestamp >= c.last_boost + (10 * ONE_MINUTE))?;
+    mint_info.has_address(&MINT_ADDRESS)?.as_mint()?;
+    reserve_tokens_info
+        .has_address(&BOOST_RESERVE_TOKEN)?
+        .as_token_account()?
+        .assert(|t| t.mint() == MINT_ADDRESS)?;
+    system_program.is_program(&system_program::ID)?;
+    token_program.is_program(&spl_token::ID)?;
+
+    // Update config.
+    config.last_boost = clock.unix_timestamp;
+
+    // Mint tokens to the boost program.
+    mint_to_signed(
+        mint_info,
+        reserve_tokens_info,
+        treasury_info,
+        token_program,
+        ONE_ORE * 3,
+        &[TREASURY],
+    )?;
+
+    Ok(())
+}

program/src/initialize.rs

@@ -54,9 +54,9 @@ pub fn process_initialize(accounts: &[AccountInfo<'_>], _data: &[u8]) -> Program
         )?;
         let config = config_info.as_account_mut::<Config>(&ore_api::ID)?;
         config.admin = *signer_info.key;
-        config.block_duration = 0;
-        config.sniper_fee_duration = 0;
+        config.last_boost = 0;
         config.fee_collector = *signer_info.key;
+        config.sniper_fee_duration = 0;
         config.fee_rate = 0;
     } else {
         config_info.as_account::<Config>(&ore_api::ID)?;

program/src/lib.rs

@@ -1,3 +1,4 @@
+mod boost;
 mod claim_ore;
 mod claim_seeker;
 mod claim_sol;
@@ -10,6 +11,7 @@ mod set_admin;
 mod set_fee_collector;
 mod whitelist;
 
+use boost::*;
 use claim_ore::*;
 use claim_seeker::*;
 use claim_sol::*;
@@ -33,6 +35,7 @@ pub fn process_instruction(
 
     match ix {
         // User
+        OreInstruction::Boost => process_boost(accounts, data)?,
         OreInstruction::ClaimSOL => process_claim_sol(accounts, data)?,
         OreInstruction::ClaimORE => process_claim_ore(accounts, data)?,
         OreInstruction::Initialize => process_initialize(accounts, data)?,

program/src/reset.rs

@@ -7,7 +7,7 @@ pub fn process_reset(accounts: &[AccountInfo<'_>], _data: &[u8]) -> ProgramResul
     // Load accounts.
     let clock = Clock::get()?;
     let (required_accounts, miner_accounts) = accounts.split_at(9);
-    let [signer_info, board_info, mint_info, reserve_tokens_info, treasury_info, treasury_tokens_info, system_program, token_program, slot_hashes_sysvar] =
+    let [signer_info, board_info, mint_info, treasury_info, treasury_tokens_info, system_program, token_program, slot_hashes_sysvar] =
         required_accounts
     else {
         return Err(ProgramError::NotEnoughAccountKeys);
@@ -18,26 +18,12 @@ pub fn process_reset(accounts: &[AccountInfo<'_>], _data: &[u8]) -> ProgramResul
         .assert_mut(|b| b.slot_hash == [0; 32])?
         .assert_mut(|b| clock.slot >= b.end_slot)?;
     let mint = mint_info.has_address(&MINT_ADDRESS)?.as_mint()?;
-    reserve_tokens_info
-        .has_address(&BOOST_RESERVE_TOKEN)?
-        .as_token_account()?
-        .assert(|t| t.mint() == MINT_ADDRESS)?;
     let treasury = treasury_info.as_account_mut::<Treasury>(&ore_api::ID)?;
     treasury_tokens_info.as_associated_token_account(&treasury_info.key, &mint_info.key)?;
     system_program.is_program(&system_program::ID)?;
     token_program.is_program(&spl_token::ID)?;
     slot_hashes_sysvar.is_sysvar(&sysvar::slot_hashes::ID)?;
 
-    // Mint tokens to the boost program.
-    mint_to_signed(
-        mint_info,
-        reserve_tokens_info,
-        treasury_info,
-        token_program,
-        ONE_ORE / 3,
-        &[TREASURY],
-    )?;
-
     // Sample slot hash.
     let (winning_square, square_prospects) =
         if let Ok(slot_hash) = get_slot_hash(board.end_slot, slot_hashes_sysvar) {