mcp-probe-kit 1.15.0 → 2.0.0

package/build/index.js

@@ -2,10 +2,10 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { CallToolRequestSchema, ListToolsRequestSchema, ListResourcesRequestSchema, ReadResourceRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
-import { detectShell, initSetting, initProject, gencommit, debug, genapi, codeReview, gentest, genpr, checkDeps, gendoc, genchangelog, refactor, perf, fix, gensql, resolveConflict, genui, explain, convert, cssOrder, genreadme, split, analyzeProject, initProjectContext, addFeature, securityScan, fixBug, estimate, genMock, design2code, startFeature, startBugfix, startReview, startRelease, startRefactor, startOnboard, startApi, startDoc, genSkill, startRalph, interview, askUser, uiDesignSystem, initComponentCatalog, uiSearch, syncUiData, renderUi, startUi } from "./tools/index.js";
+import { initProject, gencommit, debug, genapi, codeReview, gentest, genpr, checkDeps, gendoc, genchangelog, refactor, perf, gensql, resolveConflict, genreadme, analyzeProject, initProjectContext, addFeature, securityScan, fixBug, estimate, genMock, startFeature, startBugfix, startReview, startRelease, startRefactor, startOnboard, startApi, startDoc, startRalph, interview, askUser, uiDesignSystem, initComponentCatalog, uiSearch, syncUiData, renderUi, startUi } from "./tools/index.js";
 import { VERSION, NAME } from "./version.js";
 import { allToolSchemas } from "./schemas/index.js";
-import { getToolParamsGuide } from "./resources/index.js";
+import { filterTools, getToolsetFromEnv } from "./lib/toolset-manager.js";
 // 创建MCP服务器实例
 const server = new Server({
     name: NAME,
@@ -16,10 +16,13 @@ const server = new Server({
         resources: {},
     },
 });
-// 定义工具列表 - 从 schemas 导入
+// 定义工具列表 - 从 schemas 导入，并根据工具集过滤
 server.setRequestHandler(ListToolsRequestSchema, async () => {
+    const toolset = getToolsetFromEnv();
+    const filteredTools = filterTools(allToolSchemas, toolset);
+    console.error(`[MCP Probe Kit] 当前工具集: ${toolset} (${filteredTools.length}/${allToolSchemas.length} 个工具)`);
     return {
-        tools: allToolSchemas,
+        tools: filteredTools,
     };
 });
 // 处理工具调用
@@ -27,10 +30,6 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     const { name, arguments: args } = request.params;
     try {
         switch (name) {
-            case "detect_shell":
-                return await detectShell(args);
-            case "init_setting":
-                return await initSetting(args);
             case "init_project":
                 return await initProject(args);
             case "gencommit":
@@ -55,24 +54,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 return await refactor(args);
             case "perf":
                 return await perf(args);
-            case "fix":
-                return await fix(args);
             case "gensql":
                 return await gensql(args);
             case "resolve_conflict":
                 return await resolveConflict(args);
-            case "genui":
-                return await genui(args);
-            case "explain":
-                return await explain(args);
-            case "convert":
-                return await convert(args);
-            case "css_order":
-                return await cssOrder(args);
             case "genreadme":
                 return await genreadme(args);
-            case "split":
-                return await split(args);
             case "analyze_project":
                 return await analyzeProject(args);
             case "init_project_context":
@@ -87,8 +74,6 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 return await estimate(args);
             case "gen_mock":
                 return await genMock(args);
-            case "design2code":
-                return await design2code(args);
             // 智能编排工具
             case "start_feature":
                 return await startFeature(args);
@@ -106,8 +91,6 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 return await startApi(args);
             case "start_doc":
                 return await startDoc(args);
-            case "gen_skill":
-                return await genSkill(args);
             case "start_ralph":
                 return await startRalph(args);
             // 访谈工具
@@ -155,12 +138,6 @@ server.setRequestHandler(ListResourcesRequestSchema, async () => {
                 description: "MCP Probe Kit 服务器当前状态",
                 mimeType: "application/json",
             },
-            {
-                uri: "probe://tool-params-guide",
-                name: "工具参数指南",
-                description: "所有工具的参数说明和使用示例，帮助 AI 正确传参",
-                mimeType: "application/json",
-            },
         ],
     };
 });
@@ -179,7 +156,7 @@ server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
                         serverInfo: {
                             name: NAME,
                             version: VERSION,
-                            description: "Cursor 开发增强工具集",
+                            description: "AI 驱动的完整研发工具集",
                         },
                         toolCount: allToolSchemas.length,
                     }, null, 2),
