a3s-code-core 1.8.0

//! Orchestrator configuration

use serde::{Deserialize, Serialize};

/// Orchestrator 配置
#[derive(Debug, Clone)]
pub struct OrchestratorConfig {
    /// 事件通道缓冲区大小
    pub event_buffer_size: usize,

    /// 控制信号通道缓冲区大小
    pub control_buffer_size: usize,

    /// SubAgent 最大并发数
    pub max_concurrent_subagents: usize,

    /// 事件主题前缀
    pub subject_prefix: String,
}

impl Default for OrchestratorConfig {
    fn default() -> Self {
        Self {
            event_buffer_size: 1000,
            control_buffer_size: 100,
            max_concurrent_subagents: 50,
            subject_prefix: "agent".to_string(),
        }
    }
}

/// SubAgent 配置
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SubAgentConfig {
    /// Agent 类型 (general, explore, plan, etc.)
    pub agent_type: String,

    /// 任务描述
    pub description: String,

    /// 执行提示词
    pub prompt: String,

    /// 是否启用宽松模式（绕过 HITL）
    #[serde(default)]
    pub permissive: bool,

    /// Deny rules to enforce even in permissive mode (e.g., ["mcp__longvt__*"])
    #[serde(default)]
    pub permissive_deny: Vec<String>,

    /// 最大执行步数
    pub max_steps: Option<usize>,

    /// 执行超时（毫秒）
    pub timeout_ms: Option<u64>,

    /// 父 SubAgent ID（用于嵌套）
    #[serde(skip_serializing_if = "Option::is_none")]
    pub parent_id: Option<String>,

    /// 自定义元数据
    #[serde(default)]
    pub metadata: serde_json::Value,

    /// Workspace directory for the SubAgent (defaults to ".")
    #[serde(default = "default_workspace")]
    pub workspace: String,

    /// Extra directories to scan for agent definition files
    #[serde(default)]
    pub agent_dirs: Vec<String>,

    /// Extra directories to scan for skill definition files
    #[serde(default)]
    pub skill_dirs: Vec<String>,

    /// Lane queue configuration for External/Hybrid tool dispatch.
    ///
    /// When set, the SubAgent's session is created with this queue config,
    /// enabling tools to be routed to external workers.  Any lane not
    /// explicitly configured falls back to Internal mode.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub lane_config: Option<crate::queue::SessionQueueConfig>,
}

/// SubAgent 信息（元数据）
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SubAgentInfo {
    /// SubAgent ID
    pub id: String,

    /// Agent 类型
    pub agent_type: String,

    /// 任务描述
    pub description: String,

    /// 当前状态
    pub state: String,

    /// 父 SubAgent ID
    pub parent_id: Option<String>,

    /// 创建时间（Unix 时间戳，毫秒）
    pub created_at: u64,

    /// 最后更新时间（Unix 时间戳，毫秒）
    pub updated_at: u64,

    /// 当前活动
    pub current_activity: Option<SubAgentActivity>,
}

/// SubAgent 当前活动
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum SubAgentActivity {
    /// 空闲
    Idle,

    /// 正在调用工具
    CallingTool {
        tool_name: String,
        args: serde_json::Value,
    },

    /// 正在请求 LLM
    RequestingLlm { message_count: usize },

    /// 正在等待控制信号
    WaitingForControl { reason: String },
}

impl SubAgentConfig {
    /// 创建新的 SubAgent 配置
    pub fn new(agent_type: impl Into<String>, prompt: impl Into<String>) -> Self {
        Self {
            agent_type: agent_type.into(),
            description: String::new(),
            prompt: prompt.into(),
            permissive: false,
            permissive_deny: Vec::new(),
            max_steps: None,
            timeout_ms: None,
            parent_id: None,
            metadata: serde_json::Value::Null,
            workspace: default_workspace(),
            agent_dirs: Vec::new(),
            skill_dirs: Vec::new(),
            lane_config: None,
        }
    }

