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
100
101
102
103

use crate::linter::{Rule, RuleResult};
use sv_parser::{unwrap_locate, RefNode, StatementItem, StatementOrNull, SyntaxTree};

pub struct IfWithBegin;

impl Rule for IfWithBegin {
    fn check(&self, syntax_tree: &SyntaxTree, node: &RefNode) -> RuleResult {
        match node {
            RefNode::ConditionalStatement(x) => {
                let (ref a, ref b, ref c, ref d, ref e, ref f) = x.nodes;

                // if statement
                let mut if_str = String::from("");
                syntax_tree.get_str(a).map(|x| if_str.push_str(x));
                syntax_tree.get_str(b).map(|x| if_str.push_str(x));
                syntax_tree.get_str(c).map(|x| if_str.push_str(x));
                syntax_tree.get_str(d).map(|x| if_str.push_str(x));

                match d {
                    StatementOrNull::Statement(x) => {
                        let (_, _, ref x) = x.nodes;
                        match x {
                            StatementItem::SeqBlock(_) => (),
                            _ => {
                                if if_str.trim_end().contains("\n") {
                                    return RuleResult::Fail;
                                }
                            }
                        }
                    }
                    _ => (),
                }

                // else if statement
                for e in e {
                    let (ref a, ref b, ref c, ref d) = e;

                    let mut elsif_str = String::from("");
                    syntax_tree.get_str(a).map(|x| elsif_str.push_str(x));
                    syntax_tree.get_str(b).map(|x| elsif_str.push_str(x));
                    syntax_tree.get_str(c).map(|x| elsif_str.push_str(x));
                    syntax_tree.get_str(d).map(|x| elsif_str.push_str(x));

                    match d {
                        StatementOrNull::Statement(x) => {
                            let (_, _, ref x) = x.nodes;
                            match x {
                                StatementItem::SeqBlock(_) => (),
                                _ => {
                                    if elsif_str.trim_end().contains("\n") {
                                        let locate = unwrap_locate!(a).unwrap();
                                        return RuleResult::FailLocate(locate.clone());
                                    }
                                }
                            }
                        }
                        _ => (),
                    };
                }

                // else statement
                if let Some(f) = f {
                    let (ref a, ref b) = f;

                    let mut else_str = String::from("");
                    syntax_tree.get_str(a).map(|x| else_str.push_str(x));
                    syntax_tree.get_str(b).map(|x| else_str.push_str(x));

                    match b {
                        StatementOrNull::Statement(x) => {
                            let (_, _, ref x) = x.nodes;
                            match x {
                                StatementItem::SeqBlock(_) => (),
                                _ => {
                                    if else_str.trim_end().contains("\n") {
                                        let locate = unwrap_locate!(a).unwrap();
                                        return RuleResult::FailLocate(locate.clone());
                                    }
                                }
                            }
                        }
                        _ => (),
                    };
                }

                RuleResult::Pass
            }
            _ => RuleResult::Pass,
        }
    }

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

    fn hint(&self) -> String {
        String::from("multiline `if` statement must have `begin`")
    }

    fn reason(&self) -> String {
        String::from("if there is not `begin`, the second statatement are confusing")
    }
}