You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
3.2 KiB
90 lines
3.2 KiB
use std::error::Error; |
|
use std::io::Read; |
|
use std::process::{Command, Stdio}; |
|
|
|
use cargo_metadata::Target; |
|
use serde::Deserialize; |
|
|
|
use crate::print_command; |
|
|
|
/// In lieu of <https://github.com/oli-obk/cargo_metadata/issues/107> |
|
/// and to avoid pulling in the real `cargo` |
|
/// [data structures](https://docs.rs/cargo/latest/cargo/core/compiler/unit_graph/type.UnitGraph.html) |
|
/// as a dependency, we define the subset of the build graph we care about. |
|
#[derive(Deserialize)] |
|
pub struct UnitGraph { |
|
pub version: i32, |
|
pub units: Vec<Unit>, |
|
} |
|
|
|
impl UnitGraph { |
|
/// Collect the unit graph via Cargo's `--unit-graph` flag. |
|
/// This runs the same command as the actual build, except nothing is actually |
|
/// build and the graph is output instead. |
|
/// |
|
/// See <https://doc.rust-lang.org/nightly/cargo/reference/unstable.html#unit-graph>. |
|
pub fn from_cargo(cargo_cmd: &Command, verbose: bool) -> Result<Self, Box<dyn Error>> { |
|
// Since Command isn't Clone, copy it "by hand", by copying its args and envs |
|
let mut cmd = Command::new(cargo_cmd.get_program()); |
|
|
|
let mut args = cargo_cmd.get_args(); |
|
cmd.args(args.next()) |
|
// These options must be added before any possible `--`, so the best |
|
// place is to just stick them immediately after the first arg (subcommand) |
|
.args(["-Z", "unstable-options", "--unit-graph"]) |
|
.args(args) |
|
.envs(cargo_cmd.get_envs().filter_map(|(k, v)| Some((k, v?)))) |
|
.stdout(Stdio::piped()) |
|
.stderr(Stdio::piped()); |
|
|
|
if verbose { |
|
print_command(&cmd); |
|
} |
|
|
|
let mut proc = cmd.spawn()?; |
|
let stdout = proc.stdout.take().unwrap(); |
|
let mut stderr = proc.stderr.take().unwrap(); |
|
|
|
let result: Self = serde_json::from_reader(stdout).map_err(|err| { |
|
let mut stderr_str = String::new(); |
|
let _ = stderr.read_to_string(&mut stderr_str); |
|
|
|
let _ = proc.wait(); |
|
format!("unable to parse `--unit-graph` json: {err}\nstderr: `{stderr_str}`") |
|
})?; |
|
|
|
let _status = proc.wait()?; |
|
// TODO: with cargo 1.74.0-nightly (b4ddf95ad 2023-09-18), |
|
// `cargo run --unit-graph` panics at src/cargo/ops/cargo_run.rs:83:5 |
|
// It seems to have been fixed as of cargo 1.76.0-nightly (71cd3a926 2023-11-20) |
|
// so maybe we can stop ignoring it once we bump the minimum toolchain version, |
|
// and certainly we should once `--unit-graph` is ever stabilized. |
|
// |
|
// if !status.success() { |
|
// return Err(format!("`cargo --unit-graph` exited with status {status:?}").into()); |
|
// } |
|
|
|
if result.version == 1 { |
|
Ok(result) |
|
} else { |
|
Err(format!( |
|
"unknown `cargo --unit-graph` output version {}", |
|
result.version |
|
))? |
|
} |
|
} |
|
} |
|
|
|
#[derive(Deserialize)] |
|
pub struct Unit { |
|
pub target: Target, |
|
pub profile: Profile, |
|
} |
|
|
|
/// This struct is very similar to [`cargo_metadata::ArtifactProfile`], but seems |
|
/// to have some slight differences so we define a different version. We only |
|
/// really care about `debuginfo` anyway. |
|
#[derive(Deserialize)] |
|
pub struct Profile { |
|
pub debuginfo: Option<u32>, |
|
}
|
|
|