muthr 0.1.36

A zero-trust orchestrator that automates secure inference and isolated execution of local AI agents.
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
99
100
101
102
103

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

#[derive(Subcommand)]
pub enum ConfigCommands {
    Init {
        #[arg(long, help = "Force overwrite existing muthr.toml")]
        force: bool,
    },
    Show,
}

#[derive(Serialize, Deserialize, Debug, Clone, Default)]
pub struct MuthrConfig {
    pub server_port: Option<u16>,
    pub workspace_root: Option<String>,
    pub model_dir: Option<String>,
    pub default_provision_profile: Option<String>,
}

impl MuthrConfig {
    fn resolve(self) -> (u16, String, String, String) {
        let server_port = self.server_port.unwrap_or(8080);
        let workspace_root = match self.workspace_root {
            Some(v) => v,
            None => std::env::var("HOME")
                .ok()
                .map(|h| format!("{}/src", h))
                .unwrap_or_else(|| "~/src".to_string()),
        };
        let model_dir = match self.model_dir {
            Some(v) => v,
            None => std::env::var("HOME")
                .ok()
                .map(|h| format!("{}/opt/models", h))
                .unwrap_or_else(|| "~/opt/models".to_string()),
        };
        let provision_profile = self
            .default_provision_profile
            .unwrap_or_else(|| "base".to_string());
        (server_port, workspace_root, model_dir, provision_profile)
    }

    pub fn print_resolved(&self) {
        let (server_port, workspace_root, model_dir, provision_profile) = self.clone().resolve();
        println!("server_port       {}", server_port);
        println!("workspace_root    {}", workspace_root);
        println!("model_dir         {}", model_dir);
        println!("provision_profile {}", provision_profile);
    }
}

pub fn load() -> Result<MuthrConfig, color_eyre::Report> {
    let home = std::env::var("HOME")?;
    let config_path = PathBuf::from(&home).join(".config/muthr/muthr.toml");

    let mut config = if config_path.exists() {
        let content = fs::read_to_string(&config_path)?;
        toml::from_str(&content)?
    } else {
        MuthrConfig::default()
    };

    if let Ok(v) = std::env::var("MUTHR_SERVER_PORT") {
        config.server_port = v.parse().ok();
    }
    if let Ok(v) = std::env::var("MUTHR_WORKSPACE_ROOT") {
        config.workspace_root = Some(v);
    }
    if let Ok(v) = std::env::var("MUTHR_MODEL_DIR") {
        config.model_dir = Some(v);
    }
    if let Ok(v) = std::env::var("MUTHR_PROVISION_PROFILE") {
        config.default_provision_profile = Some(v);
    }

    Ok(config)
}

pub fn init_config(force: bool) -> Result<(), color_eyre::Report> {
    let home = std::env::var("HOME")?;
    let config_dir = PathBuf::from(&home).join(".config/muthr");
    let config_path = config_dir.join("muthr.toml");

    if config_path.exists() && !force {
        return Ok(());
    }

    fs::create_dir_all(&config_dir)?;

    let template = r##"# muthr configuration
server_port = 8080
workspace_root = "~/src"
model_dir = "~/opt/models"
default_provision_profile = "base"
"##;

    fs::write(&config_path, template)?;
    eprintln!("info: created {}", config_path.display());
    Ok(())
}