use crate::core::compiler::RustcTargetData;
use crate::core::resolver::{HasDevUnits, ResolveOpts};
use crate::core::{Shell, Workspace};
use crate::ops;
use crate::util::CargoResult;
use std::collections::HashMap;
use std::path::Path;
use std::process::Command;
#[derive(Debug)]
pub struct DocOptions<'a> {
pub open_result: bool,
pub compile_opts: ops::CompileOptions<'a>,
}
pub fn doc(ws: &Workspace<'_>, options: &DocOptions<'_>) -> CargoResult<()> {
let specs = options.compile_opts.spec.to_package_id_specs(ws)?;
let opts = ResolveOpts::new(
true,
&options.compile_opts.features,
options.compile_opts.all_features,
!options.compile_opts.no_default_features,
);
let requested_kind = options.compile_opts.build_config.requested_kind;
let target_data = RustcTargetData::new(ws, requested_kind)?;
let ws_resolve = ops::resolve_ws_with_opts(
ws,
&target_data,
requested_kind,
&opts,
&specs,
HasDevUnits::No,
)?;
let ids = specs
.iter()
.map(|s| s.query(ws_resolve.targeted_resolve.iter()))
.collect::<CargoResult<Vec<_>>>()?;
let pkgs = ws_resolve.pkg_set.get_many(ids)?;
let mut lib_names = HashMap::new();
let mut bin_names = HashMap::new();
let mut names = Vec::new();
for package in &pkgs {
for target in package.targets().iter().filter(|t| t.documented()) {
if target.is_lib() {
if let Some(prev) = lib_names.insert(target.crate_name(), package) {
anyhow::bail!(
"The library `{}` is specified by packages `{}` and \
`{}` but can only be documented once. Consider renaming \
or marking one of the targets as `doc = false`.",
target.crate_name(),
prev,
package
);
}
} else if let Some(prev) = bin_names.insert(target.crate_name(), package) {
anyhow::bail!(
"The binary `{}` is specified by packages `{}` and \
`{}` but can be documented only once. Consider renaming \
or marking one of the targets as `doc = false`.",
target.crate_name(),
prev,
package
);
}
names.push(target.crate_name());
}
}
let compilation = ops::compile(ws, &options.compile_opts)?;
if options.open_result {
let name = match names.first() {
Some(s) => s.to_string(),
None => return Ok(()),
};
let path = compilation
.root_output
.with_file_name("doc")
.join(&name)
.join("index.html");
if path.exists() {
let mut shell = options.compile_opts.config.shell();
shell.status("Opening", path.display())?;
open_docs(&path, &mut shell)?;
}
}
Ok(())
}
fn open_docs(path: &Path, shell: &mut Shell) -> CargoResult<()> {
match std::env::var_os("BROWSER") {
Some(browser) => {
if let Err(e) = Command::new(&browser).arg(path).status() {
shell.warn(format!(
"Couldn't open docs with {}: {}",
browser.to_string_lossy(),
e
))?;
}
}
None => {
if let Err(e) = opener::open(&path) {
shell.warn(format!("Couldn't open docs: {}", e))?;
for cause in anyhow::Error::new(e).chain().skip(1) {
shell.warn(format!("Caused by:\n {}", cause))?;
}
}
}
};
Ok(())
}