DevFlow/src-tauri/src/commands/ai/tools/idea.rs

//! 灵感类 AI 工具声明式注册(list + create + update + delete 共 4 个)。
//!
//! 迁自 `tool_registry.rs::register_idea_tools`(原 2 个 list_ideas/create_idea),
//! 改用 `declare_tool!` 宏。新增 update_idea / delete_idea(AI 改/删灵感的专用工具,
//! 后端 idea_repo.update_field / soft_delete 早已具备,工具层此前缺失,致 AI 改/删灵感
//! 时被迫误用 update_task/delete_task)。
//!
//! 迁移策略(handler 逻辑零变更):
//! - handler body 逐字照搬原 `register_idea_tools` 内 async move 块(逻辑等价),
//!   仅闭包包装(`{ let db = db.clone(); Box::new(move |args| { let db = db.clone();
//!   Box::pin(async move { ... }) }) }`)改由 `declare_tool!` 宏生成。
//! - name/desc/schema/risk 与原手写定义逐字一致。
//! - 复用类型/常量:new_id/IdeaStatus/IdeaRecord/now_millis(各 pub(crate) 来源,与原同源) +
//!   MAX_LIST_RESULTS(super::tool_registry 单真相源)。
//!
//! 等价性验证:基线测试 `test_build_ai_tool_registry_baseline_tool_count` 断言 50 总量 +
//! 工具名集合稳定(防 rename / 漏注册)。

use std::sync::Arc;

use df_ai::ai_tools::{object_schema, AiToolRegistry, GoalMeta, GoalStyle, RiskLevel, ToolMeta};
use df_ai::declare_tool;
use df_storage::db::Database;
use df_storage::models::IdeaRecord;
use df_types::types::{new_id, IdeaStatus};

use crate::commands::now_millis;
use crate::commands::ai::tool_registry::MAX_LIST_RESULTS;

