lorum 0.1.2-alpha.1

//! Configuration schema, loading, merging, and resolution for lorum.
//!
//! This module defines the unified configuration structs, handles reading
//! global (`~/.config/lorum/config.yaml`) and project-level (`.lorum/config.yaml`)
//! configuration files, merging them, and resolving the effective configuration
//! based on command-line arguments.

use std::collections::BTreeMap;
use std::path::{Path, PathBuf};

use crate::error::LorumError;

use serde::de::DeserializeOwned;

/// Output format for configuration display.
#[derive(Debug, Clone, Copy, Default, PartialEq)]
pub enum OutputFormat {
    /// YAML output (default).
    #[default]
    Yaml,
    /// JSON output.
    Json,
}

/// Single MCP server configuration entry.
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct McpServer {
    /// Command to start the MCP server (e.g. `"npx"` or `"python"`).
    pub command: String,
    /// Arguments passed to the command.
    #[serde(default)]
    pub args: Vec<String>,
    /// Environment variables set for the server process.
    #[serde(default)]
    pub env: BTreeMap<String, String>,
}

/// MCP server collection.
#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct McpConfig {
    /// Map of server name to server configuration.
    #[serde(default)]
    pub servers: BTreeMap<String, McpServer>,
}

/// A single hook handler entry.
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct HookHandler {
    /// Matcher pattern (e.g., tool name or regex).
    pub matcher: String,
    /// Command to execute (or URL / prompt text / agent instruction).
    pub command: String,
    /// Optional timeout in seconds.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub timeout: Option<u64>,
    /// Handler type: `command`, `http`, `prompt`, `agent`, `mcp_tool`.
    /// Defaults to `command` when not specified.
    #[serde(default, rename = "type", skip_serializing_if = "Option::is_none")]
    pub handler_type: Option<String>,
}

/// Hooks configuration: event name → list of handlers.
///
/// Event names use kebab-case in the unified config (e.g., `pre-tool-use`).
#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct HooksConfig {
    /// Map of event name to ordered list of handlers.
    #[serde(default, flatten)]
    pub events: BTreeMap<String, Vec<HookHandler>>,
}

/// Lorum unified global configuration file structure.
#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct LorumConfig {
    /// MCP server configurations.
    #[serde(default)]
    pub mcp: McpConfig,
    /// Hooks configurations.
    #[serde(default)]
    pub hooks: HooksConfig,
}

/// Project-level configuration with additional fields.
#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize, PartialEq)]
pub struct ProjectConfig {
    /// MCP server configurations (overrides global on name collision).
    #[serde(default)]
    pub mcp: McpConfig,
    /// Hooks configurations (overrides global on event name collision).
    #[serde(default)]
    pub hooks: HooksConfig,
    /// Server names to exclude from the merged result.
    #[serde(default)]
    pub exclude: Vec<String>,
}

/// Resolves the normalized XDG configuration directory.
///
/// Uses `$XDG_CONFIG_HOME` if set, otherwise falls back to
/// `$HOME/.config` on all platforms.
///
/// The returned path is normalized (`.` and `..` components are resolved)
/// and verified to be absolute.  This prevents malicious paths containing
/// `..` from escaping the intended directory without requiring the path
/// to already exist on disk.
pub fn resolve_config_dir() -> Result<PathBuf, LorumError> {
    let config_dir = if let Ok(xdg) = std::env::var("XDG_CONFIG_HOME") {
        PathBuf::from(xdg)
    } else {
        let home = dirs::home_dir().ok_or_else(|| LorumError::Other {
            message: "cannot determine home directory".into(),
        })?;
        home.join(".config")
    };

    if !config_dir.is_absolute() {
        return Err(LorumError::Other {
            message: "config directory path must be absolute".into(),
        });
    }

    // Normalize `.` and `..` components manually so the path is
    // trustworthy even when the directory does not exist yet.
    let mut normalized = PathBuf::new();
    for component in config_dir.components() {
        match component {
            std::path::Component::ParentDir => {
                normalized.pop();
            }
            std::path::Component::CurDir => {}
            other => normalized.push(other.as_os_str()),
        }
    }

    Ok(normalized)
}

/// Returns the global configuration file path: `~/.config/lorum/config.yaml`
/// (XDG Base Directory Specification).
///
/// Uses `$XDG_CONFIG_HOME/lorum` if that environment variable is set.
/// Otherwise falls back to `$HOME/.config/lorum` on all platforms
/// (including macOS), overriding the macOS-native `dirs::config_dir()` path.
pub fn global_config_path() -> Result<PathBuf, LorumError> {
    Ok(resolve_config_dir()?.join("lorum").join("config.yaml"))
}

/// Load and deserialize a YAML file at `path`.
///
/// # Errors
///
/// - [`LorumError::ConfigParse`] if the file cannot be parsed as YAML.
/// - [`LorumError::Io`] for other I/O failures.
fn load_yaml<T: DeserializeOwned>(path: &Path) -> Result<T, LorumError> {
    let contents = std::fs::read_to_string(path)?;
    serde_yaml::from_str(&contents).map_err(|e| LorumError::ConfigParse {
        format: "yaml".into(),
        path: path.to_path_buf(),
        source: Box::new(e),
    })
}

