valkyrie-ast 0.0.11

Strong typed abstract syntax tree of valkyrie language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use super::*;

impl ControlType {
    pub fn as_str(&self) -> &'static str {
        match self {
            ControlType::Break => "break",
            ControlType::Continue => "continue",
            ControlType::Fallthrough => "fallthrough",
            ControlType::Return => "return",
            ControlType::Raise => "raise",
            ControlType::YieldReturn => "yield",
            ControlType::YieldBreak => "yield break",
            ControlType::YieldFrom => "yield from",
        }
    }
}

impl Display for ControlType {
    fn fmt(&self, f: &mut Formatter<'_>) -> core::fmt::Result {
        f.write_str(self.as_str())
    }
}

impl PrettyPrint for ControlNode {
    fn build<'a>(&self, allocator: &'a PrettyProvider<'a>) -> PrettyTree<'a> {
        let mut terms = Vec::with_capacity(3);
        terms.push(self.r#type.build(allocator));
        if let Some(s) = &self.expression {
            terms.push(allocator.space());
            terms.push(s.build(allocator));
        }
        allocator.concat(terms)
    }
}

impl PrettyPrint for ControlType {
    fn build<'a>(&self, allocator: &'a PrettyProvider<'a>) -> PrettyTree<'a> {
        allocator.keyword(self.as_str())
    }
}