mcp-probe-kit 3.0.5 → 3.0.7

package/build/tools/start_ui.js

@@ -13,6 +13,7 @@ import { okStructured } from "../lib/response.js";
 import { renderOrchestrationHeader } from "../lib/orchestration-guidance.js";
 import { UIReportSchema, RequirementsLoopSchema } from "../schemas/structured-output.js";
 import { detectProjectType } from "../lib/project-detector.js";
+import { reportToolProgress, throwIfAborted, } from "../lib/tool-execution-context.js";
 function inferProductType(description) {
     const text = (description || '').toLowerCase();
     if (/电商|e-?commerce|shop|商城|购物/.test(text))
@@ -204,11 +205,11 @@ const PROMPT_TEMPLATE_GUIDED = `# 快速开始
 在 "## 📚 文档导航" 部分添加：
 
 \`\`\`markdown
-### [UI 设计系统](./design-system.md)
-项目的 UI 设计规范，包括颜色、字体、组件样式等
-
-### [UI 组件目录](./ui/component-catalog.json)
-可用的 UI 组件及其属性定义
+### [UI 设计系统](./design-system.md)
+项目的 UI 设计规范，包括颜色、字体、组件样式等
+
+### [UI 组件目录](./ui/component-catalog.json)
+可用的 UI 组件及其属性定义
 \`\`\`
 
 在 "## 💡 开发时查看对应文档" 部分的 "添加新功能" 下添加：
@@ -238,19 +239,19 @@ A: 不需要。如果文件已存在，直接跳过步骤 1。
 **Q: 如何使用自定义模板？**
 A: 在 \`docs/ui/\` 目录创建 JSON 模板文件，然后在步骤 4 中指定模板路径。
 `;
-const PROMPT_TEMPLATE_STRICT = `# UI 开发编排（严格）
-
-**职责说明**: 本工具仅提供执行指导，不执行实际操作。请按顺序调用以下 MCP 工具。
-
-## ✅ 执行计划
-
-1. 检查 \`docs/project-context.md\`，缺失则调用 \`init_project_context\`
-2. 检查 \`docs/design-system.md\`，缺失则调用 \`ui_design_system --product_type="{productType}" --stack="{framework}"\`
-3. 检查 \`docs/ui/component-catalog.json\`，缺失则调用 \`init_component_catalog\`
-4. \`ui_search --mode=template --query="{description}"\`
-5. 选择模板并保存到 \`docs/ui/{templateName}.json\`
-6. \`render_ui --template="docs/ui/{templateName}.json" --framework="{framework}"\`
-7. 将生成的 UI 文档添加到 \`docs/project-context.md\` 索引中
+const PROMPT_TEMPLATE_STRICT = `# UI 开发编排（严格）
+
+**职责说明**: 本工具仅提供执行指导，不执行实际操作。请按顺序调用以下 MCP 工具。
+
+## ✅ 执行计划
+
+1. 检查 \`docs/project-context.md\`，缺失则调用 \`init_project_context\`
+2. 检查 \`docs/design-system.md\`，缺失则调用 \`ui_design_system --product_type="{productType}" --stack="{framework}"\`
+3. 检查 \`docs/ui/component-catalog.json\`，缺失则调用 \`init_component_catalog\`
+4. \`ui_search --mode=template --query="{description}"\`
+5. 选择模板并保存到 \`docs/ui/{templateName}.json\`
+6. \`render_ui --template="docs/ui/{templateName}.json" --framework="{framework}"\`
+7. 将生成的 UI 文档添加到 \`docs/project-context.md\` 索引中
 
 ---
 
@@ -264,16 +265,16 @@ const PROMPT_TEMPLATE_STRICT = `# UI 开发编排（严格）
 
 ## 步骤 2: 生成设计系统（如不存在）
 
-**检查**: \`docs/design-system.md\`
-**缺失则调用**: \`ui_design_system\`
-\`\`\`json
-{
-  "product_type": "{productType}",
-  "stack": "{framework}",
-  "description": "{description}"
-}
-\`\`\`
-**预期输出**: \`docs/design-system.json\`、\`docs/design-system.md\`
+**检查**: \`docs/design-system.md\`
+**缺失则调用**: \`ui_design_system\`
+\`\`\`json
+{
+  "product_type": "{productType}",
+  "stack": "{framework}",
+  "description": "{description}"
+}
+\`\`\`
+**预期输出**: \`docs/design-system.json\`、\`docs/design-system.md\`
 
 ---
 
@@ -285,35 +286,35 @@ const PROMPT_TEMPLATE_STRICT = `# UI 开发编排（严格）
 
 ---
 
-## 步骤 4: 搜索模板
-
-**工具**: \`ui_search\`
-\`\`\`json
-{ "mode": "template", "query": "{description}" }
-\`\`\`
-
----
-
-## 步骤 5: 保存模板文件
-
-**操作**: 从搜索结果选择模板或创建最小模板
-
-**保存路径**: \`docs/ui/{templateName}.json\`
-
----
-
-## 步骤 6: 渲染代码
-
-**工具**: \`render_ui\`
-\`\`\`json
-{ "template": "docs/ui/{templateName}.json", "framework": "{framework}" }
-\`\`\`
-
----
-
-## 步骤 7: 更新项目上下文索引
-
-将 UI 文档链接添加到 \`docs/project-context.md\`
+## 步骤 4: 搜索模板
+
+**工具**: \`ui_search\`
+\`\`\`json
+{ "mode": "template", "query": "{description}" }
+\`\`\`
+
+---
+
+## 步骤 5: 保存模板文件
+
+**操作**: 从搜索结果选择模板或创建最小模板
+
+**保存路径**: \`docs/ui/{templateName}.json\`
+
+---
+
+## 步骤 6: 渲染代码
+
+**工具**: \`render_ui\`
+\`\`\`json
+{ "template": "docs/ui/{templateName}.json", "framework": "{framework}" }
+\`\`\`
+
+---
+
+## 步骤 7: 更新项目上下文索引
+
+将 UI 文档链接添加到 \`docs/project-context.md\`
 `;
 const LOOP_PROMPT_TEMPLATE_GUIDED = `# 🧭 UI 需求澄清与补全（Requirements Loop）
 
@@ -344,8 +345,8 @@ UI 需求：{description}
 ---
 
 ## ✅ 结束后继续
-当满足结束条件时，按 delegated plan 执行：
-- 设计系统 → 组件目录 → 模板搜索 → 保存模板 → 渲染代码 → 更新上下文
+当满足结束条件时，按 delegated plan 执行：
+- 设计系统 → 组件目录 → 模板搜索 → 保存模板 → 渲染代码 → 更新上下文
 
 ---
 
@@ -374,7 +375,7 @@ UI 需求：{description}
 ---
 
 ## ✅ 结束后继续
-当满足结束条件时，按 delegated plan 执行 UI 计划
+当满足结束条件时，按 delegated plan 执行 UI 计划
 
 ---
 
@@ -422,8 +423,10 @@ function getFrameworkFromContext(projectRoot) {
 /**
  * 统一 UI 开发编排工具
  */
-export async function startUi(args) {
+export async function startUi(args, context) {
     try {
+        throwIfAborted(context?.signal, "start_ui 已取消");
+        await reportToolProgress(context, 10, "start_ui: 解析参数与检测项目框架");
         const projectRoot = process.cwd();
         // 优先从 project-context.md 读取框架信息
         let detectedFramework = 'html'; // 默认值
@@ -494,6 +497,8 @@ export async function startUi(args) {
         const assumptionCap = getNumber(parsedArgs.loop_assumption_cap, 3);
         let templateName = getString(parsedArgs.template);
         templateName = normalizeTemplateName(templateName || description || 'ui-template', 'ui-template');
+        throwIfAborted(context?.signal, "start_ui 已取消");
+        await reportToolProgress(context, 35, "start_ui: 参数解析完成");
         const profileDecision = resolveTemplateProfile(rawProfile, description || "");
         const templateMeta = {
             profile: profileDecision.resolved,
@@ -538,6 +543,8 @@ start_ui "用户列表" --mode=auto
         }
         // requirements loop 模式
         if (requirementsMode === "loop") {
+            throwIfAborted(context?.signal, "start_ui(loop) 已取消");
+            await reportToolProgress(context, 70, "start_ui: 生成 loop 计划");
             if (!description) {
                 return {
                     content: [
@@ -691,6 +698,7 @@ start_ui <描述> --requirements_mode=loop
                     template: templateMeta,
                 },
             };
+            await reportToolProgress(context, 95, "start_ui: loop 输出已生成");
             return okStructured(guide, loopReport, {
                 schema: RequirementsLoopSchema,
                 note: 'AI 应按轮次澄清 UI 需求并更新结构化输出，满足结束条件后再执行 UI 计划',
@@ -698,6 +706,8 @@ start_ui <描述> --requirements_mode=loop
         }
         // 自动模式实现
         if (mode === "auto") {
+            throwIfAborted(context?.signal, "start_ui(auto) 已取消");
+            await reportToolProgress(context, 55, "start_ui: 生成智能推荐");
             // 1. 获取推理引擎
             const engine = await getReasoningEngine();
             // 2. 构造设计请求
@@ -708,13 +718,15 @@ start_ui <描述> --requirements_mode=loop
             };
             // 3. 生成推荐
             const recommendation = engine.generateRecommendation(request);
+            throwIfAborted(context?.signal, "start_ui(auto) 已取消");
+            await reportToolProgress(context, 80, "start_ui: 智能计划已生成");
             // 4. 提取推理结果
             const inferredProductType = recommendation.target;
             const inferredKeywords = recommendation.style.keywords.join(", ");
             const inferredStack = framework; // 保持用户指定的技术栈，或默认为 react
             // 5. 生成智能执行计划
             const searchQuery = description || templateName;
-            const smartPlanGuided = `# 🚀 智能 UI 开发计划
+            const smartPlanGuided = `# 🚀 智能 UI 开发计划
 
 基于您的描述 "**${description}**"，AI 引擎已为您规划了最佳开发路径。
 
@@ -746,25 +758,25 @@ ui_design_system --product_type="${inferredProductType}" --stack="${inferredStac
 init_component_catalog
 \`\`\`
 
-### 4. 生成 UI 模板 📄
-\`\`\`bash
-# 搜索现有模板或生成新模板
-ui_search --mode=template --query="${searchQuery}"
-\`\`\`
-
-### 5. 保存模板文件 🧩
-\`\`\`bash
-# 将选中的模板保存到本地
-# 目标路径：docs/ui/${templateName}.json
-\`\`\`
-
-### 6. 渲染代码 💻
-\`\`\`bash
-render_ui docs/ui/${templateName}.json --framework="${inferredStack}"
-\`\`\`
-
-### 7. 更新项目上下文 📝
-将生成的 UI 文档链接添加到 \`docs/project-context.md\` 的文档导航部分。
+### 4. 生成 UI 模板 📄
+\`\`\`bash
+# 搜索现有模板或生成新模板
+ui_search --mode=template --query="${searchQuery}"
+\`\`\`
+
+### 5. 保存模板文件 🧩
+\`\`\`bash
+# 将选中的模板保存到本地
+# 目标路径：docs/ui/${templateName}.json
+\`\`\`
+
+### 6. 渲染代码 💻
+\`\`\`bash
+render_ui docs/ui/${templateName}.json --framework="${inferredStack}"
+\`\`\`
+
+### 7. 更新项目上下文 📝
+将生成的 UI 文档链接添加到 \`docs/project-context.md\` 的文档导航部分。
 
 ---
 
@@ -785,13 +797,13 @@ ${recommendation.reasoning}
 
 ## 📋 执行步骤
 
-1) init_project_context
-2) ui_design_system --product_type="${inferredProductType}" --stack="${inferredStack}" --keywords="${inferredKeywords}" --description="${description}"
-3) init_component_catalog
-4) ui_search --mode=template --query="${searchQuery}"
-5) 保存模板到 docs/ui/${templateName}.json
-6) render_ui docs/ui/${templateName}.json --framework="${inferredStack}"
-7) 更新 project-context.md 索引
+1) init_project_context
+2) ui_design_system --product_type="${inferredProductType}" --stack="${inferredStack}" --keywords="${inferredKeywords}" --description="${description}"
+3) init_component_catalog
+4) ui_search --mode=template --query="${searchQuery}"
+5) 保存模板到 docs/ui/${templateName}.json
+6) render_ui docs/ui/${templateName}.json --framework="${inferredStack}"
+7) 更新 project-context.md 索引
 `;
             const plan = {
                 mode: 'delegated',
@@ -931,6 +943,7 @@ ${recommendation.reasoning}
                     template: templateMeta,
                 },
             };
