1use fvm_shared2::clock::ChainEpoch as ChainEpoch_v2;
5use fvm_shared3::clock::ChainEpoch as ChainEpoch_v3;
6use fvm_shared4::clock::ChainEpoch as ChainEpoch_v4;
7use fvm2::externs::Rand as Rand_v2;
8use fvm3::externs::Rand as Rand_v3;
9use fvm4::externs::Rand as Rand_v4;
10
11#[derive(Clone, Debug)]
12pub struct RandWrapper<T> {
13 pub chain_rand: T,
14}
15
16pub trait Rand {
17 fn get_chain_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]>;
18 fn get_beacon_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]>;
19}
20
21impl Rand for Box<dyn Rand> {
22 fn get_chain_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]> {
23 self.as_ref().get_chain_randomness(round)
24 }
25 fn get_beacon_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]> {
26 self.as_ref().get_beacon_randomness(round)
27 }
28}
29
30impl<T: Rand> Rand_v2 for RandWrapper<T> {
31 fn get_chain_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]> {
32 self.chain_rand.get_chain_randomness(round)
33 }
34
35 fn get_beacon_randomness(&self, round: ChainEpoch_v2) -> anyhow::Result<[u8; 32]> {
36 self.chain_rand.get_beacon_randomness(round)
37 }
38}
39
40impl<T: Rand> Rand_v3 for RandWrapper<T> {
41 fn get_chain_randomness(&self, round: ChainEpoch_v3) -> anyhow::Result<[u8; 32]> {
42 self.chain_rand.get_chain_randomness(round)
43 }
44
45 fn get_beacon_randomness(&self, round: ChainEpoch_v3) -> anyhow::Result<[u8; 32]> {
46 self.chain_rand.get_beacon_randomness(round)
47 }
48}
49
50impl<T: Rand> Rand_v4 for RandWrapper<T> {
51 fn get_chain_randomness(&self, round: ChainEpoch_v4) -> anyhow::Result<[u8; 32]> {
52 self.chain_rand.get_chain_randomness(round)
53 }
54
55 fn get_beacon_randomness(&self, round: ChainEpoch_v4) -> anyhow::Result<[u8; 32]> {
56 self.chain_rand.get_beacon_randomness(round)
57 }
58}
59
60impl<T> From<T> for RandWrapper<T> {
61 fn from(chain_rand: T) -> Self {
62 RandWrapper { chain_rand }
63 }
64}