yo 1.3.5

Ask your terminal anything using AI (OpenAI or Ollama).
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

use serde::{Deserialize, Serialize};
use std::fs;
use std::path::PathBuf;

/// Resolve $XDG_CONFIG_HOME or fallback to ~/.config
fn base_dir() -> PathBuf {
    if let Ok(xdg) = std::env::var("XDG_CONFIG_HOME") {
        PathBuf::from(xdg)
    } else if let Some(home) = dirs::home_dir() {
        home.join(".config")
    } else {
        panic!("could not determine config directory");
    }
}

#[derive(Serialize, Deserialize, Debug)]
pub struct Config {
    /// "openai" or "ollama"
    pub source: String,
    /// model ID, e.g. "gpt-4" or "llama3:latest"
    pub model: String,
    pub openai_api_key: Option<String>,
}

pub fn get_config_path() -> PathBuf {
    let dir = base_dir().join("yo");
    fs::create_dir_all(&dir).unwrap();
    dir.join("config.toml")
}

pub fn load_or_create_config() -> Config {
    let path = get_config_path();
    if !path.exists() {
        eprintln!("no config found. run `yo setup` to get started.");
        std::process::exit(1);
    }
    let s = fs::read_to_string(&path).expect("failed to read config");
    toml::from_str(&s).expect("invalid config format")
}

pub fn save_config(cfg: &Config) {
    let path = get_config_path();
    let toml = toml::to_string_pretty(cfg).unwrap();
    fs::write(&path, toml).unwrap();
}