batty_cli/config/

mod.rs

#![cfg_attr(not(test), allow(dead_code))]

use anyhow::{Context, Result};
use serde::Deserialize;
use std::collections::HashMap;
use std::path::{Path, PathBuf};

const CONFIG_FILENAME: &str = "config.toml";
const CONFIG_DIR: &str = ".batty";

#[derive(Debug, Default, Clone, Copy, Deserialize, PartialEq)]
#[serde(rename_all = "kebab-case")]
pub enum Policy {
    #[default]
    Observe,
    Suggest,
    Act,
}

#[derive(Debug, Deserialize)]
pub struct Defaults {
    #[serde(default = "default_agent")]
    pub agent: String,
    #[serde(default)]
    pub policy: Policy,
    #[serde(default)]
    pub dod: Option<String>,
    #[serde(default = "default_max_retries")]
    pub max_retries: u32,
}

/// Policy section with auto-answer patterns.
///
/// ```toml
/// [policy.auto_answer]
/// "Continue? [y/n]" = "y"
/// "Allow tool" = "y"
/// ```
#[derive(Debug, Deserialize, Default)]
pub struct PolicyConfig {
    #[serde(default)]
    pub auto_answer: HashMap<String, String>,
}

fn default_agent() -> String {
    "claude".to_string()
}

fn default_max_retries() -> u32 {
    3
}

fn default_supervisor_enabled() -> bool {
    true
}

fn default_supervisor_program() -> String {
    "claude".to_string()
}

fn default_supervisor_args() -> Vec<String> {
    vec![
        "-p".to_string(),
        "--output-format".to_string(),
        "text".to_string(),
    ]
}

fn default_supervisor_timeout_secs() -> u64 {
    60
}

fn default_supervisor_trace_io() -> bool {
    true
}

fn default_detector_silence_timeout_secs() -> u64 {
    3
}

fn default_detector_answer_cooldown_millis() -> u64 {
    1000
}

fn default_detector_unknown_request_fallback() -> bool {
    true
}

fn default_detector_idle_input_fallback() -> bool {
    true
}

fn default_dangerous_mode_enabled() -> bool {
    false
}

impl Default for Defaults {
    fn default() -> Self {
        Self {
            agent: default_agent(),
            policy: Policy::default(),
            dod: None,
            max_retries: default_max_retries(),
        }
    }
}

#[derive(Debug, Deserialize)]
pub struct SupervisorConfig {
    #[serde(default = "default_supervisor_enabled")]
    pub enabled: bool,
    #[serde(default = "default_supervisor_program")]
    pub program: String,
    #[serde(default = "default_supervisor_args")]
    pub args: Vec<String>,
    #[serde(default = "default_supervisor_timeout_secs")]
    pub timeout_secs: u64,
    #[serde(default = "default_supervisor_trace_io")]
    pub trace_io: bool,
}

impl Default for SupervisorConfig {
    fn default() -> Self {
        Self {
            enabled: default_supervisor_enabled(),
            program: default_supervisor_program(),
            args: default_supervisor_args(),
            timeout_secs: default_supervisor_timeout_secs(),
            trace_io: default_supervisor_trace_io(),
        }
    }
}

#[derive(Debug, Deserialize)]
pub struct DetectorSettings {
    #[serde(default = "default_detector_silence_timeout_secs")]
    pub silence_timeout_secs: u64,
    #[serde(default = "default_detector_answer_cooldown_millis")]
    pub answer_cooldown_millis: u64,
    #[serde(default = "default_detector_unknown_request_fallback")]
    pub unknown_request_fallback: bool,
    #[serde(default = "default_detector_idle_input_fallback")]
    pub idle_input_fallback: bool,
}

impl Default for DetectorSettings {
    fn default() -> Self {
        Self {
            silence_timeout_secs: default_detector_silence_timeout_secs(),
            answer_cooldown_millis: default_detector_answer_cooldown_millis(),
            unknown_request_fallback: default_detector_unknown_request_fallback(),
            idle_input_fallback: default_detector_idle_input_fallback(),
        }
    }
}

#[derive(Debug, Deserialize)]
pub struct DangerousModeConfig {
    #[serde(default = "default_dangerous_mode_enabled")]
    pub enabled: bool,
}

impl Default for DangerousModeConfig {
    fn default() -> Self {
        Self {
            enabled: default_dangerous_mode_enabled(),
        }
    }
}

#[derive(Debug, Deserialize, Default)]
pub struct ProjectConfig {
    #[serde(default)]
    pub defaults: Defaults,
    #[serde(default)]
    pub policy: PolicyConfig,
    #[serde(default)]
    pub supervisor: SupervisorConfig,
    #[serde(default)]
    pub detector: DetectorSettings,
    #[serde(default)]
    pub dangerous_mode: DangerousModeConfig,
}

impl ProjectConfig {
    /// Search upward from `start` for a `.batty/config.toml` file and load it.
    /// Returns the default config if no file is found.
    pub fn load(start: &Path) -> Result<(Self, Option<PathBuf>)> {
        if let Some(path) = Self::find_config_file(start) {
            let contents = std::fs::read_to_string(&path)
                .with_context(|| format!("failed to read {}", path.display()))?;
            let config: ProjectConfig = toml::from_str(&contents)
                .with_context(|| format!("failed to parse {}", path.display()))?;
            Ok((config, Some(path)))
        } else {
            Ok((ProjectConfig::default(), None))
        }
    }