+            await reportToolProgress(context, 95, "start_ui: auto 输出已生成");
             return okStructured(smartPlan, uiReport, {
                 schema: UIReportSchema,
                 note: 'AI 应该按照智能计划执行步骤，并在每个步骤完成后更新 structuredContent',
@@ -1121,6 +1134,7 @@ start_ui "设置页面" --framework=react
                 template: templateMeta,
             },
         };
+        await reportToolProgress(context, 95, "start_ui: manual 输出已生成");
         return okStructured(guide, uiReport, {
             schema: UIReportSchema,
             note: 'AI 应该按照指南执行步骤，并在每个步骤完成后更新 structuredContent',

package/build/tools/ui-ux-tools.d.ts

@@ -5,6 +5,7 @@
  * 使用模板类实现文档生成的原子化
  */
 import { DesignReasoningEngine, DesignSystemRecommendation } from '../utils/design-reasoning-engine.js';
+import { type ToolExecutionContext } from '../lib/tool-execution-context.js';
 /**
  * 文件索引接口
  * 定义需要创建的文件及其元数据
@@ -77,4 +78,4 @@ export declare function uiSearch(args: any): Promise<import("../lib/response.js"
 /**
  * UI 数据同步工具
  */
-export declare function syncUiData(args: any): Promise<import("../lib/response.js").ToolResponse>;
+export declare function syncUiData(args: any, context?: ToolExecutionContext): Promise<import("../lib/response.js").ToolResponse>;

package/build/tools/ui-ux-tools.js

@@ -10,6 +10,7 @@ import { ASCIIBoxFormatter } from '../utils/ascii-box-formatter.js';
 import { syncUIDataToCache } from '../utils/ui-sync.js';
 import { formatDesignSystemJson } from '../utils/design-system-json-formatter.js';
 import { okStructured } from '../lib/response.js';
+import { reportToolProgress, throwIfAborted, } from '../lib/tool-execution-context.js';
 // 全局数据加载器实例
 let dataLoader = null;
 let reasoningEngine = null;
@@ -734,17 +735,22 @@ ${formattedResults}
 /**
  * UI 数据同步工具
  */
-export async function syncUiData(args) {
+export async function syncUiData(args, context) {
     try {
+        throwIfAborted(context?.signal, 'sync_ui_data 已取消');
+        await reportToolProgress(context, 5, 'sync_ui_data: 开始同步流程');
         const force = args.force || false;
         const verbose = args.verbose || false;
         // 检查是否需要更新
         if (!force) {
+            throwIfAborted(context?.signal, 'sync_ui_data 已取消');
+            await reportToolProgress(context, 15, 'sync_ui_data: 检查上游版本');
             const loader = await getDataLoader();
             const cacheManager = loader.getCacheManager();
             try {
                 const updateInfo = await cacheManager.checkUpdate();
                 if (!updateInfo.hasUpdate) {
+                    await reportToolProgress(context, 100, 'sync_ui_data: 数据已是最新');
                     const upToDateData = {
                         summary: "UI/UX 数据已是最新版本",
                         status: 'success',
@@ -769,8 +775,17 @@ export async function syncUiData(args) {
             }
         }
         // 执行同步
-        await syncUIDataToCache(force, verbose);
+        throwIfAborted(context?.signal, 'sync_ui_data 已取消');
+        await reportToolProgress(context, 30, 'sync_ui_data: 下载并处理数据');
+        await syncUIDataToCache(force, verbose, {
+            signal: context?.signal,
+            onProgress: async (progress, message) => {
+                await reportToolProgress(context, 30 + Math.round(progress * 0.6), `sync_ui_data: ${message}`);
+            },
+        });
         // 重新加载数据
+        throwIfAborted(context?.signal, 'sync_ui_data 已取消');
+        await reportToolProgress(context, 92, 'sync_ui_data: 重载本地缓存');
         if (dataLoader) {
             await dataLoader.reload();
         }
@@ -789,7 +804,8 @@ export async function syncUiData(args) {
             },
             timestamp: new Date().toISOString(),
         };
-        return okStructured(`✅ UI/UX 数据同步成功
+        await reportToolProgress(context, 100, 'sync_ui_data: 同步完成');
+        return okStructured(`✅ UI/UX 数据同步成功
 
 数据已更新到缓存目录: ${cacheDir}
 

package/build/utils/ui-sync.d.ts

@@ -3,11 +3,15 @@
  *
  * 从 npm 包 uipro-cli 同步数据到缓存目录
  */
+export interface SyncRuntimeOptions {
+    signal?: AbortSignal;
+    onProgress?: (progress: number, message: string) => Promise<void> | void;
+}
 /**
  * 同步 UI/UX 数据到指定目录（通用函数）
  */
-export declare function syncUIDataTo(outputDir: string, verbose?: boolean): Promise<void>;
+export declare function syncUIDataTo(outputDir: string, verbose?: boolean, options?: SyncRuntimeOptions): Promise<void>;
 /**
  * 同步 UI/UX 数据到缓存
  */
-export declare function syncUIDataToCache(force?: boolean, verbose?: boolean): Promise<void>;
+export declare function syncUIDataToCache(force?: boolean, verbose?: boolean, options?: SyncRuntimeOptions): Promise<void>;