stake

src/processor/claim.rs

@@ -56,12 +56,9 @@ pub fn process_claim<'a, 'info>(
         .checked_sub(amount)
         .ok_or(OreError::ClaimTooLarge)?;
 
-    // Update lifetime status
+    // Distribute tokens from treasury to beneficiary
     let mut treasury_data = treasury_info.data.borrow_mut();
     let treasury = Treasury::try_from_bytes_mut(&mut treasury_data)?;
-    treasury.total_claimed_rewards = treasury.total_claimed_rewards.saturating_add(amount);
-
-    // Distribute tokens from treasury to beneficiary
     let treasury_bump = treasury.bump;
     drop(treasury_data);
     solana_program::program::invoke_signed(

src/processor/initialize.rs

@@ -148,7 +148,7 @@ pub fn process_initialize<'a, 'info>(
     treasury.bump = args.treasury_bump as u64;
     // treasury.last_reset_at = 0;
     // treasury.reward_rate = INITIAL_REWARD_RATE;
-    treasury.total_claimed_rewards = 0;
+    // treasury.total_claimed_rewards = 0;
     drop(treasury_data);
 
     // Initialize mint

src/processor/mod.rs

@@ -3,6 +3,7 @@ mod initialize;
 mod mine;
 mod register;
 mod reset;
+mod stake;
 mod update_admin;
 
 pub use claim::*;
@@ -10,4 +11,5 @@ pub use initialize::*;
 pub use mine::*;
 pub use register::*;
 pub use reset::*;
+pub use stake::*;
 pub use update_admin::*;

src/processor/stake.rs

@@ -1 +1,67 @@
+use solana_program::{
+    account_info::AccountInfo, clock::Clock, entrypoint::ProgramResult,
+    program_error::ProgramError, pubkey::Pubkey, sysvar::Sysvar,
+};
 
+use crate::{
+    error::OreError, instruction::StakeArgs, loaders::*, state::Proof, utils::AccountDeserialize,
+    MINT_ADDRESS, TREASURY_ADDRESS,
+};
+
+pub fn process_stake<'a, 'info>(
+    _program_id: &Pubkey,
+    accounts: &'a [AccountInfo<'info>],
+    data: &[u8],
+) -> ProgramResult {
+    // Parse args
+    let args = StakeArgs::try_from_bytes(data)?;
+    let amount = u64::from_le_bytes(args.amount);
+
+    // Load accounts
+    let [signer_info, sender_info, proof_info, treasury_tokens_info, token_program] = accounts
+    else {
+        return Err(ProgramError::NotEnoughAccountKeys);
+    };
+    load_signer(signer_info)?;
+    load_token_account(sender_info, Some(signer_info.key), &MINT_ADDRESS, true)?;
+    load_proof(proof_info, signer_info.key, true)?;
+    load_token_account(
+        treasury_tokens_info,
+        Some(&TREASURY_ADDRESS),
+        &MINT_ADDRESS,
+        true,
+    )?;
+    load_program(token_program, spl_token::id())?;
+
+    // Update proof balance
+    let mut proof_data = proof_info.data.borrow_mut();
+    let proof = Proof::try_from_bytes_mut(&mut proof_data)?;
+    proof.balance = proof
+        .balance
+        .checked_add(amount)
+        .ok_or(OreError::StakeTooLarge)?;
+
+    // Update deposit timestamp
+    let clock = Clock::get().or(Err(ProgramError::InvalidAccountData))?;
+    proof.last_deposit_slot = clock.slot;
+
+    // Distribute tokens from signer to treasury
+    solana_program::program::invoke(
+        &spl_token::instruction::transfer(
+            &spl_token::id(),
+            sender_info.key,
+            treasury_tokens_info.key,
+            signer_info.key,
+            &[signer_info.key],
+            amount,
+        )?,
+        &[
+            token_program.clone(),
+            sender_info.clone(),
+            treasury_tokens_info.clone(),
+            signer_info.clone(),
+        ],
+    )?;
+
+    Ok(())
+}