openc2 0.2.0

Rust types for OpenC2 commands and responses
Documentation 
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

use serde::{Deserialize, Deserializer, Serialize, Serializer, ser::SerializeMap};

use crate::Value;

/// A map containing a single key-value pair.
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Choice<K, V> {
    pub key: K,
    pub value: V,
}

impl<K, V> Choice<K, V> {
    pub fn new(key: K, value: V) -> Self {
        Self { key, value }
    }
}

impl<K, V: Value> Choice<K, V> {
    pub fn new_value(key: K, value: impl Serialize) -> Result<Self, V::Error> {
        Ok(Self {
            key,
            value: V::from_typed(&value)?,
        })
    }
}

impl<K, V: Value> Choice<K, V> {
    pub fn get<'a, T: Deserialize<'a>>(&'a self) -> Result<T, V::Error> {
        self.value.to_typed()
    }
}

impl<K: Serialize, V: Serialize> Serialize for Choice<K, V> {
    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        let mut map = serializer.serialize_map(Some(1))?;
        map.serialize_entry(&self.key, &self.value)?;
        map.end()
    }
}

impl<'de, K: Deserialize<'de>, V: Deserialize<'de>> Deserialize<'de> for Choice<K, V> {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        struct ChoiceVisitor<K, V>(std::marker::PhantomData<(K, V)>);

        impl<'de, K: Deserialize<'de>, V: Deserialize<'de>> serde::de::Visitor<'de>
            for ChoiceVisitor<K, V>
        {
            type Value = Choice<K, V>;

            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
                formatter.write_str("a map with a single key-value pair")
            }

            fn visit_map<M>(self, mut access: M) -> Result<Self::Value, M::Error>
            where
                M: serde::de::MapAccess<'de>,
            {
                if let Some((key, value)) = access.next_entry()? {
                    if access.next_key::<K>()?.is_some() {
                        return Err(serde::de::Error::custom("expected a single key-value pair"));
                    }
                    Ok(Choice { key, value })
                } else {
                    Err(serde::de::Error::custom("expected a single key-value pair"))
                }
            }
        }

        deserializer.deserialize_map(ChoiceVisitor(std::marker::PhantomData))
    }
}