use std::collections::HashSet;
use std::collections::{BTreeMap, HashMap};
use std::{cmp, env, fmt, hash};
use serde::Deserialize;
use crate::core::compiler::{CompileMode, ProfileKind};
use crate::core::interning::InternedString;
use crate::core::{Feature, Features, PackageId, PackageIdSpec, Resolve, Shell};
use crate::util::errors::CargoResultExt;
use crate::util::toml::{ProfilePackageSpec, StringOrBool, TomlProfile, TomlProfiles, U32OrBool};
use crate::util::{closest_msg, CargoResult, Config};
#[derive(Clone, Debug)]
pub struct Profiles {
incremental: Option<bool>,
dir_names: HashMap<String, String>,
by_name: HashMap<String, ProfileMaker>,
named_profiles_enabled: bool,
}
impl Profiles {
pub fn new(
profiles: Option<&TomlProfiles>,
config: &Config,
features: &Features,
warnings: &mut Vec<String>,
) -> CargoResult<Profiles> {
if let Some(profiles) = profiles {
profiles.validate(features, warnings)?;
}
let config_profiles = config.profiles()?;
let incremental = match env::var_os("CARGO_INCREMENTAL") {
Some(v) => Some(v == "1"),
None => config.build_config()?.incremental,
};
if !features.is_enabled(Feature::named_profiles()) {
let mut profile_makers = Profiles {
incremental,
named_profiles_enabled: false,
dir_names: Self::predefined_dir_names(),
by_name: HashMap::new(),
};
profile_makers.by_name.insert(
"dev".to_owned(),
ProfileMaker {
toml: profiles.and_then(|p| p.get("dev").cloned()),
inherits: vec![],
config: config_profiles.dev.clone(),
default: Profile::default_dev(),
},
);
profile_makers
.dir_names
.insert("dev".to_owned(), "debug".to_owned());
profile_makers.by_name.insert(
"release".to_owned(),
ProfileMaker {
toml: profiles.and_then(|p| p.get("release").cloned()),
inherits: vec![],
config: config_profiles.release.clone(),
default: Profile::default_release(),
},
);
profile_makers
.dir_names
.insert("release".to_owned(), "release".to_owned());
profile_makers.by_name.insert(
"test".to_owned(),
ProfileMaker {
toml: profiles.and_then(|p| p.get("test").cloned()),
inherits: vec![],
config: None,
default: Profile::default_test(),
},
);
profile_makers
.dir_names
.insert("test".to_owned(), "debug".to_owned());
profile_makers.by_name.insert(
"bench".to_owned(),
ProfileMaker {
toml: profiles.and_then(|p| p.get("bench").cloned()),
inherits: vec![],
config: None,
default: Profile::default_bench(),
},
);
profile_makers
.dir_names
.insert("bench".to_owned(), "release".to_owned());
profile_makers.by_name.insert(
"doc".to_owned(),
ProfileMaker {
toml: profiles.and_then(|p| p.get("doc").cloned()),
inherits: vec![],
config: None,
default: Profile::default_doc(),
},
);
profile_makers
.dir_names
.insert("doc".to_owned(), "debug".to_owned());
return Ok(profile_makers);
}
let mut profile_makers = Profiles {
incremental,
named_profiles_enabled: true,
dir_names: Self::predefined_dir_names(),
by_name: HashMap::new(),
};
Self::add_root_profiles(&mut profile_makers, profiles, config_profiles);
let mut profiles = if let Some(profiles) = profiles {
profiles.get_all().clone()
} else {
BTreeMap::new()
};
use std::collections::btree_map::Entry;
for (predef_name, mut predef_prof) in Self::predefined_profiles().into_iter() {
match profiles.entry(predef_name.to_owned()) {
Entry::Vacant(vac) => {
vac.insert(predef_prof);
}
Entry::Occupied(mut oc) => {
let r = oc.get_mut();
predef_prof.merge(r);
*r = predef_prof;
}
}
}
profile_makers.process_customs(&profiles)?;
Ok(profile_makers)
}
fn predefined_dir_names() -> HashMap<String, String> {
let mut dir_names = HashMap::new();
dir_names.insert("dev".to_owned(), "debug".to_owned());
dir_names.insert("check".to_owned(), "debug".to_owned());
dir_names.insert("test".to_owned(), "debug".to_owned());
dir_names.insert("bench".to_owned(), "release".to_owned());
dir_names
}
fn add_root_profiles(
profile_makers: &mut Profiles,
profiles: Option<&TomlProfiles>,
config_profiles: &ConfigProfiles,
) {
let profile_name = "dev";
profile_makers.by_name.insert(
profile_name.to_owned(),
ProfileMaker {
default: Profile::default_dev(),
toml: profiles.and_then(|p| p.get(profile_name).cloned()),
config: config_profiles.dev.clone(),
inherits: vec![],
},
);
let profile_name = "release";
profile_makers.by_name.insert(
profile_name.to_owned(),
ProfileMaker {
default: Profile::default_release(),
toml: profiles.and_then(|p| p.get(profile_name).cloned()),
config: config_profiles.release.clone(),
inherits: vec![],
},
);
}
fn predefined_profiles() -> Vec<(&'static str, TomlProfile)> {
vec![
(
"bench",
TomlProfile {
inherits: Some(String::from("release")),
..TomlProfile::default()
},
),
(
"test",
TomlProfile {
inherits: Some(String::from("dev")),
..TomlProfile::default()
},
),
(
"check",
TomlProfile {
inherits: Some(String::from("dev")),
..TomlProfile::default()
},
),
(
"doc",
TomlProfile {
inherits: Some(String::from("dev")),
..TomlProfile::default()
},
),
]
}
fn process_customs(&mut self, profiles: &BTreeMap<String, TomlProfile>) -> CargoResult<()> {
for (name, profile) in profiles {
let mut set = HashSet::new();
let mut result = Vec::new();
set.insert(name.as_str().to_owned());
match &profile.dir_name {
None => {}
Some(dir_name) => {
self.dir_names.insert(name.clone(), dir_name.to_owned());
}
}
match name.as_str() {
"dev" | "release" => {
match &profile.inherits {
None => {}
Some(_) => {
failure::bail!(
"An 'inherits' must not specified root profile '{}'",
name
);
}
}
continue;
}
_ => {}
};
let mut maker = self.process_chain(name, profile, &mut set, &mut result, profiles)?;
result.reverse();
maker.inherits = result;
self.by_name.insert(name.as_str().to_owned(), maker);
}
Ok(())
}
fn process_chain(
&mut self,
name: &str,
profile: &TomlProfile,
set: &mut HashSet<String>,
result: &mut Vec<TomlProfile>,
profiles: &BTreeMap<String, TomlProfile>,
) -> CargoResult<ProfileMaker> {
result.push(profile.clone());
match profile.inherits.as_ref().map(|x| x.as_str()) {
Some(name @ "dev") | Some(name @ "release") => {
Ok(self.by_name.get(name).unwrap().clone())
}
Some(name) => {
let name = name.to_owned();
if set.get(&name).is_some() {
failure::bail!(
"Inheritance loop of profiles cycles with profile '{}'",
name
);
}
set.insert(name.clone());
match profiles.get(&name) {
None => {
failure::bail!("Profile '{}' not found in Cargo.toml", name);
}
Some(parent) => self.process_chain(&name, parent, set, result, profiles),
}
}
None => {
failure::bail!(
"An 'inherits' directive is needed for all \
profiles that are not 'dev' or 'release'. Here \
it is missing from '{}'",
name
);
}
}
}
pub fn get_profile(
&self,
pkg_id: PackageId,
is_member: bool,
unit_for: UnitFor,
mode: CompileMode,
profile_kind: ProfileKind,
) -> Profile {
let (profile_name, inherits) = if !self.named_profiles_enabled {
let release = match profile_kind {
ProfileKind::Release => true,
ProfileKind::Custom(ref s) if s == "bench" => true,
_ => false,
};
match mode {
CompileMode::Test | CompileMode::Bench => {
if release {
("bench", Some("release"))
} else {
("test", Some("dev"))
}
}
CompileMode::Build
| CompileMode::Check { .. }
| CompileMode::Doctest
| CompileMode::RunCustomBuild => {
if release {
("release", None)
} else {
("dev", None)
}
}
CompileMode::Doc { .. } => ("doc", None),
}
} else {
(profile_kind.name(), None)
};
let maker = match self.by_name.get(profile_name) {
None => panic!("Profile {} undefined", profile_name),
Some(r) => r,
};
let mut profile = maker.get_profile(Some(pkg_id), is_member, unit_for);
match unit_for.panic_setting() {
PanicSetting::AlwaysUnwind => profile.panic = PanicStrategy::Unwind,
PanicSetting::ReadProfile => {}
PanicSetting::Inherit => {
if let Some(inherits) = inherits {
let maker = self.by_name.get(inherits).unwrap();
profile.panic = maker.get_profile(Some(pkg_id), is_member, unit_for).panic;
}
}
}
if let Some(v) = self.incremental {
profile.incremental = v;
}
if !pkg_id.source_id().is_path() {
profile.incremental = false;
}
profile
}
pub fn get_profile_run_custom_build(&self, for_unit_profile: &Profile) -> Profile {
let mut result = Profile::default();
result.root = for_unit_profile.root;
result.debuginfo = for_unit_profile.debuginfo;
result.opt_level = for_unit_profile.opt_level;
result
}
pub fn base_profile(&self, profile_kind: &ProfileKind) -> CargoResult<Profile> {
let profile_name = if !self.named_profiles_enabled {
match profile_kind {
ProfileKind::Release => "release",
ProfileKind::Custom(ref s) if s == "bench" => "bench",
_ => "dev",
}
} else {
profile_kind.name()
};
match self.by_name.get(profile_name) {
None => failure::bail!("Profile `{}` undefined", profile_kind.name()),
Some(r) => Ok(r.get_profile(None, true, UnitFor::new_normal())),
}
}
pub fn get_dir_name(&self, profile_kind: &ProfileKind) -> String {
let dest = profile_kind.name();
match self.dir_names.get(dest) {
None => dest.to_owned(),
Some(s) => s.clone(),
}
}
pub fn validate_packages(&self, shell: &mut Shell, resolve: &Resolve) -> CargoResult<()> {
for profile in self.by_name.values() {
profile.validate_packages(shell, resolve)?;
}
Ok(())
}
}
#[derive(Debug, Clone)]
struct ProfileMaker {
default: Profile,
toml: Option<TomlProfile>,
inherits: Vec<TomlProfile>,
config: Option<TomlProfile>,
}
impl ProfileMaker {
fn get_profile(
&self,
pkg_id: Option<PackageId>,
is_member: bool,
unit_for: UnitFor,
) -> Profile {
let mut profile = self.default;
let mut tomls = vec![];
if let Some(ref toml) = self.toml {
tomls.push(toml);
}
for toml in &self.inherits {
tomls.push(toml);
}
for toml in &tomls {
merge_profile(&mut profile, toml);
}
for toml in &tomls {
merge_toml_overrides(pkg_id, is_member, unit_for, &mut profile, toml);
}
if let Some(ref toml) = self.config {
merge_profile(&mut profile, toml);
merge_toml_overrides(pkg_id, is_member, unit_for, &mut profile, toml);
}
profile
}
fn validate_packages(&self, shell: &mut Shell, resolve: &Resolve) -> CargoResult<()> {
self.validate_packages_toml(shell, resolve, &self.toml, true)?;
self.validate_packages_toml(shell, resolve, &self.config, false)?;
Ok(())
}
fn validate_packages_toml(
&self,
shell: &mut Shell,
resolve: &Resolve,
toml: &Option<TomlProfile>,
warn_unmatched: bool,
) -> CargoResult<()> {
let toml = match *toml {
Some(ref toml) => toml,
None => return Ok(()),
};
let overrides = match toml.package.as_ref().or_else(|| toml.overrides.as_ref()) {
Some(overrides) => overrides,
None => return Ok(()),
};
let mut found = HashSet::new();
for pkg_id in resolve.iter() {
let matches: Vec<&PackageIdSpec> = overrides
.keys()
.filter_map(|key| match *key {
ProfilePackageSpec::All => None,
ProfilePackageSpec::Spec(ref spec) => {
if spec.matches(pkg_id) {
Some(spec)
} else {
None
}
}
})
.collect();
match matches.len() {
0 => {}
1 => {
found.insert(matches[0].clone());
}
_ => {
let specs = matches
.iter()
.map(|spec| spec.to_string())
.collect::<Vec<_>>()
.join(", ");
failure::bail!(
"multiple package overrides in profile `{}` match package `{}`\n\
found package specs: {}",
self.default.name,
pkg_id,
specs
);
}
}
}
if !warn_unmatched {
return Ok(());
}
let missing_specs = overrides.keys().filter_map(|key| {
if let ProfilePackageSpec::Spec(ref spec) = *key {
if !found.contains(spec) {
return Some(spec);
}
}
None
});
for spec in missing_specs {
let name_matches: Vec<String> = resolve
.iter()
.filter_map(|pkg_id| {
if pkg_id.name() == spec.name() {
Some(pkg_id.to_string())
} else {
None
}
})
.collect();
if name_matches.is_empty() {
let suggestion = closest_msg(&spec.name(), resolve.iter(), |p| p.name().as_str());
shell.warn(format!(
"package profile spec `{}` did not match any packages{}",
spec, suggestion
))?;
} else {
shell.warn(format!(
"version or URL in package profile spec `{}` does not \
match any of the packages: {}",
spec,
name_matches.join(", ")
))?;
}
}
Ok(())
}
}
fn merge_toml_overrides(
pkg_id: Option<PackageId>,
is_member: bool,
unit_for: UnitFor,
profile: &mut Profile,
toml: &TomlProfile,
) {
if unit_for.is_build() {
if let Some(ref build_override) = toml.build_override {
merge_profile(profile, build_override);
}
}
if let Some(overrides) = toml.package.as_ref().or_else(|| toml.overrides.as_ref()) {
if !is_member {
if let Some(all) = overrides.get(&ProfilePackageSpec::All) {
merge_profile(profile, all);
}
}
if let Some(pkg_id) = pkg_id {
let mut matches = overrides
.iter()
.filter_map(|(key, spec_profile)| match *key {
ProfilePackageSpec::All => None,
ProfilePackageSpec::Spec(ref s) => {
if s.matches(pkg_id) {
Some(spec_profile)
} else {
None
}
}
});
if let Some(spec_profile) = matches.next() {
merge_profile(profile, spec_profile);
assert!(
matches.next().is_none(),
"package `{}` matched multiple package profile overrides",
pkg_id
);
}
}
}
}
fn merge_profile(profile: &mut Profile, toml: &TomlProfile) {
if let Some(ref opt_level) = toml.opt_level {
profile.opt_level = InternedString::new(&opt_level.0);
}
match toml.lto {
Some(StringOrBool::Bool(b)) => profile.lto = Lto::Bool(b),
Some(StringOrBool::String(ref n)) => profile.lto = Lto::Named(InternedString::new(n)),
None => {}
}
if toml.codegen_units.is_some() {
profile.codegen_units = toml.codegen_units;
}
match toml.debug {
Some(U32OrBool::U32(debug)) => profile.debuginfo = Some(debug),
Some(U32OrBool::Bool(true)) => profile.debuginfo = Some(2),
Some(U32OrBool::Bool(false)) => profile.debuginfo = None,
None => {}
}
if let Some(debug_assertions) = toml.debug_assertions {
profile.debug_assertions = debug_assertions;
}
if let Some(rpath) = toml.rpath {
profile.rpath = rpath;
}
if let Some(panic) = &toml.panic {
profile.panic = match panic.as_str() {
"unwind" => PanicStrategy::Unwind,
"abort" => PanicStrategy::Abort,
_ => panic!("Unexpected panic setting `{}`", panic),
};
}
if let Some(overflow_checks) = toml.overflow_checks {
profile.overflow_checks = overflow_checks;
}
if let Some(incremental) = toml.incremental {
profile.incremental = incremental;
}
}
#[derive(Clone, Copy, Eq, PartialOrd, Ord, PartialEq, Debug)]
pub enum ProfileRoot {
Release,
Debug,
}
#[derive(Clone, Copy, Eq, PartialOrd, Ord)]
pub struct Profile {
pub name: &'static str,
pub opt_level: InternedString,
pub root: ProfileRoot,
pub lto: Lto,
pub codegen_units: Option<u32>,
pub debuginfo: Option<u32>,
pub debug_assertions: bool,
pub overflow_checks: bool,
pub rpath: bool,
pub incremental: bool,
pub panic: PanicStrategy,
}
impl Default for Profile {
fn default() -> Profile {
Profile {
name: "",
opt_level: InternedString::new("0"),
root: ProfileRoot::Debug,
lto: Lto::Bool(false),
codegen_units: None,
debuginfo: None,
debug_assertions: false,
overflow_checks: false,
rpath: false,
incremental: false,
panic: PanicStrategy::Unwind,
}
}
}
compact_debug! {
impl fmt::Debug for Profile {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
let (default, default_name) = match self.name {
"dev" => (Profile::default_dev(), "default_dev()"),
"release" => (Profile::default_release(), "default_release()"),
_ => (Profile::default(), "default()"),
};
[debug_the_fields(
name
opt_level
lto
root
codegen_units
debuginfo
debug_assertions
overflow_checks
rpath
incremental
panic
)]
}
}
}
impl fmt::Display for Profile {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "Profile({})", self.name)
}
}
impl hash::Hash for Profile {
fn hash<H>(&self, state: &mut H)
where
H: hash::Hasher,
{
self.comparable().hash(state);
}
}
impl cmp::PartialEq for Profile {
fn eq(&self, other: &Self) -> bool {
self.comparable() == other.comparable()
}
}
impl Profile {
fn default_dev() -> Profile {
Profile {
name: "dev",
root: ProfileRoot::Debug,
debuginfo: Some(2),
debug_assertions: true,
overflow_checks: true,
incremental: true,
..Profile::default()
}
}
fn default_release() -> Profile {
Profile {
name: "release",
root: ProfileRoot::Release,
opt_level: InternedString::new("3"),
..Profile::default()
}
}
fn default_test() -> Profile {
Profile {
name: "test",
..Profile::default_dev()
}
}
fn default_bench() -> Profile {
Profile {
name: "bench",
..Profile::default_release()
}
}
fn default_doc() -> Profile {
Profile {
name: "doc",
..Profile::default_dev()
}
}
fn comparable(
&self,
) -> (
InternedString,
Lto,
Option<u32>,
Option<u32>,
bool,
bool,
bool,
bool,
PanicStrategy,
) {
(
self.opt_level,
self.lto,
self.codegen_units,
self.debuginfo,
self.debug_assertions,
self.overflow_checks,
self.rpath,
self.incremental,
self.panic,
)
}
}
#[derive(Clone, Copy, PartialEq, Eq, Debug, Hash, PartialOrd, Ord)]
pub enum Lto {
Bool(bool),
Named(InternedString),
}
#[derive(Clone, Copy, PartialEq, Eq, Debug, Hash, PartialOrd, Ord)]
pub enum PanicStrategy {
Unwind,
Abort,
}
impl fmt::Display for PanicStrategy {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match *self {
PanicStrategy::Unwind => "unwind",
PanicStrategy::Abort => "abort",
}
.fmt(f)
}
}
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, Ord, PartialOrd)]
pub struct UnitFor {
build: bool,
panic_setting: PanicSetting,
}
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, Ord, PartialOrd)]
enum PanicSetting {
AlwaysUnwind,
ReadProfile,
Inherit,
}
impl UnitFor {
pub fn new_normal() -> UnitFor {
UnitFor {
build: false,
panic_setting: PanicSetting::ReadProfile,
}
}
pub fn new_build() -> UnitFor {
UnitFor {
build: true,
panic_setting: PanicSetting::AlwaysUnwind,
}
}
pub fn new_compiler() -> UnitFor {
UnitFor {
build: false,
panic_setting: PanicSetting::AlwaysUnwind,
}
}
pub fn new_test(config: &Config) -> UnitFor {
UnitFor {
build: false,
panic_setting: if config.cli_unstable().panic_abort_tests {
PanicSetting::Inherit
} else {
PanicSetting::AlwaysUnwind
},
}
}
pub fn with_for_host(self, for_host: bool) -> UnitFor {
UnitFor {
build: self.build || for_host,
panic_setting: if for_host {
PanicSetting::AlwaysUnwind
} else {
self.panic_setting
},
}
}
pub fn is_build(self) -> bool {
self.build
}
fn panic_setting(self) -> PanicSetting {
self.panic_setting
}
pub fn all_values() -> &'static [UnitFor] {
static ALL: &[UnitFor] = &[
UnitFor {
build: false,
panic_setting: PanicSetting::ReadProfile,
},
UnitFor {
build: true,
panic_setting: PanicSetting::AlwaysUnwind,
},
UnitFor {
build: false,
panic_setting: PanicSetting::AlwaysUnwind,
},
UnitFor {
build: false,
panic_setting: PanicSetting::Inherit,
},
];
ALL
}
}
#[derive(Clone, Debug, Deserialize, Default)]
pub struct ConfigProfiles {
dev: Option<TomlProfile>,
release: Option<TomlProfile>,
}
impl ConfigProfiles {
pub fn validate(&self, features: &Features, warnings: &mut Vec<String>) -> CargoResult<()> {
if let Some(ref profile) = self.dev {
profile
.validate("dev", features, warnings)
.chain_err(|| failure::format_err!("config profile `profile.dev` is not valid"))?;
}
if let Some(ref profile) = self.release {
profile
.validate("release", features, warnings)
.chain_err(|| {
failure::format_err!("config profile `profile.release` is not valid")
})?;
}
Ok(())
}
}