Skip to main content

cu29_value/
de.rs

1use crate::Value;
2use core::error::Error;
3use core::fmt;
4use core::marker::PhantomData;
5use cu29_clock::CuDuration;
6use serde::{de, forward_to_deserialize_any};
7
8use alloc::borrow::ToOwned;
9use alloc::boxed::Box;
10use alloc::collections::BTreeMap;
11use alloc::string::{String, ToString};
12use alloc::vec::Vec;
13
14#[derive(Debug)]
15pub enum Unexpected {
16    Bool(bool),
17    Unsigned(u64),
18    Signed(i64),
19    Float(f64),
20    Char(char),
21    Str(String),
22    Bytes(Vec<u8>),
23    Unit,
24    Option,
25    NewtypeStruct,
26    Seq,
27    Map,
28    Enum,
29    UnitVariant,
30    NewtypeVariant,
31    TupleVariant,
32    StructVariant,
33    Other(String),
34}
35
36impl From<de::Unexpected<'_>> for Unexpected {
37    fn from(unexp: de::Unexpected) -> Unexpected {
38        match unexp {
39            de::Unexpected::Bool(v) => Unexpected::Bool(v),
40            de::Unexpected::Unsigned(v) => Unexpected::Unsigned(v),
41            de::Unexpected::Signed(v) => Unexpected::Signed(v),
42            de::Unexpected::Float(v) => Unexpected::Float(v),
43            de::Unexpected::Char(v) => Unexpected::Char(v),
44            de::Unexpected::Str(v) => Unexpected::Str(v.to_owned()),
45            de::Unexpected::Bytes(v) => Unexpected::Bytes(v.to_owned()),
46            de::Unexpected::Unit => Unexpected::Unit,
47            de::Unexpected::Option => Unexpected::Option,
48            de::Unexpected::NewtypeStruct => Unexpected::NewtypeStruct,
49            de::Unexpected::Seq => Unexpected::Seq,
50            de::Unexpected::Map => Unexpected::Map,
51            de::Unexpected::Enum => Unexpected::Enum,
52            de::Unexpected::UnitVariant => Unexpected::UnitVariant,
53            de::Unexpected::NewtypeVariant => Unexpected::NewtypeVariant,
54            de::Unexpected::TupleVariant => Unexpected::TupleVariant,
55            de::Unexpected::StructVariant => Unexpected::StructVariant,
56            de::Unexpected::Other(v) => Unexpected::Other(v.to_owned()),
57        }
58    }
59}
60
61impl Unexpected {
62    pub fn to_unexpected(&self) -> de::Unexpected<'_> {
63        match *self {
64            Unexpected::Bool(v) => de::Unexpected::Bool(v),
65            Unexpected::Unsigned(v) => de::Unexpected::Unsigned(v),
66            Unexpected::Signed(v) => de::Unexpected::Signed(v),
67            Unexpected::Float(v) => de::Unexpected::Float(v),
68            Unexpected::Char(v) => de::Unexpected::Char(v),
69            Unexpected::Str(ref v) => de::Unexpected::Str(v),
70            Unexpected::Bytes(ref v) => de::Unexpected::Bytes(v),
71            Unexpected::Unit => de::Unexpected::Unit,
72            Unexpected::Option => de::Unexpected::Option,
73            Unexpected::NewtypeStruct => de::Unexpected::NewtypeStruct,
74            Unexpected::Seq => de::Unexpected::Seq,
75            Unexpected::Map => de::Unexpected::Map,
76            Unexpected::Enum => de::Unexpected::Enum,
77            Unexpected::UnitVariant => de::Unexpected::UnitVariant,
78            Unexpected::NewtypeVariant => de::Unexpected::NewtypeVariant,
79            Unexpected::TupleVariant => de::Unexpected::TupleVariant,
80            Unexpected::StructVariant => de::Unexpected::StructVariant,
81            Unexpected::Other(ref v) => de::Unexpected::Other(v),
82        }
83    }
84}
85
86#[derive(Debug)]
87pub enum DeserializerError {
88    Custom(String),
89    InvalidType(Unexpected, String),
90    InvalidValue(Unexpected, String),
91    InvalidLength(usize, String),
92    UnknownVariant(String, &'static [&'static str]),
93    UnknownField(String, &'static [&'static str]),
94    MissingField(&'static str),
95    DuplicateField(&'static str),
96}
97
98impl de::Error for DeserializerError {
99    fn custom<T: fmt::Display>(msg: T) -> Self {
100        DeserializerError::Custom(msg.to_string())
101    }
102
103    fn invalid_type(unexp: de::Unexpected, exp: &dyn de::Expected) -> Self {
104        DeserializerError::InvalidType(unexp.into(), exp.to_string())
105    }
106
107    fn invalid_value(unexp: de::Unexpected, exp: &dyn de::Expected) -> Self {
108        DeserializerError::InvalidValue(unexp.into(), exp.to_string())
109    }
110
111    fn invalid_length(len: usize, exp: &dyn de::Expected) -> Self {
112        DeserializerError::InvalidLength(len, exp.to_string())
113    }
114
115    fn unknown_variant(field: &str, expected: &'static [&'static str]) -> Self {
116        DeserializerError::UnknownVariant(field.into(), expected)
117    }
118
119    fn unknown_field(field: &str, expected: &'static [&'static str]) -> Self {
120        DeserializerError::UnknownField(field.into(), expected)
121    }
122
123    fn missing_field(field: &'static str) -> Self {
124        DeserializerError::MissingField(field)
125    }
126
127    fn duplicate_field(field: &'static str) -> Self {
128        DeserializerError::DuplicateField(field)
129    }
130}
131
132impl DeserializerError {
133    pub fn to_error<E: de::Error>(&self) -> E {
134        match *self {
135            DeserializerError::Custom(ref msg) => E::custom(msg.clone()),
136            DeserializerError::InvalidType(ref unexp, ref exp) => {
137                E::invalid_type(unexp.to_unexpected(), &&**exp)
138            }
139            DeserializerError::InvalidValue(ref unexp, ref exp) => {
140                E::invalid_value(unexp.to_unexpected(), &&**exp)
141            }
142            DeserializerError::InvalidLength(len, ref exp) => E::invalid_length(len, &&**exp),
143            DeserializerError::UnknownVariant(ref field, exp) => E::unknown_variant(field, exp),
144            DeserializerError::UnknownField(ref field, exp) => E::unknown_field(field, exp),
145            DeserializerError::MissingField(field) => E::missing_field(field),
146            DeserializerError::DuplicateField(field) => E::missing_field(field),
147        }
148    }
149
150    pub fn into_error<E: de::Error>(self) -> E {
151        self.to_error()
152    }
153}
154
155impl Error for DeserializerError {
156    fn description(&self) -> &str {
157        "Value deserializer error"
158    }
159}
160
161impl fmt::Display for DeserializerError {
162    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
163        match *self {
164            DeserializerError::Custom(ref msg) => write!(f, "{msg}"),
165            DeserializerError::InvalidType(ref unexp, ref exp) => {
166                write!(
167                    f,
168                    "Invalid type {}. Expected {}",
169                    unexp.to_unexpected(),
170                    exp
171                )
172            }
173            DeserializerError::InvalidValue(ref unexp, ref exp) => {
174                write!(
175                    f,
176                    "Invalid value {}. Expected {}",
177                    unexp.to_unexpected(),
178                    exp
179                )
180            }
181            DeserializerError::InvalidLength(len, ref exp) => {
182                write!(f, "Invalid length {len}. Expected {exp}")
183            }
184            DeserializerError::UnknownVariant(ref field, exp) => {
185                write!(
186                    f,
187                    "Unknown variant {}. Expected one of {}",
188                    field,
189                    exp.join(", ")
190                )
191            }
192            DeserializerError::UnknownField(ref field, exp) => {
193                write!(
194                    f,
195                    "Unknown field {}. Expected one of {}",
196                    field,
197                    exp.join(", ")
198                )
199            }
200            DeserializerError::MissingField(field) => write!(f, "Missing field {field}"),
201            DeserializerError::DuplicateField(field) => write!(f, "Duplicate field {field}"),
202        }
203    }
204}
205
206impl From<de::value::Error> for DeserializerError {
207    fn from(e: de::value::Error) -> DeserializerError {
208        DeserializerError::Custom(e.to_string())
209    }
210}
211
212pub struct ValueVisitor;
213
214impl<'de> de::Visitor<'de> for ValueVisitor {
215    type Value = Value;
216
217    fn expecting(&self, fmt: &mut fmt::Formatter) -> fmt::Result {
218        fmt.write_str("any value")
219    }
220
221    fn visit_bool<E>(self, value: bool) -> Result<Value, E> {
222        Ok(Value::Bool(value))
223    }
224
225    fn visit_i8<E>(self, value: i8) -> Result<Value, E> {
226        Ok(Value::I8(value))
227    }
228
229    fn visit_i16<E>(self, value: i16) -> Result<Value, E> {
230        Ok(Value::I16(value))
231    }
232
233    fn visit_i32<E>(self, value: i32) -> Result<Value, E> {
234        Ok(Value::I32(value))
235    }
236
237    fn visit_i64<E>(self, value: i64) -> Result<Value, E> {
238        Ok(Value::I64(value))
239    }
240
241    fn visit_u8<E>(self, value: u8) -> Result<Value, E> {
242        Ok(Value::U8(value))
243    }
244
245    fn visit_u16<E>(self, value: u16) -> Result<Value, E> {
246        Ok(Value::U16(value))
247    }
248
249    fn visit_u32<E>(self, value: u32) -> Result<Value, E> {
250        Ok(Value::U32(value))
251    }
252
253    fn visit_u64<E>(self, value: u64) -> Result<Value, E> {
254        Ok(Value::U64(value))
255    }
256
257    fn visit_u128<E>(self, value: u128) -> Result<Value, E> {
258        Ok(Value::U128(value))
259    }
260
261    fn visit_i128<E>(self, value: i128) -> Result<Value, E> {
262        Ok(Value::I128(value))
263    }
264
265    fn visit_f32<E>(self, value: f32) -> Result<Value, E> {
266        Ok(Value::F32(value))
267    }
268
269    fn visit_f64<E>(self, value: f64) -> Result<Value, E> {
270        Ok(Value::F64(value))
271    }
272
273    fn visit_char<E>(self, value: char) -> Result<Value, E> {
274        Ok(Value::Char(value))
275    }
276
277    fn visit_str<E>(self, value: &str) -> Result<Value, E> {
278        Ok(Value::String(value.into()))
279    }
280
281    fn visit_string<E>(self, value: String) -> Result<Value, E> {
282        Ok(Value::String(value))
283    }
284
285    fn visit_unit<E>(self) -> Result<Value, E> {
286        Ok(Value::Unit)
287    }
288
289    fn visit_none<E>(self) -> Result<Value, E> {
290        Ok(Value::Option(None))
291    }
292
293    fn visit_some<D: de::Deserializer<'de>>(self, d: D) -> Result<Value, D::Error> {
294        d.deserialize_any(ValueVisitor)
295            .map(|v| Value::Option(Some(Box::new(v))))
296    }
297
298    fn visit_newtype_struct<D: de::Deserializer<'de>>(self, d: D) -> Result<Value, D::Error> {
299        d.deserialize_any(ValueVisitor)
300            .map(|v| Value::Newtype(Box::new(v)))
301    }
302
303    fn visit_seq<V: de::SeqAccess<'de>>(self, mut visitor: V) -> Result<Value, V::Error> {
304        let mut values = Vec::new();
305        while let Some(elem) = visitor.next_element()? {
306            values.push(elem);
307        }
308        Ok(Value::Seq(values))
309    }
310
311    fn visit_map<V: de::MapAccess<'de>>(self, mut visitor: V) -> Result<Value, V::Error> {
312        let mut values = BTreeMap::new();
313        while let Some((key, value)) = visitor.next_entry()? {
314            values.insert(key, value);
315        }
316        Ok(Value::Map(values))
317    }
318
319    fn visit_bytes<E>(self, v: &[u8]) -> Result<Value, E> {
320        Ok(Value::Bytes(v.into()))
321    }
322
323    fn visit_byte_buf<E>(self, v: Vec<u8>) -> Result<Value, E> {
324        Ok(Value::Bytes(v))
325    }
326}
327
328impl<'de> de::Deserialize<'de> for Value {
329    fn deserialize<D: de::Deserializer<'de>>(d: D) -> Result<Self, D::Error> {
330        d.deserialize_any(ValueVisitor)
331    }
332}
333
334impl de::IntoDeserializer<'_, DeserializerError> for Value {
335    type Deserializer = Value;
336
337    fn into_deserializer(self) -> Value {
338        self
339    }
340}
341
342pub struct ValueDeserializer<E> {
343    value: Value,
344    error: PhantomData<fn() -> E>,
345}
346
347impl<E> ValueDeserializer<E> {
348    pub fn new(value: Value) -> Self {
349        ValueDeserializer {
350            value,
351            error: Default::default(),
352        }
353    }
354
355    pub fn into_value(self) -> Value {
356        self.value
357    }
358}
359
360impl<'de, E> de::Deserializer<'de> for ValueDeserializer<E>
361where
362    E: de::Error,
363{
364    type Error = E;
365
366    fn deserialize_any<V: de::Visitor<'de>>(self, visitor: V) -> Result<V::Value, Self::Error> {
367        match self.value {
368            Value::Bool(v) => visitor.visit_bool(v),
369            Value::U8(v) => visitor.visit_u8(v),
370            Value::U16(v) => visitor.visit_u16(v),
371            Value::U32(v) => visitor.visit_u32(v),
372            Value::U64(v) => visitor.visit_u64(v),
373            Value::U128(v) => visitor.visit_u128(v),
374            Value::I8(v) => visitor.visit_i8(v),
375            Value::I16(v) => visitor.visit_i16(v),
376            Value::I32(v) => visitor.visit_i32(v),
377            Value::I64(v) => visitor.visit_i64(v),
378            Value::I128(v) => visitor.visit_i128(v),
379            Value::F32(v) => visitor.visit_f32(v),
380            Value::F64(v) => visitor.visit_f64(v),
381            Value::Char(v) => visitor.visit_char(v),
382            Value::String(v) => visitor.visit_string(v),
383            Value::Unit => visitor.visit_unit(),
384            Value::Option(None) => visitor.visit_none(),
385            Value::Option(Some(v)) => visitor.visit_some(ValueDeserializer::new(*v)),
386            Value::Newtype(v) => visitor.visit_newtype_struct(ValueDeserializer::new(*v)),
387            Value::Seq(v) => visitor.visit_seq(de::value::SeqDeserializer::new(
388                v.into_iter().map(ValueDeserializer::new),
389            )),
390            Value::Map(v) => visitor
391                .visit_map(de::value::MapDeserializer::new(v.into_iter().map(
392                    |(k, v)| (ValueDeserializer::new(k), ValueDeserializer::new(v)),
393                ))),
394            Value::Bytes(v) => visitor.visit_byte_buf(v),
395            Value::CuTime(v) => {
396                let CuDuration(nanos) = v;
397                visitor.visit_u64(nanos)
398            }
399        }
400    }
401
402    fn deserialize_option<V: de::Visitor<'de>>(self, visitor: V) -> Result<V::Value, Self::Error> {
403        match self.value {
404            Value::Option(..) => self.deserialize_any(visitor),
405            Value::Unit => visitor.visit_unit(),
406            _ => visitor.visit_some(self),
407        }
408    }
409
410    fn deserialize_enum<V: de::Visitor<'de>>(
411        self,
412        _name: &'static str,
413        _variants: &'static [&'static str],
414        visitor: V,
415    ) -> Result<V::Value, Self::Error> {
416        let (variant, value) = match self.value {
417            Value::Map(value) => {
418                let mut iter = value.into_iter();
419                let (variant, value) = match iter.next() {
420                    Some(v) => v,
421                    None => {
422                        return Err(de::Error::invalid_value(
423                            de::Unexpected::Map,
424                            &"map with a single key",
425                        ));
426                    }
427                };
428                // enums are encoded as maps with a single key:value pair
429                if iter.next().is_some() {
430                    return Err(de::Error::invalid_value(
431                        de::Unexpected::Map,
432                        &"map with a single key",
433                    ));
434                }
435                (variant, Some(value))
436            }
437            Value::String(variant) => (Value::String(variant), None),
438            other => {
439                return Err(de::Error::invalid_type(
440                    other.unexpected(),
441                    &"string or map",
442                ));
443            }
444        };
445
446        let d = EnumDeserializer {
447            variant,
448            value,
449            error: Default::default(),
450        };
451        visitor.visit_enum(d)
452    }
453
454    fn deserialize_newtype_struct<V: de::Visitor<'de>>(
455        self,
456        _name: &'static str,
457        visitor: V,
458    ) -> Result<V::Value, Self::Error> {
459        match self.value {
460            Value::Newtype(v) => visitor.visit_newtype_struct(ValueDeserializer::new(*v)),
461            _ => visitor.visit_newtype_struct(self),
462        }
463    }
464
465    forward_to_deserialize_any! {
466        bool u8 u16 u32 u64 u128 i8 i16 i32 i64 i128 f32 f64 char str string unit
467        seq bytes byte_buf map unit_struct
468        tuple_struct struct tuple ignored_any identifier
469    }
470}
471
472impl<E> de::IntoDeserializer<'_, E> for ValueDeserializer<E>
473where
474    E: de::Error,
475{
476    type Deserializer = Self;
477
478    fn into_deserializer(self) -> Self::Deserializer {
479        self
480    }
481}
482
483impl<'de> de::Deserializer<'de> for Value {
484    type Error = DeserializerError;
485
486    fn deserialize_any<V: de::Visitor<'de>>(self, visitor: V) -> Result<V::Value, Self::Error> {
487        ValueDeserializer::new(self).deserialize_any(visitor)
488    }
489
490    fn deserialize_option<V: de::Visitor<'de>>(self, visitor: V) -> Result<V::Value, Self::Error> {
491        ValueDeserializer::new(self).deserialize_option(visitor)
492    }
493
494    fn deserialize_enum<V: de::Visitor<'de>>(
495        self,
496        name: &'static str,
497        variants: &'static [&'static str],
498        visitor: V,
499    ) -> Result<V::Value, Self::Error> {
500        ValueDeserializer::new(self).deserialize_enum(name, variants, visitor)
501    }
502
503    fn deserialize_newtype_struct<V: de::Visitor<'de>>(
504        self,
505        name: &'static str,
506        visitor: V,
507    ) -> Result<V::Value, Self::Error> {
508        ValueDeserializer::new(self).deserialize_newtype_struct(name, visitor)
509    }
510
511    forward_to_deserialize_any! {
512        bool u8 u16 u32 u64 u128 i8 i16 i32 i64 i128 f32 f64 char str string unit
513        seq bytes byte_buf map unit_struct
514        tuple_struct struct tuple ignored_any identifier
515    }
516}
517
518struct EnumDeserializer<E> {
519    variant: Value,
520    value: Option<Value>,
521    error: PhantomData<fn() -> E>,
522}
523
524impl<'de, E> de::EnumAccess<'de> for EnumDeserializer<E>
525where
526    E: de::Error,
527{
528    type Error = E;
529    type Variant = VariantDeserializer<Self::Error>;
530
531    fn variant_seed<V>(
532        self,
533        seed: V,
534    ) -> Result<(V::Value, VariantDeserializer<Self::Error>), Self::Error>
535    where
536        V: de::DeserializeSeed<'de>,
537    {
538        let visitor = VariantDeserializer {
539            value: self.value,
540            error: Default::default(),
541        };
542        seed.deserialize(ValueDeserializer::new(self.variant))
543            .map(|v| (v, visitor))
544    }
545}
546
547struct VariantDeserializer<E> {
548    value: Option<Value>,
549    error: PhantomData<fn() -> E>,
550}
551
552impl<'de, E> de::VariantAccess<'de> for VariantDeserializer<E>
553where
554    E: de::Error,
555{
556    type Error = E;
557
558    fn unit_variant(self) -> Result<(), Self::Error> {
559        match self.value {
560            Some(value) => de::Deserialize::deserialize(ValueDeserializer::new(value)),
561            None => Ok(()),
562        }
563    }
564
565    fn newtype_variant_seed<T>(self, seed: T) -> Result<T::Value, Self::Error>
566    where
567        T: de::DeserializeSeed<'de>,
568    {
569        match self.value {
570            Some(value) => seed.deserialize(ValueDeserializer::new(value)),
571            None => Err(de::Error::invalid_type(
572                de::Unexpected::UnitVariant,
573                &"newtype variant",
574            )),
575        }
576    }
577
578    fn tuple_variant<V>(self, _len: usize, visitor: V) -> Result<V::Value, Self::Error>
579    where
580        V: de::Visitor<'de>,
581    {
582        match self.value {
583            Some(Value::Seq(v)) => de::Deserializer::deserialize_any(
584                de::value::SeqDeserializer::new(v.into_iter().map(ValueDeserializer::new)),
585                visitor,
586            ),
587            Some(other) => Err(de::Error::invalid_type(
588                other.unexpected(),
589                &"tuple variant",
590            )),
591            None => Err(de::Error::invalid_type(
592                de::Unexpected::UnitVariant,
593                &"tuple variant",
594            )),
595        }
596    }
597
598    fn struct_variant<V>(
599        self,
600        _fields: &'static [&'static str],
601        visitor: V,
602    ) -> Result<V::Value, Self::Error>
603    where
604        V: de::Visitor<'de>,
605    {
606        match self.value {
607            Some(Value::Map(v)) => de::Deserializer::deserialize_any(
608                de::value::MapDeserializer::new(
609                    v.into_iter()
610                        .map(|(k, v)| (ValueDeserializer::new(k), ValueDeserializer::new(v))),
611                ),
612                visitor,
613            ),
614            Some(other) => Err(de::Error::invalid_type(
615                other.unexpected(),
616                &"struct variant",
617            )),
618            None => Err(de::Error::invalid_type(
619                de::Unexpected::UnitVariant,
620                &"struct variant",
621            )),
622        }
623    }
624}