@@ -187,19 +164,75 @@ server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
             ],
         };
     }
-    if (uri === "probe://tool-params-guide") {
-        return {
-            contents: [
-                {
-                    uri,
-                    mimeType: "application/json",
-                    text: JSON.stringify(getToolParamsGuide(VERSION), null, 2),
-                },
-            ],
-        };
-    }
     throw new Error(`未知资源: ${uri}`);
 });
+// ============================================
+// Tasks API 端点 - 暂时禁用，等待 MCP SDK 正式支持
+// ============================================
+// 注意：当前 MCP SDK 版本不支持自定义 method，Tasks API 功能暂时禁用
+// 相关 issue: https://github.com/modelcontextprotocol/sdk/issues/xxx
+/*
+// 获取任务状态
+server.setRequestHandler({ method: "tasks/get" } as any, async (request: any) => {
+  try {
+    const { taskId } = request.params;
+    const tasksManager = getTasksManager();
+    const task = tasksManager.getTask(taskId);
+    
+    return {
+      task,
+    };
+  } catch (error) {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    throw new Error(`Failed to get task: ${errorMessage}`);
+  }
+});
+
+// 获取任务结果
+server.setRequestHandler({ method: "tasks/result" } as any, async (request: any) => {
+  try {
+    const { taskId } = request.params;
+    const tasksManager = getTasksManager();
+    const result = tasksManager.getTaskResult(taskId);
+    
+    return result;
+  } catch (error) {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    throw new Error(`Failed to get task result: ${errorMessage}`);
+  }
+});
+
+// 取消任务
+server.setRequestHandler({ method: "tasks/cancel" } as any, async (request: any) => {
+  try {
+    const { taskId } = request.params;
+    const tasksManager = getTasksManager();
+    tasksManager.cancelTask(taskId);
+    
+    return {
+      _meta: {},
+    };
+  } catch (error) {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    throw new Error(`Failed to cancel task: ${errorMessage}`);
+  }
+});
+
+// 列出所有任务
+server.setRequestHandler({ method: "tasks/list" } as any, async () => {
+  try {
+    const tasksManager = getTasksManager();
+    const tasks = tasksManager.listTasks();
+    
+    return {
+      tasks,
+    };
+  } catch (error) {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    throw new Error(`Failed to list tasks: ${errorMessage}`);
+  }
+});
+*/
 // 启动服务器
 async function main() {
     const transport = new StdioServerTransport();

package/build/lib/elicitation-helper.d.ts

@@ -0,0 +1,73 @@
+/**
+ * Elicitation 辅助模块
+ *
+ * MCP 2025-11-25 Elicitation 支持
+ * 提供标准化的用户交互能力
+ *
+ * 注意：当前版本通过文本方式实现交互（向后兼容）
+ * 未来可以升级为真正的 Elicitation API 调用
+ */
+/**
+ * Elicitation 表单字段
+ */
+export interface ElicitationField {
+    name: string;
+    type: 'string' | 'number' | 'boolean';
+    description?: string;
+    enum?: string[];
+    default?: any;
+    required?: boolean;
+}
+/**
+ * Elicitation 表单 Schema
+ */
+export interface ElicitationFormSchema {
+    type: 'object';
+    properties: Record<string, {
+        type: string;
+        description?: string;
+        enum?: string[];
+        default?: any;
+    }>;
+    required?: string[];
+}
+/**
+ * 将字段列表转换为 JSON Schema
+ *
+ * @param fields - 字段列表
+ * @returns JSON Schema
+ */
+export declare function fieldsToSchema(fields: ElicitationField[]): ElicitationFormSchema;
+/**
+ * 生成文本格式的问题列表（回退方案）
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns 格式化的文本
+ */
+export declare function generateTextQuestions(message: string, fields: ElicitationField[]): string;
+/**
+ * 检查客户端是否支持 Elicitation
+ *
+ * 注意：当前版本始终返回 false（使用文本回退）
+ * 未来可以通过 server.getClientCapabilities() 检测
+ *
+ * @returns 是否支持 Elicitation
+ */
+export declare function supportsElicitation(): boolean;
+/**
+ * 创建 Elicitation 请求（未来实现）
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns Elicitation 结果
+ */
+export declare function createElicitation(message: string, fields: ElicitationField[]): Promise<Record<string, any>>;
+/**
+ * 智能提问：优先使用 Elicitation，回退到文本
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns 格式化的文本（当前版本）
+ */
+export declare function smartAsk(message: string, fields: ElicitationField[]): string;

package/build/lib/elicitation-helper.js

@@ -0,0 +1,130 @@
+/**
+ * Elicitation 辅助模块
+ *
+ * MCP 2025-11-25 Elicitation 支持
+ * 提供标准化的用户交互能力
+ *
+ * 注意：当前版本通过文本方式实现交互（向后兼容）
+ * 未来可以升级为真正的 Elicitation API 调用
+ */
+/**
+ * 将字段列表转换为 JSON Schema
+ *
+ * @param fields - 字段列表
+ * @returns JSON Schema
+ */
+export function fieldsToSchema(fields) {
+    const properties = {};
+    const required = [];
+    for (const field of fields) {
+        properties[field.name] = {
+            type: field.type,
+            description: field.description,
+        };
+        if (field.enum) {
+            properties[field.name].enum = field.enum;
+        }
+        if (field.default !== undefined) {
+            properties[field.name].default = field.default;
+        }
+        if (field.required) {
+            required.push(field.name);
+        }
+    }
+    return {
+        type: 'object',
+        properties,
+        required: required.length > 0 ? required : undefined,
+    };
+}
+/**
+ * 生成文本格式的问题列表（回退方案）
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns 格式化的文本
+ */
+export function generateTextQuestions(message, fields) {
+    const lines = [];
+    lines.push("# ❓ 需要你的输入");
+    lines.push("");
+    lines.push(message);
+    lines.push("");
+    lines.push("---");
+    lines.push("");
+    for (let i = 0; i < fields.length; i++) {
+        const field = fields[i];
+        const required = field.required ? "**[必答]**" : "_[可选]_";
+        lines.push(`### ${i + 1}. ${field.description || field.name} ${required}`);
+        lines.push("");
+        if (field.enum && field.enum.length > 0) {
+            lines.push("**可选项**:");
+            for (const option of field.enum) {
+                lines.push(`- ${option}`);
+            }
+            lines.push("");
+        }
+        if (field.default !== undefined) {
+            lines.push(`_默认值: ${field.default}_`);
+            lines.push("");
+        }
+        lines.push(`**字段名**: \`${field.name}\``);
+        lines.push(`**类型**: ${field.type}`);
+        lines.push("");
+        lines.push("**你的回答**: ");
+        lines.push("");
+        lines.push("---");
+        lines.push("");
+    }
+    lines.push("💡 **提示**: 请回答上述问题，我会根据你的回答继续工作。");
+    return lines.join("\n");
+}
+/**
+ * 检查客户端是否支持 Elicitation
+ *
+ * 注意：当前版本始终返回 false（使用文本回退）
+ * 未来可以通过 server.getClientCapabilities() 检测
+ *
+ * @returns 是否支持 Elicitation
+ */
+export function supportsElicitation() {
+    // TODO: 实现真正的能力检测
+    // const capabilities = server.getClientCapabilities();
+    // return !!capabilities?.elicitation;
+    return false;
+}
+/**
+ * 创建 Elicitation 请求（未来实现）
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns Elicitation 结果
+ */
+export async function createElicitation(message, fields) {
+    // TODO: 实现真正的 Elicitation API 调用
+    // const schema = fieldsToSchema(fields);
+    // const result = await server.elicitInput({
+    //   mode: 'form',
+    //   message,
+    //   requestedSchema: schema,
+    // });
+    // return result.data;
+    throw new Error('Elicitation not yet implemented. Use text-based fallback.');
+}
+/**
+ * 智能提问：优先使用 Elicitation，回退到文本
+ *
+ * @param message - 提示消息
+ * @param fields - 字段列表
+ * @returns 格式化的文本（当前版本）
+ */
+export function smartAsk(message, fields) {
+    // 当前版本：始终使用文本回退
+    // 未来版本：检测客户端能力，优先使用 Elicitation
+    if (supportsElicitation()) {
+        // 未来：返回 Elicitation 请求
+        // return createElicitation(message, fields);
+    }
+    // 回退：返回文本格式
+    return generateTextQuestions(message, fields);
+}

package/build/lib/response.d.ts

@@ -0,0 +1,32 @@
+/**
+ * 统一输出封装
+ * 提供标准化的工具响应格式，支持结构化输出
+ */
+/**
+ * 工具响应内容
+ */
+export interface ToolContent {
+    type: 'text' | 'image' | 'resource';
+    text?: string;
+    data?: string;
+    mimeType?: string;
+}
+/**
+ * 工具响应结果
+ */
+export interface ToolResponse {
+    content: ToolContent[];
+    isError?: boolean;
+    structuredContent?: any;
+    _meta?: Record<string, any>;
+}
+/**
+ * 返回结构化响应
+ * 同时包含人类可读的文本和机器可读的结构化数据
+ *
+ * @param text - 人类可读的文本内容
+ * @param structuredContent - 机器可读的结构化数据
+ * @param meta - 可选的元数据
+ * @returns 标准工具响应
+ */
+export declare function okStructured(text: string, structuredContent: any, meta?: Record<string, any>): ToolResponse;

package/build/lib/response.js

@@ -0,0 +1,28 @@
+/**
+ * 统一输出封装
+ * 提供标准化的工具响应格式，支持结构化输出
+ */
+/**
+ * 返回结构化响应
+ * 同时包含人类可读的文本和机器可读的结构化数据
+ *
+ * @param text - 人类可读的文本内容
+ * @param structuredContent - 机器可读的结构化数据
+ * @param meta - 可选的元数据
+ * @returns 标准工具响应
+ */
+export function okStructured(text, structuredContent, meta) {
+    const response = {
+        content: [
+            {
+                type: 'text',
+                text,
+            },
+        ],
+        structuredContent,
+    };
+    if (meta) {
+        response._meta = meta;
+    }
+    return response;
+}

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

@@ -0,0 +1,116 @@
+/**
+ * Tasks 管理器
+ * 实现 MCP 2025-11-25 Tasks API
+ *
+ * 支持长时间运行的工具任务化执行：
+ * - 创建任务
+ * - 查询任务状态
+ * - 获取任务结果
+ * - 取消任务
+ * - 列出任务
+ */
+/**
+ * 任务状态
+ */
+export type TaskStatus = 'working' | 'completed' | 'failed' | 'cancelled';
+/**
+ * 任务元数据
+ */
+export interface TaskMetadata {
+    taskId: string;
+    status: TaskStatus;
+    statusMessage?: string;
+    createdAt: string;
+    lastUpdatedAt: string;
+    ttl?: number;
+    pollInterval?: number;
+}
+/**
+ * 任务结果
+ */
+export interface TaskResult {
+    content: Array<{
+        type: string;
+        text?: string;
+        [key: string]: any;
+    }>;
+    isError?: boolean;
+    structuredContent?: any;
+    _meta?: Record<string, any>;
+}
+/**
+ * Tasks 管理器类
+ */
+export declare class TasksManager {
+    private tasks;
+    private cleanupInterval?;
+    constructor();
+    /**
+     * 创建新任务
+     *
+     * @param ttl - 任务生存时间（毫秒），null 表示无限
+     * @param pollInterval - 建议的轮询间隔（毫秒）
+     * @returns 任务元数据
+     */
+    createTask(ttl?: number | null, pollInterval?: number): TaskMetadata;
+    /**
+     * 更新任务状态
+     *
+     * @param taskId - 任务 ID
+     * @param status - 新状态
+     * @param statusMessage - 状态消息
+     */
+    updateTaskStatus(taskId: string, status: TaskStatus, statusMessage?: string): void;
+    /**
+     * 设置任务结果
+     *
+     * @param taskId - 任务 ID
+     * @param result - 任务结果
+     */
+    setTaskResult(taskId: string, result: TaskResult): void;
+    /**
+     * 获取任务元数据
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务元数据
+     */
+    getTask(taskId: string): TaskMetadata;
+    /**
+     * 获取任务结果
+     *
+     * @param taskId - 任务 ID
+     * @returns 任务结果
+     */
+    getTaskResult(taskId: string): TaskResult;
+    /**
+     * 取消任务
+     *
+     * @param taskId - 任务 ID
+     */
+    cancelTask(taskId: string): void;
+    /**
+     * 列出所有任务
+     *
+     * @returns 任务元数据列表
+     */
+    listTasks(): TaskMetadata[];
+    /**
+     * 执行异步任务
+     *
+     * @param taskId - 任务 ID
+     * @param executor - 任务执行函数
+     */
+    executeTask(taskId: string, executor: () => Promise<TaskResult>): Promise<void>;
+    /**
+     * 清理过期任务
+     */
+    private cleanupExpiredTasks;
+    /**
+     * 销毁管理器
+     */
+    destroy(): void;
+}
+/**
+ * 获取全局 Tasks 管理器实例
+ */
+export declare function getTasksManager(): TasksManager;