slick 0.14.1

async ZSH prompt
Documentation 
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
134
135
136
137
138
139
140

pub mod precmd;
pub mod prompt;

use std::env;
use std::sync::OnceLock;

// Cache for environment variable defaults to avoid repeated env::var() calls
static ENV_CACHE: OnceLock<EnvDefaults> = OnceLock::new();

struct EnvDefaults {
    cmd_max_exec_time: String,
    error_color: String,
    git_action_color: String,
    git_auth_color: String,
    git_auth_symbol: String,
    git_branch_color: String,
    git_fetch: String,
    git_master_branch_color: String,
    git_remote_color: String,
    git_remote_ahead: String,
    git_remote_behind: String,
    git_staged_color: String,
    git_status_color: String,
    git_uname_color: String,
    non_breaking_space: String,
    path_color: String,
    root_color: String,
    root_symbol: String,
    ssh_color: String,
    symbol: String,
    symbol_color: String,
    time_elapsed_color: String,
    vicmd_color: String,
    vicmd_symbol: String,
}

impl EnvDefaults {
    fn new() -> Self {
        Self {
            cmd_max_exec_time: env::var("SLICK_PROMPT_CMD_MAX_EXEC_TIME")
                .unwrap_or_else(|_| "5".into()),
            error_color: env::var("SLICK_PROMPT_ERROR_COLOR").unwrap_or_else(|_| "196".into()),
            git_action_color: env::var("SLICK_PROMPT_GIT_ACTION_COLOR")
                .unwrap_or_else(|_| "3".into()),
            git_auth_color: env::var("SLICK_PROMPT_GIT_AUTH_COLOR")
                .unwrap_or_else(|_| "red".into()),
            git_auth_symbol: env::var("SLICK_PROMPT_GIT_AUTH_SYMBOL")
                .unwrap_or_else(|_| "🔒".into()),
            git_branch_color: env::var("SLICK_PROMPT_GIT_BRANCH_COLOR")
                .unwrap_or_else(|_| "3".into()),
            git_fetch: env::var("SLICK_PROMPT_GIT_FETCH").unwrap_or_else(|_| "1".into()),
            git_master_branch_color: env::var("SLICK_PROMPT_GIT_MASTER_BRANCH_COLOR")
                .unwrap_or_else(|_| "160".into()),
            git_remote_color: env::var("SLICK_PROMPT_GIT_REMOTE_COLOR")
                .unwrap_or_else(|_| "6".into()),
            git_remote_ahead: env::var("SLICK_PROMPT_GIT_REMOTE_AHEAD")
                .unwrap_or_else(|_| "\u{21e1}".into()),
            git_remote_behind: env::var("SLICK_PROMPT_GIT_REMOTE_BEHIND")
                .unwrap_or_else(|_| "\u{21e3}".into()),
            git_staged_color: env::var("SLICK_PROMPT_GIT_STAGED_COLOR")
                .unwrap_or_else(|_| "7".into()),
            git_status_color: env::var("SLICK_PROMPT_GIT_STATUS_COLOR")
                .unwrap_or_else(|_| "5".into()),
            git_uname_color: env::var("SLICK_PROMPT_GIT_UNAME_COLOR")
                .unwrap_or_else(|_| "8".into()),
            non_breaking_space: env::var("SLICK_PROMPT_NON_BREAKING_SPACE")
                .unwrap_or_else(|_| "\u{a0}".into()),
            path_color: env::var("SLICK_PROMPT_PATH_COLOR").unwrap_or_else(|_| "74".into()),
            root_color: env::var("SLICK_PROMPT_ROOT_COLOR").unwrap_or_else(|_| "1".into()),
            root_symbol: env::var("SLICK_PROMPT_ROOT_SYMBOL").unwrap_or_else(|_| "#".into()),
            ssh_color: env::var("SLICK_PROMPT_SSH_COLOR").unwrap_or_else(|_| "8".into()),
            symbol: env::var("SLICK_PROMPT_SYMBOL").unwrap_or_else(|_| "$".into()),
            symbol_color: env::var("SLICK_PROMPT_SYMBOL_COLOR").unwrap_or_else(|_| "5".into()),
            time_elapsed_color: env::var("SLICK_PROMPT_TIME_ELAPSED_COLOR")
                .unwrap_or_else(|_| "3".into()),
            vicmd_color: env::var("SLICK_PROMPT_VICMD_COLOR").unwrap_or_else(|_| "3".into()),
            vicmd_symbol: env::var("SLICK_PROMPT_VICMD_SYMBOL").unwrap_or_else(|_| ">".into()),
        }
    }
}

