rrubocop 0.8.15

10x faster RuboCop drop-in. Same output format and configs, without maintaining RuboCop gems.
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
104
105
106
107
108
109
110
111
112
113
114

//! Line / source / file-model lint phases.

use crate::cli::AutocorrectMode;
use crate::correction::Correction;
use crate::diagnostic::Diagnostic;
use crate::model;
use crate::parse::codemap::CodeMap;
use crate::parse::source::SourceFile;

use super::syntax_gate::ActiveCop;
use super::{allow_corr, finish_cop_pass};

pub(super) fn run_line_phase(
    source: &SourceFile,
    active: &[ActiveCop<'_>],
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    for (cop, cfg, idx) in active.iter().filter(|(c, _, _)| c.uses_line_phase()) {
        let mut corr_buf = allow_corr(mode, *cop).then(Vec::new);
        let before = diagnostics.len();
        cop.check_lines(source, cfg, diagnostics, corr_buf.as_mut());
        finish_cop_pass(diagnostics, before, cfg, &mut corr_buf, *idx, corrections);
    }
}

pub(super) fn run_source_phase(
    source: &SourceFile,
    tree: &tree_sitter::Tree,
    active: &[ActiveCop<'_>],
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    run_file_model_cops(source, tree, active, mode, diagnostics, corrections);
    run_plain_source_cops(source, tree, active, mode, diagnostics, corrections);
}

fn run_file_model_cops(
    source: &SourceFile,
    tree: &tree_sitter::Tree,
    active: &[ActiveCop<'_>],
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    let model_cops: Vec<_> = active
        .iter()
        .filter(|(c, _, _)| c.needs_file_model())
        .collect();
    if model_cops.is_empty() {
        return;
    }
    let file_model = model::build(source.as_bytes(), tree.clone());
    for (cop, cfg, idx) in model_cops {
        one_file_model_cop(source, &file_model, *cop, cfg, *idx, mode, diagnostics, corrections);
    }
}

fn one_file_model_cop(
    source: &SourceFile,
    file_model: &crate::model::FileModel,
    cop: &dyn crate::cop::Cop,
    cfg: &crate::cop::CopConfig,
    idx: usize,
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    let mut corr_buf = allow_corr(mode, cop).then(Vec::new);
    let before = diagnostics.len();
    cop.check_file_model(source, file_model, cfg, diagnostics, corr_buf.as_mut());
    finish_cop_pass(diagnostics, before, cfg, &mut corr_buf, idx, corrections);
}

fn run_plain_source_cops(
    source: &SourceFile,
    tree: &tree_sitter::Tree,
    active: &[ActiveCop<'_>],
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    let plain: Vec<_> = active
        .iter()
        .filter(|(c, _, _)| c.uses_source_phase() && !c.needs_file_model())
        .collect();
    if plain.is_empty() {
        return;
    }
    // Build CodeMap once, only when plain source cops need it.
    let code_map = CodeMap::from_tree(tree.root_node(), source.as_bytes());
    for (cop, cfg, idx) in plain {
        one_plain_source_cop(source, tree, &code_map, *cop, cfg, *idx, mode, diagnostics, corrections);
    }
}

fn one_plain_source_cop(
    source: &SourceFile,
    tree: &tree_sitter::Tree,
    code_map: &CodeMap,
    cop: &dyn crate::cop::Cop,
    cfg: &crate::cop::CopConfig,
    idx: usize,
    mode: AutocorrectMode,
    diagnostics: &mut Vec<Diagnostic>,
    corrections: &mut Option<Vec<Correction>>,
) {
    let mut corr_buf = allow_corr(mode, cop).then(Vec::new);
    let before = diagnostics.len();
    cop.check_source(source, tree, code_map, cfg, diagnostics, corr_buf.as_mut());
    finish_cop_pass(diagnostics, before, cfg, &mut corr_buf, idx, corrections);
}