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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

pub type NodeList = &'static [&'static Node];

#[derive(Debug, Clone, PartialEq)]
pub struct Node {
    pub camelcase_name: &'static str,
    pub wqp_name: &'static str,
    pub fields: NodeFieldList,
    pub comment: &'static [&'static str],
}

impl Node {
    pub fn render_comment(&self, prefix: &str, offset: usize) -> String {
        crate::comment::Comment::new(self.comment, prefix).to_string(offset)
    }

    pub fn upper_name(&self) -> String {
        crate::helpers::camel_case_to_underscored(self.camelcase_name).to_uppercase()
    }

    pub fn lower_name(&self) -> String {
        crate::helpers::camel_case_to_underscored(self.camelcase_name).to_lowercase()
    }
}

pub type NodeFieldList = &'static [&'static NodeField];

#[derive(Debug, Clone, PartialEq)]
pub struct NodeField {
    pub node: &'static Node,
    pub snakecase_name: &'static str,
    pub field_type: NodeFieldType,
    pub always_print: bool,
    pub comment: &'static [&'static str],
}

impl NodeField {
    pub fn render_comment(&self, prefix: &str, offset: usize) -> String {
        crate::comment::Comment::new(self.comment, prefix).to_string(offset)
    }
}

#[derive(PartialEq, Clone, Debug)]
pub enum NodeFieldType {
    Node,
    Nodes,
    MaybeNode { regexp_options: bool },
    Loc,
    MaybeLoc,
    Str { raw: bool },
    MaybeStr { chars: bool },
    StringValue,
    U8,
}

impl NodeFieldType {}

#[macro_export]
macro_rules! node_has_field {
    ($node:ident, $( $pattern:pat_param )|+) => {
        $node.fields.iter().any(|f| matches!(f.field_type, $( $pattern )|+))
    };
}

pub use node_has_field;

#[cfg(test)]
mod tests {
    use super::*;

    static TEST_NODE: Node = Node {
        camelcase_name: "",
        wqp_name: "",
        fields: &[&NodeField {
            node: &TEST_NODE,
            snakecase_name: "",
            field_type: NodeFieldType::Loc,
            always_print: true,
            comment: &[],
        }],
        comment: &[],
    };

    #[test]
    fn test_node_has_field() {
        assert!(node_has_field!(TEST_NODE, NodeFieldType::Loc));
        assert!(!node_has_field!(TEST_NODE, NodeFieldType::MaybeNode { .. }))
    }
}