1use serde::{Deserialize, Serialize};
2use steel::*;
3
4use crate::state::round_pda;
5
6use super::OilAccount;
7
8#[repr(C)]
9#[derive(Clone, Copy, Debug, PartialEq, Pod, Zeroable, Serialize, Deserialize, Default)]
10pub struct Round {
11 pub id: u64,
13
14 pub deployed: [u64; 25],
16
17 pub slot_hash: [u8; 32],
19
20 pub count: [u64; 25],
22
23 pub expires_at: u64,
25
26 pub rent_payer: Pubkey,
28
29 pub top_miner: Pubkey,
31
32 pub top_miner_reward: u64,
34
35 pub total_deployed: u64,
37
38 pub total_miners: u64,
40
41 pub total_vaulted: u64,
43
44 pub total_winnings: u64,
46
47 pub gusher_sol: u64,
49
50 pub deployed_pooled: [u64; 25],
52
53 pub total_pooled: u64,
55
56 pub pool_rewards_sol: u64,
58
59 pub pool_rewards_oil: u64,
61
62 pub pool_members: u64,
64
65 pub pool_cumulative: [u64; 25],
67}
68
69impl Round {
70 pub fn pda(&self) -> (Pubkey, u8) {
71 round_pda(self.id)
72 }
73
74 pub fn rng(&self) -> Option<u64> {
75 if self.slot_hash == [0; 32] || self.slot_hash == [u8::MAX; 32] {
76 return None;
77 }
78 let r1 = u64::from_le_bytes(self.slot_hash[0..8].try_into().unwrap());
79 let r2 = u64::from_le_bytes(self.slot_hash[8..16].try_into().unwrap());
80 let r3 = u64::from_le_bytes(self.slot_hash[16..24].try_into().unwrap());
81 let r4 = u64::from_le_bytes(self.slot_hash[24..32].try_into().unwrap());
82 let r = r1 ^ r2 ^ r3 ^ r4;
83 Some(r)
84 }
85
86 pub fn winning_square(&self, rng: u64) -> usize {
87 (rng % 25) as usize
88 }
89
90 pub fn top_miner_sample(&self, rng: u64, winning_square: usize) -> u64 {
91 if self.deployed[winning_square] == 0 {
92 return 0;
93 }
94 rng.reverse_bits() % self.deployed[winning_square]
95 }
96
97 pub fn calculate_total_winnings(&self, winning_square: usize) -> u64 {
98 let mut total_winnings = 0;
99 for (i, &deployed) in self.deployed.iter().enumerate() {
100 if i != winning_square {
101 total_winnings += deployed;
102 }
103 }
104 total_winnings
105 }
106
107 pub fn is_split_reward(&self, rng: u64) -> bool {
108 let rng = rng.reverse_bits().to_le_bytes();
110 let r1 = u16::from_le_bytes(rng[0..2].try_into().unwrap());
111 let r2 = u16::from_le_bytes(rng[2..4].try_into().unwrap());
112 let r3 = u16::from_le_bytes(rng[4..6].try_into().unwrap());
113 let r4 = u16::from_le_bytes(rng[6..8].try_into().unwrap());
114 let r = r1 ^ r2 ^ r3 ^ r4;
115 r % 2 == 0
116 }
117
118 pub fn did_hit_gusher_sol_only(&self, rng: u64) -> bool {
119 rng.reverse_bits() % 625 == 0
120 }
121}
122
123account!(OilAccount, Round);
124
125#[cfg(test)]
126mod tests {
127 use solana_program::rent::Rent;
128
129 use super::*;
130
131 #[test]
132 fn test_rent() {
133 let size_of_round = 8 + std::mem::size_of::<Round>();
134 let required_rent = Rent::default().minimum_balance(size_of_round);
135 println!("Round account size: {} bytes", size_of_round);
136 println!("Required rent: {} lamports ({} SOL)", required_rent, required_rent as f64 / 1_000_000_000.0);
137
138 assert_eq!(size_of_round, 1008, "Round account size should be 1008 bytes (v4 with pool_cumulative, no padding)");
141
142 assert!(required_rent > 0, "Required rent should be greater than 0");
144 assert!(required_rent < 10_000_000_000, "Required rent should be less than 10 SOL");
145 }
146}