mcp-probe-kit 3.0.16 → 3.0.17

package/build/tools/start_ui.js

@@ -14,7 +14,9 @@ import { renderOrchestrationHeader } from "../lib/orchestration-guidance.js";
 import { buildSkillBridgePlanStep, buildSkillHeaderNote, detectSkillBridge, renderSkillBridgeSection, } from "../lib/skill-bridge.js";
 import { UIReportSchema, RequirementsLoopSchema } from "../schemas/structured-output.js";
 import { detectProjectType } from "../lib/project-detector.js";
+import { resolveWorkspaceRoot, isLikelyProjectNamedRelativePath, buildProjectRootRetryHint } from "../lib/workspace-root.js";
 import { reportToolProgress, throwIfAborted, } from "../lib/tool-execution-context.js";
+import { buildMemoryPlanStep, loadMemoryInjectionContext, renderMemoryGuideSection, } from "../lib/memory-orchestration.js";
 function inferProductType(description) {
     const text = (description || '').toLowerCase();
     if (/电商|e-?commerce|shop|商城|购物/.test(text))
@@ -428,7 +430,49 @@ export async function startUi(args, context) {
     try {
         throwIfAborted(context?.signal, "start_ui 已取消");
         await reportToolProgress(context, 10, "start_ui: 解析参数与检测项目框架");
-        const projectRoot = process.cwd();
+        // 智能参数解析
+        const parsedArgs = parseArgs(args, {
+            defaultValues: {
+                description: "",
+                framework: "html",
+                template: "",
+                mode: "manual",
+                template_profile: "auto",
+                requirements_mode: "steady",
+                loop_max_rounds: 2,
+                loop_question_budget: 5,
+                loop_assumption_cap: 3,
+            },
+            primaryField: "description",
+            fieldAliases: {
+                description: ["desc", "ui", "page", "需求", "描述"],
+                framework: ["stack", "lib", "框架"],
+                template: ["name", "模板名"],
+                project_root: ["projectRoot", "project_path", "projectPath", "root", "project_root", "path", "dir", "directory", "项目路径", "项目根目录"],
+                mode: ["模式"],
+                template_profile: ["profile", "template_profile", "模板档位", "模板模式"],
+                requirements_mode: ["requirements_mode", "loop", "需求模式"],
+                loop_max_rounds: ["max_rounds", "rounds", "最大轮次"],
+                loop_question_budget: ["question_budget", "问题数量", "问题预算"],
+                loop_assumption_cap: ["assumption_cap", "假设上限"],
+            },
+        });
+        const explicitProjectRoot = getString(parsedArgs.project_root);
+        if (isLikelyProjectNamedRelativePath(explicitProjectRoot)) {
+            return {
+                content: [{
+                        type: "text",
+                        text: `拒绝执行 UI 编排：project_root 不能传带项目名的半相对路径，例如 ${explicitProjectRoot}。请改为传项目根目录绝对路径。`,
+                    }],
+                isError: true,
+                structuredContent: {
+                    error_code: "INVALID_PROJECT_ROOT",
+                    rejected_project_root: explicitProjectRoot,
+                    retry_hint: buildProjectRootRetryHint(explicitProjectRoot),
+                },
+            };
+        }
+        const projectRoot = resolveWorkspaceRoot(explicitProjectRoot);
         // 优先从 project-context.md 读取框架信息
         let detectedFramework = 'html'; // 默认值
         const contextFramework = getFrameworkFromContext(projectRoot);
@@ -461,32 +505,6 @@ export async function startUi(args, context) {
                 }
             }
         }
-        // 智能参数解析
-        const parsedArgs = parseArgs(args, {
-            defaultValues: {
-                description: "",
-                framework: detectedFramework, // 使用检测到的框架
-                template: "",
-                mode: "manual",
-                template_profile: "auto",
-                requirements_mode: "steady",
-                loop_max_rounds: 2,
-                loop_question_budget: 5,
-                loop_assumption_cap: 3,
-            },
-            primaryField: "description",
-            fieldAliases: {
-                description: ["desc", "ui", "page", "需求", "描述"],
-                framework: ["stack", "lib", "框架"],
-                template: ["name", "模板名"],
-                mode: ["模式"],
-                template_profile: ["profile", "template_profile", "模板档位", "模板模式"],
-                requirements_mode: ["requirements_mode", "loop", "需求模式"],
-                loop_max_rounds: ["max_rounds", "rounds", "最大轮次"],
-                loop_question_budget: ["question_budget", "问题数量", "问题预算"],
-                loop_assumption_cap: ["assumption_cap", "假设上限"],
-            },
-        });
         const description = getString(parsedArgs.description);
         const productType = inferProductType(description);
         const framework = getString(parsedArgs.framework) || detectedFramework;
@@ -524,6 +542,8 @@ export async function startUi(args, context) {
         const skillBridgeStep = buildSkillBridgePlanStep(skillBridge);
         const skillBridgeSection = renderSkillBridgeSection(skillBridge);
         headerNotes.push(buildSkillHeaderNote(skillBridge));
+        const memoryContext = await loadMemoryInjectionContext(description || templateName);
+        const memoryGuideSection = renderMemoryGuideSection(memoryContext);
         // 验证 mode 参数
         const validModes = ["auto", "manual"];
         if (mode && !validModes.includes(mode)) {
@@ -658,6 +678,7 @@ start_ui <描述> --requirements_mode=loop
                         action: 'update_project_context',
                         outputs: ['docs/project-context.md'],
                     },
+                    ...(memoryContext.enabled ? [buildMemoryPlanStep()] : []),
                 ],
             };
             const header = renderOrchestrationHeader({
@@ -667,7 +688,10 @@ start_ui <描述> --requirements_mode=loop
                     '按 Requirements Loop 规则提问并更新结构化输出',
                     '满足结束条件后按 delegated plan 执行 UI 计划',
                 ],
-                notes: headerNotes,
+                notes: [
+                    ...headerNotes,
+                    ...(memoryContext.enabled ? ['记忆系统: 已启用，先复用历史 UI 资产，再决定是否沉淀本次结果'] : []),
+                ],
             });
             const loopTemplate = profileDecision.resolved === 'strict'
                 ? LOOP_PROMPT_TEMPLATE_STRICT
