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
89
90
91
92
93
94
95
96
97
98
99

use crate::config::ConfigOption;
use crate::linter::{SyntaxRule, SyntaxRuleResult};
use regex::Regex;
use sv_parser::{NodeEvent, RefNode, SyntaxTree};

#[derive(Default)]
pub struct StyleOperatorInteger {
    re_split: Option<Regex>,
    re_op: Option<Regex>,
    re_succ: Option<Regex>,
}

impl SyntaxRule for StyleOperatorInteger {
    fn check(
        &mut self,
        syntax_tree: &SyntaxTree,
        event: &NodeEvent,
        _option: &ConfigOption,
    ) -> SyntaxRuleResult {
        /*
        re_split extracts operator from anything following it.
        re_op is used to selectively apply this rule to specific operators.
        re_succ matches what is allowed after the operator.
            - newline
            - exactly 1space, then comment
            - exactly 1space, then nothing
        */
        if self.re_split.is_none() {
            self.re_split = Some(Regex::new(r"(?P<op>\S+)(?P<succ>(?s:.)*)").unwrap());
        }
        if self.re_op.is_none() {
            let operators =
                [ "&" // {{{
                , "\\|"
                , "\\^"
                , "\\^~"
                , "~\\^"
                , ">>"
                , "<<"
                , ">>>"
                , "<<<"
                ].join("|"); // }}}

            self.re_op = Some(Regex::new(format!("^({})$", operators).as_str()).unwrap());
        }
        if self.re_succ.is_none() {
            self.re_succ = Some(Regex::new(r"^([\n\v\f\r]| /| $)").unwrap());
        }

        let node = match event {
            NodeEvent::Enter(x) => x,
            NodeEvent::Leave(_) => {
                return SyntaxRuleResult::Pass;
            }
        };

        let s: Option<&str> = match node {
            RefNode::BinaryOperator(x) => {
                Some(syntax_tree.get_str(*x).unwrap())
            }
            RefNode::BinaryModulePathOperator(x) => {
                Some(syntax_tree.get_str(*x).unwrap())
            }
            _ => None,
        };

        if let Some(t) = s {
            let re_split = self.re_split.as_ref().unwrap();
            let re_op = self.re_op.as_ref().unwrap();
            let caps = re_split.captures(&t).unwrap();

            if re_op.is_match(&caps[1]) {
                let re_succ = self.re_succ.as_ref().unwrap();

                if re_succ.is_match(&caps[2]) {
                    SyntaxRuleResult::Pass
                } else {
                    SyntaxRuleResult::Fail
                }
            } else {
                SyntaxRuleResult::Pass
            }
        } else {
            SyntaxRuleResult::Pass
        }
    }

    fn name(&self) -> String {
        String::from("style_operator_integer")
    }

    fn hint(&self, _option: &ConfigOption) -> String {
        String::from("Follow operator with a newline or exactly 1 space.")
    }

    fn reason(&self) -> String {
        String::from("Consistent use of whitespace enhances readability by reducing visual noise.")
    }
}