    /// 设置描述
    pub fn with_description(mut self, description: impl Into<String>) -> Self {
        self.description = description.into();
        self
    }

    /// 设置宽松模式
    pub fn with_permissive(mut self, permissive: bool) -> Self {
        self.permissive = permissive;
        self
    }

    /// Set deny rules to enforce even in permissive mode
    pub fn with_permissive_deny(mut self, deny_rules: Vec<String>) -> Self {
        self.permissive_deny = deny_rules;
        self
    }

    /// 设置最大步数
    pub fn with_max_steps(mut self, max_steps: usize) -> Self {
        self.max_steps = Some(max_steps);
        self
    }

    /// 设置超时
    pub fn with_timeout_ms(mut self, timeout_ms: u64) -> Self {
        self.timeout_ms = Some(timeout_ms);
        self
    }

    /// 设置父 ID
    pub fn with_parent_id(mut self, parent_id: impl Into<String>) -> Self {
        self.parent_id = Some(parent_id.into());
        self
    }

    /// 设置元数据
    pub fn with_metadata(mut self, metadata: serde_json::Value) -> Self {
        self.metadata = metadata;
        self
    }

    /// Set the workspace directory for this SubAgent
    pub fn with_workspace(mut self, workspace: impl Into<String>) -> Self {
        self.workspace = workspace.into();
        self
    }

    /// Add extra directories to scan for agent definition files
    pub fn with_agent_dirs(mut self, dirs: Vec<String>) -> Self {
        self.agent_dirs = dirs;
        self
    }

    /// Add extra directories to scan for skill definition files
    pub fn with_skill_dirs(mut self, dirs: Vec<String>) -> Self {
        self.skill_dirs = dirs;
        self
    }

    /// Set lane queue configuration for External/Hybrid tool dispatch
    pub fn with_lane_config(mut self, config: crate::queue::SessionQueueConfig) -> Self {
        self.lane_config = Some(config);
        self
    }
}

fn default_workspace() -> String {
    ".".to_string()
}

/// Unified agent slot declaration — used for both standalone subagents and team members.
///
/// `AgentSlot` is a superset of [`SubAgentConfig`] that also carries an optional
/// team role.  When `role` is `None` the slot describes a standalone subagent;
/// when set, [`crate::orchestrator::AgentOrchestrator::run_team`] uses it to
/// build the team layout.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentSlot {
    /// Agent type (general, explore, plan, etc.)
    pub agent_type: String,

    /// Team role for this slot; `None` means standalone subagent.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub role: Option<crate::agent_teams::TeamRole>,

    /// Task description
    pub description: String,

    /// Execution prompt
    pub prompt: String,

    /// Whether to bypass HITL (permissive mode)
    #[serde(default)]
    pub permissive: bool,

    /// Deny rules to enforce even in permissive mode
    #[serde(default)]
    pub permissive_deny: Vec<String>,

    /// Maximum execution steps
    pub max_steps: Option<usize>,

    /// Execution timeout in milliseconds
    pub timeout_ms: Option<u64>,

    /// Parent subagent ID (for nesting)
    #[serde(skip_serializing_if = "Option::is_none")]
    pub parent_id: Option<String>,

    /// Custom metadata
    #[serde(default)]
    pub metadata: serde_json::Value,

    /// Workspace directory for this slot (defaults to ".")
    #[serde(default = "default_workspace")]
    pub workspace: String,

    /// Extra directories to scan for agent definition files
    #[serde(default)]
    pub agent_dirs: Vec<String>,

    /// Extra directories to scan for skill definition files
    #[serde(default)]
    pub skill_dirs: Vec<String>,

    /// Lane queue configuration for External/Hybrid tool dispatch
    #[serde(skip_serializing_if = "Option::is_none")]
    pub lane_config: Option<crate::queue::SessionQueueConfig>,
}

