yaak 0.1.2

Translate natural language to bash commands using an OpenAI-compatible LLM
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

use serde::Deserialize;

#[derive(Deserialize, Default)]
pub struct Config {
    pub api_base: Option<String>,
    pub api_key: Option<String>,
    pub model: Option<String>,
    pub language: Option<String>,
    #[serde(default)]
    pub allow_destructive: bool,
}

pub fn load_config() -> Config {
    let mut candidates = Vec::new();
    // XDG-style ~/.config (works on all platforms)
    if let Some(home) = dirs::home_dir() {
        candidates.push(home.join(".config").join("yaak").join("config.toml"));
    }
    // Platform-native config dir (e.g. ~/Library/Application Support on macOS)
    if let Some(config_dir) = dirs::config_dir() {
        candidates.push(config_dir.join("yaak").join("config.toml"));
    }
    for path in candidates {
        if let Ok(contents) = std::fs::read_to_string(&path) {
            if let Ok(config) = toml::from_str(&contents) {
                return config;
            }
        }
    }
    Config::default()
}

pub fn config_path() -> std::path::PathBuf {
    if let Some(home) = dirs::home_dir() {
        let xdg_path = home.join(".config").join("yaak").join("config.toml");
        if xdg_path.exists() {
            return xdg_path;
        }
    }
    if let Some(config_dir) = dirs::config_dir() {
        let native_path = config_dir.join("yaak").join("config.toml");
        if native_path.exists() {
            return native_path;
        }
    }
    // Default: prefer ~/.config on Unix, platform-native elsewhere
    if cfg!(unix) {
        dirs::home_dir()
            .unwrap_or_default()
            .join(".config")
            .join("yaak")
            .join("config.toml")
    } else {
        dirs::config_dir()
            .unwrap_or_default()
            .join("yaak")
            .join("config.toml")
    }
}

pub fn resolve(cli: Option<String>, config: Option<String>, fallback: &str) -> String {
    cli.or(config).unwrap_or_else(|| fallback.to_string())
}

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

    #[test]
    fn resolve_prefers_cli() {
        assert_eq!(
            resolve(Some("cli".into()), Some("config".into()), "fallback"),
            "cli"
        );
    }

    #[test]
    fn resolve_falls_back_to_config() {
        assert_eq!(resolve(None, Some("config".into()), "fallback"), "config");
    }

    #[test]
    fn resolve_falls_back_to_default() {
        assert_eq!(resolve(None, None, "fallback"), "fallback");
    }
}