@@ -675,7 +699,8 @@ start_ui <描述> --requirements_mode=loop
             const guide = header + skillBridgeSection + loopTemplate
                 .replace(/{description}/g, description)
                 .replace(/{question_budget}/g, String(questionBudget))
-                .replace(/{assumption_cap}/g, String(assumptionCap));
+                .replace(/{assumption_cap}/g, String(assumptionCap))
+                + memoryGuideSection;
             const loopReport = {
                 mode: 'loop',
                 round: 1,
@@ -869,6 +894,7 @@ ${recommendation.reasoning}
                         action: 'update_project_context',
                         outputs: ['docs/project-context.md'],
                     },
+                    ...(memoryContext.enabled ? [buildMemoryPlanStep()] : []),
                 ],
             };
             const header = renderOrchestrationHeader({
@@ -878,9 +904,12 @@ ${recommendation.reasoning}
                     '按 delegated plan 顺序调用工具',
                     '生成设计系统、模板并渲染 UI 代码',
                 ],
-                notes: headerNotes,
+                notes: [
+                    ...headerNotes,
+                    ...(memoryContext.enabled ? ['记忆系统: 已启用，必要时先读取相似历史 UI 资产'] : []),
+                ],
             });
-            const smartPlan = header + skillBridgeSection + (profileDecision.resolved === 'strict' ? smartPlanStrict : smartPlanGuided);
+            const smartPlan = header + skillBridgeSection + (profileDecision.resolved === 'strict' ? smartPlanStrict : smartPlanGuided) + memoryGuideSection;
             // Create structured UI report for auto mode
             const uiReport = {
                 summary: `智能 UI 开发：${description}`,
@@ -1006,7 +1035,10 @@ start_ui "设置页面" --framework=react
                 '按 delegated plan 顺序调用工具',
                 '生成设计系统、模板并渲染 UI 代码',
             ],
-            notes: headerNotes,
+            notes: [
+                ...headerNotes,
+                ...(memoryContext.enabled ? ['记忆系统: 已启用，必要时先读取相似历史 UI 资产'] : []),
+            ],
         });
         const baseTemplate = profileDecision.resolved === 'strict'
             ? PROMPT_TEMPLATE_STRICT
@@ -1016,6 +1048,7 @@ start_ui "设置页面" --framework=react
         guide = safeReplace(guide, '{productType}', productType);
         guide = safeReplace(guide, '{framework}', framework);
         guide = safeReplace(guide, '{templateName}', templateName);
+        guide += memoryGuideSection;
         const plan = {
             mode: 'delegated',
             steps: [
@@ -1072,6 +1105,7 @@ start_ui "设置页面" --framework=react
                     action: 'update_project_context',
                     outputs: ['docs/project-context.md'],
                 },
+                ...(memoryContext.enabled ? [buildMemoryPlanStep()] : []),
             ],
         };
         // Create structured UI report for manual mode