2020-11-13 10:38:26 -06:00
|
|
|
//! Handles lowering of build-system specific workspace information (`cargo
|
|
|
|
//! metadata` or `rust-project.json`) into representation stored in the salsa
|
|
|
|
//! database -- `CrateGraph`.
|
|
|
|
|
2021-03-07 04:18:01 -06:00
|
|
|
use std::{collections::VecDeque, fmt, fs, path::Path, process::Command};
|
2020-11-13 09:44:48 -06:00
|
|
|
|
|
|
|
use anyhow::{Context, Result};
|
2020-12-07 10:16:50 -06:00
|
|
|
use base_db::{CrateDisplayName, CrateGraph, CrateId, CrateName, Edition, Env, FileId, ProcMacro};
|
2020-11-13 09:44:48 -06:00
|
|
|
use cfg::CfgOptions;
|
|
|
|
use paths::{AbsPath, AbsPathBuf};
|
|
|
|
use proc_macro_api::ProcMacroClient;
|
|
|
|
use rustc_hash::{FxHashMap, FxHashSet};
|
|
|
|
|
|
|
|
use crate::{
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data::{BuildData, BuildDataMap, BuildDataResult},
|
|
|
|
cargo_workspace,
|
|
|
|
cfg_flag::CfgFlag,
|
|
|
|
rustc_cfg,
|
|
|
|
sysroot::SysrootCrate,
|
|
|
|
utf8_stdout, BuildDataCollector, CargoConfig, CargoWorkspace, ProjectJson, ProjectManifest,
|
|
|
|
Sysroot, TargetKind,
|
2020-11-13 09:44:48 -06:00
|
|
|
};
|
|
|
|
|
|
|
|
/// `PackageRoot` describes a package root folder.
|
|
|
|
/// Which may be an external dependency, or a member of
|
|
|
|
/// the current workspace.
|
|
|
|
#[derive(Debug, Clone, Eq, PartialEq, Hash)]
|
|
|
|
pub struct PackageRoot {
|
|
|
|
/// Is a member of the current workspace
|
|
|
|
pub is_member: bool,
|
|
|
|
pub include: Vec<AbsPathBuf>,
|
|
|
|
pub exclude: Vec<AbsPathBuf>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone, Eq, PartialEq)]
|
|
|
|
pub enum ProjectWorkspace {
|
|
|
|
/// Project workspace was discovered by running `cargo metadata` and `rustc --print sysroot`.
|
2021-01-18 05:52:12 -06:00
|
|
|
Cargo {
|
|
|
|
cargo: CargoWorkspace,
|
|
|
|
sysroot: Sysroot,
|
|
|
|
rustc: Option<CargoWorkspace>,
|
|
|
|
/// Holds cfg flags for the current target. We get those by running
|
|
|
|
/// `rustc --print cfg`.
|
|
|
|
///
|
|
|
|
/// FIXME: make this a per-crate map, as, eg, build.rs might have a
|
|
|
|
/// different target.
|
|
|
|
rustc_cfg: Vec<CfgFlag>,
|
|
|
|
},
|
2020-11-13 09:44:48 -06:00
|
|
|
/// Project workspace was manually specified using a `rust-project.json` file.
|
2021-01-18 05:52:12 -06:00
|
|
|
Json { project: ProjectJson, sysroot: Option<Sysroot>, rustc_cfg: Vec<CfgFlag> },
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
impl fmt::Debug for ProjectWorkspace {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
2021-01-28 08:04:44 -06:00
|
|
|
// Make sure this isn't too verbose.
|
2020-11-13 09:44:48 -06:00
|
|
|
match self {
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Cargo { cargo, sysroot, rustc, rustc_cfg } => f
|
2020-11-13 09:44:48 -06:00
|
|
|
.debug_struct("Cargo")
|
|
|
|
.field("n_packages", &cargo.packages().len())
|
|
|
|
.field("n_sysroot_crates", &sysroot.crates().len())
|
|
|
|
.field(
|
|
|
|
"n_rustc_compiler_crates",
|
|
|
|
&rustc.as_ref().map_or(0, |rc| rc.packages().len()),
|
|
|
|
)
|
2021-01-28 08:04:44 -06:00
|
|
|
.field("n_rustc_cfg", &rustc_cfg.len())
|
2020-11-13 09:44:48 -06:00
|
|
|
.finish(),
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Json { project, sysroot, rustc_cfg } => {
|
2020-11-13 09:44:48 -06:00
|
|
|
let mut debug_struct = f.debug_struct("Json");
|
|
|
|
debug_struct.field("n_crates", &project.n_crates());
|
|
|
|
if let Some(sysroot) = sysroot {
|
|
|
|
debug_struct.field("n_sysroot_crates", &sysroot.crates().len());
|
|
|
|
}
|
2021-01-28 08:04:44 -06:00
|
|
|
debug_struct.field("n_rustc_cfg", &rustc_cfg.len());
|
2020-11-13 09:44:48 -06:00
|
|
|
debug_struct.finish()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ProjectWorkspace {
|
2021-01-07 11:08:46 -06:00
|
|
|
pub fn load(
|
|
|
|
manifest: ProjectManifest,
|
|
|
|
config: &CargoConfig,
|
|
|
|
progress: &dyn Fn(String),
|
|
|
|
) -> Result<ProjectWorkspace> {
|
2020-11-13 09:44:48 -06:00
|
|
|
let res = match manifest {
|
|
|
|
ProjectManifest::ProjectJson(project_json) => {
|
|
|
|
let file = fs::read_to_string(&project_json).with_context(|| {
|
|
|
|
format!("Failed to read json file {}", project_json.display())
|
|
|
|
})?;
|
|
|
|
let data = serde_json::from_str(&file).with_context(|| {
|
|
|
|
format!("Failed to deserialize json file {}", project_json.display())
|
|
|
|
})?;
|
|
|
|
let project_location = project_json.parent().unwrap().to_path_buf();
|
2020-11-17 04:31:40 -06:00
|
|
|
let project_json = ProjectJson::new(&project_location, data);
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::load_inline(project_json, config.target.as_deref())?
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
ProjectManifest::CargoToml(cargo_toml) => {
|
|
|
|
let cargo_version = utf8_stdout({
|
|
|
|
let mut cmd = Command::new(toolchain::cargo());
|
|
|
|
cmd.arg("--version");
|
|
|
|
cmd
|
|
|
|
})?;
|
|
|
|
|
2021-01-07 11:08:46 -06:00
|
|
|
let cargo = CargoWorkspace::from_cargo_metadata(&cargo_toml, config, progress)
|
|
|
|
.with_context(|| {
|
2020-11-13 09:44:48 -06:00
|
|
|
format!(
|
|
|
|
"Failed to read Cargo metadata from Cargo.toml file {}, {}",
|
|
|
|
cargo_toml.display(),
|
|
|
|
cargo_version
|
|
|
|
)
|
2021-01-07 11:08:46 -06:00
|
|
|
})?;
|
2021-02-11 10:34:56 -06:00
|
|
|
|
2020-11-13 10:38:26 -06:00
|
|
|
let sysroot = if config.no_sysroot {
|
|
|
|
Sysroot::default()
|
|
|
|
} else {
|
2020-11-13 09:44:48 -06:00
|
|
|
Sysroot::discover(&cargo_toml).with_context(|| {
|
|
|
|
format!(
|
|
|
|
"Failed to find sysroot for Cargo.toml file {}. Is rust-src installed?",
|
|
|
|
cargo_toml.display()
|
|
|
|
)
|
|
|
|
})?
|
|
|
|
};
|
|
|
|
|
2021-02-11 10:34:56 -06:00
|
|
|
let rustc_dir = if let Some(rustc_source) = &config.rustc_source {
|
|
|
|
use cargo_workspace::RustcSource;
|
|
|
|
match rustc_source {
|
|
|
|
RustcSource::Path(path) => Some(path.clone()),
|
|
|
|
RustcSource::Discover => Sysroot::discover_rustc(&cargo_toml),
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
|
|
|
|
|
|
|
let rustc = if let Some(rustc_dir) = rustc_dir {
|
2021-01-07 11:08:46 -06:00
|
|
|
Some(
|
|
|
|
CargoWorkspace::from_cargo_metadata(&rustc_dir, config, progress)
|
|
|
|
.with_context(|| {
|
|
|
|
format!("Failed to read Cargo metadata for Rust sources")
|
|
|
|
})?,
|
|
|
|
)
|
2020-11-13 09:44:48 -06:00
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
2021-01-18 05:52:12 -06:00
|
|
|
let rustc_cfg = rustc_cfg::get(config.target.as_deref());
|
|
|
|
ProjectWorkspace::Cargo { cargo, sysroot, rustc, rustc_cfg }
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
|
2021-01-18 05:52:12 -06:00
|
|
|
pub fn load_inline(
|
|
|
|
project_json: ProjectJson,
|
|
|
|
target: Option<&str>,
|
|
|
|
) -> Result<ProjectWorkspace> {
|
2020-11-13 09:44:48 -06:00
|
|
|
let sysroot = match &project_json.sysroot_src {
|
|
|
|
Some(path) => Some(Sysroot::load(path)?),
|
|
|
|
None => None,
|
|
|
|
};
|
2021-01-18 05:52:12 -06:00
|
|
|
let rustc_cfg = rustc_cfg::get(target);
|
|
|
|
Ok(ProjectWorkspace::Json { project: project_json, sysroot, rustc_cfg })
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Returns the roots for the current `ProjectWorkspace`
|
|
|
|
/// The return type contains the path and whether or not
|
|
|
|
/// the root is a member of the current workspace
|
2021-01-28 09:33:02 -06:00
|
|
|
pub fn to_roots(&self, build_data: Option<&BuildDataResult>) -> Vec<PackageRoot> {
|
2020-11-13 09:44:48 -06:00
|
|
|
match self {
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Json { project, sysroot, rustc_cfg: _ } => project
|
2020-11-13 09:44:48 -06:00
|
|
|
.crates()
|
|
|
|
.map(|(_, krate)| PackageRoot {
|
|
|
|
is_member: krate.is_workspace_member,
|
|
|
|
include: krate.include.clone(),
|
|
|
|
exclude: krate.exclude.clone(),
|
|
|
|
})
|
|
|
|
.collect::<FxHashSet<_>>()
|
|
|
|
.into_iter()
|
|
|
|
.chain(sysroot.as_ref().into_iter().flat_map(|sysroot| {
|
|
|
|
sysroot.crates().map(move |krate| PackageRoot {
|
|
|
|
is_member: false,
|
|
|
|
include: vec![sysroot[krate].root_dir().to_path_buf()],
|
|
|
|
exclude: Vec::new(),
|
|
|
|
})
|
|
|
|
}))
|
|
|
|
.collect::<Vec<_>>(),
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Cargo { cargo, sysroot, rustc, rustc_cfg: _ } => cargo
|
2020-11-17 04:31:40 -06:00
|
|
|
.packages()
|
|
|
|
.map(|pkg| {
|
|
|
|
let is_member = cargo[pkg].is_member;
|
|
|
|
let pkg_root = cargo[pkg].root().to_path_buf();
|
|
|
|
|
|
|
|
let mut include = vec![pkg_root.clone()];
|
2021-01-28 09:33:02 -06:00
|
|
|
include.extend(
|
|
|
|
build_data
|
|
|
|
.and_then(|it| it.get(cargo.workspace_root()))
|
|
|
|
.and_then(|map| map.get(&cargo[pkg].id))
|
|
|
|
.and_then(|it| it.out_dir.clone()),
|
|
|
|
);
|
2020-11-17 04:31:40 -06:00
|
|
|
|
|
|
|
let mut exclude = vec![pkg_root.join(".git")];
|
|
|
|
if is_member {
|
|
|
|
exclude.push(pkg_root.join("target"));
|
|
|
|
} else {
|
|
|
|
exclude.push(pkg_root.join("tests"));
|
|
|
|
exclude.push(pkg_root.join("examples"));
|
|
|
|
exclude.push(pkg_root.join("benches"));
|
|
|
|
}
|
|
|
|
PackageRoot { is_member, include, exclude }
|
|
|
|
})
|
|
|
|
.chain(sysroot.crates().map(|krate| PackageRoot {
|
|
|
|
is_member: false,
|
|
|
|
include: vec![sysroot[krate].root_dir().to_path_buf()],
|
|
|
|
exclude: Vec::new(),
|
|
|
|
}))
|
|
|
|
.chain(rustc.into_iter().flat_map(|rustc| {
|
|
|
|
rustc.packages().map(move |krate| PackageRoot {
|
2020-11-13 09:44:48 -06:00
|
|
|
is_member: false,
|
2020-11-17 04:31:40 -06:00
|
|
|
include: vec![rustc[krate].root().to_path_buf()],
|
2020-11-13 09:44:48 -06:00
|
|
|
exclude: Vec::new(),
|
2020-11-17 04:31:40 -06:00
|
|
|
})
|
|
|
|
}))
|
|
|
|
.collect(),
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn n_packages(&self) -> usize {
|
|
|
|
match self {
|
|
|
|
ProjectWorkspace::Json { project, .. } => project.n_crates(),
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Cargo { cargo, sysroot, rustc, .. } => {
|
2020-11-13 09:44:48 -06:00
|
|
|
let rustc_package_len = rustc.as_ref().map_or(0, |rc| rc.packages().len());
|
|
|
|
cargo.packages().len() + sysroot.crates().len() + rustc_package_len
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn to_crate_graph(
|
|
|
|
&self,
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data: Option<&BuildDataResult>,
|
2020-12-07 10:16:50 -06:00
|
|
|
proc_macro_client: Option<&ProcMacroClient>,
|
2020-11-13 09:44:48 -06:00
|
|
|
load: &mut dyn FnMut(&AbsPath) -> Option<FileId>,
|
|
|
|
) -> CrateGraph {
|
2021-01-18 04:25:57 -06:00
|
|
|
let _p = profile::span("ProjectWorkspace::to_crate_graph");
|
2020-12-07 10:16:50 -06:00
|
|
|
let proc_macro_loader = |path: &Path| match proc_macro_client {
|
|
|
|
Some(client) => client.by_dylib_path(path),
|
|
|
|
None => Vec::new(),
|
|
|
|
};
|
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
let mut crate_graph = match self {
|
2021-01-18 05:52:12 -06:00
|
|
|
ProjectWorkspace::Json { project, sysroot, rustc_cfg } => project_json_to_crate_graph(
|
|
|
|
rustc_cfg.clone(),
|
|
|
|
&proc_macro_loader,
|
|
|
|
load,
|
|
|
|
project,
|
|
|
|
sysroot,
|
|
|
|
),
|
|
|
|
ProjectWorkspace::Cargo { cargo, sysroot, rustc, rustc_cfg } => cargo_to_crate_graph(
|
|
|
|
rustc_cfg.clone(),
|
|
|
|
&proc_macro_loader,
|
|
|
|
load,
|
|
|
|
cargo,
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data.and_then(|it| it.get(cargo.workspace_root())),
|
2021-01-18 05:52:12 -06:00
|
|
|
sysroot,
|
|
|
|
rustc,
|
2021-01-28 09:33:02 -06:00
|
|
|
rustc.as_ref().zip(build_data).and_then(|(it, map)| map.get(it.workspace_root())),
|
2021-01-18 05:52:12 -06:00
|
|
|
),
|
2020-11-17 06:35:25 -06:00
|
|
|
};
|
|
|
|
if crate_graph.patch_cfg_if() {
|
|
|
|
log::debug!("Patched std to depend on cfg-if")
|
|
|
|
} else {
|
|
|
|
log::debug!("Did not patch std to depend on cfg-if")
|
|
|
|
}
|
|
|
|
crate_graph
|
|
|
|
}
|
2021-01-28 09:33:02 -06:00
|
|
|
|
|
|
|
pub fn collect_build_data_configs(&self, collector: &mut BuildDataCollector) {
|
|
|
|
match self {
|
|
|
|
ProjectWorkspace::Cargo { cargo, rustc, .. } => {
|
|
|
|
collector.add_config(&cargo.workspace_root(), cargo.build_data_config().clone());
|
|
|
|
if let Some(rustc) = rustc {
|
|
|
|
collector.add_config(rustc.workspace_root(), rustc.build_data_config().clone());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
fn project_json_to_crate_graph(
|
2021-01-18 05:52:12 -06:00
|
|
|
rustc_cfg: Vec<CfgFlag>,
|
2020-12-07 10:16:50 -06:00
|
|
|
proc_macro_loader: &dyn Fn(&Path) -> Vec<ProcMacro>,
|
2020-11-17 06:35:25 -06:00
|
|
|
load: &mut dyn FnMut(&AbsPath) -> Option<FileId>,
|
|
|
|
project: &ProjectJson,
|
|
|
|
sysroot: &Option<Sysroot>,
|
|
|
|
) -> CrateGraph {
|
|
|
|
let mut crate_graph = CrateGraph::default();
|
|
|
|
let sysroot_deps = sysroot
|
|
|
|
.as_ref()
|
2021-01-18 05:52:12 -06:00
|
|
|
.map(|sysroot| sysroot_to_crate_graph(&mut crate_graph, sysroot, rustc_cfg.clone(), load));
|
2020-11-17 06:35:25 -06:00
|
|
|
|
2021-01-18 05:52:12 -06:00
|
|
|
let mut cfg_cache: FxHashMap<&str, Vec<CfgFlag>> = FxHashMap::default();
|
2020-11-17 06:35:25 -06:00
|
|
|
let crates: FxHashMap<CrateId, CrateId> = project
|
|
|
|
.crates()
|
|
|
|
.filter_map(|(crate_id, krate)| {
|
|
|
|
let file_path = &krate.root_module;
|
|
|
|
let file_id = load(&file_path)?;
|
|
|
|
Some((crate_id, krate, file_id))
|
|
|
|
})
|
|
|
|
.map(|(crate_id, krate, file_id)| {
|
|
|
|
let env = krate.env.clone().into_iter().collect();
|
2020-12-07 10:16:50 -06:00
|
|
|
let proc_macro = krate.proc_macro_dylib_path.clone().map(|it| proc_macro_loader(&it));
|
2020-11-17 06:35:25 -06:00
|
|
|
|
2021-01-18 05:52:12 -06:00
|
|
|
let target_cfgs = match krate.target.as_deref() {
|
|
|
|
Some(target) => {
|
|
|
|
cfg_cache.entry(target).or_insert_with(|| rustc_cfg::get(Some(target)))
|
|
|
|
}
|
|
|
|
None => &rustc_cfg,
|
|
|
|
};
|
2020-11-17 06:35:25 -06:00
|
|
|
|
|
|
|
let mut cfg_options = CfgOptions::default();
|
|
|
|
cfg_options.extend(target_cfgs.iter().chain(krate.cfg.iter()).cloned());
|
|
|
|
(
|
|
|
|
crate_id,
|
|
|
|
crate_graph.add_crate_root(
|
|
|
|
file_id,
|
|
|
|
krate.edition,
|
|
|
|
krate.display_name.clone(),
|
|
|
|
cfg_options,
|
|
|
|
env,
|
|
|
|
proc_macro.unwrap_or_default(),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
})
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
for (from, krate) in project.crates() {
|
|
|
|
if let Some(&from) = crates.get(&from) {
|
|
|
|
if let Some((public_deps, _proc_macro)) = &sysroot_deps {
|
|
|
|
for (name, to) in public_deps.iter() {
|
|
|
|
add_dep(&mut crate_graph, from, name.clone(), *to)
|
|
|
|
}
|
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
for dep in &krate.deps {
|
|
|
|
if let Some(&to) = crates.get(&dep.crate_id) {
|
|
|
|
add_dep(&mut crate_graph, from, dep.name.clone(), to)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
crate_graph
|
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
fn cargo_to_crate_graph(
|
2021-01-18 05:52:12 -06:00
|
|
|
rustc_cfg: Vec<CfgFlag>,
|
2020-12-07 10:16:50 -06:00
|
|
|
proc_macro_loader: &dyn Fn(&Path) -> Vec<ProcMacro>,
|
2020-11-17 06:35:25 -06:00
|
|
|
load: &mut dyn FnMut(&AbsPath) -> Option<FileId>,
|
|
|
|
cargo: &CargoWorkspace,
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data_map: Option<&BuildDataMap>,
|
2020-11-17 06:35:25 -06:00
|
|
|
sysroot: &Sysroot,
|
|
|
|
rustc: &Option<CargoWorkspace>,
|
2021-01-28 09:33:02 -06:00
|
|
|
rustc_build_data_map: Option<&BuildDataMap>,
|
2020-11-17 06:35:25 -06:00
|
|
|
) -> CrateGraph {
|
2021-01-18 04:25:57 -06:00
|
|
|
let _p = profile::span("cargo_to_crate_graph");
|
2020-11-17 06:35:25 -06:00
|
|
|
let mut crate_graph = CrateGraph::default();
|
|
|
|
let (public_deps, libproc_macro) =
|
2021-01-18 05:52:12 -06:00
|
|
|
sysroot_to_crate_graph(&mut crate_graph, sysroot, rustc_cfg.clone(), load);
|
2020-11-17 06:35:25 -06:00
|
|
|
|
|
|
|
let mut cfg_options = CfgOptions::default();
|
2021-01-18 05:52:12 -06:00
|
|
|
cfg_options.extend(rustc_cfg);
|
2020-11-17 06:35:25 -06:00
|
|
|
|
|
|
|
let mut pkg_to_lib_crate = FxHashMap::default();
|
|
|
|
|
|
|
|
// Add test cfg for non-sysroot crates
|
|
|
|
cfg_options.insert_atom("test".into());
|
|
|
|
cfg_options.insert_atom("debug_assertions".into());
|
|
|
|
|
|
|
|
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 cargo[pkg].targets.iter() {
|
2020-11-17 08:25:46 -06:00
|
|
|
if let Some(file_id) = load(&cargo[tgt].root) {
|
|
|
|
let crate_id = add_target_crate_root(
|
|
|
|
&mut crate_graph,
|
|
|
|
&cargo[pkg],
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data_map.and_then(|it| it.get(&cargo[pkg].id)),
|
2020-11-17 08:25:46 -06:00
|
|
|
&cfg_options,
|
2020-12-07 10:16:50 -06:00
|
|
|
proc_macro_loader,
|
2020-11-17 08:25:46 -06:00
|
|
|
file_id,
|
|
|
|
);
|
2020-11-17 06:35:25 -06:00
|
|
|
if cargo[tgt].kind == TargetKind::Lib {
|
|
|
|
lib_tgt = Some((crate_id, cargo[tgt].name.clone()));
|
|
|
|
pkg_to_lib_crate.insert(pkg, crate_id);
|
|
|
|
}
|
|
|
|
if cargo[tgt].is_proc_macro {
|
|
|
|
if let Some(proc_macro) = libproc_macro {
|
|
|
|
add_dep(
|
2020-11-13 09:44:48 -06:00
|
|
|
&mut crate_graph,
|
2020-11-17 06:35:25 -06:00
|
|
|
crate_id,
|
|
|
|
CrateName::new("proc_macro").unwrap(),
|
|
|
|
proc_macro,
|
|
|
|
);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
pkg_crates.entry(pkg).or_insert_with(Vec::new).push(crate_id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set deps to the core, std and to the lib target of the current package
|
|
|
|
for &from in pkg_crates.get(&pkg).into_iter().flatten() {
|
|
|
|
if let Some((to, name)) = lib_tgt.clone() {
|
|
|
|
if to != from {
|
|
|
|
// For root projects with dashes in their name,
|
|
|
|
// cargo metadata does not do any normalization,
|
|
|
|
// so we do it ourselves currently
|
|
|
|
let name = CrateName::normalize_dashes(&name);
|
|
|
|
add_dep(&mut crate_graph, from, name, to);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
}
|
|
|
|
for (name, krate) in public_deps.iter() {
|
|
|
|
add_dep(&mut crate_graph, from, name.clone(), *krate);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2020-11-17 06:35:25 -06:00
|
|
|
// Now add a dep edge from all targets of upstream to the lib
|
|
|
|
// target of downstream.
|
|
|
|
for pkg in cargo.packages() {
|
|
|
|
for dep in cargo[pkg].dependencies.iter() {
|
|
|
|
let name = CrateName::new(&dep.name).unwrap();
|
|
|
|
if let Some(&to) = pkg_to_lib_crate.get(&dep.pkg) {
|
|
|
|
for &from in pkg_crates.get(&pkg).into_iter().flatten() {
|
|
|
|
add_dep(&mut crate_graph, from, name.clone(), to)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut rustc_pkg_crates = FxHashMap::default();
|
|
|
|
|
|
|
|
// If the user provided a path to rustc sources, we add all the rustc_private crates
|
2021-03-07 04:18:01 -06:00
|
|
|
// and create dependencies on them for the crates which opt-in to that
|
2020-11-17 06:35:25 -06:00
|
|
|
if let Some(rustc_workspace) = rustc {
|
2021-03-07 04:18:01 -06:00
|
|
|
// rustc-dev crates start from 'rustc_driver'
|
|
|
|
// Therefore, we collect all crates which are transitive dependencies of rustc_driver
|
|
|
|
if let Some(root_pkg) = rustc_workspace
|
|
|
|
.packages()
|
|
|
|
.find(|package| rustc_workspace[*package].name == "rustc_driver")
|
|
|
|
{
|
|
|
|
let mut queue = VecDeque::new();
|
|
|
|
queue.push_back(root_pkg);
|
|
|
|
while let Some(pkg) = queue.pop_front() {
|
2021-03-07 05:17:14 -06:00
|
|
|
if rustc_pkg_crates.contains_key(&pkg) {
|
|
|
|
continue;
|
|
|
|
}
|
2021-03-07 04:18:01 -06:00
|
|
|
for dep in &rustc_workspace[pkg].dependencies {
|
|
|
|
queue.push_back(dep.pkg);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2021-03-07 04:18:01 -06:00
|
|
|
for &tgt in rustc_workspace[pkg].targets.iter() {
|
|
|
|
if rustc_workspace[tgt].kind != TargetKind::Lib {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if let Some(file_id) = load(&rustc_workspace[tgt].root) {
|
|
|
|
let crate_id = add_target_crate_root(
|
|
|
|
&mut crate_graph,
|
|
|
|
&rustc_workspace[pkg],
|
|
|
|
rustc_build_data_map.and_then(|it| it.get(&rustc_workspace[pkg].id)),
|
|
|
|
&cfg_options,
|
|
|
|
proc_macro_loader,
|
|
|
|
file_id,
|
|
|
|
);
|
|
|
|
pkg_to_lib_crate.insert(pkg, crate_id);
|
|
|
|
// Add dependencies on the core / std / alloc for rustc
|
|
|
|
for (name, krate) in public_deps.iter() {
|
|
|
|
add_dep(&mut crate_graph, crate_id, name.clone(), *krate);
|
|
|
|
}
|
|
|
|
rustc_pkg_crates.entry(pkg).or_insert_with(Vec::new).push(crate_id);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Now add a dep edge from all targets of upstream to the lib
|
|
|
|
// target of downstream.
|
2021-03-07 05:17:14 -06:00
|
|
|
for pkg in rustc_pkg_crates.keys().copied() {
|
2020-11-17 06:35:25 -06:00
|
|
|
for dep in rustc_workspace[pkg].dependencies.iter() {
|
|
|
|
let name = CrateName::new(&dep.name).unwrap();
|
|
|
|
if let Some(&to) = pkg_to_lib_crate.get(&dep.pkg) {
|
|
|
|
for &from in rustc_pkg_crates.get(&pkg).into_iter().flatten() {
|
|
|
|
add_dep(&mut crate_graph, from, name.clone(), to);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
|
2021-03-07 04:18:01 -06:00
|
|
|
// Add dependencies for all crates which opt in to rustc_private libraries
|
2020-11-17 06:35:25 -06:00
|
|
|
for dep in rustc_workspace.packages() {
|
|
|
|
let name = CrateName::normalize_dashes(&rustc_workspace[dep].name);
|
|
|
|
|
|
|
|
if let Some(&to) = pkg_to_lib_crate.get(&dep) {
|
|
|
|
for pkg in cargo.packages() {
|
2021-03-06 06:17:22 -06:00
|
|
|
let package = &cargo[pkg];
|
|
|
|
if matches!(
|
|
|
|
(package.is_member, package.metadata.rustc_private),
|
|
|
|
(true, Some(false)) | (false, Some(false)) | (false, None)
|
|
|
|
) {
|
2020-11-17 06:35:25 -06:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
for &from in pkg_crates.get(&pkg).into_iter().flatten() {
|
2021-03-07 04:18:01 -06:00
|
|
|
// Avoid creating duplicate dependencies
|
2021-03-06 07:56:42 -06:00
|
|
|
if !crate_graph[from].dependencies.iter().any(|d| d.name == name) {
|
|
|
|
add_dep(&mut crate_graph, from, name.clone(), to);
|
|
|
|
} else {
|
2021-03-07 04:18:01 -06:00
|
|
|
eprintln!(
|
|
|
|
"Skipped {} for {:?}",
|
|
|
|
&name, &crate_graph[from].display_name
|
|
|
|
);
|
2021-03-06 07:56:42 -06:00
|
|
|
}
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2021-03-07 05:17:14 -06:00
|
|
|
} else {
|
|
|
|
eprintln!("No cargo workspace");
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 06:35:25 -06:00
|
|
|
crate_graph
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
fn add_target_crate_root(
|
|
|
|
crate_graph: &mut CrateGraph,
|
|
|
|
pkg: &cargo_workspace::PackageData,
|
2021-01-28 09:33:02 -06:00
|
|
|
build_data: Option<&BuildData>,
|
2020-11-13 09:44:48 -06:00
|
|
|
cfg_options: &CfgOptions,
|
2020-12-07 10:16:50 -06:00
|
|
|
proc_macro_loader: &dyn Fn(&Path) -> Vec<ProcMacro>,
|
2020-11-17 08:25:46 -06:00
|
|
|
file_id: FileId,
|
|
|
|
) -> CrateId {
|
|
|
|
let edition = pkg.edition;
|
|
|
|
let cfg_options = {
|
|
|
|
let mut opts = cfg_options.clone();
|
2021-01-21 05:12:19 -06:00
|
|
|
for feature in pkg.active_features.iter() {
|
2020-11-17 08:25:46 -06:00
|
|
|
opts.insert_key_value("feature".into(), feature.into());
|
|
|
|
}
|
2021-01-28 09:33:02 -06:00
|
|
|
if let Some(cfgs) = build_data.as_ref().map(|it| &it.cfgs) {
|
|
|
|
opts.extend(cfgs.iter().cloned());
|
|
|
|
}
|
2020-11-17 08:25:46 -06:00
|
|
|
opts
|
|
|
|
};
|
2020-12-07 13:52:31 -06:00
|
|
|
|
2020-11-17 08:25:46 -06:00
|
|
|
let mut env = Env::default();
|
2021-01-28 09:33:02 -06:00
|
|
|
if let Some(envs) = build_data.map(|it| &it.envs) {
|
|
|
|
for (k, v) in envs {
|
|
|
|
env.set(k, v.clone());
|
|
|
|
}
|
2020-12-07 13:52:31 -06:00
|
|
|
}
|
|
|
|
|
2021-01-28 09:33:02 -06:00
|
|
|
let proc_macro = build_data
|
2021-01-22 05:11:01 -06:00
|
|
|
.as_ref()
|
2021-01-28 09:33:02 -06:00
|
|
|
.and_then(|it| it.proc_macro_dylib_path.as_ref())
|
2021-01-22 05:11:01 -06:00
|
|
|
.map(|it| proc_macro_loader(&it))
|
|
|
|
.unwrap_or_default();
|
2020-11-17 08:25:46 -06:00
|
|
|
|
|
|
|
let display_name = CrateDisplayName::from_canonical_name(pkg.name.clone());
|
|
|
|
let crate_id = crate_graph.add_crate_root(
|
|
|
|
file_id,
|
|
|
|
edition,
|
|
|
|
Some(display_name),
|
|
|
|
cfg_options,
|
|
|
|
env,
|
2020-12-12 11:27:09 -06:00
|
|
|
proc_macro,
|
2020-11-17 08:25:46 -06:00
|
|
|
);
|
|
|
|
|
|
|
|
crate_id
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
2020-11-17 08:25:46 -06:00
|
|
|
|
2020-11-13 09:44:48 -06:00
|
|
|
fn sysroot_to_crate_graph(
|
|
|
|
crate_graph: &mut CrateGraph,
|
|
|
|
sysroot: &Sysroot,
|
2021-01-18 05:52:12 -06:00
|
|
|
rustc_cfg: Vec<CfgFlag>,
|
2020-11-13 09:44:48 -06:00
|
|
|
load: &mut dyn FnMut(&AbsPath) -> Option<FileId>,
|
|
|
|
) -> (Vec<(CrateName, CrateId)>, Option<CrateId>) {
|
2021-01-18 04:25:57 -06:00
|
|
|
let _p = profile::span("sysroot_to_crate_graph");
|
2020-11-13 09:44:48 -06:00
|
|
|
let mut cfg_options = CfgOptions::default();
|
2021-01-18 05:52:12 -06:00
|
|
|
cfg_options.extend(rustc_cfg);
|
2020-11-17 04:50:54 -06:00
|
|
|
let sysroot_crates: FxHashMap<SysrootCrate, CrateId> = sysroot
|
2020-11-13 09:44:48 -06:00
|
|
|
.crates()
|
|
|
|
.filter_map(|krate| {
|
|
|
|
let file_id = load(&sysroot[krate].root)?;
|
|
|
|
|
|
|
|
let env = Env::default();
|
|
|
|
let proc_macro = vec![];
|
2020-11-17 04:50:54 -06:00
|
|
|
let display_name = CrateDisplayName::from_canonical_name(sysroot[krate].name.clone());
|
2020-11-13 09:44:48 -06:00
|
|
|
let crate_id = crate_graph.add_crate_root(
|
|
|
|
file_id,
|
|
|
|
Edition::Edition2018,
|
2020-11-17 04:50:54 -06:00
|
|
|
Some(display_name),
|
2020-11-13 09:44:48 -06:00
|
|
|
cfg_options.clone(),
|
|
|
|
env,
|
|
|
|
proc_macro,
|
|
|
|
);
|
|
|
|
Some((krate, crate_id))
|
|
|
|
})
|
|
|
|
.collect();
|
|
|
|
|
|
|
|
for from in sysroot.crates() {
|
|
|
|
for &to in sysroot[from].deps.iter() {
|
|
|
|
let name = CrateName::new(&sysroot[to].name).unwrap();
|
|
|
|
if let (Some(&from), Some(&to)) = (sysroot_crates.get(&from), sysroot_crates.get(&to)) {
|
2020-11-17 04:50:54 -06:00
|
|
|
add_dep(crate_graph, from, name, to);
|
2020-11-13 09:44:48 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let public_deps = sysroot
|
|
|
|
.public_deps()
|
|
|
|
.map(|(name, idx)| (CrateName::new(name).unwrap(), sysroot_crates[&idx]))
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
|
|
|
|
let libproc_macro = sysroot.proc_macro().and_then(|it| sysroot_crates.get(&it).copied());
|
|
|
|
(public_deps, libproc_macro)
|
|
|
|
}
|
|
|
|
|
2020-11-17 04:50:54 -06:00
|
|
|
fn add_dep(graph: &mut CrateGraph, from: CrateId, name: CrateName, to: CrateId) {
|
|
|
|
if let Err(err) = graph.add_dep(from, name, to) {
|
|
|
|
log::error!("{}", err)
|
|
|
|
}
|
|
|
|
}
|