treadmill-cli 0.3.1

CLI client for the Treadmill distributed hardware testbed
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

use anyhow::{Context, Result};
use serde::{Deserialize, Serialize};
use std::fs;
use std::path::PathBuf;
use xdg::BaseDirectories;

#[derive(Debug, Deserialize, Serialize)]
pub struct Config {
    pub api: Api,
    pub ssh_keys: Option<Vec<String>>,
}

#[derive(Debug, Deserialize, Serialize)]
pub struct Api {
    pub url: String,
}

impl Default for Config {
    fn default() -> Self {
        Config {
            api: Api {
                url: "https://swb.treadmill.ci".to_string(),
            },
            ssh_keys: None,
        }
    }
}

pub fn load_config(config_path: Option<&str>) -> Result<Config> {
    let xdg_dirs = BaseDirectories::with_prefix("treadmill-tb")?;

    let config_path = match config_path {
        Some(path) => PathBuf::from(path),
        None => xdg_dirs.find_config_file("config.toml").unwrap_or_else(|| {
            xdg_dirs
                .place_config_file("config.toml")
                .expect("Failed to create config file")
        }),
    };

    if !config_path.exists() {
        let default_config = Config::default();
        let toml = toml::to_string(&default_config)?;
        fs::write(&config_path, toml)?;
    }

    let config_str = fs::read_to_string(&config_path)
        .with_context(|| format!("Failed to read config file: {config_path:?}"))?;
    let config: Config = toml::from_str(&config_str)
        .with_context(|| format!("Failed to parse config file: {config_path:?}"))?;

    Ok(config)
}