Skip to main content

mangofetch_core/core/
process.rs

1fn enhanced_path() -> Option<String> {
2    use std::sync::OnceLock;
3    static CACHED: OnceLock<Option<String>> = OnceLock::new();
4    CACHED
5        .get_or_init(|| {
6            let bin_dir = crate::core::paths::app_data_dir()?.join("bin");
7            let sep = if cfg!(windows) { ";" } else { ":" };
8            let current = std::env::var("PATH").unwrap_or_default();
9
10            #[allow(unused_mut)]
11            let mut extra_dirs: Vec<String> = vec![bin_dir.display().to_string()];
12
13            #[cfg(target_os = "macos")]
14            {
15                extra_dirs.push("/opt/homebrew/bin".into());
16                extra_dirs.push("/usr/local/bin".into());
17            }
18
19            #[cfg(target_os = "linux")]
20            {
21                if let Some(home) = dirs::home_dir() {
22                    extra_dirs.push(home.join(".local").join("bin").display().to_string());
23                }
24                extra_dirs.push("/usr/local/bin".into());
25            }
26
27            Some(format!("{}{}{}", extra_dirs.join(sep), sep, current))
28        })
29        .clone()
30}
31
32pub fn command<S: AsRef<std::ffi::OsStr>>(program: S) -> tokio::process::Command {
33    let mut cmd = tokio::process::Command::new(program);
34    #[cfg(target_os = "windows")]
35    cmd.creation_flags(0x08000000);
36    if let Some(path) = enhanced_path() {
37        cmd.env("PATH", path);
38    }
39    cmd.env_remove("PYTHONHOME");
40    cmd.env_remove("PYTHONPATH");
41    cmd.env("PYTHONIOENCODING", "utf-8");
42    cmd.env("PYTHONUTF8", "1");
43    cmd.stdin(std::process::Stdio::null());
44    cmd
45}
46
47pub fn std_command<S: AsRef<std::ffi::OsStr>>(program: S) -> std::process::Command {
48    let mut cmd = std::process::Command::new(program);
49    #[cfg(target_os = "windows")]
50    {
51        use std::os::windows::process::CommandExt;
52        cmd.creation_flags(0x08000000);
53    }
54    if let Some(path) = enhanced_path() {
55        cmd.env("PATH", path);
56    }
57    cmd.env_remove("PYTHONHOME");
58    cmd.env_remove("PYTHONPATH");
59    cmd.env("PYTHONIOENCODING", "utf-8");
60    cmd.env("PYTHONUTF8", "1");
61    cmd.stdin(std::process::Stdio::null());
62    cmd
63}