    fn find_config_file(start: &Path) -> Option<PathBuf> {
        let mut dir = start.to_path_buf();
        loop {
            let candidate = dir.join(CONFIG_DIR).join(CONFIG_FILENAME);
            if candidate.is_file() {
                return Some(candidate);
            }
            if !dir.pop() {
                return None;
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::fs;

    #[test]
    fn default_config_values() {
        let config = ProjectConfig::default();
        assert_eq!(config.defaults.agent, "claude");
        assert_eq!(config.defaults.policy, Policy::Observe);
        assert_eq!(config.defaults.max_retries, 3);
        assert!(config.defaults.dod.is_none());
        assert!(config.supervisor.enabled);
        assert_eq!(config.supervisor.program, "claude");
        assert_eq!(
            config.supervisor.args,
            vec!["-p", "--output-format", "text"]
        );
        assert_eq!(config.supervisor.timeout_secs, 60);
        assert!(config.supervisor.trace_io);
        assert_eq!(config.detector.silence_timeout_secs, 3);
        assert_eq!(config.detector.answer_cooldown_millis, 1000);
        assert!(config.detector.unknown_request_fallback);
        assert!(config.detector.idle_input_fallback);
        assert!(!config.dangerous_mode.enabled);
    }

    #[test]
    fn parse_full_config() {
        let toml = r#"
[defaults]
agent = "codex"
policy = "act"
dod = "cargo test --workspace"
max_retries = 5

[supervisor]
enabled = true
program = "claude"
args = ["-p", "--output-format", "text"]
timeout_secs = 45
trace_io = true

[detector]
silence_timeout_secs = 5
answer_cooldown_millis = 1500
unknown_request_fallback = true
idle_input_fallback = true

[dangerous_mode]
enabled = true
"#;
        let config: ProjectConfig = toml::from_str(toml).unwrap();
        assert_eq!(config.defaults.agent, "codex");
        assert_eq!(config.defaults.policy, Policy::Act);
        assert_eq!(
            config.defaults.dod.as_deref(),
            Some("cargo test --workspace")
        );
        assert_eq!(config.defaults.max_retries, 5);
        assert!(config.supervisor.enabled);
        assert_eq!(config.supervisor.program, "claude");
        assert_eq!(config.supervisor.timeout_secs, 45);
        assert!(config.supervisor.trace_io);
        assert_eq!(config.detector.silence_timeout_secs, 5);
        assert_eq!(config.detector.answer_cooldown_millis, 1500);
        assert!(config.detector.unknown_request_fallback);
        assert!(config.detector.idle_input_fallback);
        assert!(config.dangerous_mode.enabled);
    }

    #[test]
    fn parse_partial_config() {
        let toml = r#"
[defaults]
agent = "aider"
"#;
        let config: ProjectConfig = toml::from_str(toml).unwrap();
        assert_eq!(config.defaults.agent, "aider");
        assert_eq!(config.defaults.policy, Policy::Observe);
        assert_eq!(config.defaults.max_retries, 3);
        assert!(config.detector.unknown_request_fallback);
        assert!(config.detector.idle_input_fallback);
        assert!(!config.dangerous_mode.enabled);
    }

    #[test]
    fn load_from_directory() {
        let tmp = tempfile::tempdir().unwrap();
        let batty_dir = tmp.path().join(".batty");
        fs::create_dir_all(&batty_dir).unwrap();
        fs::write(
            batty_dir.join("config.toml"),
            r#"
[defaults]
agent = "claude"
policy = "act"
dod = "cargo test"
max_retries = 2
"#,
        )
        .unwrap();

        let (config, path) = ProjectConfig::load(tmp.path()).unwrap();
        assert!(path.is_some());
        assert_eq!(config.defaults.agent, "claude");
        assert_eq!(config.defaults.policy, Policy::Act);
        assert_eq!(config.defaults.max_retries, 2);
    }

    #[test]
    fn load_returns_default_when_no_file() {
        let tmp = tempfile::tempdir().unwrap();
        let (config, path) = ProjectConfig::load(tmp.path()).unwrap();
        assert!(path.is_none());
        assert_eq!(config.defaults.agent, "claude");
    }

    #[test]
    fn parse_auto_answer_config() {
        let toml = r#"
[defaults]
agent = "claude"
policy = "act"

[policy.auto_answer]
"Continue? [y/n]" = "y"
"Allow tool" = "y"
"#;
        let config: ProjectConfig = toml::from_str(toml).unwrap();
        assert_eq!(config.policy.auto_answer.len(), 2);
        assert_eq!(
            config.policy.auto_answer.get("Continue? [y/n]").unwrap(),
            "y"
        );
        assert_eq!(config.policy.auto_answer.get("Allow tool").unwrap(), "y");
    }

    #[test]
    fn load_walks_up_directories() {
        let tmp = tempfile::tempdir().unwrap();
        let batty_dir = tmp.path().join(".batty");
        fs::create_dir_all(&batty_dir).unwrap();
        fs::write(
            batty_dir.join("config.toml"),
            r#"
[defaults]
agent = "codex"
"#,
        )
        .unwrap();

        let nested = tmp.path().join("src").join("deep").join("nested");
        fs::create_dir_all(&nested).unwrap();

        let (config, path) = ProjectConfig::load(&nested).unwrap();
        assert!(path.is_some());
        assert_eq!(config.defaults.agent, "codex");
    }
}