/// Reads a [`LorumConfig`] from the file at `path`.
///
/// The file must be valid YAML.
///
/// # Errors
///
/// - [`LorumError::ConfigNotFound`] if the file does not exist.
/// - [`LorumError::ConfigParse`] if the file cannot be parsed as YAML.
/// - [`LorumError::Io`] for other I/O failures.
pub fn load_config(path: &Path) -> Result<LorumConfig, LorumError> {
    if !path.exists() {
        return Err(LorumError::ConfigNotFound {
            path: path.to_path_buf(),
        });
    }
    load_yaml(path)
}

/// Writes a [`LorumConfig`] to the file at `path`.
///
/// Parent directories are created automatically if they do not exist.
///
/// # Errors
///
/// - [`LorumError::ConfigWrite`] if the file cannot be written.
pub fn save_config(path: &Path, config: &LorumConfig) -> Result<(), LorumError> {
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent).map_err(|e| LorumError::ConfigWrite {
            path: path.to_path_buf(),
            source: e,
        })?;
    }
    let yaml = serde_yaml::to_string(config).map_err(|e| LorumError::ConfigSerialize {
        path: path.to_path_buf(),
        source: Box::new(e),
    })?;
    std::fs::write(path, yaml).map_err(|e| LorumError::ConfigWrite {
        path: path.to_path_buf(),
        source: e,
    })?;
    Ok(())
}

/// Recursively searches for `.lorum/config.yaml` starting from `start_dir`.
///
/// Traverses upward through parent directories until the file is found or
/// the filesystem root is reached (similar to how Git discovers `.git`).
///
/// Returns `Some(path)` when a `.lorum/config.yaml` is found, or `None` if
/// no project configuration exists above `start_dir`.
pub fn find_project_config(start_dir: &Path) -> Option<PathBuf> {
    let mut dir = start_dir;
    loop {
        let candidate = dir.join(".lorum").join("config.yaml");
        if candidate.exists() {
            return Some(candidate);
        }
        dir = dir.parent()?;
    }
}

/// Reads a [`ProjectConfig`] from the file at `path`.
///
/// The file must be valid YAML.
///
/// # Errors
///
/// - [`LorumError::ConfigParse`] if the file cannot be parsed as YAML.
/// - [`LorumError::Io`] for other I/O failures.
pub fn load_project_config(path: &Path) -> Result<Option<ProjectConfig>, LorumError> {
    if !path.exists() {
        return Ok(None);
    }
    load_yaml(path).map(Some)
}

/// Merges global and project-level configurations.
///
/// # Merge rules
///
/// 1. Project-level servers with the same name **override** global servers
///    (no deep merge -- the project entry replaces the global one entirely).
/// 2. Servers listed in the project's `exclude` field are removed from the
///    merged result.
pub fn merge_configs(global: &LorumConfig, project: Option<&ProjectConfig>) -> LorumConfig {
    let Some(project) = project else {
        return global.clone();
    };

    let mut servers = global.mcp.servers.clone();

    // Project-level servers override global ones (full replacement).
    for (name, server) in &project.mcp.servers {
        servers.insert(name.clone(), server.clone());
    }

    // Remove excluded servers.
    for name in &project.exclude {
        servers.remove(name);
    }

    // Project-level hooks override global hooks by event name (full replacement).
    let mut events = global.hooks.events.clone();
    for (name, handlers) in &project.hooks.events {
        events.insert(name.clone(), handlers.clone());
    }

    LorumConfig {
        mcp: McpConfig { servers },
        hooks: HooksConfig { events },
    }
}

/// Resolves the effective configuration based on command-line arguments.
///
/// - If `config_path` is provided, only that file is loaded (no global or
///   project-level lookup is performed).
/// - Otherwise the global config is loaded, any project config is discovered
///   from `search_start`, and the two are merged via [`merge_configs`].
///
/// # Errors
///
/// Propagates any error from [`load_config`], [`load_project_config`], or
/// I/O operations.
pub fn resolve_effective_config(
    config_path: Option<&Path>,
    search_start: &Path,
) -> Result<LorumConfig, LorumError> {
    if let Some(path) = config_path {
        return load_config(path);
    }

    let global = match load_config(&global_config_path()?) {
        Ok(cfg) => cfg,
        Err(LorumError::ConfigNotFound { .. }) => LorumConfig::default(),
        Err(e) => return Err(e),
    };

    let project = find_project_config(search_start);

    let project_config = project
        .as_ref()
        .map(|p| load_project_config(p))
        .transpose()?
        .flatten();

    Ok(merge_configs(&global, project_config.as_ref()))
}

/// Resolves the effective configuration starting from the current working directory.
///
/// Convenience wrapper around [`resolve_effective_config`] that uses
/// `std::env::current_dir()` as the search start point.
///
/// # Errors
///
/// Propagates any error from [`resolve_effective_config`] or if the current
/// working directory cannot be determined.
pub fn resolve_effective_config_from_cwd(
    config_path: Option<&Path>,
) -> Result<LorumConfig, LorumError> {
    let cwd = std::env::current_dir().map_err(|e| LorumError::Io { source: e })?;
    resolve_effective_config(config_path, &cwd)
}

#[cfg(test)]
mod tests;