blob: ba0a14226abdf1ed7fac0773ab6be1ba328a1329 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#[cfg(any(windows, test))]
use std::path::Path;
#[cfg(windows)]
use std::path::PathBuf;
use std::process::Command;
pub(crate) fn command_for_argv(program: &str, args: &[String]) -> Command {
let mut command = command_for_program(program);
command.args(args);
command
}
#[cfg(not(windows))]
fn command_for_program(program: &str) -> Command {
Command::new(program)
}
#[cfg(windows)]
fn command_for_program(program: &str) -> Command {
let resolved = resolve_windows_program(program);
let command_program = resolved.as_ref().map_or_else(
|| std::ffi::OsString::from(program),
|path| path.as_os_str().to_os_string(),
);
if is_windows_batch_file_name(program)
|| resolved
.as_ref()
.is_some_and(|path| is_windows_batch_path(path))
{
let shell =
std::env::var_os("ComSpec").unwrap_or_else(|| r"C:\Windows\System32\cmd.exe".into());
let mut command = Command::new(shell);
command.arg("/d").arg("/c").arg(command_program);
command
} else {
Command::new(command_program)
}
}
#[cfg(windows)]
fn resolve_windows_program(program: &str) -> Option<PathBuf> {
if has_path_separator(program) {
return None;
}
let path = Path::new(program);
if path.extension().is_some() {
return std::env::var_os("PATH").and_then(|path_var| {
std::env::split_paths(&path_var)
.map(|dir| dir.join(program))
.find(|candidate| candidate.is_file())
});
}
let extensions = windows_path_extensions();
std::env::var_os("PATH").and_then(|path_var| {
std::env::split_paths(&path_var).find_map(|dir| {
extensions
.iter()
.map(|extension| dir.join(format!("{program}{extension}")))
.find(|candidate| candidate.is_file())
})
})
}
#[cfg(windows)]
fn windows_path_extensions() -> Vec<String> {
std::env::var_os("PATHEXT")
.map(|value| {
value
.to_string_lossy()
.split(';')
.map(str::trim)
.filter(|part| !part.is_empty())
.map(|part| {
if part.starts_with('.') {
part.to_string()
} else {
format!(".{part}")
}
})
.collect::<Vec<_>>()
})
.filter(|extensions| !extensions.is_empty())
.unwrap_or_else(|| {
vec![
".COM".to_string(),
".EXE".to_string(),
".BAT".to_string(),
".CMD".to_string(),
]
})
}
#[cfg(windows)]
fn has_path_separator(program: &str) -> bool {
program.contains(['/', '\\'])
}
#[cfg(windows)]
fn is_windows_batch_path(path: &Path) -> bool {
path.extension()
.and_then(std::ffi::OsStr::to_str)
.is_some_and(is_windows_batch_extension)
}
#[cfg(any(windows, test))]
fn is_windows_batch_file_name(program: &str) -> bool {
Path::new(program)
.extension()
.and_then(std::ffi::OsStr::to_str)
.is_some_and(is_windows_batch_extension)
}
#[cfg(any(windows, test))]
fn is_windows_batch_extension(extension: &str) -> bool {
extension.eq_ignore_ascii_case("cmd") || extension.eq_ignore_ascii_case("bat")
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn recognizes_windows_batch_extensions_case_insensitively() {
assert!(is_windows_batch_file_name("npm.cmd"));
assert!(is_windows_batch_file_name("script.BAT"));
assert!(!is_windows_batch_file_name("node.exe"));
assert!(!is_windows_batch_file_name("node"));
}
}
|
