Arcrun/mcp/src/tools/arcrun_search_workflows.ts

import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { toolName } from "../brand.js";
import { z } from "zod";
import { Env } from "../types.js";

/**
 * arcrun_search_workflows — 用自然語言找現成工作流（workflow-discovery R2）
 *
 * 北極星入口：AI 先查「有沒有現成工作流能做這件事」→ 找到就執行，別重造。
 * 呼叫 cypher GET /workflows/search → 轉發 KBDB /entries/search（entry_type=workflow + 本租戶）。
 * 優先語意搜尋；KBDB 未開 Vectorize → 自動降級關鍵字 + 回 capability_hint（不假裝語義）。
 *
 * 薄殼（rule 07）：只做參數轉換 + 呼叫 + 格式化，零業務邏輯。形態對齊 arcrun_search_components。
 * flag 安全：AI 收到意圖時主動 call 一次，無輪詢/排程。
 */
export function registerSearchWorkflows(
  server: McpServer,
  env: Env,
  orgNamespace: string,
  partnerToken: string,
) {
  server.tool(
    toolName("search_workflows"),
    "用自然語言找現成的工作流（先查有沒有現成的能做這件事，找到就用，別重造）。例如：「把資料寫進 Google Sheets」、「每天抓 RSS 發通知」、「webhook 轉發到別的 API」。回傳本帳號下符合的工作流清單。",
    {
      query: z.string().describe("自然語言描述要找的工作流，如「把資料寫進 Google Sheets」"),
    },
    async ({ query }) => {
      try {
        if (!env.CYPHER_EXECUTOR) {
          return {
            content: [{ type: "text", text: "Error: CYPHER_EXECUTOR service binding is not configured." }],
            isError: true,
          };
        }

        const response = await env.CYPHER_EXECUTOR.fetch(
          `http://cypher-executor/workflows/search?q=${encodeURIComponent(query)}`,
          { method: "GET", headers: { "X-Arcrun-API-Key": partnerToken } },
        );

        if (!response.ok) {
          const errorText = await response.text();
          return {
            content: [{ type: "text", text: `Search failed: ${errorText}` }],
            isError: true,
          };
        }

        const result = await response.json() as {
          entries?: Array<{ page_name?: string; content?: string }>;
          count?: number;
          mode?: string;
          capability_hint?: string;
        };
        const entries = result.entries ?? [];
        const count = result.count ?? entries.length;

        if (count === 0) {
          const hint = result.capability_hint ? `\n\n（${result.capability_hint}）` : "";
          return {
            content: [{
              type: "text",
              text: `找不到符合「${query}」的現成工作流。可以用 arcrun_push_workflow 部署一個新的。${hint}`,
            }],
          };
        }

        // capability_hint 透傳給 AI：未開語義時 AI 看到就能主動問用戶要不要開 Vectorize（R2.3 閉環）。
        const hintLine = result.capability_hint
          ? `\n\n⚠️ ${result.capability_hint}`
          : "";
        return {
          content: [{
            type: "text",
            text: `找到 ${count} 個工作流（mode: ${result.mode ?? "keyword"}）：\n${JSON.stringify(entries, null, 2)}${hintLine}`,
          }],
        };
      } catch (error) {
        return {
          content: [{ type: "text", text: `Internal Error: ${error instanceof Error ? error.message : String(error)}` }],
          isError: true,
        };
      }
    }
  );
}