teo_runtime/arguments/
mod.rs

1use std::collections::BTreeMap;
2use std::fmt::{Debug, Display, Formatter};
3use std::sync::Arc;
4use maplit::btreemap;
5use serde::{Serialize, Serializer};
6use teo_result::Error;
7use teo_result::Result;
8use crate::value::Value;
9
10#[repr(transparent)]
11#[derive(Clone, PartialEq)]
12pub struct Arguments {
13    inner: Arc<Inner>
14}
15
16impl Serialize for Arguments {
17
18    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error> where S: Serializer {
19        serializer.collect_map(self.inner.map.iter())
20    }
21}
22
23impl Debug for Arguments {
24
25    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
26        let mut debug_struct = f.debug_struct("Arguments");
27        for (k, v) in &self.inner.map {
28            debug_struct.field(k.as_str(), &v);
29        }
30        debug_struct.finish()
31    }
32}
33
34impl Display for Arguments {
35    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
36        f.write_str("(")?;
37        for (i, (k, v)) in self.inner.map.iter().enumerate() {
38            f.write_str(k)?;
39            f.write_str(": ")?;
40            Display::fmt(v, f)?;
41            if i != self.inner.map.len() - 1 {
42                f.write_str(", ")?;
43            }
44        }
45        f.write_str(")")?;
46        Ok(())
47    }
48}
49
50impl Arguments {
51
52    pub(crate) fn new(map: BTreeMap<String, Value>) -> Self {
53        Self {
54            inner: Arc::new(Inner { map })
55        }
56    }
57
58    pub fn is_empty(&self) -> bool {
59        self.inner.map.is_empty()
60    }
61
62    pub fn has(&self, key: impl AsRef<str>) -> bool {
63        self.inner.map.contains_key(key.as_ref())
64    }
65
66    pub fn get<'a, T: 'a, E>(&'a self, key: impl AsRef<str>) -> Result<T> where E: core::error::Error, Error: From<E>, T: TryFrom<&'a Value, Error = E> {
67        let object = self.get_value_ref(key)?;
68        match object.try_into() {
69            Ok(v) => Ok(v),
70            Err(e) => Err(Error::new(format!("{e}")))
71        }
72    }
73
74    pub fn get_optional<'a, T: 'a, E>(&'a self, key: impl AsRef<str>) -> Result<Option<T>> where E: core::error::Error, Error: From<E>, T: TryFrom<&'a Value, Error = E> {
75        if let Ok(object) = self.get_value_ref(key) {
76            if object.is_null() {
77                Ok(None)
78            } else {
79                match object.try_into() {
80                    Ok(v) => Ok(Some(v)),
81                    Err(e) => Err(Error::new(format!("{e}")))
82                }
83            }
84        } else {
85            Ok(None)
86        }
87    }
88
89    pub fn get_value_ref(&self, key: impl AsRef<str>) -> Result<&Value> {
90        if let Some(object) = self.inner.map.get(key.as_ref()) {
91            Ok(object)
92        } else {
93            Err(Error::new(format!("argument '{}' is not present", key.as_ref())))
94        }
95    }
96
97    pub fn get_value(&self, key: impl AsRef<str>) -> Result<Value> {
98        self.get_value_ref(key).map(|o| o.clone())
99    }
100
101    pub fn iter(&self) -> std::collections::btree_map::Iter<String, Value> {
102        self.inner.map.iter()
103    }
104}
105
106impl Default for Arguments {
107
108    fn default() -> Self {
109        Arguments {
110            inner: Arc::new(Inner {
111                map: btreemap!{}
112            })
113        }
114    }
115}
116
117#[repr(transparent)]
118#[derive(PartialEq)]
119struct Inner {
120    map: BTreeMap<String, Value>
121}
teo_runtime/arguments/mod.rs

teo_runtime/arguments/
mod.rs
