starfire 0.4.0

CLI router and tool manager — manage API keys and run CLI tools with credentials automatically injected
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

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

use crate::errors::StarfireError;

#[derive(Serialize, Deserialize, Default)]
pub struct Credentials {
    #[serde(default)]
    pub keys: HashMap<String, String>,
}

/// Returns ~/.starfire, creating it if needed.
pub fn starfire_dir() -> Result<PathBuf, StarfireError> {
    let home = dirs::home_dir().ok_or_else(|| {
        StarfireError::IoError(std::io::Error::new(
            std::io::ErrorKind::NotFound,
            "could not determine home directory",
        ))
    })?;
    let dir = home.join(".starfire");
    fs::create_dir_all(&dir)?;
    Ok(dir)
}

fn credentials_path() -> Result<PathBuf, StarfireError> {
    Ok(starfire_dir()?.join("credentials.json"))
}

pub fn load_credentials() -> Result<Credentials, StarfireError> {
    let path = credentials_path()?;
    if !path.exists() {
        return Ok(Credentials::default());
    }
    let data = fs::read_to_string(&path)?;
    Ok(serde_json::from_str(&data)?)
}

pub fn save_credentials(creds: &Credentials) -> Result<(), StarfireError> {
    let path = credentials_path()?;
    let data = serde_json::to_string_pretty(creds)?;
    fs::write(&path, data)?;

    // Lock down permissions to owner-only
    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        let perms = fs::Permissions::from_mode(0o600);
        fs::set_permissions(&path, perms)?;
    }

    Ok(())
}