saya-cli 0.2.0

Database-aware AI agent for the terminal: full-screen TUI, schema discovery, and bounded read-only SQL over PostgreSQL, MySQL, DuckDB, and Snowflake.
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

use super::runtime::RuntimeError;
use saya_config::{ConfigFile, ConnectionsFile, parse_explicit_env_file};
use std::{
    collections::BTreeMap,
    fs,
    path::{Path, PathBuf},
};

pub(crate) struct Paths {
    pub(crate) user_config: Option<PathBuf>,
    pub(crate) user_connections: Option<PathBuf>,
    pub(crate) project_config: Option<PathBuf>,
    pub(crate) project_connections: Option<PathBuf>,
}

impl Paths {
    pub(crate) fn discover(cwd: &Path, user: &Path) -> Self {
        let project = cwd.join(".saya");
        Self {
            user_config: Some(user.join("config.toml")),
            user_connections: Some(user.join("connections.toml")),
            project_config: Some(project.join("config.toml")),
            project_connections: Some(project.join("connections.toml")),
        }
    }
}

pub(crate) fn user_config_dir() -> PathBuf {
    if let Some(path) = std::env::var_os("SAYA_CONFIG_HOME") {
        return PathBuf::from(path).join("saya");
    }
    if let Some(path) = std::env::var_os("XDG_CONFIG_HOME") {
        return PathBuf::from(path).join("saya");
    }
    if let Some(path) = std::env::var_os("APPDATA") {
        return PathBuf::from(path).join("saya");
    }
    PathBuf::from(std::env::var_os("HOME").unwrap_or_else(|| ".".into())).join(".config/saya")
}

pub(crate) fn read_config(path: &Option<PathBuf>) -> Result<Option<ConfigFile>, RuntimeError> {
    let Some(path) = path else { return Ok(None) };
    if !path.exists() {
        return Ok(None);
    }
    let value = fs::read_to_string(path).map_err(|source| RuntimeError::Read {
        path: path.clone(),
        source,
    })?;
    Ok(Some(ConfigFile::from_toml(&value)?))
}

pub(crate) fn read_required_config(path: &Path) -> Result<ConfigFile, RuntimeError> {
    if !path.exists() {
        return Err(RuntimeError::Missing(path.into()));
    }
    let value = fs::read_to_string(path).map_err(|source| RuntimeError::Read {
        path: path.into(),
        source,
    })?;
    Ok(ConfigFile::from_toml(&value)?)
}

pub(crate) fn read_connections(
    path: Option<&PathBuf>,
    explicit: bool,
) -> Result<ConnectionsFile, RuntimeError> {
    let Some(path) = path else {
        return Ok(ConnectionsFile::default());
    };
    if !path.exists() {
        return if explicit {
            Err(RuntimeError::Missing(path.clone()))
        } else {
            Ok(ConnectionsFile::default())
        };
    }
    let value = fs::read_to_string(path).map_err(|source| RuntimeError::Read {
        path: path.clone(),
        source,
    })?;
    Ok(ConnectionsFile::from_toml(&value)?)
}

pub(crate) fn read_env_file(path: &Path) -> Result<BTreeMap<String, String>, RuntimeError> {
    if !path.exists() {
        return Err(RuntimeError::Missing(path.into()));
    }
    let value = fs::read_to_string(path).map_err(|source| RuntimeError::Read {
        path: path.into(),
        source,
    })?;
    Ok(parse_explicit_env_file(&value)?)
}

pub(crate) fn process_env() -> BTreeMap<String, String> {
    std::env::vars().collect()
}