teo_parser/ast/
pipeline_item_declaration.rs

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

teo_parser/ast/
pipeline_item_declaration.rs
