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}