scaffolding

program/src/withdraw.rs

@@ -0,0 +1,46 @@
+use ore_api::prelude::*;
+use steel::*;
+
+/// Withdraws stake.
+pub fn process_withdraw(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
+    // Parse data.
+    let args = Withdraw::try_from_bytes(data)?;
+    let amount = u64::from_le_bytes(args.amount);
+
+    // Load accounts.
+    let [signer_info, miner_info, recipient_info, treasury_info, treasury_tokens_info, token_program] =
+        accounts
+    else {
+        return Err(ProgramError::NotEnoughAccountKeys);
+    };
+    signer_info.is_signer()?;
+    let miner = miner_info
+        .as_account_mut::<Miner>(&ore_api::ID)?
+        .assert_mut(|m| m.authority == *signer_info.key)?;
+    recipient_info.as_associated_token_account(&signer_info.key, &MINT_ADDRESS)?;
+    let treasury = treasury_info.as_account_mut::<Treasury>(&ore_api::ID)?;
+    treasury_tokens_info.as_associated_token_account(&treasury_info.key, &MINT_ADDRESS)?;
+    token_program.is_program(&spl_token::ID)?;
+
+    // Update account state.
+    miner.stake -= amount;
+    treasury.total_stake -= amount;
+
+    // Asset miner has enough stake to cover the withdrawal.
+    assert!(
+        miner.stake >= miner.deployed,
+        "Withdrawal cannot reduce capacity below deployment levels."
+    );
+
+    // Execute transfer.
+    transfer_signed(
+        signer_info,
+        treasury_tokens_info,
+        recipient_info,
+        token_program,
+        amount,
+        &[TREASURY],
+    )?;
+
+    Ok(())
+}