impl AgentSlot {
    /// Create a new agent slot with the given agent type and prompt.
    pub fn new(agent_type: impl Into<String>, prompt: impl Into<String>) -> Self {
        Self {
            agent_type: agent_type.into(),
            role: None,
            description: String::new(),
            prompt: prompt.into(),
            permissive: false,
            permissive_deny: Vec::new(),
            max_steps: None,
            timeout_ms: None,
            parent_id: None,
            metadata: serde_json::Value::Null,
            workspace: default_workspace(),
            agent_dirs: Vec::new(),
            skill_dirs: Vec::new(),
            lane_config: None,
        }
    }

    /// Set the team role for this slot.
    pub fn with_role(mut self, role: crate::agent_teams::TeamRole) -> Self {
        self.role = Some(role);
        self
    }

    /// Set the task description.
    pub fn with_description(mut self, description: impl Into<String>) -> Self {
        self.description = description.into();
        self
    }

    /// Set permissive mode (bypass HITL).
    pub fn with_permissive(mut self, permissive: bool) -> Self {
        self.permissive = permissive;
        self
    }

    /// Set deny rules to enforce even in permissive mode.
    pub fn with_permissive_deny(mut self, deny_rules: Vec<String>) -> Self {
        self.permissive_deny = deny_rules;
        self
    }

    /// Set the maximum number of execution steps.
    pub fn with_max_steps(mut self, max_steps: usize) -> Self {
        self.max_steps = Some(max_steps);
        self
    }

    /// Set the execution timeout in milliseconds.
    pub fn with_timeout_ms(mut self, timeout_ms: u64) -> Self {
        self.timeout_ms = Some(timeout_ms);
        self
    }

    /// Set the parent subagent ID.
    pub fn with_parent_id(mut self, parent_id: impl Into<String>) -> Self {
        self.parent_id = Some(parent_id.into());
        self
    }

    /// Set custom metadata.
    pub fn with_metadata(mut self, metadata: serde_json::Value) -> Self {
        self.metadata = metadata;
        self
    }

    /// Set the workspace directory.
    pub fn with_workspace(mut self, workspace: impl Into<String>) -> Self {
        self.workspace = workspace.into();
        self
    }

    /// Add extra directories to scan for agent definition files.
    pub fn with_agent_dirs(mut self, dirs: Vec<String>) -> Self {
        self.agent_dirs = dirs;
        self
    }

    /// Add extra directories to scan for skill definition files.
    pub fn with_skill_dirs(mut self, dirs: Vec<String>) -> Self {
        self.skill_dirs = dirs;
        self
    }

    /// Set lane queue configuration for External/Hybrid tool dispatch.
    pub fn with_lane_config(mut self, config: crate::queue::SessionQueueConfig) -> Self {
        self.lane_config = Some(config);
        self
    }
}

impl From<SubAgentConfig> for AgentSlot {
    fn from(c: SubAgentConfig) -> Self {
        Self {
            agent_type: c.agent_type,
            role: None,
            description: c.description,
            prompt: c.prompt,
            permissive: c.permissive,
            permissive_deny: c.permissive_deny,
            max_steps: c.max_steps,
            timeout_ms: c.timeout_ms,
            parent_id: c.parent_id,
            metadata: c.metadata,
            workspace: c.workspace,
            agent_dirs: c.agent_dirs,
            skill_dirs: c.skill_dirs,
            lane_config: c.lane_config,
        }
    }
}

impl From<AgentSlot> for SubAgentConfig {
    fn from(s: AgentSlot) -> Self {
        Self {
            agent_type: s.agent_type,
            description: s.description,
            prompt: s.prompt,
            permissive: s.permissive,
            permissive_deny: s.permissive_deny,
            max_steps: s.max_steps,
            timeout_ms: s.timeout_ms,
            parent_id: s.parent_id,
            metadata: s.metadata,
            workspace: s.workspace,
            agent_dirs: s.agent_dirs,
            skill_dirs: s.skill_dirs,
            lane_config: s.lane_config,
        }
    }
}