lib-ruby-parser 3.0.5

Ruby parser
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
41
42
43
44
45
46
47
48
49
50
51
52
53

use lib_ruby_parser_nodes::Node;

pub(crate) struct NodeEnum<'a> {
    nodes: &'a [Node],
}

impl<'a> NodeEnum<'a> {
    pub(crate) fn new(nodes: &'a [Node]) -> Self {
        Self { nodes }
    }

    pub(crate) fn write(&self) {
        let contents = format!(
            "use crate::nodes::InnerNode;
use crate::nodes::*;

/// Generic combination of all known nodes.
#[allow(missing_docs)]
#[derive(Debug, Clone, PartialEq)]
#[repr(C)]
pub enum Node {{
    {variants}
}}

impl Node {{
    pub(crate) fn inner_ref(&self) -> &dyn InnerNode {{
        match &self {{
            {match_branches}
        }}
    }}
}}
",
            variants = self.variants().join(",\n    "),
            match_branches = self.match_branches().join("\n            ")
        );

        std::fs::write("src/nodes/node_enum_gen.rs", contents).unwrap();
    }

    fn variants(&self) -> Vec<String> {
        self.nodes
            .iter()
            .map(|node| format!("{name}({name})", name = node.struct_name))
            .collect()
    }

    fn match_branches(&self) -> Vec<String> {
        self.nodes
            .iter()
            .map(|node| format!("Node::{name}(inner) => inner,", name = node.struct_name))
            .collect()
    }
}