use crate::CompatibilityMode;
use std::ffi::OsStr;
use std::path::Path;
pub fn resolve(
strict_flag: Option<bool>,
env_strict: Option<&OsStr>,
argv0: Option<&OsStr>,
) -> CompatibilityMode {
if let Some(flag) = strict_flag {
return if flag {
CompatibilityMode::Strict
} else {
CompatibilityMode::Default
};
}
if let Some(value) = env_strict {
if env_var_is_truthy(value) {
return CompatibilityMode::Strict;
}
}
if let Some(arg0) = argv0 {
if argv0_implies_strict(arg0) {
return CompatibilityMode::Strict;
}
}
CompatibilityMode::Default
}
fn env_var_is_truthy(value: &OsStr) -> bool {
let Some(s) = value.to_str() else {
return false;
};
matches!(
s.trim().to_ascii_lowercase().as_str(),
"1" | "true" | "yes" | "on"
)
}
fn argv0_implies_strict(arg0: &OsStr) -> bool {
let Some(stem) = Path::new(arg0).file_stem() else {
return false;
};
stem == OsStr::new("sponge")
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn explicit_strict_flag_wins() {
assert_eq!(resolve(Some(true), None, None), CompatibilityMode::Strict);
assert_eq!(
resolve(
Some(false),
Some(OsStr::new("1")),
Some(OsStr::new("sponge"))
),
CompatibilityMode::Default,
"explicit --no-strict beats env and argv[0]"
);
}
#[test]
fn env_var_truthy_implies_strict() {
for v in ["1", "true", "yes", "on", "TRUE", " 1 ", "On"] {
assert_eq!(
resolve(None, Some(OsStr::new(v)), None),
CompatibilityMode::Strict,
"env value {v:?} should imply strict"
);
}
}
#[test]
fn env_var_falsy_does_not_imply_strict() {
for v in ["0", "false", "no", "off", ""] {
assert_eq!(
resolve(None, Some(OsStr::new(v)), None),
CompatibilityMode::Default,
"env value {v:?} should NOT imply strict"
);
}
}
#[test]
fn argv0_sponge_implies_strict() {
assert_eq!(
resolve(None, None, Some(OsStr::new("sponge"))),
CompatibilityMode::Strict
);
assert_eq!(
resolve(None, None, Some(OsStr::new("/usr/local/bin/sponge"))),
CompatibilityMode::Strict
);
assert_eq!(
resolve(None, None, Some(OsStr::new("sponge.exe"))),
CompatibilityMode::Strict,
"argv[0] = sponge.exe must imply strict per HINT-004"
);
}
#[cfg(windows)]
#[test]
fn argv0_sponge_implies_strict_windows_backslash_path() {
assert_eq!(
resolve(None, None, Some(OsStr::new("C:\\bin\\sponge.exe"))),
CompatibilityMode::Strict
);
}
#[test]
fn argv0_rusty_sponge_does_not_imply_strict() {
assert_eq!(
resolve(None, None, Some(OsStr::new("rusty-sponge"))),
CompatibilityMode::Default
);
assert_eq!(
resolve(None, None, Some(OsStr::new("rusty-sponge.exe"))),
CompatibilityMode::Default
);
}
#[test]
fn default_when_nothing_set() {
assert_eq!(resolve(None, None, None), CompatibilityMode::Default);
}
}