swarm-engine-eval 0.1.6

//! Actions Configuration
//!
//! シナリオで利用可能なアクションの定義。

use serde::{Deserialize, Serialize};

// ============================================================================
// Action Parameter Definition
// ============================================================================

/// Action のパラメータ定義（TOML 用）
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ActionParamDef {
    /// パラメータ名
    pub name: String,
    /// 説明
    pub description: String,
    /// 必須かどうか
    #[serde(default)]
    pub required: bool,
}

// ============================================================================
// Action Category
// ============================================================================

/// Action カテゴリ（TOML 用）
///
/// 探索空間への影響を定義する。
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum ScenarioActionCategory {
    /// 新しい探索対象を発見する（Grep, List など）
    #[default]
    NodeExpand,
    /// 既存 Node の状態を遷移させる（Read など）
    NodeStateChange,
}

impl ScenarioActionCategory {
    /// Core の ActionCategory に変換
    pub fn to_core(&self) -> swarm_engine_core::actions::ActionCategory {
        match self {
            Self::NodeExpand => swarm_engine_core::actions::ActionCategory::NodeExpand,
            Self::NodeStateChange => swarm_engine_core::actions::ActionCategory::NodeStateChange,
        }
    }
}

// ============================================================================
// Action Definition
// ============================================================================

/// Action 定義（TOML 用）
///
/// シナリオで利用可能なアクションを定義する。
/// Core の ActionsConfig に変換される。
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ScenarioActionDef {
    /// Action 名
    pub name: String,
    /// 説明（LLM プロンプト用）
    pub description: String,
    /// カテゴリ（探索空間への影響）
    #[serde(default)]
    pub category: ScenarioActionCategory,
    /// パラメータ定義
    #[serde(default)]
    pub params: Vec<ActionParamDef>,
    /// 出力例（LLM プロンプト用 JSON 形式）
    #[serde(default)]
    pub example: Option<String>,
}

// ============================================================================
// Scenario Actions
// ============================================================================

/// シナリオの Actions 設定
///
/// TOML で定義されたアクションリストを保持し、
/// Core の ActionsConfig に変換する。
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct ScenarioActions {
    /// 定義された Action リスト
    #[serde(default)]
    pub actions: Vec<ScenarioActionDef>,
}

impl ScenarioActions {
    /// Core の ActionsConfig に変換
    pub fn to_core_config(&self) -> swarm_engine_core::actions::ActionsConfig {
        use swarm_engine_core::actions::{ActionDef, ActionsConfig, ParamDef};

        let mut config = ActionsConfig::new();
        for action_def in &self.actions {
            let mut def = ActionDef::new(&action_def.name, &action_def.description)
                .category(action_def.category.to_core())
                .group("default");

            for param in &action_def.params {
                let param_def = if param.required {
                    ParamDef::required(&param.name, &param.description)
                } else {
                    ParamDef::optional(&param.name, &param.description)
                };
                def = def.param(param_def);
            }

            if let Some(ref ex) = action_def.example {
                def = def.example(ex);
            }

            config.add_action(&action_def.name, def);
        }
        config
    }

    /// Action 名のリストを取得
    pub fn action_names(&self) -> Vec<String> {
        self.actions.iter().map(|a| a.name.clone()).collect()
    }
}

// ============================================================================
// Tests
// ============================================================================

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

    #[test]
    fn test_scenario_action_category_default() {
        let category = ScenarioActionCategory::default();
        assert_eq!(category, ScenarioActionCategory::NodeExpand);
    }

    #[test]
    fn test_scenario_actions_default() {
        let actions = ScenarioActions::default();
        assert!(actions.actions.is_empty());
        assert!(actions.action_names().is_empty());
    }

    #[test]
    fn test_scenario_actions_deserialize_toml() {
        let toml_str = r#"
            [[actions]]
            name = "grep"
            description = "Search for patterns in files"
            category = "node_expand"

            [[actions.params]]
            name = "pattern"
            description = "Search pattern"
            required = true

            [[actions]]
            name = "read"
            description = "Read file contents"
            category = "node_state_change"
        "#;
        let actions: ScenarioActions = toml::from_str(toml_str).unwrap();
        assert_eq!(actions.actions.len(), 2);
        assert_eq!(actions.action_names(), vec!["grep", "read"]);

        let grep = &actions.actions[0];
        assert_eq!(grep.name, "grep");
        assert_eq!(grep.category, ScenarioActionCategory::NodeExpand);
        assert_eq!(grep.params.len(), 1);
        assert!(grep.params[0].required);
    }

    #[test]
    fn test_scenario_actions_to_core_config() {
        let actions = ScenarioActions {
            actions: vec![ScenarioActionDef {
                name: "test_action".to_string(),
                description: "A test action".to_string(),
                category: ScenarioActionCategory::NodeExpand,
                params: vec![ActionParamDef {
                    name: "param1".to_string(),
                    description: "A parameter".to_string(),
                    required: true,
                }],
                example: Some(r#"{"action": "test_action"}"#.to_string()),
            }],
        };

        let core_config = actions.to_core_config();
        let action_def = core_config.get("test_action");
        assert!(action_def.is_some());
    }
}