From 74c8a1a073471bedc15f5508624f81f9b910eade Mon Sep 17 00:00:00 2001
From: Hardhat Chad <hardhatchad@gmail.com>
Date: Tue, 24 Sep 2024 05:28:14 +0000
Subject: [PATCH] reimplement staking multiplier

---
 program/src/mine.rs | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/program/src/mine.rs b/program/src/mine.rs
index 5c05b8a..c6a0907 100644
--- a/program/src/mine.rs
+++ b/program/src/mine.rs
@@ -101,6 +101,30 @@ pub fn process_mine(accounts: &[AccountInfo], data: &[u8]) -> ProgramResult {
         .checked_mul(2u64.checked_pow(normalized_difficulty).unwrap())
         .unwrap();
 
+    // Apply staking multiplier.
+    //
+    // If user has greater than or equal to the max stake on the network, they receive 2x multiplier.
+    // Any stake less than this will receives between 1x and 2x multipler. The multipler is only active
+    // if the miner's last stake deposit was more than one minute ago to protect against flash loan attacks.
+    let mut bus_data = bus_info.data.borrow_mut();
+    let bus = Bus::try_from_bytes_mut(&mut bus_data)?;
+    if proof.balance.gt(&0) && proof.last_stake_at.saturating_add(ONE_MINUTE).lt(&t) {
+        // Calculate staking reward.
+        if config.top_balance.gt(&0) {
+            let staking_reward = (reward as u128)
+                .checked_mul(proof.balance.min(config.top_balance) as u128)
+                .unwrap()
+                .checked_div(config.top_balance as u128)
+                .unwrap() as u64;
+            reward = reward.checked_add(staking_reward).unwrap();
+        }
+
+        // Update bus stake tracker.
+        if proof.balance.gt(&bus.top_balance) {
+            bus.top_balance = proof.balance;
+        }
+    }
+
     // Apply boosts.
     //
     // If user has greater than or equal to the max stake on the network, they receive 2x multiplier.