src/tools/rust-analyzer/crates/rust-analyzer/src/cli/diagnostics.rs

   1 //! Analyze all modules in a project for diagnostics. Exits with a non-zero
   2 //! status code if any errors are found.
   3
   4 use rustc_hash::FxHashSet;
   5
   6 use hir::{db::HirDatabase, Crate, Module};
   7 use ide::{AssistResolveStrategy, DiagnosticsConfig, Severity};
   8 use ide_db::base_db::SourceDatabaseExt;
   9
  10 use crate::cli::{
  11     flags,
  12     load_cargo::{load_workspace_at, LoadCargoConfig},
  13 };
  14
  15 impl flags::Diagnostics {
  16     pub fn run(self) -> anyhow::Result<()> {
  17         let cargo_config = Default::default();
  18         let load_cargo_config = LoadCargoConfig {
  19             load_out_dirs_from_check: !self.disable_build_scripts,
  20             with_proc_macro: !self.disable_proc_macros,
  21             prefill_caches: false,
  22         };
  23         let (host, _vfs, _proc_macro) =
  24             load_workspace_at(&self.path, &cargo_config, &load_cargo_config, &|_| {})?;
  25         let db = host.raw_database();
  26         let analysis = host.analysis();
  27
  28         let mut found_error = false;
  29         let mut visited_files = FxHashSet::default();
  30
  31         let work = all_modules(db).into_iter().filter(|module| {
  32             let file_id = module.definition_source(db).file_id.original_file(db);
  33             let source_root = db.file_source_root(file_id);
  34             let source_root = db.source_root(source_root);
  35             !source_root.is_library
  36         });
  37
  38         for module in work {
  39             let file_id = module.definition_source(db).file_id.original_file(db);
  40             if !visited_files.contains(&file_id) {
  41                 let crate_name =
  42                     module.krate().display_name(db).as_deref().unwrap_or("unknown").to_string();
  43                 println!("processing crate: {}, module: {}", crate_name, _vfs.file_path(file_id));
  44                 for diagnostic in analysis
  45                     .diagnostics(
  46                         &DiagnosticsConfig::test_sample(),
  47                         AssistResolveStrategy::None,
  48                         file_id,
  49                     )
  50                     .unwrap()
  51                 {
  52                     if matches!(diagnostic.severity, Severity::Error) {
  53                         found_error = true;
  54                     }
  55
  56                     println!("{:?}", diagnostic);
  57                 }
  58
  59                 visited_files.insert(file_id);
  60             }
  61         }
  62
  63         println!();
  64         println!("diagnostic scan complete");
  65
  66         if found_error {
  67             println!();
  68             anyhow::bail!("diagnostic error detected")
  69         }
  70
  71         Ok(())
  72     }
  73 }
  74
  75 fn all_modules(db: &dyn HirDatabase) -> Vec<Module> {
  76     let mut worklist: Vec<_> =
  77         Crate::all(db).into_iter().map(|krate| krate.root_module(db)).collect();
  78     let mut modules = Vec::new();
  79
  80     while let Some(module) = worklist.pop() {
  81         modules.push(module);
  82         worklist.extend(module.children(db));
  83     }
  84
  85     modules
  86 }