clippy_dev/src/stderr_length_check.rs

   1 use std::ffi::OsStr;
   2 use walkdir::WalkDir;
   3
   4 use std::fs::File;
   5 use std::io::prelude::*;
   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 LIMIT: usize = 275;
  11
  12 pub fn check() {
  13     let stderr_files = stderr_files();
  14     let exceeding_files = exceeding_stderr_files(stderr_files).collect::<Vec<String>>();
  15
  16     if !exceeding_files.is_empty() {
  17         eprintln!("Error: stderr files exceeding limit of {} lines:", LIMIT);
  18         for path in exceeding_files {
  19             println!("{}", path);
  20         }
  21         std::process::exit(1);
  22     }
  23 }
  24
  25 fn exceeding_stderr_files(files: impl Iterator<Item = walkdir::DirEntry>) -> impl Iterator<Item = String> {
  26     files.filter_map(|file| {
  27         let path = file.path().to_str().expect("Could not convert path to str").to_string();
  28         let linecount = count_linenumbers(&path);
  29         if linecount > LIMIT {
  30             Some(path)
  31         } else {
  32             None
  33         }
  34     })
  35 }
  36
  37 fn stderr_files() -> impl Iterator<Item = walkdir::DirEntry> {
  38     // We use `WalkDir` instead of `fs::read_dir` here in order to recurse into subdirectories.
  39     WalkDir::new("../tests/ui")
  40         .into_iter()
  41         .filter_map(std::result::Result::ok)
  42         .filter(|f| f.path().extension() == Some(OsStr::new("stderr")))
  43 }
  44
  45 fn count_linenumbers(filepath: &str) -> usize {
  46     if let Ok(mut file) = File::open(filepath) {
  47         let mut content = String::new();
  48         file.read_to_string(&mut content).expect("Failed to read file?");
  49         content.lines().count()
  50     } else {
  51         0
  52     }
  53 }