teo_runtime/interface/
interface.rs

1use std::collections::BTreeMap;
2use std::sync::Arc;
3use indexmap::IndexMap;
4use maplit::btreemap;
5use serde::Serialize;
6use teo_parser::r#type::reference::Reference;
7use teo_parser::r#type::synthesized_shape::SynthesizedShape;
8use teo_parser::r#type::Type;
9use crate::comment::Comment;
10use crate::interface::field::Field;
11use crate::traits::documentable::Documentable;
12use crate::traits::named::Named;
13use crate::Value;
14
15#[derive(Debug, Clone)]
16pub struct Interface {
17    pub(super) inner: Arc<Inner>
18}
19
20#[derive(Debug, Serialize)]
21pub(super) struct Inner {
22    pub(super) path: Vec<String>,
23    pub(super) parser_path: Vec<usize>,
24    pub(super) comment: Option<Comment>,
25    pub(super) fields: IndexMap<String, Field>,
26    pub(super) generic_names: Vec<String>,
27    pub(super) extends: Vec<Type>,
28    pub(super) shape: SynthesizedShape,
29    pub(super) generate_client: bool,
30    pub(super) generate_entity: bool,
31    pub(super) data: BTreeMap<String, Value>,
32}
33
34impl Interface {
35
36    pub fn path(&self) -> &Vec<String> {
37        &self.inner.path
38    }
39
40    pub fn parser_path(&self) -> &Vec<usize> {
41        &self.inner.parser_path
42    }
43
44    pub fn comment(&self) -> Option<&Comment> {
45        self.inner.comment.as_ref()
46    }
47
48    pub fn fields(&self) -> &IndexMap<String, Field> {
49        &self.inner.fields
50    }
51
52    pub fn generic_names(&self) -> &Vec<String> {
53        &self.inner.generic_names
54    }
55
56    pub fn extends(&self) -> &Vec<Type> {
57        &self.inner.extends
58    }
59
60    pub fn shape(&self) -> &SynthesizedShape {
61        &self.inner.shape
62    }
63
64    pub fn generate_client(&self) -> bool {
65        self.inner.generate_client
66    }
67
68    pub fn generate_entity(&self) -> bool {
69        self.inner.generate_entity
70    }
71
72    pub fn data(&self) -> &BTreeMap<String, Value> {
73        &self.inner.data
74    }
75
76    pub fn shape_from_generics(&self, generics: &Vec<Type>) -> SynthesizedShape {
77        let map = self.calculate_generics_map(generics);
78        self.inner.shape.replace_generics(&map)
79    }
80
81    pub fn calculate_generics_map(&self, types: &Vec<Type>) -> BTreeMap<String, Type> {
82        if self.generic_names().len() == types.len() {
83            return self.generic_names().iter().enumerate().map(|(index, name)| (name.to_string(), types.get(index).unwrap().clone())).collect();
84        }
85        btreemap!{}
86    }
87
88    pub fn as_type_reference(&self) -> Reference {
89        Reference::new(self.inner.parser_path.clone(), self.inner.path.clone())
90    }
91}
92
93impl Named for Interface {
94
95    fn name(&self) -> &str {
96        self.inner.path.last().map(|s| s.as_str()).unwrap()
97    }
98}
99
100impl Documentable for Interface {
101
102    fn comment(&self) -> Option<&Comment> {
103        self.inner.comment.as_ref()
104    }
105
106    fn kind(&self) -> &'static str {
107        "interface"
108    }
109}
110
111impl Serialize for Interface {
112    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where S: serde::Serializer {
113        self.inner.serialize(serializer)
114    }
115}