Skip to main content

zen_types/rcvalue/
de.rs

1#[cfg(feature = "arbitrary_precision")]
2use crate::constant::NUMBER_TOKEN;
3use crate::rcvalue::RcValue;
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 RcValueVisitor;
16
17impl<'de> Visitor<'de> for RcValueVisitor {
18    type Value = RcValue;
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(RcValue::Bool(v))
29    }
30
31    fn visit_i64<E>(self, v: i64) -> Result<Self::Value, E>
32    where
33        E: Error,
34    {
35        Ok(RcValue::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(RcValue::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(RcValue::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(RcValue::String(Rc::from(v)))
63    }
64
65    fn visit_unit<E>(self) -> Result<Self::Value, E>
66    where
67        E: Error,
68    {
69        Ok(RcValue::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(RcValueDeserializer)? {
78            vec.push(value);
79        }
80
81        Ok(RcValue::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>>, RcValueDeserializer)?
94        {
95            #[cfg(feature = "arbitrary_precision")]
96            if first && key.deref() == NUMBER_TOKEN {
97                let str = match &value {
98                    RcValue::String(s) => s.as_ref(),
99                    _ => return Err(Error::custom("failed to deserialize number")),
100                };
101
102                return Ok(RcValue::Number(
103                    Decimal::from_str_exact(str)
104                        .or_else(|_| Decimal::from_scientific(str))
105                        .map_err(|_| Error::custom("invalid number"))?,
106                ));
107            }
108
109            m.insert(key, value);
110            #[cfg(feature = "arbitrary_precision")]
111            {
112                first = false;
113            }
114        }
115
116        Ok(RcValue::Object(m))
117    }
118}
119
120pub struct RcValueDeserializer;
121
122impl<'de> DeserializeSeed<'de> for RcValueDeserializer {
123    type Value = RcValue;
124
125    fn deserialize<D>(self, deserializer: D) -> Result<Self::Value, D::Error>
126    where
127        D: Deserializer<'de>,
128    {
129        deserializer.deserialize_any(RcValueVisitor)
130    }
131}
132
133impl<'de> Deserialize<'de> for RcValue {
134    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
135    where
136        D: Deserializer<'de>,
137    {
138        deserializer.deserialize_any(RcValueVisitor)
139    }
140}