#![allow(clippy::needless_doctest_main)]
#![deny(missing_docs)]
#[cfg(test)]
#[macro_use]
extern crate lazy_static;
#[cfg(test)]
mod test;
use heck::ShoutySnakeCase;
use std::collections::HashMap;
use std::env;
use std::fmt;
use std::fs;
use std::io::Read;
use std::path::{Path, PathBuf};
use std::str::FromStr;
use strum::IntoEnumIterator;
use strum_macros::{EnumIter, EnumString};
use thiserror::Error;
use version_compare::VersionCompare;
#[derive(Error, Debug)]
pub enum Error {
#[error(transparent)]
PkgConfig(#[from] pkg_config::Error),
#[error("Failed to build {0}: {1}")]
BuildInternalClosureError(String, #[source] BuildInternalClosureError),
#[error("{0}")]
FailToRead(String, #[source] std::io::Error),
#[error("{0}")]
InvalidMetadata(String),
#[error("You should define at least one lib using {} or {}", EnvVariable::new_lib(.0).to_string(), EnvVariable::new_lib_framework(.0))]
MissingLib(String),
#[error("{0}")]
BuildInternalInvalid(String),
#[error("Missing build internal closure for {0} (version {1})")]
BuildInternalNoClosure(String, String),
#[error("Internally built {0} {1} but minimum required version is {2}")]
BuildInternalWrongVersion(String, String, String),
}
#[derive(Error, Debug)]
pub enum BuildInternalClosureError {
#[error(transparent)]
PkgConfig(#[from] pkg_config::Error),
#[error("{0}")]
Failed(String),
}
impl BuildInternalClosureError {
pub fn failed(details: &str) -> Self {
Self::Failed(details.to_string())
}
}
#[derive(Debug, PartialEq, EnumIter)]
enum EnvVariable {
Lib(String),
LibFramework(String),
SearchNative(String),
SearchFramework(String),
Include(String),
NoPkgConfig(String),
BuildInternal(Option<String>),
}
impl EnvVariable {
fn new_lib(lib: &str) -> Self {
Self::Lib(lib.to_string())
}
fn new_lib_framework(lib: &str) -> Self {
Self::LibFramework(lib.to_string())
}
fn new_search_native(lib: &str) -> Self {
Self::SearchNative(lib.to_string())
}
fn new_search_framework(lib: &str) -> Self {
Self::SearchFramework(lib.to_string())
}
fn new_include(lib: &str) -> Self {
Self::Include(lib.to_string())
}
fn new_no_pkg_config(lib: &str) -> Self {
Self::NoPkgConfig(lib.to_string())
}
fn new_build_internal(lib: Option<&str>) -> Self {
Self::BuildInternal(lib.map(|l| l.to_string()))
}
fn suffix(&self) -> &'static str {
match self {
EnvVariable::Lib(_) => "LIB",
EnvVariable::LibFramework(_) => "LIB_FRAMEWORK",
EnvVariable::SearchNative(_) => "SEARCH_NATIVE",
EnvVariable::SearchFramework(_) => "SEARCH_FRAMEWORK",
EnvVariable::Include(_) => "INCLUDE",
EnvVariable::NoPkgConfig(_) => "NO_PKG_CONFIG",
EnvVariable::BuildInternal(_) => "BUILD_INTERNAL",
}
}
}
impl fmt::Display for EnvVariable {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let suffix = match self {
EnvVariable::Lib(lib)
| EnvVariable::LibFramework(lib)
| EnvVariable::SearchNative(lib)
| EnvVariable::SearchFramework(lib)
| EnvVariable::Include(lib)
| EnvVariable::NoPkgConfig(lib)
| EnvVariable::BuildInternal(Some(lib)) => {
format!("{}_{}", lib.to_shouty_snake_case(), self.suffix())
}
EnvVariable::BuildInternal(None) => self.suffix().to_string(),
};
write!(f, "SYSTEM_DEPS_{}", suffix)
}
}
type FnBuildInternal =
dyn FnOnce(&str, &str) -> std::result::Result<Library, BuildInternalClosureError>;
pub struct Config {
env: EnvVariables,
build_internals: HashMap<String, Box<FnBuildInternal>>,
}
impl Default for Config {
fn default() -> Self {
Self::new_with_env(EnvVariables::Environnement)
}
}
impl Config {
pub fn new() -> Self {
Self::default()
}
fn new_with_env(env: EnvVariables) -> Self {
Self {
env,
build_internals: HashMap::new(),
}
}
pub fn probe(self) -> Result<HashMap<String, Library>, Error> {
let (libraries, flags) = self.probe_full()?;
println!("{}", flags);
Ok(libraries)
}
pub fn add_build_internal<F>(self, name: &str, func: F) -> Self
where
F: 'static + FnOnce(&str, &str) -> std::result::Result<Library, BuildInternalClosureError>,
{
let mut build_internals = self.build_internals;
build_internals.insert(name.to_string(), Box::new(func));
Self {
env: self.env,
build_internals,
}
}
fn probe_full(mut self) -> Result<(HashMap<String, Library>, BuildFlags), Error> {
let mut libraries = self.probe_pkg_config()?;
self.override_from_flags(&mut libraries);
let flags = self.gen_flags(&libraries)?;
Ok((libraries, flags))
}
fn probe_pkg_config(&mut self) -> Result<HashMap<String, Library>, Error> {
let dir = self
.env
.get("CARGO_MANIFEST_DIR")
.ok_or_else(|| Error::InvalidMetadata("$CARGO_MANIFEST_DIR not set".into()))?;
let mut path = PathBuf::from(dir);
path.push("Cargo.toml");
let mut manifest = fs::File::open(&path)
.map_err(|e| Error::FailToRead(format!("Error opening {}", path.display()), e))?;
let mut manifest_str = String::new();
manifest
.read_to_string(&mut manifest_str)
.map_err(|e| Error::FailToRead(format!("Error reading {}", path.display()), e))?;
let toml = manifest_str.parse::<toml::Value>().map_err(|e| {
Error::InvalidMetadata(format!(
"Error parsing TOML from {}: {:?}",
path.display(),
e
))
})?;
let key = "package.metadata.system-deps";
let meta = toml
.get("package")
.and_then(|v| v.get("metadata"))
.and_then(|v| v.get("system-deps"))
.ok_or_else(|| Error::InvalidMetadata(format!("No {} in {}", key, path.display())))?;
let table = meta.as_table().ok_or_else(|| {
Error::InvalidMetadata(format!("{} not a table in {}", key, path.display()))
})?;
let mut libraries = HashMap::new();
for (name, value) in table {
let (lib_name, version) = match value {
toml::Value::String(ref s) => (name, s),
toml::Value::Table(ref t) => {
let mut feature = None;
let mut version = None;
let mut lib_name = None;
let mut enabled_feature_versions = Vec::new();
for (tname, tvalue) in t {
match (tname.as_str(), tvalue) {
("feature", &toml::Value::String(ref s)) => {
feature = Some(s);
}
("version", &toml::Value::String(ref s)) => {
version = Some(s);
}
("name", &toml::Value::String(ref s)) => {
lib_name = Some(s);
}
("feature-versions", &toml::Value::Table(ref feature_versions)) => {
for (k, v) in feature_versions {
match (k.as_str(), v) {
(_, &toml::Value::String(ref feat_vers)) => {
if self.has_feature(&k) {
enabled_feature_versions.push(feat_vers);
}
}
_ => {
return Err(Error::InvalidMetadata(format!(
"Unexpected feature-version key: {} type {}",
k,
v.type_str()
)))
}
}
}
}
_ => {
return Err(Error::InvalidMetadata(format!(
"Unexpected key {}.{}.{} type {}",
key,
name,
tname,
tvalue.type_str()
)))
}
}
}
if let Some(feature) = feature {
if !self.has_feature(feature) {
continue;
}
}
let version = {
if !enabled_feature_versions.is_empty() {
enabled_feature_versions.sort_by(|a, b| {
VersionCompare::compare(b, a)
.expect("failed to compare versions")
.ord()
.expect("invalid version")
});
Some(enabled_feature_versions[0])
} else {
version
}
};
(
lib_name.unwrap_or(name),
version.ok_or_else(|| {
Error::InvalidMetadata(format!("No version in {}.{}", key, name))
})?,
)
}
_ => {
return Err(Error::InvalidMetadata(format!(
"{}.{} not a string or table",
key, name
)))
}
};
let build_internal = self.get_build_internal_status(name)?;
let library = if self.env.contains(&EnvVariable::new_no_pkg_config(name)) {
Library::from_env_variables()
} else if build_internal == BuildInternal::Always {
self.call_build_internal(lib_name, version)?
} else {
match pkg_config::Config::new()
.atleast_version(&version)
.print_system_libs(false)
.cargo_metadata(false)
.probe(lib_name)
{
Ok(lib) => Library::from_pkg_config(lib),
Err(e) => {
if build_internal == BuildInternal::Auto {
self.call_build_internal(name, version)?
} else {
return Err(e.into());
}
}
}
};
libraries.insert(name.clone(), library);
}
Ok(libraries)
}
fn get_build_internal_env_var(&self, var: EnvVariable) -> Result<Option<BuildInternal>, Error> {
match self.env.get(&var).as_deref() {
Some(s) => {
let b = BuildInternal::from_str(s).map_err(|_| {
Error::BuildInternalInvalid(format!(
"Invalid value in {}: {} (allowed: 'auto', 'always', 'never')",
var, s
))
})?;
Ok(Some(b))
}
None => Ok(None),
}
}
fn get_build_internal_status(&self, name: &str) -> Result<BuildInternal, Error> {
match self.get_build_internal_env_var(EnvVariable::new_build_internal(Some(name)))? {
Some(b) => Ok(b),
None => Ok(self
.get_build_internal_env_var(EnvVariable::new_build_internal(None))?
.unwrap_or_default()),
}
}
fn call_build_internal(&mut self, name: &str, version: &str) -> Result<Library, Error> {
let lib = match self.build_internals.remove(name) {
Some(f) => {
f(name, version).map_err(|e| Error::BuildInternalClosureError(name.into(), e))?
}
None => return Err(Error::BuildInternalNoClosure(name.into(), version.into())),
};
match VersionCompare::compare(&lib.version, version) {
Ok(version_compare::CompOp::Lt) => Err(Error::BuildInternalWrongVersion(
name.into(),
lib.version.clone(),
version.into(),
)),
_ => Ok(lib),
}
}
fn override_from_flags(&self, libraries: &mut HashMap<String, Library>) {
for (name, lib) in libraries.iter_mut() {
if let Some(value) = self.env.get(&EnvVariable::new_search_native(name)) {
lib.link_paths = split_paths(&value);
}
if let Some(value) = self.env.get(&EnvVariable::new_search_framework(name)) {
lib.framework_paths = split_paths(&value);
}
if let Some(value) = self.env.get(&EnvVariable::new_lib(name)) {
lib.libs = split_string(&value);
}
if let Some(value) = self.env.get(&EnvVariable::new_lib_framework(name)) {
lib.frameworks = split_string(&value);
}
if let Some(value) = self.env.get(&EnvVariable::new_include(name)) {
lib.include_paths = split_paths(&value);
}
}
}
fn gen_flags(&self, libraries: &HashMap<String, Library>) -> Result<BuildFlags, Error> {
let mut flags = BuildFlags::new();
let mut include_paths = Vec::new();
for (name, lib) in libraries.iter() {
include_paths.extend(lib.include_paths.clone());
if lib.source == Source::EnvVariables
&& lib.libs.is_empty()
&& lib.frameworks.is_empty()
{
return Err(Error::MissingLib(name.clone()));
}
lib.link_paths
.iter()
.for_each(|l| flags.add(BuildFlag::SearchNative(l.to_string_lossy().to_string())));
lib.framework_paths.iter().for_each(|f| {
flags.add(BuildFlag::SearchFramework(f.to_string_lossy().to_string()))
});
lib.libs
.iter()
.for_each(|l| flags.add(BuildFlag::Lib(l.clone())));
lib.frameworks
.iter()
.for_each(|f| flags.add(BuildFlag::LibFramework(f.clone())));
}
if !include_paths.is_empty() {
if let Ok(paths) = std::env::join_paths(include_paths) {
flags.add(BuildFlag::Include(paths.to_string_lossy().to_string()));
}
}
flags.add(BuildFlag::RerunIfEnvChanged(
EnvVariable::new_build_internal(None),
));
for (name, _lib) in libraries.iter() {
for var in EnvVariable::iter() {
let var = match var {
EnvVariable::Lib(_) => EnvVariable::new_lib(name),
EnvVariable::LibFramework(_) => EnvVariable::new_lib_framework(name),
EnvVariable::SearchNative(_) => EnvVariable::new_search_native(name),
EnvVariable::SearchFramework(_) => EnvVariable::new_search_framework(name),
EnvVariable::Include(_) => EnvVariable::new_include(name),
EnvVariable::NoPkgConfig(_) => EnvVariable::new_no_pkg_config(name),
EnvVariable::BuildInternal(_) => EnvVariable::new_build_internal(Some(name)),
};
flags.add(BuildFlag::RerunIfEnvChanged(var));
}
}
Ok(flags)
}
fn has_feature(&self, feature: &str) -> bool {
let var: &str = &format!("CARGO_FEATURE_{}", feature.to_uppercase().replace('-', "_"));
self.env.contains(var)
}
}
#[derive(Debug, PartialEq)]
pub enum Source {
PkgConfig,
EnvVariables,
}
#[derive(Debug)]
pub struct Library {
pub source: Source,
pub libs: Vec<String>,
pub link_paths: Vec<PathBuf>,
pub frameworks: Vec<String>,
pub framework_paths: Vec<PathBuf>,
pub include_paths: Vec<PathBuf>,
pub defines: HashMap<String, Option<String>>,
pub version: String,
}
impl Library {
fn from_pkg_config(l: pkg_config::Library) -> Self {
Self {
source: Source::PkgConfig,
libs: l.libs,
link_paths: l.link_paths,
include_paths: l.include_paths,
frameworks: l.frameworks,
framework_paths: l.framework_paths,
defines: l.defines,
version: l.version,
}
}
fn from_env_variables() -> Self {
Self {
source: Source::EnvVariables,
libs: Vec::new(),
link_paths: Vec::new(),
include_paths: Vec::new(),
frameworks: Vec::new(),
framework_paths: Vec::new(),
defines: HashMap::new(),
version: String::new(),
}
}
pub fn from_internal_pkg_config<P>(
pkg_config_dir: P,
lib: &str,
version: &str,
) -> Result<Self, BuildInternalClosureError>
where
P: AsRef<Path>,
{
let old = env::var("PKG_CONFIG_PATH");
match old {
Ok(ref s) => {
let paths = [s, &pkg_config_dir.as_ref().to_string_lossy().to_string()];
let paths = env::join_paths(paths.iter()).unwrap();
env::set_var("PKG_CONFIG_PATH", paths)
}
Err(_) => env::set_var("PKG_CONFIG_PATH", pkg_config_dir.as_ref()),
}
let lib = pkg_config::Config::new()
.atleast_version(&version)
.print_system_libs(false)
.cargo_metadata(false)
.probe(lib);
env::set_var("PKG_CONFIG_PATH", &old.unwrap_or_else(|_| "".into()));
match lib {
Ok(lib) => Ok(Self::from_pkg_config(lib)),
Err(e) => Err(e.into()),
}
}
}
#[derive(Debug)]
enum EnvVariables {
Environnement,
#[cfg(test)]
Mock(HashMap<&'static str, String>),
}
trait EnvVariablesExt<T> {
fn contains(&self, var: T) -> bool {
self.get(var).is_some()
}
fn get(&self, var: T) -> Option<String>;
}
impl EnvVariablesExt<&str> for EnvVariables {
fn get(&self, var: &str) -> Option<String> {
match self {
EnvVariables::Environnement => env::var(var).ok(),
#[cfg(test)]
EnvVariables::Mock(vars) => vars.get(var).cloned(),
}
}
}
impl EnvVariablesExt<&EnvVariable> for EnvVariables {
fn get(&self, var: &EnvVariable) -> Option<String> {
let s = var.to_string();
let var: &str = s.as_ref();
self.get(var)
}
}
#[derive(Debug, PartialEq)]
enum BuildFlag {
Include(String),
SearchNative(String),
SearchFramework(String),
Lib(String),
LibFramework(String),
RerunIfEnvChanged(EnvVariable),
}
impl fmt::Display for BuildFlag {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
BuildFlag::Include(paths) => write!(f, "include={}", paths),
BuildFlag::SearchNative(lib) => write!(f, "rustc-link-search=native={}", lib),
BuildFlag::SearchFramework(lib) => write!(f, "rustc-link-search=framework={}", lib),
BuildFlag::Lib(lib) => write!(f, "rustc-link-lib={}", lib),
BuildFlag::LibFramework(lib) => write!(f, "rustc-link-lib=framework={}", lib),
BuildFlag::RerunIfEnvChanged(env) => write!(f, "rerun-if-env-changed={}", env),
}
}
}
#[derive(Debug, PartialEq)]
struct BuildFlags(Vec<BuildFlag>);
impl BuildFlags {
fn new() -> Self {
Self(Vec::new())
}
fn add(&mut self, flag: BuildFlag) {
self.0.push(flag);
}
}
impl fmt::Display for BuildFlags {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
for flag in self.0.iter() {
writeln!(f, "cargo:{}", flag)?;
}
Ok(())
}
}
fn split_paths(value: &str) -> Vec<PathBuf> {
if !value.is_empty() {
let paths = env::split_paths(&value);
paths.map(|p| Path::new(&p).into()).collect()
} else {
Vec::new()
}
}
fn split_string(value: &str) -> Vec<String> {
if !value.is_empty() {
value.split(' ').map(|s| s.to_string()).collect()
} else {
Vec::new()
}
}
#[derive(Debug, PartialEq, EnumString)]
#[strum(serialize_all = "snake_case")]
enum BuildInternal {
Auto,
Always,
Never,
}
impl Default for BuildInternal {
fn default() -> Self {
BuildInternal::Never
}
}