Skip to content

Commit

Permalink
Fix issue-11010
Browse files Browse the repository at this point in the history
  • Loading branch information
LuuuXXX committed Nov 30, 2023
1 parent 35ea623 commit 7e3d375
Show file tree
Hide file tree
Showing 22 changed files with 195 additions and 0 deletions.
28 changes: 28 additions & 0 deletions src/cargo/ops/cargo_add/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ use std::collections::BTreeSet;
use std::collections::VecDeque;
use std::fmt::Write;
use std::path::Path;
use std::str::FromStr;

use anyhow::Context as _;
use cargo_util::paths;
Expand Down Expand Up @@ -196,6 +197,13 @@ pub fn add(workspace: &Workspace<'_>, options: &AddOptions<'_>) -> CargoResult<(
print_dep_table_msg(&mut options.config.shell(), &dep)?;

manifest.insert_into_table(&dep_table, &dep)?;
if let Some(option) = dep.optional {
let rust_version_check =
check_rust_version_for_optional_dependency(options.spec.rust_version())?;
if option && rust_version_check {
manifest.insert_into_feature_table(&dep)?;
}
}
manifest.gc_dep(dep.toml_key());
}

Expand Down Expand Up @@ -469,6 +477,26 @@ fn check_invalid_ws_keys(toml_key: &str, arg: &DepOp) -> CargoResult<()> {
Ok(())
}

/// When the `--optional` option is added using `cargo add`, we need to
/// check the current rust-version. As the `dep:` syntax is only avaliable
/// starting with Rust 1.60.0
///
/// `true` means that the rust-version is None or the rust-version is higher
/// than the version needed.
///
/// Note: Previous versions can only use the implicit feature name.
fn check_rust_version_for_optional_dependency(
rust_version: Option<&RustVersion>,
) -> CargoResult<bool> {
match rust_version {
Some(version) => {
let syntax_support_version = RustVersion::from_str("1.60.0")?;
Ok(version.cmp(&syntax_support_version).is_ge())
}
None => Ok(true),
}
}

/// Provide the existing dependency for the target table
///
/// If it doesn't exist but exists in another table, let's use that as most likely users
Expand Down
41 changes: 41 additions & 0 deletions src/cargo/util/toml_mut/dependency.rs
Original file line number Diff line number Diff line change
Expand Up @@ -921,6 +921,47 @@ impl Display for WorkspaceSource {
}
}

impl Dependency {
/// Convert denpency as feature to TOML.
/// Panics if the path is relative
pub fn to_toml_as_feature(&self, crate_root: &Path) -> toml_edit::Item {
assert!(
crate_root.is_absolute(),
"Absolute path needed, got: {}",
crate_root.display()
);
let features: toml_edit::Value = vec![self.as_feature()].iter().collect();
toml_edit::value(features)
}

pub fn as_feature(&self) -> String {
if let Some(rename) = &self.rename {
format!("dep:{}", rename)
} else {
format!("dep:{}", self.name)
}
}

/// Check whether `dep:<dep>` is defined in the features section.
///
/// `true` if there is a define `dep:<dep>`
/// `false` if `dep:<dep>` is not used ever.
///
/// Make sure that `dep:<dep>` is included in one of features in the features table.
pub fn is_active_as_feature(&self, item: &mut toml_edit::Item) -> bool {
item.as_table_like_mut().unwrap().iter().any(|(_, values)| {
if let Some(values) = &values.as_array() {
values.iter().any(|value| match value.as_str() {
Some(val) => val.to_string().eq(&self.as_feature()),
None => false,
})
} else {
false
}
})
}
}

