tuff-core 0.14.0

Core lifecycle engine and adapter contract for Tuff.
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

use std::path::Path;

use serde::{Deserialize, Serialize};

use crate::error::Result;
use crate::paths;

pub const DEFAULT_AGENT: &str = "open-agents";

#[derive(Debug, Serialize, Deserialize)]
#[serde(default)]
pub struct TuffConfig {
    #[serde(rename = "agents", alias = "targets", alias = "harnesses")]
    pub agents: Vec<String>,
    #[serde(rename = "defaultAgent", alias = "default_agent")]
    pub default_agent: String,
}

impl Default for TuffConfig {
    fn default() -> Self {
        Self {
            agents: vec![],
            default_agent: DEFAULT_AGENT.to_string(),
        }
    }
}

pub fn read_config(repo_root: &Path) -> Result<TuffConfig> {
    read_config_at(&paths::project_config(repo_root))
}

pub fn read_global_config(home: &Path) -> Result<TuffConfig> {
    read_config_at(&paths::user_config(home))
}

fn read_config_at(path: &Path) -> Result<TuffConfig> {
    if path.exists() {
        let config: TuffConfig = serde_json::from_str(&std::fs::read_to_string(path)?)?;
        Ok(config)
    } else {
        let default = TuffConfig::default();
        write_config_at(path, &default)?;
        Ok(default)
    }
}

pub fn write_config(repo_root: &Path, config: &TuffConfig) -> Result<()> {
    write_config_at(&paths::project_config(repo_root), config)
}

pub fn write_global_config(home: &Path, config: &TuffConfig) -> Result<()> {
    write_config_at(&paths::user_config(home), config)
}

fn write_config_at(path: &Path, config: &TuffConfig) -> Result<()> {
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)?;
    }
    std::fs::write(path, serde_json::to_string_pretty(config)? + "\n")?;
    Ok(())
}