zv 0.2.1

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

use super::ShellType;
use crate::tools::is_tty;
use sysinfo::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()
        && 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")
            && 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")
        && term_program == "vscode" {
            // VS Code integrated terminal, check for shell preference
            if let Ok(vscode_shell) = std::env::var("VSCODE_SHELL_INTEGRATION")
                && 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()
        && 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")
        && 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")
        && term_program == "vscode" {
            // VS Code integrated terminal, check for shell preference
            if let Ok(vscode_shell) = std::env::var("VSCODE_SHELL_INTEGRATION")
                && let Some(shell) = detect_shell_from_string(&vscode_shell) {
                    return shell;
                }
        }

    ShellType::Unknown
}