#[cfg(test)]
mod tests {
use std::path::Path;
Expand Down
19 changes: 19 additions & 0 deletions src/cargo/util/toml_mut/manifest.rs
Original file line number Diff line number Diff line change
Expand Up @@ -363,6 +363,25 @@ impl LocalManifest {
Ok(())
}

/// Add feature entry to a Cargo.toml.
pub fn insert_into_feature_table(&mut self, dep: &Dependency) -> CargoResult<()> {
let crate_root = self
.path
.parent()
.expect("manifest path is absolute")
.to_owned();
let dep_key = dep.toml_key();
let table = self.get_table_mut(&vec![String::from("features")])?;

// Check whether `dep:<dep>` is defined in the [features] section.
if !dep.is_active_as_feature(table) {
let new_feature = dep.to_toml_as_feature(&crate_root);
table[dep_key] = new_feature;
}

Ok(())
}

/// Remove entry from a Cargo.toml.
pub fn remove_from_table(&mut self, table_path: &[String], name: &str) -> CargoResult<()> {
let parent_table = self.get_table_mut(table_path)?;
Expand Down
3 changes: 3 additions & 0 deletions tests/testsuite/cargo_add/change_rename_target/out/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
some-package = { package = "my-package2", version = "99999.0.0", optional = true }

[features]
some-package = ["dep:some-package"]
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,6 @@ version = "0.0.0"

[dependencies]
foo = { workspace = true, optional = true }

[features]
foo = ["dep:foo"]
4 changes: 4 additions & 0 deletions tests/testsuite/cargo_add/optional/out/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,7 @@ version = "0.0.0"
[dependencies]
my-package1 = { version = "99999.0.0", optional = true }
my-package2 = { version = "0.4.1", optional = true }

[features]
my-package1 = ["dep:my-package1"]
my-package2 = ["dep:my-package2"]
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
cargo-list-test-fixture-dependency = { optional = true, path = "../dependency", version = "0.0.0" }

[features]
cargo-list-test-fixture-dependency = ["dep:cargo-list-test-fixture-dependency"]
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,6 @@ version = "0.0.0"

[dependencies]
foo = { workspace = true, optional = true }

[features]
foo = ["dep:foo"]
3 changes: 3 additions & 0 deletions tests/testsuite/cargo_add/overwrite_name_noop/out/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,6 @@ version = "0.0.0"

[dependencies]
your-face = { version = "0.0.0", path = "dependency", optional = true, default-features = false, features = ["nose", "mouth"], registry = "alternative" }

[features]
your-face = ["dep:your-face"]
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,7 @@ version = "0.0.0"
[dependencies]
my-package1 = { version = "99999.0.0", optional = true }
my-package2 = { version = "0.4.1", optional = true }

[features]
my-package1 = ["dep:my-package1"]
my-package2 = ["dep:my-package2"]
4 changes: 4 additions & 0 deletions tests/testsuite/cargo_add/overwrite_optional/out/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,7 @@ version = "0.0.0"
[dependencies]
my-package1 = { version = "99999.0.0", optional = true }
my-package2 = { version = "0.4.1", optional = true }

[features]
my-package1 = ["dep:my-package1"]
my-package2 = ["dep:my-package2"]
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
[workspace]

[package]
name = "cargo-list-test-fixture"
version = "0.0.0"

[dependencies]
my-package1 = "99999.0.0"
my-package2 = "0.4.1"

[features]
default = ["dep:my-package1", "dep:my-package2"]
Empty file.
38 changes: 38 additions & 0 deletions tests/testsuite/cargo_add/overwrite_optional_with_optional/mod.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
use cargo_test_support::compare::assert_ui;
use cargo_test_support::prelude::*;
use cargo_test_support::Project;

use cargo_test_support::curr_dir;

#[cargo_test]
fn case() {
cargo_test_support::registry::init();
for name in ["my-package1", "my-package2"] {
for ver in [
"0.1.1+my-package",
"0.2.0+my-package",
"0.2.3+my-package",
"0.4.1+my-package",
"20.0.0+my-package",
"99999.0.0+my-package",
"99999.0.0-alpha.1+my-package",
] {
cargo_test_support::registry::Package::new(name, ver).publish();
}
}

let project = Project::from_template(curr_dir!().join("in"));
let project_root = project.root();
let cwd = &project_root;

snapbox::cmd::Command::cargo_ui()
.arg("add")
.arg_line("my-package1 [email protected] --optional")
.current_dir(cwd)
.assert()
.success()
.stdout_matches_path(curr_dir!().join("stdout.log"))
.stderr_matches_path(curr_dir!().join("stderr.log"));

assert_ui().subset_matches(curr_dir!().join("out"), &project_root);
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
[workspace]

[package]
name = "cargo-list-test-fixture"
version = "0.0.0"

[dependencies]
my-package1 = { version = "99999.0.0", optional = true }
my-package2 = { version = "0.4.1", optional = true }

[features]
default = ["dep:my-package1", "dep:my-package2"]
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
Updating `dummy-registry` index
Adding my-package1 v99999.0.0 to optional dependencies.
Adding my-package2 v0.4.1 to optional dependencies.
Empty file.
3 changes: 3 additions & 0 deletions tests/testsuite/cargo_add/overwrite_path_noop/out/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,6 @@ version = "0.0.0"

[dependencies]
your-face = { version = "0.0.0", path = "dependency", optional = true, default-features = false, features = ["nose", "mouth"], registry = "alternative" }

[features]
your-face = ["dep:your-face"]
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
cargo-list-test-fixture-dependency = { optional = true, version = "20.0" }

[features]
cargo-list-test-fixture-dependency = ["dep:cargo-list-test-fixture-dependency"]
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
a1 = { package = "versioned-package", version = "0.1.1", optional = true }

[features]
a1 = ["dep:a1"]
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
versioned-package = { version = "0.3.0", optional = true, git = "[ROOTURL]/versioned-package" }

[features]
versioned-package = ["dep:versioned-package"]
Original file line number Diff line number Diff line change
Expand Up @@ -6,3 +6,6 @@ version = "0.0.0"

[dependencies]
cargo-list-test-fixture-dependency = { version = "0.0.0", optional = true, path = "../dependency" }

[features]
cargo-list-test-fixture-dependency = ["dep:cargo-list-test-fixture-dependency"]

0 comments on commit 7e3d375

Please sign in to comment.