Skip to main content

walrus_daemon/config/
mod.rs

1//! Daemon configuration loaded from TOML.
2
3pub use crate::hook::{
4    mcp::McpServerConfig,
5    os::{PermissionConfig, ToolPermission},
6    system::SystemConfig,
7};
8pub use ::model::{ModelConfig, ProviderDef, ProviderRegistry};
9use anyhow::Result;
10use compact_str::CompactString;
11pub use loader::{load_agents_dir, scaffold_config_dir};
12use serde::{Deserialize, Serialize};
13use std::collections::BTreeMap;
14pub use wcore::{
15    AgentConfig, HeartbeatConfig,
16    paths::{AGENTS_DIR, CONFIG_DIR, DATA_DIR, HOME_DIR, MEMORY_DB, SKILLS_DIR, SOCKET_PATH},
17};
18
19mod loader;
20
21/// Top-level daemon configuration.
22#[derive(Debug, Clone, Serialize, Deserialize, Default)]
23pub struct DaemonConfig {
24    /// Provider definitions (`[provider.<name>]`).
25    #[serde(default)]
26    pub provider: BTreeMap<CompactString, ProviderDef>,
27    /// Model configuration (embedding model).
28    #[serde(default)]
29    pub model: ModelConfig,
30    /// MCP server configurations.
31    #[serde(default)]
32    pub mcps: BTreeMap<String, McpServerConfig>,
33    /// System configuration (tasks + memory).
34    #[serde(default)]
35    pub system: SystemConfig,
36    /// Per-agent configurations (name → config).
37    #[serde(default)]
38    pub agents: BTreeMap<String, AgentConfig>,
39    /// Permission configuration: global defaults + per-agent overrides.
40    #[serde(default)]
41    pub permissions: PermissionConfig,
42    /// Managed child services (`[services.<name>]`).
43    #[serde(default)]
44    pub services: BTreeMap<String, crate::service::ServiceConfig>,
45}
46
47impl DaemonConfig {
48    /// Parse a TOML string into a `DaemonConfig`.
49    pub fn from_toml(toml_str: &str) -> Result<Self> {
50        let mut config: Self = toml::from_str(toml_str)?;
51        config.mcps.iter_mut().for_each(|(name, server)| {
52            if server.name.is_empty() {
53                server.name = name.clone().into();
54            }
55        });
56        if config.system.walrus.model.is_none() {
57            config.system.walrus.model = Some(::model::default_model().into());
58        }
59        ModelConfig::validate(&config.provider)?;
60        Ok(config)
61    }
62
63    /// Load configuration from a file path.
64    pub fn load(path: &std::path::Path) -> Result<Self> {
65        let content = std::fs::read_to_string(path)?;
66        Self::from_toml(&content)
67    }
68}