rust-analyzer/crates/base-db/src/input.rs

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

1053 lines
32 KiB
Rust
Raw Normal View History

//! This module specifies the input to rust-analyzer. In some sense, this is
//! **the** most important module, because all other fancy stuff is strictly
//! derived from this input.
//!
//! Note that neither this module, nor any other part of the analyzer's core do
2020-02-18 11:33:16 +00:00
//! actual IO. See `vfs` and `project_model` in the `rust-analyzer` crate for how
//! actual IO is done and lowered to input.
use std::{fmt, mem, ops, str::FromStr};
2023-11-08 15:51:20 +00:00
use cfg::CfgOptions;
use la_arena::{Arena, Idx};
use rustc_hash::{FxHashMap, FxHashSet};
use semver::Version;
2020-08-12 16:26:51 +00:00
use syntax::SmolStr;
2023-05-02 14:12:22 +00:00
use triomphe::Arc;
2023-03-25 17:06:06 +00:00
use vfs::{file_set::FileSet, AbsPathBuf, AnchoredPath, FileId, VfsPath};
2018-10-25 14:52:50 +00:00
2023-03-26 06:39:28 +00:00
// Map from crate id to the name of the crate and path of the proc-macro. If the value is `None`,
// then the crate for the proc-macro hasn't been build yet as the build data is missing.
2023-03-29 19:29:32 +00:00
pub type ProcMacroPaths = FxHashMap<CrateId, Result<(Option<String>, AbsPathBuf), String>>;
2018-12-20 10:47:32 +00:00
/// Files are grouped into source roots. A source root is a directory on the
/// file systems which is watched for changes. Typically it corresponds to a
/// Rust crate. Source roots *might* be nested: in this case, a file belongs to
/// the nearest enclosing source root. Paths to files are always relative to a
/// source root, and the analyzer does not know the root path of the source root at
/// all. So, a file from one source root can't refer to a file in another source
2018-12-20 10:47:32 +00:00
/// root by path.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
pub struct SourceRootId(pub u32);
#[derive(Clone, Debug, PartialEq, Eq)]
2018-12-20 10:47:32 +00:00
pub struct SourceRoot {
/// Sysroot or crates.io library.
///
/// Libraries are considered mostly immutable, this assumption is used to
/// optimize salsa's query structure
pub is_library: bool,
file_set: FileSet,
2018-12-20 10:47:32 +00:00
}
2018-10-25 14:52:50 +00:00
impl SourceRoot {
2020-06-11 09:04:09 +00:00
pub fn new_local(file_set: FileSet) -> SourceRoot {
SourceRoot { is_library: false, file_set }
2019-09-05 19:36:04 +00:00
}
2020-06-11 09:04:09 +00:00
pub fn new_library(file_set: FileSet) -> SourceRoot {
SourceRoot { is_library: true, file_set }
2019-09-05 19:36:04 +00:00
}
pub fn path_for_file(&self, file: &FileId) -> Option<&VfsPath> {
self.file_set.path_for_file(file)
}
pub fn file_for_path(&self, path: &VfsPath) -> Option<&FileId> {
self.file_set.file_for_path(path)
}
pub fn resolve_path(&self, path: AnchoredPath<'_>) -> Option<FileId> {
self.file_set.resolve_path(path)
}
2020-06-11 09:04:09 +00:00
pub fn iter(&self) -> impl Iterator<Item = FileId> + '_ {
self.file_set.iter()
2019-10-11 08:37:54 +00:00
}
}
2018-12-20 10:47:32 +00:00
/// `CrateGraph` is a bit of information which turns a set of text files into a
/// number of Rust crates.
///
/// Each crate is defined by the `FileId` of its root module, the set of enabled
/// `cfg` flags and the set of dependencies.
///
/// Note that, due to cfg's, there might be several crates for a single `FileId`!
///
/// For the purposes of analysis, a crate does not have a name. Instead, names
/// are specified on dependency edges. That is, a crate might be known under
/// different names in different dependent crates.
2018-12-20 10:47:32 +00:00
///
/// Note that `CrateGraph` is build-system agnostic: it's a concept of the Rust
/// language proper, not a concept of the build system. In practice, we get
2018-12-20 10:47:32 +00:00
/// `CrateGraph` by lowering `cargo metadata` output.
///
/// `CrateGraph` is `!Serialize` by design, see
2022-07-08 13:44:49 +00:00
/// <https://github.com/rust-lang/rust-analyzer/blob/master/docs/dev/architecture.md#serialization>
#[derive(Clone, Default)]
2018-10-25 14:52:50 +00:00
pub struct CrateGraph {
arena: Arena<CrateData>,
2018-10-25 14:52:50 +00:00
}
impl fmt::Debug for CrateGraph {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_map()
.entries(self.arena.iter().map(|(id, data)| (u32::from(id.into_raw()), data)))
.finish()
}
}
2018-12-20 10:47:32 +00:00
pub type CrateId = Idx<CrateData>;
#[derive(Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
2020-02-05 09:53:54 +00:00
pub struct CrateName(SmolStr);
2020-02-05 10:47:28 +00:00
impl CrateName {
2020-03-08 13:26:57 +00:00
/// Creates a crate name, checking for dashes in the string provided.
2020-02-05 10:47:28 +00:00
/// Dashes are not allowed in the crate names,
/// hence the input string is returned as `Err` for those cases.
pub fn new(name: &str) -> Result<CrateName, &str> {
if name.contains('-') {
Err(name)
} else {
Ok(Self(SmolStr::new(name)))
}
}
2020-03-08 13:26:57 +00:00
/// Creates a crate name, unconditionally replacing the dashes with underscores.
2020-02-05 10:47:28 +00:00
pub fn normalize_dashes(name: &str) -> CrateName {
Self(SmolStr::new(name.replace('-', "_")))
2020-02-05 09:53:54 +00:00
}
pub fn as_smol_str(&self) -> &SmolStr {
&self.0
}
2020-03-16 10:03:43 +00:00
}
2020-03-16 09:47:52 +00:00
2020-06-11 09:30:06 +00:00
impl fmt::Display for CrateName {
2022-07-20 13:02:08 +00:00
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2022-03-21 08:43:36 +00:00
self.0.fmt(f)
2020-03-16 09:47:52 +00:00
}
2020-02-05 09:53:54 +00:00
}
2020-07-01 07:53:53 +00:00
impl ops::Deref for CrateName {
type Target = str;
2020-10-20 14:00:51 +00:00
fn deref(&self) -> &str {
&self.0
2020-10-20 14:00:51 +00:00
}
}
/// Origin of the crates.
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
2021-11-22 17:44:46 +00:00
pub enum CrateOrigin {
/// Crates that are from the rustc workspace.
2023-03-31 07:10:18 +00:00
Rustc { name: String },
/// Crates that are workspace members.
2023-03-31 07:10:18 +00:00
Local { repo: Option<String>, name: Option<String> },
/// Crates that are non member libraries.
Library { repo: Option<String>, name: String },
2021-11-22 17:44:46 +00:00
/// Crates that are provided by the language, like std, core, proc-macro, ...
Lang(LangCrateOrigin),
2021-11-22 17:44:46 +00:00
}
2023-06-05 11:27:19 +00:00
impl CrateOrigin {
pub fn is_local(&self) -> bool {
matches!(self, CrateOrigin::Local { .. })
}
pub fn is_lib(&self) -> bool {
matches!(self, CrateOrigin::Library { .. })
}
2023-06-05 11:27:19 +00:00
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum LangCrateOrigin {
Alloc,
Core,
ProcMacro,
Std,
Test,
Other,
2021-11-22 17:44:46 +00:00
}
2022-04-10 10:42:16 +00:00
impl From<&str> for LangCrateOrigin {
fn from(s: &str) -> Self {
match s {
"alloc" => LangCrateOrigin::Alloc,
"core" => LangCrateOrigin::Core,
"proc-macro" => LangCrateOrigin::ProcMacro,
"std" => LangCrateOrigin::Std,
"test" => LangCrateOrigin::Test,
_ => LangCrateOrigin::Other,
}
}
}
impl fmt::Display for LangCrateOrigin {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let text = match self {
LangCrateOrigin::Alloc => "alloc",
LangCrateOrigin::Core => "core",
LangCrateOrigin::ProcMacro => "proc_macro",
LangCrateOrigin::Std => "std",
LangCrateOrigin::Test => "test",
LangCrateOrigin::Other => "other",
};
f.write_str(text)
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
2020-10-20 15:04:38 +00:00
pub struct CrateDisplayName {
// The name we use to display various paths (with `_`).
crate_name: CrateName,
// The name as specified in Cargo.toml (with `-`).
canonical_name: String,
}
2020-10-20 14:00:51 +00:00
impl CrateDisplayName {
pub fn canonical_name(&self) -> &str {
&self.canonical_name
}
pub fn crate_name(&self) -> &CrateName {
&self.crate_name
}
}
2020-10-20 14:00:51 +00:00
impl From<CrateName> for CrateDisplayName {
2020-10-20 15:04:38 +00:00
fn from(crate_name: CrateName) -> CrateDisplayName {
let canonical_name = crate_name.to_string();
CrateDisplayName { crate_name, canonical_name }
2020-10-20 14:00:51 +00:00
}
}
impl fmt::Display for CrateDisplayName {
2022-07-20 13:02:08 +00:00
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
2022-03-21 08:43:36 +00:00
self.crate_name.fmt(f)
2020-10-20 14:00:51 +00:00
}
}
impl ops::Deref for CrateDisplayName {
type Target = str;
fn deref(&self) -> &str {
&self.crate_name
2020-10-20 15:04:38 +00:00
}
}
impl CrateDisplayName {
pub fn from_canonical_name(canonical_name: String) -> CrateDisplayName {
let crate_name = CrateName::normalize_dashes(&canonical_name);
CrateDisplayName { crate_name, canonical_name }
2020-07-01 07:53:53 +00:00
}
}
pub type TargetLayoutLoadResult = Result<Arc<str>, Arc<str>>;
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash, PartialOrd, Ord)]
2023-03-31 07:10:18 +00:00
pub enum ReleaseChannel {
Stable,
Beta,
Nightly,
}
impl ReleaseChannel {
pub fn as_str(self) -> &'static str {
match self {
ReleaseChannel::Stable => "stable",
ReleaseChannel::Beta => "beta",
ReleaseChannel::Nightly => "nightly",
}
}
pub fn from_str(str: &str) -> Option<Self> {
Some(match str {
"" | "stable" => ReleaseChannel::Stable,
2023-03-31 07:10:18 +00:00
"nightly" => ReleaseChannel::Nightly,
_ if str.starts_with("beta") => ReleaseChannel::Beta,
2023-03-31 07:10:18 +00:00
_ => return None,
})
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
2020-03-09 09:26:46 +00:00
pub struct CrateData {
pub root_file_id: FileId,
pub edition: Edition,
pub version: Option<String>,
/// A name used in the package's project declaration: for Cargo projects,
Fix warnings when running `cargo doc --document-private-items` These were the warnings previously: ``` warning: could not parse code block as Rust code --> crates/stdx/src/lib.rs:137:9 | 137 | /// ∀ x in slice[..idx]: pred(x) | _________^ 138 | | /// && ∀ x in slice[idx..]: !pred(x) | |____^ | = note: error from rustc: unknown start of token: \u{2200} warning: 1 warning emitted warning: unresolved link to `package` --> crates/base_db/src/input.rs:181:15 | 181 | /// it's [package].name, can be different for other project types or even | ^^^^^^^ no item named `package` in scope | = note: `#[warn(broken_intra_doc_links)]` on by default = help: to escape `[` and `]` characters, add '\' before them like `\[` or `\]` warning: unresolved link to `package` --> crates/base_db/src/input.rs:181:15 | 181 | /// it's [package].name, can be different for other project types or even | ^^^^^^^ no item named `package` in scope | = help: to escape `[` and `]` characters, add '\' before them like `\[` or `\]` warning: 2 warnings emitted warning: unresolved link to `package` --> crates/base_db/src/input.rs:181:15 | 181 | /// it's [package].name, can be different for other project types or even | ^^^^^^^ no item named `package` in scope | = note: `#[warn(broken_intra_doc_links)]` on by default = help: to escape `[` and `]` characters, add '\' before them like `\[` or `\]` warning: unresolved link to `package` --> crates/base_db/src/input.rs:181:15 | 181 | /// it's [package].name, can be different for other project types or even | ^^^^^^^ no item named `package` in scope | = help: to escape `[` and `]` characters, add '\' before them like `\[` or `\]` warning: 2 warnings emitted ``` This does *not* fix the following warning, because it is actually rust code and rustdoc is being over eager: ``` warning: Rust code block is empty --> crates/parser/src/grammar.rs:16:5 | 16 | //! ``` | _____^ 17 | | //! // test function_with_zero_parameters 18 | | //! // fn foo() {} 19 | | //! ``` | |_______^ | help: mark blocks that do not contain Rust code as text | 16 | //! ```text | ^^^^^^^ ``` https://github.com/rust-lang/rust/pull/79816 should make this configurable so the warning can be `allow`ed.
2021-01-18 21:44:40 +00:00
/// its `[package].name` can be different for other project types or even
/// absent (a dummy crate for the code snippet, for example).
///
/// For purposes of analysis, crates are anonymous (only names in
/// `Dependency` matters), this name should only be used for UI.
2020-10-20 15:04:38 +00:00
pub display_name: Option<CrateDisplayName>,
2020-03-09 10:14:51 +00:00
pub cfg_options: CfgOptions,
2023-03-31 12:10:33 +00:00
/// The cfg options that could be used by the crate
pub potential_cfg_options: Option<CfgOptions>,
2020-03-09 10:14:51 +00:00
pub env: Env,
2020-03-09 09:26:46 +00:00
pub dependencies: Vec<Dependency>,
2021-11-22 17:44:46 +00:00
pub origin: CrateOrigin,
pub is_proc_macro: bool,
2023-11-25 14:10:31 +00:00
// FIXME: These things should not be per crate! These are more per workspace crate graph level
// things. This info does need to be somewhat present though as to prevent deduplication from
// happening across different workspaces with different layouts.
2023-03-31 07:10:18 +00:00
pub target_layout: TargetLayoutLoadResult,
pub toolchain: Option<Version>,
}
impl CrateData {
2023-10-15 15:32:12 +00:00
/// Check if [`other`] is almost equal to [`self`] ignoring `CrateOrigin` value.
2023-11-08 15:51:20 +00:00
pub fn eq_ignoring_origin_and_deps(&self, other: &CrateData, ignore_dev_deps: bool) -> bool {
2023-10-15 15:32:12 +00:00
// This method has some obscure bits. These are mostly there to be compliant with
// some patches. References to the patches are given.
if self.root_file_id != other.root_file_id {
return false;
}
if self.display_name != other.display_name {
return false;
}
if self.is_proc_macro != other.is_proc_macro {
return false;
}
if self.edition != other.edition {
return false;
}
if self.version != other.version {
return false;
}
2023-11-21 13:05:31 +00:00
let mut opts = self.cfg_options.difference(&other.cfg_options);
if let Some(it) = opts.next() {
// Don't care if rust_analyzer CfgAtom is the only cfg in the difference set of self's and other's cfgs.
// https://github.com/rust-lang/rust-analyzer/blob/0840038f02daec6ba3238f05d8caa037d28701a0/crates/project-model/src/workspace.rs#L894
if it.to_string() != "rust_analyzer" {
return false;
}
2023-11-21 13:05:31 +00:00
if let Some(_) = opts.next() {
return false;
}
}
if self.env != other.env {
return false;
}
2023-11-08 15:51:20 +00:00
let slf_deps = self.dependencies.iter();
let other_deps = other.dependencies.iter();
if ignore_dev_deps {
return slf_deps
2023-11-08 15:51:20 +00:00
.clone()
2023-11-21 13:05:31 +00:00
.filter(|it| it.kind != DependencyKind::Dev)
.eq(other_deps.clone().filter(|it| it.kind != DependencyKind::Dev));
2023-11-08 15:51:20 +00:00
}
slf_deps.eq(other_deps)
}
pub fn channel(&self) -> Option<ReleaseChannel> {
self.toolchain.as_ref().and_then(|v| ReleaseChannel::from_str(&v.pre))
}
}
2021-01-01 16:22:23 +00:00
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
2019-02-10 21:34:29 +00:00
pub enum Edition {
Edition2015,
2021-01-01 16:22:23 +00:00
Edition2018,
Edition2021,
Edition2024,
2019-02-10 21:34:29 +00:00
}
impl Edition {
2023-01-31 10:39:25 +00:00
pub const CURRENT: Edition = Edition::Edition2021;
pub const DEFAULT: Edition = Edition::Edition2015;
}
#[derive(Default, Debug, Clone, PartialEq, Eq)]
pub struct Env {
entries: FxHashMap<String, String>,
2020-03-11 03:04:02 +00:00
}
2020-03-10 13:59:12 +00:00
2023-06-05 11:27:19 +00:00
impl Env {
pub fn new_for_test_fixture() -> Self {
Env {
entries: FxHashMap::from_iter([(
String::from("__ra_is_test_fixture"),
String::from("__ra_is_test_fixture"),
)]),
}
}
}
2023-11-08 15:51:20 +00:00
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
pub enum DependencyKind {
Normal,
Dev,
Build,
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
2018-12-05 13:01:18 +00:00
pub struct Dependency {
2018-12-08 22:05:49 +00:00
pub crate_id: CrateId,
2020-07-01 07:53:53 +00:00
pub name: CrateName,
kind: DependencyKind,
prelude: bool,
}
impl Dependency {
pub fn new(name: CrateName, crate_id: CrateId, kind: DependencyKind) -> Self {
Self { name, crate_id, prelude: true, kind }
}
pub fn with_prelude(
name: CrateName,
crate_id: CrateId,
prelude: bool,
kind: DependencyKind,
) -> Self {
Self { name, crate_id, prelude, kind }
}
/// Whether this dependency is to be added to the depending crate's extern prelude.
pub fn is_prelude(&self) -> bool {
self.prelude
}
2023-11-08 15:51:20 +00:00
pub fn kind(&self) -> DependencyKind {
self.kind
}
2018-12-08 21:51:06 +00:00
}
2018-12-05 13:01:18 +00:00
impl CrateGraph {
pub fn add_crate_root(
&mut self,
root_file_id: FileId,
edition: Edition,
2020-10-20 15:04:38 +00:00
display_name: Option<CrateDisplayName>,
version: Option<String>,
cfg_options: CfgOptions,
2023-03-31 12:10:33 +00:00
potential_cfg_options: Option<CfgOptions>,
env: Env,
is_proc_macro: bool,
2021-11-22 17:44:46 +00:00
origin: CrateOrigin,
target_layout: Result<Arc<str>, Arc<str>>,
toolchain: Option<Version>,
) -> CrateId {
2020-03-09 10:17:39 +00:00
let data = CrateData {
root_file_id,
2020-03-09 10:17:39 +00:00
edition,
version,
display_name,
2020-03-09 10:17:39 +00:00
cfg_options,
potential_cfg_options,
2020-03-09 10:17:39 +00:00
env,
dependencies: Vec::new(),
2021-11-22 17:44:46 +00:00
origin,
target_layout,
is_proc_macro,
toolchain,
2020-03-09 10:17:39 +00:00
};
self.arena.alloc(data)
2018-10-25 14:52:50 +00:00
}
2023-04-20 19:25:39 +00:00
/// Remove the crate from crate graph. If any crates depend on this crate, the dependency would be replaced
/// with the second input.
pub fn remove_and_replace(
&mut self,
id: CrateId,
replace_with: CrateId,
) -> Result<(), CyclicDependenciesError> {
for (x, data) in self.arena.iter() {
if x == id {
continue;
}
for edge in &data.dependencies {
if edge.crate_id == id {
self.check_cycle_after_dependency(edge.crate_id, replace_with)?;
}
}
}
// if everything was ok, start to replace
for (x, data) in self.arena.iter_mut() {
if x == id {
continue;
}
for edge in &mut data.dependencies {
if edge.crate_id == id {
edge.crate_id = replace_with;
}
}
}
Ok(())
}
pub fn add_dep(
&mut self,
from: CrateId,
dep: Dependency,
2019-11-22 11:12:45 +00:00
) -> Result<(), CyclicDependenciesError> {
2021-04-22 18:48:17 +00:00
let _p = profile::span("add_dep");
2023-04-20 19:25:39 +00:00
self.check_cycle_after_dependency(from, dep.crate_id)?;
2023-11-21 13:05:31 +00:00
self.arena[from].add_dep(dep);
2023-04-20 19:25:39 +00:00
Ok(())
}
/// Check if adding a dep from `from` to `to` creates a cycle. To figure
/// that out, look for a path in the *opposite* direction, from `to` to
/// `from`.
fn check_cycle_after_dependency(
&self,
from: CrateId,
to: CrateId,
) -> Result<(), CyclicDependenciesError> {
if let Some(path) = self.find_path(&mut FxHashSet::default(), to, from) {
let path = path.into_iter().map(|it| (it, self[it].display_name.clone())).collect();
let err = CyclicDependenciesError { path };
2023-04-20 19:25:39 +00:00
assert!(err.from().0 == from && err.to().0 == to);
return Err(err);
}
2019-07-05 02:59:28 +00:00
Ok(())
2018-12-05 13:01:18 +00:00
}
2018-12-21 16:13:26 +00:00
pub fn is_empty(&self) -> bool {
self.arena.is_empty()
}
pub fn iter(&self) -> impl Iterator<Item = CrateId> + '_ {
self.arena.iter().map(|(idx, _)| idx)
}
// FIXME: used for `handle_hack_cargo_workspace`, should be removed later
#[doc(hidden)]
2023-06-05 11:27:19 +00:00
pub fn iter_mut(&mut self) -> impl Iterator<Item = (CrateId, &mut CrateData)> + '_ {
self.arena.iter_mut()
}
2021-03-23 09:58:48 +00:00
/// Returns an iterator over all transitive dependencies of the given crate,
/// including the crate itself.
pub fn transitive_deps(&self, of: CrateId) -> impl Iterator<Item = CrateId> {
let mut worklist = vec![of];
let mut deps = FxHashSet::default();
while let Some(krate) = worklist.pop() {
if !deps.insert(krate) {
continue;
}
worklist.extend(self[krate].dependencies.iter().map(|dep| dep.crate_id));
}
deps.into_iter()
}
/// Returns all transitive reverse dependencies of the given crate,
/// including the crate itself.
pub fn transitive_rev_deps(&self, of: CrateId) -> impl Iterator<Item = CrateId> {
let mut worklist = vec![of];
let mut rev_deps = FxHashSet::default();
rev_deps.insert(of);
2021-03-23 10:01:45 +00:00
let mut inverted_graph = FxHashMap::<_, Vec<_>>::default();
self.arena.iter().for_each(|(krate, data)| {
data.dependencies
.iter()
.for_each(|dep| inverted_graph.entry(dep.crate_id).or_default().push(krate))
});
while let Some(krate) = worklist.pop() {
if let Some(krate_rev_deps) = inverted_graph.get(&krate) {
krate_rev_deps
.iter()
.copied()
.filter(|&rev_dep| rev_deps.insert(rev_dep))
.for_each(|rev_dep| worklist.push(rev_dep));
}
}
rev_deps.into_iter()
}
/// Returns all crates in the graph, sorted in topological order (ie. dependencies of a crate
/// come before the crate itself).
pub fn crates_in_topological_order(&self) -> Vec<CrateId> {
let mut res = Vec::new();
let mut visited = FxHashSet::default();
for krate in self.iter() {
go(self, &mut visited, &mut res, krate);
}
return res;
fn go(
graph: &CrateGraph,
visited: &mut FxHashSet<CrateId>,
res: &mut Vec<CrateId>,
source: CrateId,
) {
if !visited.insert(source) {
return;
}
for dep in graph[source].dependencies.iter() {
go(graph, visited, res, dep.crate_id)
}
res.push(source)
}
}
2019-03-23 07:53:48 +00:00
// FIXME: this only finds one crate with the given root; we could have multiple
pub fn crate_id_for_crate_root(&self, file_id: FileId) -> Option<CrateId> {
let (crate_id, _) =
2020-03-09 09:26:46 +00:00
self.arena.iter().find(|(_crate_id, data)| data.root_file_id == file_id)?;
Some(crate_id)
2018-11-26 21:12:43 +00:00
}
pub fn sort_deps(&mut self) {
self.arena
.iter_mut()
.for_each(|(_, data)| data.dependencies.sort_by_key(|dep| dep.crate_id));
}
/// Extends this crate graph by adding a complete disjoint second crate
2023-03-25 17:06:06 +00:00
/// graph and adjust the ids in the [`ProcMacroPaths`] accordingly.
///
/// This will deduplicate the crates of the graph where possible.
/// Note that for deduplication to fully work, `self`'s crate dependencies must be sorted by crate id.
/// If the crate dependencies were sorted, the resulting graph from this `extend` call will also have the crate dependencies sorted.
pub fn extend(&mut self, mut other: CrateGraph, proc_macros: &mut ProcMacroPaths) {
let topo = other.crates_in_topological_order();
let mut id_map: FxHashMap<CrateId, CrateId> = FxHashMap::default();
for topo in topo {
let crate_data = &mut other.arena[topo];
2023-10-15 15:32:12 +00:00
crate_data.dependencies.iter_mut().for_each(|dep| dep.crate_id = id_map[&dep.crate_id]);
crate_data.dependencies.sort_by_key(|dep| dep.crate_id);
let res = self.arena.iter().find_map(|(id, data)| {
2023-11-08 15:51:20 +00:00
match (&data.origin, &crate_data.origin) {
(a, b) if a == b => {
if data.eq_ignoring_origin_and_deps(&crate_data, false) {
return Some((id, false));
}
}
(a @ CrateOrigin::Local { .. }, CrateOrigin::Library { .. })
| (a @ CrateOrigin::Library { .. }, CrateOrigin::Local { .. }) => {
2023-11-21 13:05:31 +00:00
// If the origins differ, check if the two crates are equal without
// considering the dev dependencies, if they are, they most likely are in
// different loaded workspaces which may cause issues. We keep the local
2023-11-23 01:08:30 +00:00
// version and discard the library one as the local version may have
2023-11-21 13:05:31 +00:00
// dev-dependencies that we want to keep resolving. See #15656 for more
// information.
2023-11-08 15:51:20 +00:00
if data.eq_ignoring_origin_and_deps(&crate_data, true) {
return Some((id, if a.is_local() { false } else { true }));
2023-11-08 15:51:20 +00:00
}
}
(_, _) => return None,
}
2023-11-08 15:51:20 +00:00
None
});
2023-10-15 15:32:12 +00:00
if let Some((res, should_update_lib_to_local)) = res {
id_map.insert(topo, res);
if should_update_lib_to_local {
2023-11-21 13:05:31 +00:00
assert!(self.arena[res].origin.is_lib());
assert!(crate_data.origin.is_local());
self.arena[res].origin = crate_data.origin.clone();
// Move local's dev dependencies into the newly-local-formerly-lib crate.
2023-11-21 13:05:31 +00:00
self.arena[res].dependencies = crate_data.dependencies.clone();
2023-10-15 15:32:12 +00:00
}
} else {
let id = self.arena.alloc(crate_data.clone());
id_map.insert(topo, id);
}
}
2023-03-25 17:06:06 +00:00
*proc_macros =
mem::take(proc_macros).into_iter().map(|(id, macros)| (id_map[&id], macros)).collect();
}
fn find_path(
&self,
visited: &mut FxHashSet<CrateId>,
from: CrateId,
to: CrateId,
) -> Option<Vec<CrateId>> {
if !visited.insert(from) {
return None;
2018-12-22 07:30:58 +00:00
}
if from == to {
return Some(vec![to]);
}
2020-03-09 10:11:59 +00:00
for dep in &self[from].dependencies {
2020-03-09 10:18:41 +00:00
let crate_id = dep.crate_id;
if let Some(mut path) = self.find_path(visited, crate_id, to) {
path.push(from);
return Some(path);
}
}
None
}
2022-07-08 13:44:49 +00:00
// Work around for https://github.com/rust-lang/rust-analyzer/issues/6038.
// As hacky as it gets.
pub fn patch_cfg_if(&mut self) -> bool {
// we stupidly max by version in an attempt to have all duplicated std's depend on the same cfg_if so that deduplication still works
let cfg_if =
self.hacky_find_crate("cfg_if").max_by_key(|&it| self.arena[it].version.clone());
let std = self.hacky_find_crate("std").next();
match (cfg_if, std) {
(Some(cfg_if), Some(std)) => {
self.arena[cfg_if].dependencies.clear();
self.arena[std].dependencies.push(Dependency::new(
CrateName::new("cfg_if").unwrap(),
cfg_if,
DependencyKind::Normal,
));
true
}
_ => false,
}
}
fn hacky_find_crate<'a>(&'a self, display_name: &'a str) -> impl Iterator<Item = CrateId> + 'a {
self.iter().filter(move |it| self[*it].display_name.as_deref() == Some(display_name))
}
2018-10-25 14:52:50 +00:00
}
2020-03-09 10:11:59 +00:00
impl ops::Index<CrateId> for CrateGraph {
type Output = CrateData;
fn index(&self, crate_id: CrateId) -> &CrateData {
&self.arena[crate_id]
2019-11-22 11:12:45 +00:00
}
}
impl CrateData {
/// Add a dependency to `self` without checking if the dependency
// is existent among `self.dependencies`.
2023-11-21 13:05:31 +00:00
fn add_dep(&mut self, dep: Dependency) {
self.dependencies.push(dep)
2019-11-22 11:12:45 +00:00
}
2019-11-22 11:08:18 +00:00
}
impl FromStr for Edition {
type Err = ParseEditionError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let res = match s {
"2015" => Edition::Edition2015,
"2018" => Edition::Edition2018,
2021-01-01 16:22:23 +00:00
"2021" => Edition::Edition2021,
"2024" => Edition::Edition2024,
_ => return Err(ParseEditionError { invalid_input: s.to_string() }),
2019-11-22 11:08:18 +00:00
};
Ok(res)
}
}
impl fmt::Display for Edition {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str(match self {
Edition::Edition2015 => "2015",
Edition::Edition2018 => "2018",
2021-01-01 16:22:23 +00:00
Edition::Edition2021 => "2021",
Edition::Edition2024 => "2024",
})
}
}
impl Extend<(String, String)> for Env {
fn extend<T: IntoIterator<Item = (String, String)>>(&mut self, iter: T) {
self.entries.extend(iter);
}
}
2020-07-21 15:17:21 +00:00
impl FromIterator<(String, String)> for Env {
fn from_iter<T: IntoIterator<Item = (String, String)>>(iter: T) -> Self {
Env { entries: FromIterator::from_iter(iter) }
}
}
impl Env {
pub fn set(&mut self, env: &str, value: String) {
self.entries.insert(env.to_owned(), value);
}
pub fn get(&self, env: &str) -> Option<String> {
self.entries.get(env).cloned()
}
2020-12-11 13:57:50 +00:00
pub fn iter(&self) -> impl Iterator<Item = (&str, &str)> {
self.entries.iter().map(|(k, v)| (k.as_str(), v.as_str()))
}
2020-03-11 03:04:02 +00:00
}
2020-03-10 13:59:12 +00:00
2019-11-22 11:12:45 +00:00
#[derive(Debug)]
pub struct ParseEditionError {
invalid_input: String,
}
2019-11-22 11:08:18 +00:00
impl fmt::Display for ParseEditionError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "invalid edition: {:?}", self.invalid_input)
}
}
impl std::error::Error for ParseEditionError {}
2019-11-22 11:12:45 +00:00
#[derive(Debug)]
2020-11-17 10:50:54 +00:00
pub struct CyclicDependenciesError {
path: Vec<(CrateId, Option<CrateDisplayName>)>,
}
impl CyclicDependenciesError {
fn from(&self) -> &(CrateId, Option<CrateDisplayName>) {
self.path.first().unwrap()
}
fn to(&self) -> &(CrateId, Option<CrateDisplayName>) {
self.path.last().unwrap()
}
2020-11-17 10:50:54 +00:00
}
impl fmt::Display for CyclicDependenciesError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let render = |(id, name): &(CrateId, Option<CrateDisplayName>)| match name {
Some(it) => format!("{it}({id:?})"),
None => format!("{id:?}"),
2020-11-17 10:50:54 +00:00
};
let path = self.path.iter().rev().map(render).collect::<Vec<String>>().join(" -> ");
write!(
f,
"cyclic deps: {} -> {}, alternative path: {}",
2021-09-03 14:00:50 +00:00
render(self.from()),
render(self.to()),
path
)
2020-11-17 10:50:54 +00:00
}
}
2019-11-22 11:12:45 +00:00
#[cfg(test)]
mod tests {
use crate::{CrateOrigin, DependencyKind};
2023-03-31 12:10:33 +00:00
use super::{CrateGraph, CrateName, Dependency, Edition::Edition2018, Env, FileId};
#[test]
fn detect_cyclic_dependency_indirect() {
let mut graph = CrateGraph::default();
2020-03-08 13:26:57 +00:00
let crate1 = graph.add_crate_root(
FileId::from_raw(1u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
let crate2 = graph.add_crate_root(
FileId::from_raw(2u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
let crate3 = graph.add_crate_root(
FileId::from_raw(3u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
assert!(graph
.add_dep(
crate1,
Dependency::new(CrateName::new("crate2").unwrap(), crate2, DependencyKind::Normal)
)
.is_ok());
assert!(graph
.add_dep(
crate2,
Dependency::new(CrateName::new("crate3").unwrap(), crate3, DependencyKind::Normal)
)
.is_ok());
assert!(graph
.add_dep(
crate3,
Dependency::new(CrateName::new("crate1").unwrap(), crate1, DependencyKind::Normal)
)
.is_err());
}
#[test]
fn detect_cyclic_dependency_direct() {
let mut graph = CrateGraph::default();
let crate1 = graph.add_crate_root(
FileId::from_raw(1u32),
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
);
let crate2 = graph.add_crate_root(
FileId::from_raw(2u32),
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
);
assert!(graph
.add_dep(
crate1,
Dependency::new(CrateName::new("crate2").unwrap(), crate2, DependencyKind::Normal)
)
.is_ok());
assert!(graph
.add_dep(
crate2,
Dependency::new(CrateName::new("crate2").unwrap(), crate2, DependencyKind::Normal)
)
.is_err());
}
#[test]
fn it_works() {
let mut graph = CrateGraph::default();
2020-03-08 13:26:57 +00:00
let crate1 = graph.add_crate_root(
FileId::from_raw(1u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
let crate2 = graph.add_crate_root(
FileId::from_raw(2u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
let crate3 = graph.add_crate_root(
FileId::from_raw(3u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
assert!(graph
.add_dep(
crate1,
Dependency::new(CrateName::new("crate2").unwrap(), crate2, DependencyKind::Normal)
)
.is_ok());
assert!(graph
.add_dep(
crate2,
Dependency::new(CrateName::new("crate3").unwrap(), crate3, DependencyKind::Normal)
)
.is_ok());
2020-02-05 09:53:54 +00:00
}
#[test]
fn dashes_are_normalized() {
let mut graph = CrateGraph::default();
2020-03-08 13:26:57 +00:00
let crate1 = graph.add_crate_root(
FileId::from_raw(1u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
let crate2 = graph.add_crate_root(
FileId::from_raw(2u32),
2020-03-08 13:26:57 +00:00
Edition2018,
None,
None,
2023-03-31 12:10:33 +00:00
Default::default(),
Default::default(),
2020-03-08 13:26:57 +00:00
Env::default(),
false,
2023-03-31 07:10:18 +00:00
CrateOrigin::Local { repo: None, name: None },
Err("".into()),
2023-03-31 07:10:18 +00:00
None,
2020-03-08 13:26:57 +00:00
);
2020-02-05 10:47:28 +00:00
assert!(graph
.add_dep(
crate1,
Dependency::new(
CrateName::normalize_dashes("crate-name-with-dashes"),
crate2,
DependencyKind::Normal
)
)
2020-02-05 10:47:28 +00:00
.is_ok());
2020-02-05 09:53:54 +00:00
assert_eq!(
2020-03-09 10:11:59 +00:00
graph[crate1].dependencies,
vec![Dependency::new(
CrateName::new("crate_name_with_dashes").unwrap(),
crate2,
DependencyKind::Normal
)]
2020-02-05 09:53:54 +00:00
);
}
}