Consolidate fn load_cargo(…) parameters into struct LoadCargoConfig { … }

This commit is contained in:
Vincent Esche 2021-02-08 11:30:16 +01:00
parent 6d9c13c710
commit 8f461ffe8f
5 changed files with 70 additions and 31 deletions

View File

@ -16,7 +16,10 @@
};
use vfs::AbsPathBuf;
use crate::cli::{load_cargo::load_cargo, print_memory_usage, Verbosity};
use crate::cli::{
load_cargo::{load_cargo, LoadCargoConfig},
print_memory_usage, Verbosity,
};
pub struct BenchCmd {
pub path: PathBuf,
@ -59,12 +62,14 @@ pub fn run(self, verbosity: Verbosity) -> Result<()> {
let start = Instant::now();
eprint!("loading: ");
let (mut host, vfs) = load_cargo(
&self.path,
&Default::default(),
self.load_output_dirs,
self.with_proc_macro,
)?;
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check: self.load_output_dirs,
with_proc_macro: self.with_proc_macro,
};
let (mut host, vfs) = load_cargo(&self.path, &load_cargo_config)?;
eprintln!("{:?}\n", start.elapsed());
let file_id = {

View File

@ -25,8 +25,10 @@
use syntax::AstNode;
use crate::cli::{
load_cargo::load_cargo, print_memory_usage, progress_report::ProgressReport, report_metric,
Result, Verbosity,
load_cargo::{load_cargo, LoadCargoConfig},
print_memory_usage,
progress_report::ProgressReport,
report_metric, Result, Verbosity,
};
use profile::StopWatch;
@ -57,12 +59,12 @@ pub fn run(self, verbosity: Verbosity) -> Result<()> {
};
let mut db_load_sw = self.stop_watch();
let (host, vfs) = load_cargo(
&self.path,
&Default::default(),
self.load_output_dirs,
self.with_proc_macro,
)?;
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check: self.load_output_dirs,
with_proc_macro: self.with_proc_macro,
};
let (host, vfs) = load_cargo(&self.path, &load_cargo_config)?;
let db = host.raw_database();
eprintln!("{:<20} {}", "Database loaded:", db_load_sw.elapsed());

View File

@ -10,7 +10,10 @@
use ide::{DiagnosticsConfig, Severity};
use ide_db::base_db::SourceDatabaseExt;
use crate::cli::{load_cargo::load_cargo, Result};
use crate::cli::{
load_cargo::{load_cargo, LoadCargoConfig},
Result,
};
fn all_modules(db: &dyn HirDatabase) -> Vec<Module> {
let mut worklist: Vec<_> =
@ -25,8 +28,17 @@ fn all_modules(db: &dyn HirDatabase) -> Vec<Module> {
modules
}
pub fn diagnostics(path: &Path, load_output_dirs: bool, with_proc_macro: bool) -> Result<()> {
let (host, _vfs) = load_cargo(path, &Default::default(), load_output_dirs, with_proc_macro)?;
pub fn diagnostics(
path: &Path,
load_out_dirs_from_check: bool,
with_proc_macro: bool,
) -> Result<()> {
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check,
with_proc_macro,
};
let (host, _vfs) = load_cargo(path, &load_cargo_config)?;
let db = host.raw_database();
let analysis = host.analysis();

View File

@ -13,15 +13,16 @@
use crate::reload::{ProjectFolders, SourceRootConfig};
pub fn load_cargo(
root: &Path,
config: &CargoConfig,
load_out_dirs_from_check: bool,
with_proc_macro: bool,
) -> Result<(AnalysisHost, vfs::Vfs)> {
pub struct LoadCargoConfig {
pub cargo_config: CargoConfig,
pub load_out_dirs_from_check: bool,
pub with_proc_macro: bool,
}
pub fn load_cargo(root: &Path, config: &LoadCargoConfig) -> Result<(AnalysisHost, vfs::Vfs)> {
let root = AbsPathBuf::assert(std::env::current_dir()?.join(root));
let root = ProjectManifest::discover_single(&root)?;
let ws = ProjectWorkspace::load(root, config, &|_| {})?;
let ws = ProjectWorkspace::load(root, &config.cargo_config, &|_| {})?;
let (sender, receiver) = unbounded();
let mut vfs = vfs::Vfs::default();
@ -31,14 +32,14 @@ pub fn load_cargo(
Box::new(loader)
};
let proc_macro_client = if with_proc_macro {
let proc_macro_client = if config.with_proc_macro {
let path = std::env::current_exe()?;
Some(ProcMacroClient::extern_process(path, &["proc-macro"]).unwrap())
} else {
None
};
let build_data = if load_out_dirs_from_check {
let build_data = if config.load_out_dirs_from_check {
let mut collector = BuildDataCollector::default();
ws.collect_build_data_configs(&mut collector);
Some(collector.collect(&|_| {})?)
@ -117,7 +118,13 @@ mod tests {
#[test]
fn test_loading_rust_analyzer() {
let path = Path::new(env!("CARGO_MANIFEST_DIR")).parent().unwrap().parent().unwrap();
let (host, _vfs) = load_cargo(path, &Default::default(), false, false).unwrap();
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check: false,
with_proc_macro: false,
};
let (host, _vfs) = load_cargo(path, &load_cargo_config).unwrap();
let n_crates = Crate::all(host.raw_database()).len();
// RA has quite a few crates, but the exact count doesn't matter
assert!(n_crates > 20);

View File

@ -1,11 +1,19 @@
//! Applies structured search replace rules from the command line.
use crate::cli::{load_cargo::load_cargo, Result};
use crate::cli::{
load_cargo::{load_cargo, LoadCargoConfig},
Result,
};
use ssr::{MatchFinder, SsrPattern, SsrRule};
pub fn apply_ssr_rules(rules: Vec<SsrRule>) -> Result<()> {
use ide_db::base_db::SourceDatabaseExt;
let (host, vfs) = load_cargo(&std::env::current_dir()?, &Default::default(), true, true)?;
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check: true,
with_proc_macro: true,
};
let (host, vfs) = load_cargo(&std::env::current_dir()?, &load_cargo_config)?;
let db = host.raw_database();
let mut match_finder = MatchFinder::at_first_file(db)?;
for rule in rules {
@ -28,7 +36,12 @@ pub fn apply_ssr_rules(rules: Vec<SsrRule>) -> Result<()> {
pub fn search_for_patterns(patterns: Vec<SsrPattern>, debug_snippet: Option<String>) -> Result<()> {
use ide_db::base_db::SourceDatabaseExt;
use ide_db::symbol_index::SymbolsDatabase;
let (host, _vfs) = load_cargo(&std::env::current_dir()?, &Default::default(), true, true)?;
let load_cargo_config = LoadCargoConfig {
cargo_config: Default::default(),
load_out_dirs_from_check: true,
with_proc_macro: true,
};
let (host, _vfs) = load_cargo(&std::env::current_dir()?, &load_cargo_config)?;
let db = host.raw_database();
let mut match_finder = MatchFinder::at_first_file(db)?;
for pattern in patterns {