mcp-probe-kit 1.15.0 → 2.0.0

package/build/lib/tasks-manager.js

@@ -0,0 +1,217 @@
+/**
+ * Tasks 管理器
+ * 实现 MCP 2025-11-25 Tasks API
+ *
+ * 支持长时间运行的工具任务化执行：
+ * - 创建任务
+ * - 查询任务状态
+ * - 获取任务结果
+ * - 取消任务
+ * - 列出任务
+ */
+import { randomUUID } from 'crypto';
+/**
+ * Tasks 管理器类
+ */
+export class TasksManager {
+    tasks = new Map();
+    cleanupInterval;
+    constructor() {
+        // 每分钟清理过期任务
+        this.cleanupInterval = setInterval(() => {
+            this.cleanupExpiredTasks();
+        }, 60000);
+    }
+    /**
+     * 创建新任务
+     *
+     * @param ttl - 任务生存时间（毫秒），null 表示无限
+     * @param pollInterval - 建议的轮询间隔（毫秒）
+     * @returns 任务元数据
+     */
+    createTask(ttl, pollInterval = 5000) {
+        const taskId = randomUUID();
+        const now = new Date().toISOString();
+        const metadata = {
+            taskId,
+            status: 'working',
+            statusMessage: 'The operation is now in progress.',
+            createdAt: now,
+            lastUpdatedAt: now,
+            pollInterval,
+        };
+        if (ttl !== null && ttl !== undefined) {
+            metadata.ttl = ttl;
+        }
+        const task = {
+            metadata,
+        };
+        // 设置过期时间
+        if (ttl) {
+            task.expiresAt = Date.now() + ttl;
+        }
+        this.tasks.set(taskId, task);
+        console.error(`[Tasks] Created task ${taskId} (TTL: ${ttl || 'unlimited'}ms)`);
+        return metadata;
+    }
+    /**
+     * 更新任务状态
+     *
+     * @param taskId - 任务 ID
+     * @param status - 新状态
+     * @param statusMessage - 状态消息
+     */
+    updateTaskStatus(taskId, status, statusMessage) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        task.metadata.status = status;
+        task.metadata.lastUpdatedAt = new Date().toISOString();
+        if (statusMessage) {
+            task.metadata.statusMessage = statusMessage;
+        }
+        console.error(`[Tasks] Updated task ${taskId} status: ${status}`);
+    }
+    /**
+     * 设置任务结果
+     *
+     * @param taskId - 任务 ID
+     * @param result - 任务结果
+     */
+    setTaskResult(taskId, result) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        task.result = result;
+        task.metadata.status = result.isError ? 'failed' : 'completed';
+        task.metadata.lastUpdatedAt = new Date().toISOString();
+        console.error(`[Tasks] Set result for task ${taskId}`);
+    }
+    /**
+     * 获取任务元数据
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务元数据
+     */
+    getTask(taskId) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        return task.metadata;
+    }
+    /**
+     * 获取任务结果
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务结果
+     */
+    getTaskResult(taskId) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        if (task.metadata.status === 'working') {
+            throw new Error(`Task is still in progress: ${taskId}`);
+        }
+        if (!task.result) {
+            throw new Error(`Task result not available: ${taskId}`);
+        }
+        return task.result;
+    }
+    /**
+     * 取消任务
+     *
+     * @param taskId - 任务 ID
+     */
+    cancelTask(taskId) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        if (task.metadata.status !== 'working') {
+            throw new Error(`Task is not in progress: ${taskId}`);
+        }
+        task.metadata.status = 'cancelled';
+        task.metadata.statusMessage = 'Task was cancelled by user.';
+        task.metadata.lastUpdatedAt = new Date().toISOString();
+        console.error(`[Tasks] Cancelled task ${taskId}`);
+    }
+    /**
+     * 列出所有任务
+     *
+     * @returns 任务元数据列表
+     */
+    listTasks() {
+        return Array.from(this.tasks.values()).map(task => task.metadata);
+    }
+    /**
+     * 执行异步任务
+     *
+     * @param taskId - 任务 ID
+     * @param executor - 任务执行函数
+     */
+    async executeTask(taskId, executor) {
+        const task = this.tasks.get(taskId);
+        if (!task) {
+            throw new Error(`Task not found: ${taskId}`);
+        }
+        try {
+            console.error(`[Tasks] Executing task ${taskId}...`);
+            const result = await executor();
+            this.setTaskResult(taskId, result);
+            console.error(`[Tasks] Task ${taskId} completed successfully`);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            console.error(`[Tasks] Task ${taskId} failed:`, errorMessage);
+            this.setTaskResult(taskId, {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Task failed: ${errorMessage}`,
+                    },
+                ],
+                isError: true,
+            });
+        }
+    }
+    /**
+     * 清理过期任务
+     */
+    cleanupExpiredTasks() {
+        const now = Date.now();
+        let cleanedCount = 0;
+        for (const [taskId, task] of this.tasks.entries()) {
+            if (task.expiresAt && task.expiresAt < now) {
+                this.tasks.delete(taskId);
+                cleanedCount++;
+            }
+        }
+        if (cleanedCount > 0) {
+            console.error(`[Tasks] Cleaned up ${cleanedCount} expired tasks`);
+        }
+    }
+    /**
+     * 销毁管理器
+     */
+    destroy() {
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+        }
+        this.tasks.clear();
+    }
+}
+// 全局单例
+let globalTasksManager = null;
+/**
+ * 获取全局 Tasks 管理器实例
+ */
+export function getTasksManager() {
+    if (!globalTasksManager) {
+        globalTasksManager = new TasksManager();
+    }
+    return globalTasksManager;
+}

package/build/lib/toolset-manager.d.ts

@@ -0,0 +1,48 @@
+/**
+ * 工具集管理器
+ * 根据环境变量 MCP_TOOLSET 过滤工具列表
+ */
+export type ToolsetType = 'core' | 'ui' | 'workflow' | 'full';
+/**
+ * 工具集定义
+ *
+ * - core: 14 个核心工具（高频使用）
+ * - ui: 5 个 UI/UX 工具（推荐使用 start_ui 统一入口）
+ * - workflow: 27 个工作流工具（包含 core + 编排工具）
+ * - full: 所有 39 个工具（默认）
+ */
+export declare const TOOLSET_DEFINITIONS: {
+    core: string[];
+    ui: string[];
+    workflow: string[];
+    full: "all";
+};
+/**
+ * 工具接口（简化版，用于过滤）
+ */
+export interface Tool {
+    name: string;
+    description: string;
+    inputSchema: any;
+}
+/**
+ * 根据工具集类型过滤工具列表
+ *
+ * @param tools - 完整的工具列表
+ * @param toolset - 工具集类型
+ * @returns 过滤后的工具列表
+ */
+export declare function filterTools(tools: Tool[], toolset: ToolsetType): Tool[];
+/**
+ * 从环境变量获取工具集类型
+ *
+ * @returns 工具集类型，默认为 'full'
+ */
+export declare function getToolsetFromEnv(): ToolsetType;
+/**
+ * 获取工具集的工具数量
+ *
+ * @param toolset - 工具集类型
+ * @returns 工具数量
+ */
+export declare function getToolsetSize(toolset: ToolsetType): number;

package/build/lib/toolset-manager.js

@@ -0,0 +1,112 @@
+/**
+ * 工具集管理器
+ * 根据环境变量 MCP_TOOLSET 过滤工具列表
+ */
+/**
+ * 工具集定义
+ *
+ * - core: 14 个核心工具（高频使用）
+ * - ui: 5 个 UI/UX 工具（推荐使用 start_ui 统一入口）
+ * - workflow: 27 个工作流工具（包含 core + 编排工具）
+ * - full: 所有 39 个工具（默认）
+ */
+export const TOOLSET_DEFINITIONS = {
+    // 核心工具集（14 个）- 最常用的基础工具
+    core: [
+        'gencommit',
+        'code_review',
+        'debug',
+        'gentest',
+        'gendoc',
+        'refactor',
+        'perf',
+        'security_scan',
+        'fix_bug',
+        'genapi',
+        'gensql',
+        'genreadme',
+        'check_deps',
+        'gen_mock',
+    ],
+    // UI/UX 工具集（5 个）- 对外工具，推荐使用 start_ui 统一入口
+    ui: [
+        'start_ui', // ⭐ 统一入口（编排工具）
+        'ui_search', // 搜索 UI/UX 数据库
+        'ui_design_system', // 生成设计系统
+        'sync_ui_data', // 同步 UI 数据
+        // 注意：不包含内部工具 init_component_catalog 和 render_ui
+    ],
+    // 工作流工具集（27 个）- 包含核心工具 + 编排工具
+    workflow: [
+        // 核心工具（复用）
+        'gencommit',
+        'code_review',
+        'debug',
+        'gentest',
+        'gendoc',
+        'refactor',
+        'perf',
+        'security_scan',
+        'fix_bug',
+        'genapi',
+        'gensql',
+        'genreadme',
+        'check_deps',
+        'gen_mock',
+        // 编排工具（10 个）
+        'start_feature',
+        'start_bugfix',
+        'start_review',
+        'start_release',
+        'start_refactor',
+        'start_onboard',
+        'start_api',
+        'start_doc',
+        'start_ralph',
+        'start_ui',
+        // 交互工具（2 个）
+        'interview',
+        'ask_user',
+    ],
+    // 完整工具集（39 个）- 包含所有工具（包括内部工具）
+    full: 'all',
+};
+/**
+ * 根据工具集类型过滤工具列表
+ *
+ * @param tools - 完整的工具列表
+ * @param toolset - 工具集类型
+ * @returns 过滤后的工具列表
+ */
+export function filterTools(tools, toolset) {
+    if (toolset === 'full') {
+        return tools;
+    }
+    const allowedTools = TOOLSET_DEFINITIONS[toolset];
+    return tools.filter(tool => allowedTools.includes(tool.name));
+}
+/**
+ * 从环境变量获取工具集类型
+ *
+ * @returns 工具集类型，默认为 'full'
+ */
+export function getToolsetFromEnv() {
+    const toolset = process.env.MCP_TOOLSET?.toLowerCase();
+    if (toolset === 'core' || toolset === 'ui' || toolset === 'workflow') {
+        return toolset;
+    }
+    return 'full';
+}
+/**
+ * 获取工具集的工具数量
+ *
+ * @param toolset - 工具集类型
+ * @returns 工具数量
+ */
+export function getToolsetSize(toolset) {
+    if (toolset === 'full') {
+        return 39;
+    }
+    const allowedTools = TOOLSET_DEFINITIONS[toolset];
+    return allowedTools.length;
+}

package/build/schemas/basic-tools.d.ts

@@ -2,38 +2,6 @@
  * 基础工具的 Schema 定义
  */
 export declare const basicToolSchemas: readonly [{
-    readonly name: "detect_shell";
-    readonly description: "当用户询问当前 AI 环境、是否为套壳产品时使用。检测 AI 应用环境指纹，返回 JSON 检测报告";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly nonce: {
-                readonly type: "string";
-                readonly description: "随机数，用于网络检测。可选";
-            };
-            readonly skip_network: {
-                readonly type: "boolean";
-                readonly description: "是否跳过网络检测。可选，默认 false";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
-}, {
-    readonly name: "init_setting";
-    readonly description: "当用户需要初始化或配置 Cursor IDE 设置时使用。写入推荐的 AI 配置到 .cursor/settings.json";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly project_path: {
-                readonly type: "string";
-                readonly description: "项目路径。可选，默认为当前工作区路径";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
-}, {
     readonly name: "init_project";
     readonly description: "当用户需要创建新项目、生成项目结构时使用。按 Spec-Driven Development 方式生成需求/设计/任务文档";
     readonly inputSchema: {

package/build/schemas/basic-tools.js

@@ -2,40 +2,6 @@
  * 基础工具的 Schema 定义
  */
 export const basicToolSchemas = [
-    {
-        name: "detect_shell",
-        description: "当用户询问当前 AI 环境、是否为套壳产品时使用。检测 AI 应用环境指纹，返回 JSON 检测报告",
-        inputSchema: {
-            type: "object",
-            properties: {
-                nonce: {
-                    type: "string",
-                    description: "随机数，用于网络检测。可选",
-                },
-                skip_network: {
-                    type: "boolean",
-                    description: "是否跳过网络检测。可选，默认 false",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
-    {
-        name: "init_setting",
-        description: "当用户需要初始化或配置 Cursor IDE 设置时使用。写入推荐的 AI 配置到 .cursor/settings.json",
-        inputSchema: {
-            type: "object",
-            properties: {
-                project_path: {
-                    type: "string",
-                    description: "项目路径。可选，默认为当前工作区路径",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
     {
         name: "init_project",
         description: "当用户需要创建新项目、生成项目结构时使用。按 Spec-Driven Development 方式生成需求/设计/任务文档",

package/build/schemas/code-analysis-tools.d.ts

@@ -37,24 +37,6 @@ export declare const codeAnalysisToolSchemas: readonly [{
         readonly required: readonly [];
         readonly additionalProperties: true;
     };
-}, {
-    readonly name: "explain";
-    readonly description: "当用户不理解某段代码、需要代码解释时使用。解释代码逻辑和实现原理，包含执行流程、关键概念";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly code: {
-                readonly type: "string";
-                readonly description: "要解释的代码。可以是代码片段或完整函数";
-            };
-            readonly context: {
-                readonly type: "string";
-                readonly description: "业务背景或上下文。可选，有助于更好的解释";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
 }, {
     readonly name: "perf";
     readonly description: "当用户关注代码性能、需要优化性能时使用。分析性能瓶颈（算法/内存/数据库/React渲染），输出瓶颈清单和优化建议";
@@ -113,24 +95,6 @@ export declare const codeAnalysisToolSchemas: readonly [{
         readonly required: readonly [];
         readonly additionalProperties: true;
     };
-}, {
-    readonly name: "fix";
-    readonly description: "当用户需要自动修复代码问题（Lint/格式化/类型错误）时使用。自动修复可机械化问题，输出补丁（unified diff）";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly code: {
-                readonly type: "string";
-                readonly description: "要修复的代码";
-            };
-            readonly type: {
-                readonly type: "string";
-                readonly description: "修复类型：lint（代码规范）、ts（TypeScript错误）、format（格式化）、import（导入）。可选，会自动识别";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
 }, {
     readonly name: "fix_bug";
     readonly description: "当用户需要修复 Bug、获取修复指导时使用。提供 Bug 修复流程指导，包含根因分析、修复方案、验证步骤";

package/build/schemas/code-analysis-tools.js

@@ -40,25 +40,6 @@ export const codeAnalysisToolSchemas = [
             additionalProperties: true,
         },
     },
-    {
-        name: "explain",
-        description: "当用户不理解某段代码、需要代码解释时使用。解释代码逻辑和实现原理，包含执行流程、关键概念",
-        inputSchema: {
-            type: "object",
-            properties: {
-                code: {
-                    type: "string",
-                    description: "要解释的代码。可以是代码片段或完整函数",
-                },
-                context: {
-                    type: "string",
-                    description: "业务背景或上下文。可选，有助于更好的解释",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
     {
         name: "perf",
         description: "当用户关注代码性能、需要优化性能时使用。分析性能瓶颈（算法/内存/数据库/React渲染），输出瓶颈清单和优化建议",
@@ -120,25 +101,6 @@ export const codeAnalysisToolSchemas = [
             additionalProperties: true,
         },
     },
-    {
-        name: "fix",
-        description: "当用户需要自动修复代码问题（Lint/格式化/类型错误）时使用。自动修复可机械化问题，输出补丁（unified diff）",
-        inputSchema: {
-            type: "object",
-            properties: {
-                code: {
-                    type: "string",
-                    description: "要修复的代码",
-                },
-                type: {
-                    type: "string",
-                    description: "修复类型：lint（代码规范）、ts（TypeScript错误）、format（格式化）、import（导入）。可选，会自动识别",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
     {
         name: "fix_bug",
         description: "当用户需要修复 Bug、获取修复指导时使用。提供 Bug 修复流程指导，包含根因分析、修复方案、验证步骤",

package/build/schemas/code-gen-tools.d.ts

@@ -59,24 +59,6 @@ export declare const codeGenToolSchemas: readonly [{
         readonly required: readonly [];
         readonly additionalProperties: true;
     };
-}, {
-    readonly name: "genui";
-    readonly description: "当用户需要生成 UI 组件代码时使用。根据描述生成 UI 组件代码（React/Vue/HTML），包含 Props 和样式";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly description: {
-                readonly type: "string";
-                readonly description: "组件描述。可以是简短描述（如'登录表单组件'）或详细的UI需求";
-            };
-            readonly framework: {
-                readonly type: "string";
-                readonly description: "前端框架：react、vue、html。可选，默认 react";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
 }, {
     readonly name: "gensql";
     readonly description: "当用户需要根据自然语言生成 SQL 查询时使用。将自然语言描述转换为 SQL 语句（PostgreSQL/MySQL/SQLite）";
@@ -121,30 +103,4 @@ export declare const codeGenToolSchemas: readonly [{
         readonly required: readonly [];
         readonly additionalProperties: true;
     };
-}, {
-    readonly name: "design2code";
-    readonly description: "当用户需要将设计稿转换为代码时使用。将设计稿（图片URL/描述/HTML）转换为前端代码（React/Vue），1:1 还原布局和样式";
-    readonly inputSchema: {
-        readonly type: "object";
-        readonly properties: {
-            readonly input: {
-                readonly type: "string";
-                readonly description: "设计稿输入：图片 URL、base64 图片、HTML 代码或设计稿描述";
-            };
-            readonly framework: {
-                readonly type: "string";
-                readonly description: "目标框架：vue、react。可选，默认为 vue";
-            };
-            readonly style_solution: {
-                readonly type: "string";
-                readonly description: "样式方案：tailwind、css-modules、styled-components。可选，默认为 tailwind";
-            };
-            readonly component_type: {
-                readonly type: "string";
-                readonly description: "组件类型：page（页面组件）、component（通用组件）。可选，默认为 page";
-            };
-        };
-        readonly required: readonly [];
-        readonly additionalProperties: true;
-    };
 }];

package/build/schemas/code-gen-tools.js

@@ -63,25 +63,6 @@ export const codeGenToolSchemas = [
             additionalProperties: true,
         },
     },
-    {
-        name: "genui",
-        description: "当用户需要生成 UI 组件代码时使用。根据描述生成 UI 组件代码（React/Vue/HTML），包含 Props 和样式",
-        inputSchema: {
-            type: "object",
-            properties: {
-                description: {
-                    type: "string",
-                    description: "组件描述。可以是简短描述（如'登录表单组件'）或详细的UI需求",
-                },
-                framework: {
-                    type: "string",
-                    description: "前端框架：react、vue、html。可选，默认 react",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
     {
         name: "gensql",
         description: "当用户需要根据自然语言生成 SQL 查询时使用。将自然语言描述转换为 SQL 语句（PostgreSQL/MySQL/SQLite）",
@@ -128,31 +109,4 @@ export const codeGenToolSchemas = [
             additionalProperties: true,
         },
     },
-    {
-        name: "design2code",
-        description: "当用户需要将设计稿转换为代码时使用。将设计稿（图片URL/描述/HTML）转换为前端代码（React/Vue），1:1 还原布局和样式",
-        inputSchema: {
-            type: "object",
-            properties: {
-                input: {
-                    type: "string",
-                    description: "设计稿输入：图片 URL、base64 图片、HTML 代码或设计稿描述",
-                },
-                framework: {
-                    type: "string",
-                    description: "目标框架：vue、react。可选，默认为 vue",
-                },
-                style_solution: {
-                    type: "string",
-                    description: "样式方案：tailwind、css-modules、styled-components。可选，默认为 tailwind",
-                },
-                component_type: {
-                    type: "string",
-                    description: "组件类型：page（页面组件）、component（通用组件）。可选，默认为 page",
-                },
-            },
-            required: [],
-            additionalProperties: true,
-        },
-    },
 ];