Skip to main content

zen_types/variable/
ser.rs

1use crate::variable::Variable;
2#[cfg(not(feature = "arbitrary_precision"))]
3use rust_decimal::prelude::ToPrimitive;
4use serde::{Serialize, Serializer};
5
6#[cfg(feature = "arbitrary_precision")]
7use crate::constant::NUMBER_TOKEN;
8#[cfg(feature = "arbitrary_precision")]
9use serde::ser::SerializeStruct;
10
11impl Serialize for Variable {
12    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
13    where
14        S: Serializer,
15    {
16        match self {
17            Variable::Null => serializer.serialize_unit(),
18            Variable::Bool(v) => serializer.serialize_bool(*v),
19            #[cfg(feature = "arbitrary_precision")]
20            Variable::Number(v) => {
21                let str = v.normalize().to_string();
22
23                let mut s = serializer.serialize_struct(NUMBER_TOKEN, 1)?;
24                s.serialize_field(NUMBER_TOKEN, &str)?;
25                s.end()
26            }
27            #[cfg(not(feature = "arbitrary_precision"))]
28            Variable::Number(v) => {
29                if let Some(u) = v.to_u64() {
30                    return serializer.serialize_u64(u);
31                }
32
33                if let Some(i) = v.to_i64() {
34                    return serializer.serialize_i64(i);
35                }
36
37                if let Some(f) = v.to_f64() {
38                    return serializer.serialize_f64(f);
39                }
40
41                Err(serde::ser::Error::custom("cannot convert to f64"))
42            }
43            Variable::String(v) => serializer.serialize_str(v),
44            Variable::Array(v) => {
45                let borrowed = v.borrow();
46                serializer.collect_seq(borrowed.iter())
47            }
48            Variable::Object(v) => {
49                let borrowed = v.borrow();
50                serializer.collect_map(borrowed.iter())
51            }
52            Variable::Dynamic(d) => serializer.serialize_str(d.to_string().as_str()),
53        }
54    }
55}