mod cargo_workspace; mod json_project; mod sysroot; use std::{ fs::File, io::BufReader, path::{Path, PathBuf}, }; use failure::bail; use rustc_hash::FxHashMap; use ra_db::{CrateGraph, FileId, Edition}; use serde_json::from_reader; pub use crate::{ cargo_workspace::{CargoWorkspace, Package, Target, TargetKind}, json_project::JsonProject, sysroot::Sysroot, }; // TODO use proper error enum pub type Result = ::std::result::Result; #[derive(Debug, Clone)] pub enum ProjectWorkspace { /// Project workspace was discovered by running `cargo metadata` and `rustc --print sysroot`. Cargo { cargo: CargoWorkspace, sysroot: Sysroot }, /// Project workspace was manually specified using a `rust-project.json` file. Json { project: JsonProject }, } impl ProjectWorkspace { pub fn discover(path: &Path) -> Result { match find_rust_project_json(path) { Some(json_path) => { let file = File::open(json_path)?; let reader = BufReader::new(file); Ok(ProjectWorkspace::Json { project: from_reader(reader)? }) } None => { let cargo_toml = find_cargo_toml(path)?; Ok(ProjectWorkspace::Cargo { cargo: CargoWorkspace::from_cargo_metadata(&cargo_toml)?, sysroot: Sysroot::discover(&cargo_toml)?, }) } } } pub fn add_roots(&self, roots: &mut Vec) { match self { ProjectWorkspace::Json { project } => { for root in &project.roots { roots.push(root.path.clone()); } } ProjectWorkspace::Cargo { cargo, sysroot } => { for pkg in cargo.packages() { roots.push(pkg.root(&cargo).to_path_buf()); } for krate in sysroot.crates() { roots.push(krate.root_dir(&sysroot).to_path_buf()) } } } } pub fn count(&self) -> usize { match self { ProjectWorkspace::Json { project } => project.crates.len(), ProjectWorkspace::Cargo { cargo, .. } => cargo.packages().count(), } } pub fn to_crate_graph(&self, load: &mut dyn FnMut(&Path) -> Option) -> CrateGraph { let mut crate_graph = CrateGraph::default(); match self { ProjectWorkspace::Json { project } => { let mut crates = FxHashMap::default(); for (id, krate) in project.crates.iter().enumerate() { let crate_id = json_project::CrateId(id); if let Some(file_id) = load(&krate.root_module) { let edition = match krate.edition { json_project::Edition::Edition2015 => Edition::Edition2015, json_project::Edition::Edition2018 => Edition::Edition2018, }; crates.insert(crate_id, crate_graph.add_crate_root(file_id, edition)); } } for (id, krate) in project.crates.iter().enumerate() { for dep in &krate.deps { let from_crate_id = json_project::CrateId(id); let to_crate_id = dep.krate; if let (Some(&from), Some(&to)) = (crates.get(&from_crate_id), crates.get(&to_crate_id)) { if let Err(_) = crate_graph.add_dep(from, dep.name.clone().into(), to) { log::error!( "cyclic dependency {:?} -> {:?}", from_crate_id, to_crate_id ); } } } } } ProjectWorkspace::Cargo { cargo, sysroot } => { let mut sysroot_crates = FxHashMap::default(); for krate in sysroot.crates() { if let Some(file_id) = load(krate.root(&sysroot)) { sysroot_crates.insert( krate, crate_graph.add_crate_root(file_id, Edition::Edition2015), ); } } for from in sysroot.crates() { for to in from.deps(&sysroot) { let name = to.name(&sysroot); if let (Some(&from), Some(&to)) = (sysroot_crates.get(&from), sysroot_crates.get(&to)) { if let Err(_) = crate_graph.add_dep(from, name.into(), to) { log::error!("cyclic dependency between sysroot crates") } } } } let libstd = sysroot.std().and_then(|it| sysroot_crates.get(&it).map(|&it| it)); let mut pkg_to_lib_crate = FxHashMap::default(); let mut pkg_crates = FxHashMap::default(); // Next, create crates for each package, target pair for pkg in cargo.packages() { let mut lib_tgt = None; for tgt in pkg.targets(&cargo) { let root = tgt.root(&cargo); if let Some(file_id) = load(root) { let edition = pkg.edition(&cargo); let crate_id = crate_graph.add_crate_root(file_id, edition); if tgt.kind(&cargo) == TargetKind::Lib { lib_tgt = Some(crate_id); pkg_to_lib_crate.insert(pkg, crate_id); } pkg_crates.entry(pkg).or_insert_with(Vec::new).push(crate_id); } } // Set deps to the std and to the lib target of the current package for &from in pkg_crates.get(&pkg).into_iter().flatten() { if let Some(to) = lib_tgt { if to != from { if let Err(_) = crate_graph.add_dep(from, pkg.name(&cargo).into(), to) { log::error!( "cyclic dependency between targets of {}", pkg.name(&cargo) ) } } } if let Some(std) = libstd { if let Err(_) = crate_graph.add_dep(from, "std".into(), std) { log::error!("cyclic dependency on std for {}", pkg.name(&cargo)) } } } } // Now add a dep ednge from all targets of upstream to the lib // target of downstream. for pkg in cargo.packages() { for dep in pkg.dependencies(&cargo) { if let Some(&to) = pkg_to_lib_crate.get(&dep.pkg) { for &from in pkg_crates.get(&pkg).into_iter().flatten() { if let Err(_) = crate_graph.add_dep(from, dep.name.clone().into(), to) { log::error!( "cyclic dependency {} -> {}", pkg.name(&cargo), dep.pkg.name(&cargo) ) } } } } } } } crate_graph } } fn find_rust_project_json(path: &Path) -> Option { if path.ends_with("rust-project.json") { return Some(path.to_path_buf()); } let mut curr = Some(path); while let Some(path) = curr { let candidate = path.join("rust-project.json"); if candidate.exists() { return Some(candidate); } curr = path.parent(); } None } fn find_cargo_toml(path: &Path) -> Result { if path.ends_with("Cargo.toml") { return Ok(path.to_path_buf()); } let mut curr = Some(path); while let Some(path) = curr { let candidate = path.join("Cargo.toml"); if candidate.exists() { return Ok(candidate); } curr = path.parent(); } bail!("can't find Cargo.toml at {}", path.display()) }