Graphite/tools/third-party-licenses/src/cargo.rs

102 lines
2.5 KiB
Rust

use crate::{Error, LicenceSource, LicenseEntry, Package};
use serde::Deserialize;
use std::fs;
use std::hash::{Hash, Hasher};
use std::path::PathBuf;
use std::process::Command;
pub struct CargoLicenseSource {}
impl CargoLicenseSource {
pub fn new() -> Self {
Self {}
}
}
impl LicenceSource for CargoLicenseSource {
fn licenses(&self) -> Result<Vec<LicenseEntry>, Error> {
Ok(parse(run()?))
}
}
impl Hash for CargoLicenseSource {
fn hash<H: Hasher>(&self, state: &mut H) {
let lock_path = PathBuf::from(env!("CARGO_WORKSPACE_DIR")).join("Cargo.lock");
fs::read_to_string(lock_path).unwrap().hash(state)
}
}
#[derive(Deserialize)]
struct Output {
licenses: Vec<License>,
}
#[derive(Deserialize)]
struct License {
name: Option<String>,
text: Option<String>,
used_by: Vec<UsedBy>,
}
#[derive(Deserialize)]
struct UsedBy {
#[serde(rename = "crate")]
crate_info: Crate,
}
#[derive(Deserialize)]
struct Crate {
name: Option<String>,
version: Option<String>,
authors: Option<Vec<String>>,
repository: Option<String>,
}
fn parse(parsed: Output) -> Vec<LicenseEntry> {
parsed
.licenses
.into_iter()
.map(|license| {
let packages = license
.used_by
.into_iter()
.map(|used| {
let name = used.crate_info.name.as_deref().unwrap_or_default();
let version = used.crate_info.version.as_deref().unwrap_or_default();
let display_name = if version.is_empty() { name.to_string() } else { format!("{name}@{version}") };
let repository = used.crate_info.repository.filter(|s| !s.is_empty());
Package {
name: display_name,
authors: used.crate_info.authors.unwrap_or_default(),
url: repository,
}
})
.collect();
LicenseEntry {
name: license.name,
text: license.text.as_deref().unwrap_or_default().to_string(),
packages,
}
})
.collect()
}
fn run() -> Result<Output, Error> {
let output = Command::new("cargo")
.args(["about", "generate", "--format", "json", "--frozen"])
.current_dir(env!("CARGO_WORKSPACE_DIR"))
.output()
.map_err(|e| Error::Io(e, "Failed to run cargo about generate".into()))?;
if !output.status.success() {
return Err(Error::Command(format!("cargo about generate failed:\n{}", String::from_utf8_lossy(&output.stderr))));
}
let stdout = String::from_utf8(output.stdout).map_err(|e| Error::Utf8(e, "cargo about generate returned invalid UTF-8".into()))?;
serde_json::from_str(&stdout).map_err(|e| Error::Json(e, "Failed to parse cargo about generate JSON".into()))
}