mirror of
https://github.com/uutils/coreutils
synced 2024-12-13 23:02:38 +00:00
181 lines
7.2 KiB
Rust
181 lines
7.2 KiB
Rust
// spell-checker:ignore (vars) krate
|
|
|
|
use std::env;
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::path::Path;
|
|
|
|
pub fn main() {
|
|
// println!("cargo:warning=Running build.rs");
|
|
|
|
if let Ok(profile) = env::var("PROFILE") {
|
|
println!("cargo:rustc-cfg=build={:?}", profile);
|
|
}
|
|
|
|
let env_feature_prefix: &str = "CARGO_FEATURE_";
|
|
let feature_prefix: &str = "feat_";
|
|
let override_prefix: &str = "uu_";
|
|
|
|
let out_dir = env::var("OUT_DIR").unwrap();
|
|
// println!("cargo:warning=out_dir={}", out_dir);
|
|
let manifest_dir = env::var("CARGO_MANIFEST_DIR").unwrap().replace('\\', "/");
|
|
// println!("cargo:warning=manifest_dir={}", manifest_dir);
|
|
let util_tests_dir = format!("{}/tests/by-util", manifest_dir);
|
|
// println!("cargo:warning=util_tests_dir={}", util_tests_dir);
|
|
|
|
let mut crates = Vec::new();
|
|
for (key, val) in env::vars() {
|
|
if val == "1" && key.starts_with(env_feature_prefix) {
|
|
let krate = key[env_feature_prefix.len()..].to_lowercase();
|
|
match krate.as_ref() {
|
|
"default" | "macos" | "unix" | "windows" | "selinux" => continue, // common/standard feature names
|
|
"nightly" | "test_unimplemented" => continue, // crate-local custom features
|
|
"test" => continue, // over-ridden with 'uu_test' to avoid collision with rust core crate 'test'
|
|
s if s.starts_with(feature_prefix) => continue, // crate feature sets
|
|
_ => {} // util feature name
|
|
}
|
|
crates.push(krate.to_string());
|
|
}
|
|
}
|
|
crates.sort();
|
|
|
|
let mut mf = File::create(Path::new(&out_dir).join("uutils_map.rs")).unwrap();
|
|
let mut tf = File::create(Path::new(&out_dir).join("test_modules.rs")).unwrap();
|
|
|
|
mf.write_all(
|
|
"type UtilityMap<T> = HashMap<&'static str, (fn(T) -> i32, fn() -> App<'static, 'static>)>;\n\
|
|
\n\
|
|
fn util_map<T: uucore::Args>() -> UtilityMap<T> {\n\
|
|
\t#[allow(unused_mut)]\n\
|
|
\t#[allow(clippy::let_and_return)]\n\
|
|
\tlet mut map = UtilityMap::new();\n\
|
|
"
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
|
|
for krate in crates {
|
|
match krate.as_ref() {
|
|
// 'test' is named uu_test to avoid collision with rust core crate 'test'.
|
|
// It can also be invoked by name '[' for the '[ expr ] syntax'.
|
|
"uu_test" => {
|
|
mf.write_all(
|
|
format!(
|
|
"\
|
|
\tmap.insert(\"test\", ({krate}::uumain, {krate}::uu_app));\n\
|
|
\t\tmap.insert(\"[\", ({krate}::uumain, {krate}::uu_app));\n\
|
|
",
|
|
krate = krate
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
tf.write_all(
|
|
format!(
|
|
"#[path=\"{dir}/test_test.rs\"]\nmod test_test;\n",
|
|
dir = util_tests_dir,
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap()
|
|
}
|
|
k if k.starts_with(override_prefix) => {
|
|
mf.write_all(
|
|
format!(
|
|
"\tmap.insert(\"{k}\", ({krate}::uumain, {krate}::uu_app));\n",
|
|
k = &krate[override_prefix.len()..],
|
|
krate = krate
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
tf.write_all(
|
|
format!(
|
|
"#[path=\"{dir}/test_{k}.rs\"]\nmod test_{k};\n",
|
|
k = &krate[override_prefix.len()..],
|
|
dir = util_tests_dir,
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap()
|
|
}
|
|
"false" | "true" => {
|
|
mf.write_all(
|
|
format!(
|
|
"\tmap.insert(\"{krate}\", (r#{krate}::uumain, r#{krate}::uu_app));\n",
|
|
krate = krate
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
tf.write_all(
|
|
format!(
|
|
"#[path=\"{dir}/test_{krate}.rs\"]\nmod test_{krate};\n",
|
|
krate = krate,
|
|
dir = util_tests_dir,
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap()
|
|
}
|
|
"hashsum" => {
|
|
mf.write_all(
|
|
format!(
|
|
"\
|
|
\tmap.insert(\"{krate}\", ({krate}::uumain, {krate}::uu_app_custom));\n\
|
|
\t\tmap.insert(\"md5sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha1sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha224sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha256sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha384sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha512sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha3sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha3-224sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha3-256sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha3-384sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"sha3-512sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"shake128sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
\t\tmap.insert(\"shake256sum\", ({krate}::uumain, {krate}::uu_app_common));\n\
|
|
",
|
|
krate = krate
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
tf.write_all(
|
|
format!(
|
|
"#[path=\"{dir}/test_{krate}.rs\"]\nmod test_{krate};\n",
|
|
krate = krate,
|
|
dir = util_tests_dir,
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap()
|
|
}
|
|
_ => {
|
|
mf.write_all(
|
|
format!(
|
|
"\tmap.insert(\"{krate}\", ({krate}::uumain, {krate}::uu_app));\n",
|
|
krate = krate
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap();
|
|
tf.write_all(
|
|
format!(
|
|
"#[path=\"{dir}/test_{krate}.rs\"]\nmod test_{krate};\n",
|
|
krate = krate,
|
|
dir = util_tests_dir,
|
|
)
|
|
.as_bytes(),
|
|
)
|
|
.unwrap()
|
|
}
|
|
}
|
|
}
|
|
|
|
mf.write_all(b"map\n}\n").unwrap();
|
|
|
|
mf.flush().unwrap();
|
|
tf.flush().unwrap();
|
|
}
|