teo_parser/ast/
interface.rs

1use std::cell::RefCell;
2use std::collections::BTreeMap;
3use maplit::btreemap;
4use serde::Serialize;
5use crate::ast::doc_comment::DocComment;
6use crate::ast::field::Field;
7use crate::ast::generics::{GenericsConstraint, GenericsDeclaration};
8use crate::ast::identifier::Identifier;
9use crate::ast::type_expr::TypeExpr;
10use crate::{declare_container_node, impl_container_node_defaults, node_child_fn, node_children_iter, node_children_iter_fn, node_optional_child_fn};
11use crate::ast::partial_field::PartialField;
12use crate::format::Writer;
13use crate::r#type::synthesized_shape::SynthesizedShape;
14use crate::r#type::Type;
15use crate::traits::info_provider::InfoProvider;
16use crate::traits::resolved::Resolve;
17use crate::traits::write::Write;
18
19use super::decorator::Decorator;
20use super::span::Span;
21
22declare_container_node!(InterfaceDeclaration, named, availability,
23    pub(crate) comment: Option<usize>,
24    pub(crate) identifier: usize,
25    pub(crate) generics_declaration: Option<usize>,
26    pub(crate) generics_constraint: Option<usize>,
27    pub(crate) extends: Vec<usize>,
28    pub(crate) fields: Vec<usize>,
29    pub(crate) partial_fields: Vec<usize>,
30    pub(crate) decorators: Vec<usize>,
31    pub(crate) empty_decorator_spans: Vec<Span>,
32    pub(crate) empty_field_decorator_spans: Vec<Span>,
33    pub(crate) unattached_field_decorators: Vec<Decorator>,
34    pub(crate) resolved: RefCell<Option<InterfaceDeclarationResolved>>,
35);
36
37impl_container_node_defaults!(InterfaceDeclaration, named, availability);
38
39node_children_iter!(InterfaceDeclaration, TypeExpr, ExtendsIter, extends);
40
41node_children_iter!(InterfaceDeclaration, Field, FieldsIter, fields);
42
43node_children_iter!(InterfaceDeclaration, PartialField, PartialFieldsIter, partial_fields);
44
45node_children_iter!(InterfaceDeclaration, Decorator, DecoratorsIter, decorators);
46
47impl InterfaceDeclaration {
48
49    node_optional_child_fn!(comment, DocComment);
50
51    node_child_fn!(identifier, Identifier);
52
53    node_optional_child_fn!(generics_declaration, GenericsDeclaration);
54
55    node_optional_child_fn!(generics_constraint, GenericsConstraint);
56
57    node_children_iter_fn!(extends, ExtendsIter);
58
59    node_children_iter_fn!(fields, FieldsIter);
60
61    node_children_iter_fn!(partial_fields, PartialFieldsIter);
62
63    node_children_iter_fn!(decorators, DecoratorsIter);
64
65    pub fn shape_from_generics(&self, generics: &Vec<Type>) -> SynthesizedShape {
66        let map = self.calculate_generics_map(generics);
67        self.resolved().shape().replace_generics(&map)
68    }
69
70    pub fn calculate_generics_map(&self, types: &Vec<Type>) -> BTreeMap<String, Type> {
71        if let Some(generics_declaration) = self.generics_declaration() {
72            generics_declaration.calculate_generics_map(types)
73        } else {
74            btreemap!{}
75        }
76    }
77}
78
79impl InfoProvider for InterfaceDeclaration {
80    fn namespace_skip(&self) -> usize {
81        1
82    }
83}
84
85impl Resolve<InterfaceDeclarationResolved> for InterfaceDeclaration {
86    fn resolved_ref_cell(&self) -> &RefCell<Option<InterfaceDeclarationResolved>> {
87        &self.resolved
88    }
89}
90
91#[derive(Debug, Clone, Serialize)]
92pub struct InterfaceDeclarationResolved {
93    pub base_shape: SynthesizedShape,
94    #[serde(skip)]
95    pub shape: Option<SynthesizedShape>,
96}
97
98impl InterfaceDeclarationResolved {
99
100    pub fn base_shape(&self) -> &SynthesizedShape {
101        &self.base_shape
102    }
103
104    pub fn shape(&self) -> &SynthesizedShape {
105        self.shape.as_ref().unwrap()
106    }
107}
108
109impl InterfaceDeclarationResolved {
110
111    pub fn new(base_shape: SynthesizedShape) -> Self {
112        Self {
113            base_shape,
114            shape: None,
115        }
116    }
117}
118
119impl Write for InterfaceDeclaration {
120    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
121        writer.write_children(self, self.children.values())
122    }
123
124    fn is_block_level_element(&self) -> bool {
125        true
126    }
127}