1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

use chain_impl_mockchain::stake;
use serde::{Deserialize, Deserializer, Serialize, Serializer};
use std::{fmt, str::FromStr};

/// Stake in the blockchain, always printed as absolute Lovelace
///
/// Stake has some property to be human readable on standard display
///
/// ```
/// # use jormungandr_lib::interfaces::Stake;
/// # use chain_impl_mockchain::stake::Stake as StdStake;
///
/// let stake: Stake = StdStake(64).into();
///
/// println!("stake: {}", stake);
///
/// # assert_eq!(stake.to_string(), "64");
/// ```
///
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub struct Stake(stake::Stake);

/* ---------------- Display ------------------------------------------------ */

impl fmt::Display for Stake {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.fmt(f)
    }
}

impl FromStr for Stake {
    type Err = std::num::ParseIntError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        s.parse().map(|v| Stake(stake::Stake(v)))
    }
}

/* ---------------- AsRef -------------------------------------------------- */

impl AsRef<stake::Stake> for Stake {
    fn as_ref(&self) -> &stake::Stake {
        &self.0
    }
}
/* ---------------- Conversion --------------------------------------------- */

impl From<stake::Stake> for Stake {
    fn from(v: stake::Stake) -> Self {
        Stake(v)
    }
}

impl From<Stake> for stake::Stake {
    fn from(v: Stake) -> Self {
        v.0
    }
}

impl From<u64> for Stake {
    fn from(v: u64) -> Self {
        Stake(stake::Stake(v))
    }
}

impl From<Stake> for u64 {
    fn from(stake: Stake) -> u64 {
        (stake.0).0
    }
}

/* ------------------- Serde ----------------------------------------------- */

impl Serialize for Stake {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        self.0.as_ref().serialize(serializer)
    }
}

impl<'de> Deserialize<'de> for Stake {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        let v = u64::deserialize(deserializer)?;
        Ok(Stake(stake::Stake(v)))
    }
}

#[derive(Deserialize, Serialize)]
#[serde(transparent, remote = "stake::Stake")]
pub struct StakeDef(u64);

#[cfg(test)]
mod test {
    use super::*;
    use quickcheck::{Arbitrary, Gen, TestResult};

    impl Arbitrary for Stake {
        fn arbitrary<G>(g: &mut G) -> Self
        where
            G: Gen,
        {
            Stake(stake::Stake(u64::arbitrary(g)))
        }
    }

    #[test]
    fn stake_display_as_u64() {
        const STAKE: u64 = 928_170;
        let stake = Stake(stake::Stake(STAKE));

        assert_eq!(stake.to_string(), STAKE.to_string());
    }

    #[test]
    fn stake_serde_as_u64() {
        const STAKE: u64 = 928_170;
        let stake = Stake(stake::Stake(STAKE));

        assert_eq!(
            serde_yaml::to_string(&stake).unwrap(),
            format!("---\n{}\n", STAKE)
        );
    }

    quickcheck! {
        fn stake_display_parse(stake: Stake) -> TestResult {
            let s = stake.to_string();
            let stake_dec: Stake = s.parse().unwrap();

            TestResult::from_bool(stake_dec == stake)
        }

        fn stake_serde_human_readable_encode_decode(stake: Stake) -> TestResult {
            let s = serde_yaml::to_string(&stake).unwrap();
            let stake_dec: Stake = serde_yaml::from_str(&s).unwrap();

            TestResult::from_bool(stake_dec == stake)
        }

        fn stake_serde_binary_encode_decode(stake: Stake) -> TestResult {
            let s = bincode::serialize(&stake).unwrap();
            let stake_dec: Stake = bincode::deserialize(&s).unwrap();

            TestResult::from_bool(stake_dec == stake)
        }
    }
}