teo_parser/ast/
unit.rs

1use crate::ast::expression::Expression;
2use crate::{declare_container_node, impl_container_node_defaults, node_children_iter, node_children_iter_fn, node_optional_child_fn};
3use crate::ast::empty_dot::EmptyDot;
4use crate::format::Writer;
5use crate::traits::write::Write;
6
7declare_container_node!(Unit,
8    pub(crate) expressions: Vec<usize>,
9    pub(crate) empty_dot: Option<usize>,
10);
11
12impl_container_node_defaults!(Unit);
13
14node_children_iter!(Unit, Expression, ExpressionsIter, expressions);
15
16impl Unit {
17
18    node_optional_child_fn!(empty_dot, EmptyDot);
19
20    node_children_iter_fn!(expressions, ExpressionsIter);
21
22    pub fn expression_at(&self, idx: usize) -> Option<&Expression> {
23        self.expressions.get(idx).map(|idx| self.children.get(idx)).flatten().map(|n| n.as_expression()).flatten()
24    }
25
26    pub fn last_expression(&self) -> Option<&Expression> {
27        self.expressions.last().map(|idx| self.children.get(idx)).flatten().map(|n| n.as_expression()).flatten()
28    }
29}
30
31impl Write for Unit {
32    fn write<'a>(&'a self, writer: &mut Writer<'a>) {
33        writer.write_children(self, self.children.values())
34    }
35}
36
37impl Unit {
38
39    pub fn unwrap_enumerable_enum_member_strings(&self) -> Option<Vec<&str>> {
40        if self.expressions.len() != 1 {
41            None
42        } else {
43            self.expressions().next().unwrap().unwrap_enumerable_enum_member_strings()
44        }
45    }
46
47    pub fn unwrap_enumerable_enum_member_string(&self) -> Option<&str> {
48        if self.expressions.len() != 1 {
49            None
50        } else {
51            self.expressions().next().unwrap().unwrap_enumerable_enum_member_string()
52        }
53    }
54}
teo_parser/ast/unit.rs

teo_parser/ast/
unit.rs
