cuprate_types/rpc/
pool_info_extent.rs

1//! TODO
2
3//---------------------------------------------------------------------------------------------------- Use
4#[cfg(feature = "serde")]
5use serde::{Deserialize, Serialize};
6
7#[cfg(feature = "epee")]
8use cuprate_epee_encoding::{
9    error,
10    macros::bytes::{Buf, BufMut},
11    EpeeValue, Marker,
12};
13
14//---------------------------------------------------------------------------------------------------- PoolInfoExtent
15/// Used in RPC's `get_blocks.bin`.
16#[derive(Copy, Clone, Default, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
17#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
18#[repr(u8)]
19pub enum PoolInfoExtent {
20    #[default]
21    None = 0,
22    Incremental = 1,
23    Full = 2,
24}
25
26impl PoolInfoExtent {
27    /// Convert [`Self`] to a [`u8`].
28    ///
29    /// ```rust
30    /// use cuprate_types::rpc::PoolInfoExtent as P;
31    ///
32    /// assert_eq!(P::None.to_u8(), 0);
33    /// assert_eq!(P::Incremental.to_u8(), 1);
34    /// assert_eq!(P::Full.to_u8(), 2);
35    /// ```
36    pub const fn to_u8(self) -> u8 {
37        match self {
38            Self::None => 0,
39            Self::Incremental => 1,
40            Self::Full => 2,
41        }
42    }
43
44    /// Convert a [`u8`] to a [`Self`].
45    ///
46    /// # Errors
47    /// This returns [`None`] if `u > 2`.
48    ///
49    /// ```rust
50    /// use cuprate_types::rpc::PoolInfoExtent as P;
51    ///
52    /// assert_eq!(P::from_u8(0), Some(P::None));
53    /// assert_eq!(P::from_u8(1), Some(P::Incremental));
54    /// assert_eq!(P::from_u8(2), Some(P::Full));
55    /// assert_eq!(P::from_u8(3), None);
56    /// ```
57    pub const fn from_u8(u: u8) -> Option<Self> {
58        Some(match u {
59            0 => Self::None,
60            1 => Self::Incremental,
61            2 => Self::Full,
62            _ => return None,
63        })
64    }
65}
66
67#[cfg(feature = "epee")]
68impl EpeeValue for PoolInfoExtent {
69    const MARKER: Marker = <u8 as EpeeValue>::MARKER;
70
71    fn read<B: Buf>(r: &mut B, marker: &Marker) -> error::Result<Self> {
72        let u = u8::read(r, marker)?;
73        Self::from_u8(u).ok_or(error::Error::Format("u8 was greater than 2"))
74    }
75
76    fn write<B: BufMut>(self, w: &mut B) -> error::Result<()> {
77        let u = self.to_u8();
78        u8::write(u, w)?;
79        Ok(())
80    }
81}