]> git.lizzy.rs Git - rust.git/blob - clippy_dev/src/stderr_length_check.rs
dev: Use bytecount for faster line count
[rust.git] / clippy_dev / src / stderr_length_check.rs
1 use std::ffi::OsStr;
2 use std::fs;
3 use std::path::{Path, PathBuf};
4
5 use walkdir::WalkDir;
6
7 // The maximum length allowed for stderr files.
8 //
9 // We limit this because small files are easier to deal with than bigger files.
10 const LENGTH_LIMIT: usize = 200;
11
12 pub fn check() {
13     let exceeding_files: Vec<_> = exceeding_stderr_files();
14
15     if !exceeding_files.is_empty() {
16         eprintln!("Error: stderr files exceeding limit of {} lines:", LENGTH_LIMIT);
17         for path in exceeding_files {
18             println!("{}", path.display());
19         }
20         std::process::exit(1);
21     }
22 }
23
24 fn exceeding_stderr_files() -> Vec<PathBuf> {
25     // We use `WalkDir` instead of `fs::read_dir` here in order to recurse into subdirectories.
26     WalkDir::new("../tests/ui")
27         .into_iter()
28         .filter_map(Result::ok)
29         .filter_map(|e| {
30             let p = e.into_path();
31             if p.extension() == Some(OsStr::new("stderr")) && count_linenumbers(&p) > LENGTH_LIMIT {
32                 Some(p)
33             } else {
34                 None
35             }
36         })
37         .collect()
38 }
39
40 #[must_use]
41 fn count_linenumbers(filepath: &Path) -> usize {
42     match fs::read(filepath) {
43         Ok(content) => bytecount::count(&content, b'\n'),
44         Err(e) => {
45             eprintln!("Failed to read file: {}", e);
46             0
47         },
48     }
49 }