zen_types/variable/
de.rs

1#[cfg(feature = "arbitrary_precision")]
2use crate::constant::NUMBER_TOKEN;
3use crate::variable::Variable;
4use ahash::{HashMap, HashMapExt};
5use rust_decimal::Decimal;
6use rust_decimal::prelude::FromPrimitive;
7use serde::de::{DeserializeSeed, Error, MapAccess, SeqAccess, Unexpected, Visitor};
8use serde::{Deserialize, Deserializer};
9use std::fmt::Formatter;
10use std::marker::PhantomData;
11#[cfg(feature = "arbitrary_precision")]
12use std::ops::Deref;
13use std::rc::Rc;
14
15struct VariableVisitor;
16
17impl<'de> Visitor<'de> for VariableVisitor {
18    type Value = Variable;
19
20    fn expecting(&self, formatter: &mut Formatter) -> std::fmt::Result {
21        formatter.write_str("A valid type")
22    }
23
24    fn visit_bool<E>(self, v: bool) -> Result<Self::Value, E>
25    where
26        E: Error,
27    {
28        Ok(Variable::Bool(v))
29    }
30
31    fn visit_i64<E>(self, v: i64) -> Result<Self::Value, E>
32    where
33        E: Error,
34    {
35        Ok(Variable::Number(Decimal::from_i64(v).ok_or_else(|| {
36            Error::invalid_value(Unexpected::Signed(v), &self)
37        })?))
38    }
39
40    fn visit_u64<E>(self, v: u64) -> Result<Self::Value, E>
41    where
42        E: Error,
43    {
44        Ok(Variable::Number(Decimal::from_u64(v).ok_or_else(|| {
45            Error::invalid_value(Unexpected::Unsigned(v), &self)
46        })?))
47    }
48
49    fn visit_f64<E>(self, v: f64) -> Result<Self::Value, E>
50    where
51        E: Error,
52    {
53        Ok(Variable::Number(Decimal::from_f64(v).ok_or_else(|| {
54            Error::invalid_value(Unexpected::Float(v), &self)
55        })?))
56    }
57
58    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
59    where
60        E: Error,
61    {
62        Ok(Variable::String(Rc::from(v)))
63    }
64
65    fn visit_unit<E>(self) -> Result<Self::Value, E>
66    where
67        E: Error,
68    {
69        Ok(Variable::Null)
70    }
71
72    fn visit_seq<A>(self, mut seq: A) -> Result<Self::Value, A::Error>
73    where
74        A: SeqAccess<'de>,
75    {
76        let mut vec = Vec::with_capacity(seq.size_hint().unwrap_or_default());
77        while let Some(value) = seq.next_element_seed(VariableDeserializer)? {
78            vec.push(value);
79        }
80
81        Ok(Variable::from_array(vec))
82    }
83
84    fn visit_map<A>(self, mut map: A) -> Result<Self::Value, A::Error>
85    where
86        A: MapAccess<'de>,
87    {
88        let mut m = HashMap::with_capacity(map.size_hint().unwrap_or_default());
89        #[cfg(feature = "arbitrary_precision")]
90        let mut first = true;
91
92        while let Some((key, value)) =
93            map.next_entry_seed(PhantomData::<Rc<str>>, VariableDeserializer)?
94        {
95            #[cfg(feature = "arbitrary_precision")]
96            if first && key.deref() == NUMBER_TOKEN {
97                let str = value
98                    .as_str()
99                    .ok_or_else(|| Error::custom("failed to deserialize number"))?;
100
101                return Ok(Variable::Number(
102                    Decimal::from_str_exact(str)
103                        .or_else(|_| Decimal::from_scientific(str))
104                        .map_err(|_| Error::custom("invalid number"))?,
105                ));
106            }
107
108            m.insert(key, value);
109            #[cfg(feature = "arbitrary_precision")]
110            {
111                first = false;
112            }
113        }
114
115        Ok(Variable::from_object(m))
116    }
117}
118
119pub struct VariableDeserializer;
120
121impl<'de> DeserializeSeed<'de> for VariableDeserializer {
122    type Value = Variable;
123
124    fn deserialize<D>(self, deserializer: D) -> Result<Self::Value, D::Error>
125    where
126        D: Deserializer<'de>,
127    {
128        deserializer.deserialize_any(VariableVisitor)
129    }
130}
131
132impl<'de> Deserialize<'de> for Variable {
133    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
134    where
135        D: Deserializer<'de>,
136    {
137        deserializer.deserialize_any(VariableVisitor)
138    }
139}
zen_types/variable/de.rs

zen_types/variable/
de.rs
