shellchat 1.0.39

Transforms natural language into shell commands for execution or explanation.
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

use anyhow::Result;
use std::io::IsTerminal;
use std::{env, process::Command};

lazy_static::lazy_static! {
    pub static ref OS: String = detect_os();
    pub static ref SHELL: Shell = detect_shell();
    pub static ref IS_STDOUT_TERMINAL: bool = std::io::stdout().is_terminal();
}

pub fn detect_os() -> String {
    let os = env::consts::OS;
    if os == "linux" {
        if let Ok(contents) = std::fs::read_to_string("/etc/os-release") {
            for line in contents.lines() {
                if let Some(id) = line.strip_prefix("ID=") {
                    return format!("{os} ({id})");
                }
            }
        }
    }
    os.to_string()
}

pub struct Shell {
    pub name: String,
    pub cmd: String,
    pub arg: String,
    pub history_cmd: Option<String>,
}

impl Shell {
    pub fn new(name: &str, cmd: &str, arg: &str, history_cmd: Option<&str>) -> Self {
        Self {
            name: name.to_string(),
            cmd: cmd.to_string(),
            arg: arg.to_string(),
            history_cmd: history_cmd.map(|cmd| cmd.to_string()),
        }
    }

    pub fn run_command(&self, eval_str: &str) -> Result<i32> {
        let status = Command::new(&self.cmd)
            .arg(&self.arg)
            .arg(eval_str)
            .status()?;

        if status.success() {
            if let Some(history_cmd) = &self.history_cmd {
                let _ = Command::new(&self.cmd)
                    .arg(&self.arg)
                    .arg(format!("{} \"{}\"", history_cmd, eval_str))
                    .status();
            }
        }

        Ok(status.code().unwrap_or_default())
    }
}

pub fn detect_shell() -> Shell {
    let os = env::consts::OS;
    if os == "windows" {
        if let Some(ret) = env::var("PSModulePath").ok().and_then(|v| {
            let v = v.to_lowercase();
            if v.split(';').count() >= 3 {
                if v.contains("powershell\\7\\") {
                    Some(Shell::new("pwsh", "pwsh.exe", "-c", None))
                } else {
                    Some(Shell::new("powershell", "powershell.exe", "-Command", None))
                }
            } else {
                None
            }
        }) {
            ret
        } else {
            Shell::new("cmd", "cmd.exe", "/C", None)
        }
    } else {
        let shell = env::var("SHELL").unwrap_or_else(|_| "/bin/sh".to_string());
        let shell = match shell.rsplit_once('/') {
            Some((_, v)) => v,
            None => &shell,
        };
        match shell {
            "bash" => Shell::new(shell, shell, "-c", Some("history -s")),
            "zsh" => Shell::new(shell, shell, "-c", Some("print -s")),
            "fish" | "pwsh" => Shell::new(shell, shell, "-c", None),
            _ => Shell::new("sh", "sh", "-c", None),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_detect_os() {
        let os = detect_os();
        assert!(!os.is_empty());
    }

    #[test]
    fn test_detect_shell() {
        let shell = detect_shell();
        assert!(!shell.name.is_empty());
        assert!(!shell.cmd.is_empty());
        assert!(!shell.arg.is_empty());
    }

    #[test]
    fn test_run_command() {
        let result = SHELL.run_command("echo Hello, world!");
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), 0);
    }
}