teo_parser/ast/
decorator_declaration.rs

1use crate::ast::argument_list_declaration::ArgumentListDeclaration;
2
3use crate::ast::callable_variant::CallableVariant;
4use crate::ast::doc_comment::DocComment;
5use crate::ast::generics::{GenericsConstraint, GenericsDeclaration};
6use crate::ast::identifier::Identifier;
7use crate::ast::reference_space::ReferenceSpace;
8use crate::{declare_container_node, impl_container_node_defaults, node_child_fn, node_children_iter, node_children_iter_fn, node_optional_child_fn};
9use crate::format::Writer;
10use crate::traits::info_provider::InfoProvider;
11use crate::traits::write::Write;
12
13declare_container_node!(DecoratorDeclaration, named, availability,
14    pub exclusive: bool,
15    pub unique: bool,
16    pub decorator_class: ReferenceSpace,
17    pub(crate) comment: Option<usize>,
18    pub(crate) identifier: usize,
19    pub(crate) generics_declaration: Option<usize>,
20    pub(crate) argument_list_declaration: Option<usize>,
21    pub(crate) generics_constraint: Option<usize>,
22    pub(crate) variants: Vec<usize>,
23);
24
25impl_container_node_defaults!(DecoratorDeclaration, named, availability);
26
27node_children_iter!(DecoratorDeclaration, DecoratorDeclarationVariant, VariantsIter, variants);
28
29impl DecoratorDeclaration {
30
31    node_optional_child_fn!(comment, DocComment);
32
33    node_child_fn!(identifier, Identifier);
34
35    node_optional_child_fn!(generics_declaration, GenericsDeclaration);
36
37    node_optional_child_fn!(argument_list_declaration, ArgumentListDeclaration);
38
39    node_optional_child_fn!(generics_constraint, GenericsConstraint);
40
41    node_children_iter_fn!(variants, VariantsIter);
42
43    pub fn has_variants(&self) -> bool {
44        !self.variants.is_empty()
45    }
46
47    pub fn callable_variants(&self) -> Vec<CallableVariant> {
48        if self.has_variants() {
49            self.variants().map(|v| CallableVariant {
50                generics_declarations: if let Some(generics_declaration) = v.generics_declaration() {
51                    vec![generics_declaration]
52                } else {
53                    vec![]
54                },
55                argument_list_declaration: v.argument_list_declaration(),
56                generics_constraints: if let Some(generics_constraint) = v.generics_constraint() {
57                    vec![generics_constraint]
58                } else {
59                    vec![]
60                },
61                pipeline_input: None,
62                pipeline_output: None,
63            }).collect()
64        } else {
65            vec![CallableVariant {
66                generics_declarations: if let Some(generics_declaration) = self.generics_declaration() {
67                    vec![generics_declaration]
68                } else {
69                    vec![]
70                },
71                argument_list_declaration: self.argument_list_declaration(),
72                generics_constraints: if let Some(generics_constraint) = self.generics_constraint() {
73                    vec![generics_constraint]
74                } else {
75                    vec![]
76                },
77                pipeline_input: None,
78                pipeline_output: None,
79            }]
80        }
81    }
82}
83
84impl InfoProvider for DecoratorDeclaration {
85    fn namespace_skip(&self) -> usize {
86        1
87    }
88}
89
90declare_container_node!(DecoratorDeclarationVariant,
91    pub(crate) comment: Option<usize>,
92    pub(crate) generics_declaration: Option<usize>,
93    pub(crate) argument_list_declaration: Option<usize>,
94    pub(crate) generics_constraint: Option<usize>,
95);
96
97impl_container_node_defaults!(DecoratorDeclarationVariant);
98
99impl DecoratorDeclarationVariant {
100
101    node_optional_child_fn!(comment, DocComment);
102
103    node_optional_child_fn!(generics_declaration, GenericsDeclaration);
104
105    node_optional_child_fn!(argument_list_declaration, ArgumentListDeclaration);
106
107    node_optional_child_fn!(generics_constraint, GenericsConstraint);
108
109}
110
111impl Write for DecoratorDeclaration {
112    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
113        writer.write_children(self, self.children.values());
114    }
115
116    fn is_block_level_element(&self) -> bool {
117        true
118    }
119}
120
121impl Write for DecoratorDeclarationVariant {
122    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
123        writer.write_children(self, self.children.values());
124    }
125
126    fn is_block_level_element(&self) -> bool {
127        true
128    }
129}
teo_parser/ast/decorator_declaration.rs

teo_parser/ast/
decorator_declaration.rs