#[must_use]
pub fn get_env(e: &str) -> String {
    let cache = ENV_CACHE.get_or_init(EnvDefaults::new);

    match e {
        "SLICK_PROMPT_CMD_MAX_EXEC_TIME" => cache.cmd_max_exec_time.clone(),
        "SLICK_PROMPT_ERROR_COLOR" => cache.error_color.clone(),
        "SLICK_PROMPT_GIT_ACTION_COLOR" => cache.git_action_color.clone(),
        "SLICK_PROMPT_GIT_AUTH_COLOR" => cache.git_auth_color.clone(),
        "SLICK_PROMPT_GIT_AUTH_SYMBOL" => cache.git_auth_symbol.clone(),
        "SLICK_PROMPT_GIT_BRANCH_COLOR" => cache.git_branch_color.clone(),
        "SLICK_PROMPT_GIT_FETCH" => cache.git_fetch.clone(),
        "SLICK_PROMPT_GIT_MASTER_BRANCH_COLOR" => cache.git_master_branch_color.clone(),
        "SLICK_PROMPT_GIT_REMOTE_COLOR" => cache.git_remote_color.clone(),
        "SLICK_PROMPT_GIT_REMOTE_AHEAD" => cache.git_remote_ahead.clone(),
        "SLICK_PROMPT_GIT_REMOTE_BEHIND" => cache.git_remote_behind.clone(),
        "SLICK_PROMPT_GIT_STAGED_COLOR" => cache.git_staged_color.clone(),
        "SLICK_PROMPT_GIT_STATUS_COLOR" => cache.git_status_color.clone(),
        "SLICK_PROMPT_GIT_UNAME_COLOR" => cache.git_uname_color.clone(),
        "SLICK_PROMPT_NON_BREAKING_SPACE" => cache.non_breaking_space.clone(),
        "SLICK_PROMPT_PATH_COLOR" => cache.path_color.clone(),
        "SLICK_PROMPT_ROOT_COLOR" => cache.root_color.clone(),
        "SLICK_PROMPT_ROOT_SYMBOL" => cache.root_symbol.clone(),
        "SLICK_PROMPT_SSH_COLOR" => cache.ssh_color.clone(),
        "SLICK_PROMPT_SYMBOL" => cache.symbol.clone(),
        "SLICK_PROMPT_SYMBOL_COLOR" => cache.symbol_color.clone(),
        "SLICK_PROMPT_TIME_ELAPSED_COLOR" => cache.time_elapsed_color.clone(),
        "SLICK_PROMPT_VICMD_COLOR" => cache.vicmd_color.clone(),
        "SLICK_PROMPT_VICMD_SYMBOL" => cache.vicmd_symbol.clone(),
        "SLICK_PROMPT_NO_GIT_UNAME" => env::var(e).unwrap_or_default(),
        "PIPENV_ACTIVE" => env::var(e).unwrap_or_default(),
        "PIPENV_ACTIVE_COLOR" => env::var(e).unwrap_or_else(|_| "7".into()),
        "VIRTUAL_ENV" => env::var(e).unwrap_or_default(),
        _ => "??".into(),
    }
}

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

    #[test]
    fn test_get_env_returns_default_symbol() {
        // Test with an env var that's unlikely to be set in test environment
        assert_eq!(get_env("SLICK_PROMPT_SYMBOL"), "$");
    }

    #[test]
    fn test_get_env_returns_default_colors() {
        assert_eq!(get_env("SLICK_PROMPT_ERROR_COLOR"), "196");
        assert_eq!(get_env("SLICK_PROMPT_PATH_COLOR"), "74");
        assert_eq!(get_env("SLICK_PROMPT_GIT_BRANCH_COLOR"), "3");
    }

    #[test]
    fn test_get_env_unknown_returns_question_marks() {
        assert_eq!(get_env("UNKNOWN_VAR"), "??");
    }
}