tau_engine/
json.rs

1// Trait implementations that allow the solver to be gneric over serde_json's `Value`.
2
3use std::borrow::Cow;
4
5use serde_json::map::Map;
6pub use serde_json::{Number, Value as Json};
7
8use crate::document::Document;
9use crate::value::{AsValue, Object, Value};
10
11impl AsValue for Json {
12    #[inline]
13    fn as_value(&self) -> Value<'_> {
14        match self {
15            Self::Null => Value::Null,
16            Self::String(s) => Value::String(Cow::Borrowed(s)),
17            Self::Number(n) => {
18                if n.is_u64() {
19                    Value::UInt(n.as_u64().unwrap())
20                } else if n.is_i64() {
21                    Value::Int(n.as_i64().unwrap())
22                } else if n.is_f64() {
23                    Value::Float(n.as_f64().unwrap())
24                } else {
25                    unreachable!()
26                }
27            }
28            Self::Bool(b) => Value::Bool(*b),
29            Self::Object(o) => Value::Object(o),
30            Self::Array(a) => Value::Array(a),
31        }
32    }
33}
34
35impl Document for Json {
36    fn find(&self, key: &str) -> Option<Value> {
37        if let Json::Object(o) = self {
38            return Object::find(o, key);
39        }
40        None
41    }
42}
43
44impl Object for Map<String, Json> {
45    #[inline]
46    fn get(&self, key: &str) -> Option<Value<'_>> {
47        self.get(key).map(|v| v.as_value())
48    }
49
50    #[inline]
51    fn keys(&self) -> Vec<Cow<'_, str>> {
52        self.keys().map(|k| Cow::Borrowed(k.as_str())).collect()
53    }
54
55    #[inline]
56    fn len(&self) -> usize {
57        self.len()
58    }
59}