zv 0.1.2

Ziglang Version Manager and Project Starter
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
130
131
132
133

use super::ShellType;
use crate::tools::is_tty;
use sysinfo::{Pid, Process, System};

/// Get the parent process name using sysinfo
pub fn get_parent_process_name() -> Option<String> {
    let mut system = System::new();
    system.refresh_processes_specifics(
        sysinfo::ProcessesToUpdate::All,
        true,
        sysinfo::ProcessRefreshKind::everything(),
    );

    let current_pid = sysinfo::get_current_pid().ok()?;
    let current_process = system.process(current_pid)?;
    let parent_pid = current_process.parent()?;
    let parent_process = system.process(parent_pid)?;

    Some(parent_process.name().to_string_lossy().to_string())
}

/// Detect shell from parent process name
pub fn detect_shell_from_parent() -> Option<ShellType> {
    let parent_name = get_parent_process_name()?;
    detect_shell_from_string(&parent_name.to_lowercase())
}

/// Detect shell from any string containing shell information
fn detect_shell_from_string(shell_str: &str) -> Option<ShellType> {
    if shell_str.contains("bash") {
        Some(ShellType::Bash)
    } else if shell_str.contains("zsh") {
        Some(ShellType::Zsh)
    } else if shell_str.contains("fish") {
        Some(ShellType::Fish)
    } else if shell_str.contains("powershell") || shell_str.contains("pwsh") {
        Some(ShellType::PowerShell)
    } else if shell_str.contains("cmd") {
        Some(ShellType::Cmd)
    } else if shell_str.contains("tcsh") || shell_str.contains("csh") {
        Some(ShellType::Tcsh)
    } else if shell_str.contains("nu") {
        Some(ShellType::Nu)
    } else if shell_str.contains("sh") && !shell_str.contains("bash") && !shell_str.contains("zsh")
    {
        Some(ShellType::Posix)
    } else {
        None
    }
}

/// Main shell detection logic
pub fn detect_shell() -> ShellType {
    if cfg!(windows) {
        tracing::debug!(target: "shell_detection", "cfg!(windows)");
        detect_windows_shell()
    } else {
        tracing::debug!(target: "shell_detection", "cfg!(not(windows))");
        detect_unix_shell()
    }
}

/// Windows-specific shell detection
fn detect_windows_shell() -> ShellType {
    // First, try to detect from parent process if we're in a TTY
    if is_tty() {
        if let Some(shell) = detect_shell_from_parent() {
            return shell;
        }
    }

    // Check if we're in WSL (Unix shells on Windows)
    if std::env::var("WSL_DISTRO_NAME").is_ok() || std::env::var("WSL_INTEROP").is_ok() {
        // In WSL, SHELL variable should work properly
        if let Ok(shell) = std::env::var("SHELL") {
            if let Some(detected) = detect_shell_from_string(&shell) {
                return detected;
            }
        }
        return ShellType::Bash; // Default for WSL
    }

    // Check for PowerShell environment indicators
    if std::env::var("PSModulePath").is_ok() {
        return ShellType::PowerShell;
    }

    // Additional checks for specific environments
    if let Ok(term_program) = std::env::var("TERM_PROGRAM") {
        if term_program == "vscode" {
            // VS Code integrated terminal, check for shell preference
            if let Ok(vscode_shell) = std::env::var("VSCODE_SHELL_INTEGRATION") {
                if let Some(shell) = detect_shell_from_string(&vscode_shell) {
                    return shell;
                }
            }
        }
    }

    // Default to PowerShell on modern Windows
    ShellType::PowerShell
}

/// Unix-like systems shell detection
fn detect_unix_shell() -> ShellType {
    // First, try to detect from parent process if we're in a TTY
    if is_tty() {
        if let Some(shell) = detect_shell_from_parent() {
            return shell;
        }
    }

    // Use SHELL environment variable (standard on Unix-like systems)
    if let Ok(shell) = std::env::var("SHELL") {
        if let Some(detected) = detect_shell_from_string(&shell) {
            return detected;
        }
    }

    // Additional checks for specific environments
    if let Ok(term_program) = std::env::var("TERM_PROGRAM") {
        if term_program == "vscode" {
            // VS Code integrated terminal, check for shell preference
            if let Ok(vscode_shell) = std::env::var("VSCODE_SHELL_INTEGRATION") {
                if let Some(shell) = detect_shell_from_string(&vscode_shell) {
                    return shell;
                }
            }
        }
    }

    ShellType::Unknown
}