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

use serde::{Deserialize, Serialize};

pub const MAX_OS_SHELL_LEN: usize = 20;

pub const HEADER_API_KEY: &str = "api-key";

#[derive(Serialize, Deserialize)]
pub struct Question {
    pub os: String,
    pub shell: String,
    pub prompt: String,
    pub explain: bool,
}

#[derive(Serialize, Deserialize)]
pub struct Error {
    pub message: String,
    pub code: Option<u16>,
}

pub fn check_or_truncate_max_os_shell(value: &str) -> &str {
    if value.len() > MAX_OS_SHELL_LEN {
        &value[..MAX_OS_SHELL_LEN]
    } else {
        value
    }
}

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

    #[test]
    fn test_question_serialization() {
        let question = Question {
            os: "Linux".to_string(),
            shell: "bash".to_string(),
            prompt: "What is Rust?".to_string(),
            explain: false,
        };

        let json = serde_json::to_string(&question).unwrap();
        let deserialized_question: Question = serde_json::from_str(&json).unwrap();

        assert_eq!(question.os, deserialized_question.os);
        assert_eq!(question.shell, deserialized_question.shell);
        assert_eq!(question.prompt, deserialized_question.prompt);
        assert_eq!(question.explain, deserialized_question.explain);
    }
}