/// 注册 4 个灵感类工具(list/create/update/delete)到 `$registry`。
///
/// 与原手写 register(name, desc, schema, risk, handler) 语义 1:1:
/// - name/desc/schema 字符串与 JSON Schema 逐字照搬原定义
/// - risk 与原一致(list=Low,create/update=Medium,delete=High)
/// - handler body 与原 async move 块逐字一致(逻辑零变更)
///
/// 唯一差异:闭包包装改由 `declare_tool!` 宏生成,handler body 直接写业务逻辑。
pub fn register(registry: &mut AiToolRegistry, db: &Arc<Database>) {
    declare_tool!(
        registry,
        db: Arc<Database>,
        "list_ideas",
        "列出所有灵感，支持 offset/limit 分页。返回 items、total、has_more。默认 limit=50",
        RiskLevel::Low,
        schema: object_schema(vec![("offset", "integer", false), ("limit", "integer", false)]),
        args => {
            let repo = df_storage::crud::IdeaRepo::new(&db);
            let items = repo.list_all().await?;
            let total = items.len();
            let offset = args["offset"].as_u64().unwrap_or(0) as usize;
            let limit = args["limit"].as_u64().unwrap_or(MAX_LIST_RESULTS as u64).min(MAX_LIST_RESULTS as u64) as usize;
            let page_items: Vec<_> = items.into_iter().skip(offset).take(limit).collect();
            let has_more = (offset + page_items.len()) < total;
            Ok(serde_json::json!({ "items": page_items, "total": total, "has_more": has_more }))
        }
    );

    declare_tool!(
        registry,
        db: Arc<Database>,
        "create_idea",
        "捕获一个新灵感",
        RiskLevel::Medium,
        meta: ToolMeta {
            entity_action: Some(("idea", "create")),
            approval_hint: Some(("捕获灵感：{}", &["title"])),
            display_verb: Some("捕获灵感"),
            goal: Some(GoalMeta { verb: "捕获灵感", keys: &["title"], style: GoalStyle::Raw, fallback: None }),
            ..Default::default()
        },
        schema: object_schema(vec![
            ("title", "string", true), ("description", "string", false),
            ("tags", "string", false), ("source", "string", false),
            // priority:与 commands::idea::CreateIdeaInput 默认值一致(=1,灵感默认普通优先级)
            ("priority", "integer", false),
        ]),
        args => {
            let title = args["title"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 title"))?;
            let repo = df_storage::crud::IdeaRepo::new(&db);
            let record = IdeaRecord {
                id: new_id(), title: title.to_string(),
                description: args["description"].as_str().unwrap_or("").to_string(),
                // priority 默认 1:灵感默认普通优先级(与 commands::idea::default_priority 及 tasks 表 SQL DEFAULT 1 对齐)
                status: IdeaStatus::Draft, priority: args["priority"].as_i64().unwrap_or(1) as i32,
                score: None, tags: args["tags"].as_str().map(|s| s.to_string()),
                source: args["source"].as_str().map(|s| s.to_string()),
                promoted_to: None, ai_analysis: None, scores: None,
                related_ids: None,
                created_at: now_millis(), updated_at: now_millis(),
            };
            let id = record.id.clone();
            repo.insert(record).await?;
            Ok(serde_json::json!({ "id": id, "title": title, "status": "draft" }))
        }
    );

    // update_idea / delete_idea:AI 改/删灵感的专用工具(对标 update_task / delete_task)。
    // 后端 idea_repo.update_field / soft_delete 早已具备,工具层此前缺失,致 AI 改/删灵感
    // 时被迫误用 update_task(任务表)——既改错表又触发 status 收口误报。
    // 与 update_task 同源:白名单复用 df-storage CRUD(is_allowed_column("ideas", field)),
    // 双重防御(schema 限制 + 白名单校验)。返回 updated 透传 affected bool(不硬编码 true),
    // 0 行(id 不存在 / 已删)→ updated=false + title 空,让 AI 知道 id 不对而非假成功。
    declare_tool!(
        registry,
        db: Arc<Database>,
        "update_idea",
        "更新灵感的指定字段（title/description/priority/tags/source/score 等），需要提供灵感 ID、字段名和新值。",
        RiskLevel::Medium,
        meta: ToolMeta {
            entity_action: Some(("idea", "update")),
            display_verb: Some("修改灵感"),
            ..Default::default()
        },
        schema: object_schema(vec![("id", "string", true), ("field", "string", true), ("value", "string", true)]),
        args => {
            let id = args["id"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 id"))?;
            let field = args["field"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 field"))?;
            let value = args["value"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 value"))?;
            // 复用 df-storage CRUD 白名单(按表隔离),与 update_field 校验同源
            if !df_storage::crud::is_allowed_column("ideas", field) {
                anyhow::bail!("不允许更新字段 '{}'", field);
            }
            let repo = df_storage::crud::IdeaRepo::new(&db);
            let affected = repo.update_field(id, field, value).await?;
            // 对齐 update_task:返回 title 供前端卡片友好展示(防裸 UUID 辨识度低)。
            // 0 行(id 不存在 / 已软删)→ affected=false 且 get_by_id 返 None→title 空,
            // 透传 affected 让 AI 知道 id 不对而非假成功。
            let title = repo.get_by_id(id).await?.map(|i| i.title).unwrap_or_default();
            Ok(serde_json::json!({ "id": id, "title": title, "field": field, "updated": affected }))
        }
    );

    declare_tool!(
        registry,
        db: Arc<Database>,
        "delete_idea",
        "删除灵感（软删除，移入回收站，可恢复）",
        RiskLevel::Medium,
        meta: ToolMeta {
            entity_action: Some(("idea", "delete")),
            display_verb: Some("删除灵感"),
            ..Default::default()
        },
        schema: object_schema(vec![("id", "string", true)]),
        args => {
            let id = args["id"].as_str().ok_or_else(|| anyhow::anyhow!("缺少 id"))?;
            let repo = df_storage::crud::IdeaRepo::new(&db);
            let affected = repo.soft_delete(id).await?;
            Ok(serde_json::json!({ "id": id, "deleted": affected }))
        }
    );
}