testdenali/
dcdt_raw_check.rs

1use super::*;
2use serde::{
3    de::{self, Deserializer, MapAccess, Visitor},
4    ser::{SerializeMap, Serializer},
5    Deserialize, Serialize,
6};
7use std::{collections::BTreeMap, fmt};
8
9pub enum CheckDcdtRaw {
10    Unspecified,
11    Star,
12    Equal(BTreeMap<String, CheckBytesValueRaw>),
13}
14
15impl CheckDcdtRaw {
16    pub fn is_star(&self) -> bool {
17        matches!(self, CheckDcdtRaw::Star)
18    }
19
20    pub fn is_unspecified(&self) -> bool {
21        matches!(self, CheckDcdtRaw::Unspecified)
22    }
23}
24
25impl Default for CheckDcdtRaw {
26    fn default() -> Self {
27        CheckDcdtRaw::Unspecified
28    }
29}
30
31impl Serialize for CheckDcdtRaw {
32    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
33    where
34        S: Serializer,
35    {
36        match self {
37            CheckDcdtRaw::Unspecified => {
38                // empty map, just in case
39                // won't get serialized anyway
40                let map = serializer.serialize_map(Some(0))?;
41                map.end()
42            },
43            CheckDcdtRaw::Star => serializer.serialize_str("*"),
44            CheckDcdtRaw::Equal(m) => {
45                let mut map = serializer.serialize_map(Some(m.len()))?;
46                for (k, v) in m {
47                    map.serialize_entry(k, v)?;
48                }
49                map.end()
50            },
51        }
52    }
53}
54
55struct CheckDcdtRawVisitor;
56
57impl<'de> Visitor<'de> for CheckDcdtRawVisitor {
58    type Value = CheckDcdtRaw;
59
60    // Format a message stating what data this Visitor expects to receive.
61    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
62        formatter.write_str("serialized object JSON representation of dcdt check")
63    }
64
65    fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
66    where
67        E: de::Error,
68    {
69        if value == "*" {
70            Ok(CheckDcdtRaw::Star)
71        } else {
72            Err(de::Error::custom("only '*' allowed as dcdt string value"))
73        }
74    }
75
76    fn visit_map<M>(self, mut access: M) -> Result<Self::Value, M::Error>
77    where
78        M: MapAccess<'de>,
79    {
80        let mut map = BTreeMap::<String, CheckBytesValueRaw>::new();
81
82        // While there are entries remaining in the input, add them
83        // into our map.
84        while let Some((key, value)) = access.next_entry()? {
85            map.insert(key, value);
86        }
87
88        Ok(CheckDcdtRaw::Equal(map))
89    }
90}
91
92impl<'de> Deserialize<'de> for CheckDcdtRaw {
93    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
94    where
95        D: Deserializer<'de>,
96    {
97        deserializer.deserialize_any(CheckDcdtRawVisitor)
98    }
99}
testdenali/dcdt_raw_check.rs

testdenali/
dcdt_raw_check.rs
