deploy

program/src/deploy.rs

@@ -0,0 +1,116 @@
+use ore_api::prelude::*;
+use steel::*;
+
+use crate::whitelist::AUTHORIZED_ACCOUNTS;
+
+/// Deploys capital to prospect on a square.
+pub fn process_deploy(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
+    // Parse data.
+    let args = Deploy::try_from_bytes(data)?;
+    let amount = u64::from_le_bytes(args.amount);
+    let square_id = usize::from_le_bytes(args.square_id);
+
+    // Load accounts.
+    let clock = Clock::get()?;
+    let [signer_info, board_info, config_info, fee_collector_info, miner_info, square_info, system_program] =
+        accounts
+    else {
+        return Err(ProgramError::NotEnoughAccountKeys);
+    };
+    signer_info.is_signer()?;
+    let board = board_info
+        .as_account_mut::<Board>(&ore_api::ID)?
+        .assert_mut(|b| {
+            (clock.slot >= b.start_slot && clock.slot < b.end_slot && b.slot_hash == [0; 32])
+                || (clock.slot >= b.end_slot + 25 && b.slot_hash != [0; 32])
+        })?;
+    let config = config_info.as_account::<Config>(&ore_api::ID)?;
+    fee_collector_info
+        .has_address(&config.fee_collector)?
+        .is_writable()?;
+    miner_info.is_writable()?;
+    let square = square_info.as_account_mut::<Square>(&ore_api::ID)?;
+    system_program.is_program(&system_program::ID)?;
+
+    // Check whitelist
+    if !AUTHORIZED_ACCOUNTS.contains(&signer_info.key) {
+        return Err(OreError::NotAuthorized.into());
+    }
+
+    // Check minimum amount.
+    if amount < config.min_deploy_amount {
+        return Err(OreError::AmountTooSmall.into());
+    }
+
+    // Create miner.
+    let miner = if miner_info.data_is_empty() {
+        create_program_account::<Miner>(
+            miner_info,
+            system_program,
+            signer_info,
+            &ore_api::ID,
+            &[MINER, &signer_info.key.to_bytes()],
+        )?;
+        let miner = miner_info.as_account_mut::<Miner>(&ore_api::ID)?;
+        miner.authority = *signer_info.key;
+        miner.prospects = [0; 25];
+        miner.rewards_sol = 0;
+        miner.rewards_ore = 0;
+        miner.round_id = board.id;
+        miner.lifetime_rewards_sol = 0;
+        miner.lifetime_rewards_ore = 0;
+        miner
+    } else {
+        miner_info
+            .as_account_mut::<Miner>(&ore_api::ID)?
+            .assert_mut(|m| m.authority == *signer_info.key)?
+    };
+
+    // Reset board.
+    if board.slot_hash != [0; 32] {
+        // Reset board
+        board.prospects = [0; 25];
+        board.id += 1;
+        board.slot_hash = [0; 32];
+        board.start_slot = clock.slot;
+        board.end_slot = clock.slot + 150; // one minute
+        board.top_miner = Pubkey::default();
+        board.total_prospects = 0;
+        board.total_vaulted = 0;
+        board.total_winnings = 0;
+
+        // Reset squares
+        square.count = [0; 25];
+        square.miners = [[Pubkey::default(); 16]; 25];
+    }
+
+    // Reset miner
+    if miner.round_id != board.id {
+        miner.prospects = [0; 25];
+        miner.round_id = board.id;
+    }
+
+    // Normalize amount.
+    let fee = amount / 100;
+    let amount = amount - fee;
+
+    // Update miner
+    let is_first_move = miner.prospects[square_id] == 0;
+    miner.prospects[square_id] += amount;
+
+    // Update square
+    if is_first_move {
+        square.miners[square_id][square.count[square_id] as usize] = *signer_info.key;
+        square.count[square_id] += 1;
+    }
+
+    // Update board
+    board.prospects[square_id] += amount;
+    board.total_prospects += amount;
+
+    // Transfer prospects.
+    board_info.collect(amount, &signer_info)?;
+    fee_collector_info.collect(fee, &signer_info)?;
+
+    Ok(())
+}