feat: 插件自动注入编程AI规则文件(9种IDE) - rulesInjector.ts + 手动/自动注入命令

Author: zcj220

extension/package.json

@@ -142,6 +142,16 @@
         "command": "testpilot-ai.connectDevice.en",
         "title": "TestPilot AI: Connect Mobile Device",
         "icon": "$(device-mobile)"
+      },
+      {
+        "command": "testpilot-ai.injectRules",
+        "title": "TestPilot AI: 注入蓝本规则到项目",
+        "icon": "$(file-symlink-directory)"
+      },
+      {
+        "command": "testpilot-ai.injectRules.en",
+        "title": "TestPilot AI: Inject Blueprint Rules",
+        "icon": "$(file-symlink-directory)"
       }
     ],
     "menus": {
@@ -212,6 +222,12 @@
         },
         {
           "command": "testpilot-ai.connectDevice.en"
+        },
+        {
+          "command": "testpilot-ai.injectRules"
+        },
+        {
+          "command": "testpilot-ai.injectRules.en"
         }
       ]
     },

extension/src/extension.ts

@@ -11,6 +11,7 @@
 import * as vscode from "vscode";
 import { EngineClient } from "./engineClient";
 import { SidebarProvider } from "./sidebarProvider";
+import { autoInjectOnActivate, injectRules } from "./rulesInjector";
 
 let client: EngineClient;
 let outputChannel: vscode.OutputChannel;
@@ -506,6 +507,7 @@ export function activate(context: vscode.ExtensionContext): void {
     ["testpilot-ai.launchEngine.en", "testpilot-ai.launchEngine"],
     ["testpilot-ai.stopEngine.en", "testpilot-ai.stopEngine"],
     ["testpilot-ai.connectDevice.en", "testpilot-ai.connectDevice"],
+    ["testpilot-ai.injectRules.en", "testpilot-ai.injectRules"],
   ];
 
   for (const [alias, target] of commandAliases) {
@@ -514,6 +516,42 @@ export function activate(context: vscode.ExtensionContext): void {
     );
   }
 
+  // ── 手动注入规则命令 ──
+  context.subscriptions.push(
+    vscode.commands.registerCommand("testpilot-ai.injectRules", async () => {
+      const folders = vscode.workspace.workspaceFolders;
+      if (!folders || folders.length === 0) {
+        vscode.window.showWarningMessage("请先打开一个项目文件夹");
+        return;
+      }
+
+      let targetFolder: vscode.WorkspaceFolder;
+      if (folders.length === 1) {
+        targetFolder = folders[0];
+      } else {
+        const picked = await vscode.window.showWorkspaceFolderPick({
+          placeHolder: "选择要注入规则的项目",
+        });
+        if (!picked) { return; }
+        targetFolder = picked;
+      }
+
+      const result = injectRules(targetFolder.uri.fsPath, outputChannel);
+      if (result.created.length > 0) {
+        vscode.window.showInformationMessage(
+          `✅ 已注入 ${result.created.length} 个规则文件到 ${targetFolder.name}`,
+        );
+      } else {
+        vscode.window.showInformationMessage(
+          `${targetFolder.name} 已有所有规则文件，无需重复注入`,
+        );
+      }
+    }),
+  );
+
+  // ── 插件激活时自动注入规则到工作区 ──
+  autoInjectOnActivate(outputChannel);
+
   outputChannel.appendLine("[TestPilot AI] 所有命令已注册（中文+英文）");
   outputChannel.appendLine("[TestPilot AI] 💡 Tip: Search 'TestPilot AI' in Command Palette (Ctrl+Shift+P) for all commands");
 }

extension/src/rulesInjector.ts

@@ -0,0 +1,297 @@
+/**
+ * TestPilot AI — 编程AI规则文件注入器
+ *
+ * 插件激活时自动在用户项目中创建规则文件，
+ * 让所有主流IDE的编程AI都能读到蓝本生成规则。
+ *
+ * 覆盖9种IDE：
+ * - AGENTS.md（跨工具通用：Cursor/Cline/Augment/Kilo Code）
+ * - .github/copilot-instructions.md（VS Code Copilot）
+ * - .cursor/rules/testpilot.md（Cursor）
+ * - .windsurf/rules/testpilot.md（Windsurf）
+ * - .trae/rules/testpilot.md（Trae / 字节跳动）
+ * - .clinerules/testpilot.md（Cline）
+ * - .aiassistant/rules/testpilot.md（JetBrains AI）
+ * - .augment/rules/testpilot.md（Augment Code）
+ * - CLAUDE.md（Claude Code / Anthropic CLI）
+ */
+
+import * as vscode from "vscode";
+import * as fs from "fs";
+import * as path from "path";
+
+/** 需要注入的规则文件相对路径列表 */
+const RULES_FILES: string[] = [
+  "AGENTS.md",
+  "CLAUDE.md",
+  ".github/copilot-instructions.md",
+  ".cursor/rules/testpilot.md",
+  ".windsurf/rules/testpilot.md",
+  ".trae/rules/testpilot.md",
+  ".clinerules/testpilot.md",
+  ".aiassistant/rules/testpilot.md",
+  ".augment/rules/testpilot.md",
+];
+
+/** 规则模板内容（内嵌，避免依赖外部文件） */
+function getTemplateContent(): string {
+  return `# TestPilot AI — 编程AI蓝本自动生成规则
+
+> 本文件由 TestPilot AI 插件自动注入，指导编程AI在开发过程中自动生成和维护测试蓝本。
+> 你可以根据项目需求自由修改本文件内容。
+
+---
+
+## 一、蓝本自动生成（最重要）
+
+当项目中**不存在** \`testpilot.json\` 或 \`testpilot/\` 目录时，你在完成功能开发后**必须**自动生成蓝本文件。
+
+### 生成时机
+
+| 时机 | 动作 |
+|------|------|
+| 新项目创建完成 | 生成完整蓝本，覆盖所有已实现功能 |
+| 新增一个完整功能模块 | 在蓝本中追加对应场景 |
+| 项目初始蓝本不存在 | 扫描现有代码，生成功能全覆盖的蓝本 |
+
+### 生成规则
+
+1. 蓝本文件固定命名 \`testpilot.json\`，放在项目根目录或 \`testpilot/\` 子目录下
+2. 必须覆盖项目的**所有用户可交互功能**，不允许遗漏
+3. 每个功能模块对应一个 \`page\`，每个用户场景对应一个 \`scenario\`
+4. 如果项目功能较多，按模块拆分为多个蓝本文件放在 \`testpilot/\` 目录下：
+   - \`testpilot/auth.testpilot.json\` — 登录注册
+   - \`testpilot/product.testpilot.json\` — 商品管理
+   - \`testpilot/order.testpilot.json\` — 订单流程
+5. 每个蓝本必须包含 \`app_name\`、\`description\`、\`platform\`、\`base_url\` 字段
+
+---
+
+## 二、蓝本增量维护（日常开发）
+
+当项目已有蓝本文件时，以下6种代码变更**必须**同步更新蓝本：
+
+| # | 触发条件 | 蓝本更新动作 |
+|---|---------|------------|
+| 1 | 新增/删除 UI 元素 | 添加/删除对应场景和步骤 |
+| 2 | 修改元素 id/class/选择器 | 更新蓝本中所有用到该选择器的 \`target\` |
+| 3 | 修改文本内容（按钮文字/提示/错误信息） | 更新 \`assert_text\` 的 \`expected\` |
+| 4 | 修改业务逻辑（表单验证/跳转/计算） | 更新断言和预期结果 |
+| 5 | 修复 Bug | 更新蓝本中对应断言，确保能检测到该Bug |
+| 6 | 修改应用配置（URL/端口/路由/启动命令） | 更新 \`base_url\` / \`start_command\` |
+
+**不触发更新的情况**：纯CSS样式调整、代码注释修改、内部重构（不影响用户可见行为）、测试文件修改。
+
+---
+
+## 三、蓝本基本结构
+
+\`\`\`json
+{
+  "app_name": "应用名称",
+  "description": "应用功能的完整描述（50-200字）",
+  "base_url": "http://localhost:3000",
+  "platform": "web",
+  "start_command": "npm start",
+  "start_cwd": ".",
+  "pages": [
+    {
+      "url": "/",
+      "name": "首页",
+      "scenarios": [
+        {
+          "name": "场景名称",
+          "description": "测试目标",
+          "steps": [
+            {"action": "navigate", "value": "/", "description": "打开首页"},
+            {"action": "fill", "target": "#username", "value": "testuser", "description": "在用户名输入框输入"},
+            {"action": "click", "target": "#loginBtn", "description": "点击登录按钮"},
+            {"action": "assert_text", "expected": "欢迎", "description": "验证登录成功显示欢迎信息"},
+            {"action": "screenshot", "description": "登录成功后的页面"}
+          ]
+        }
+      ]
+    }
+  ]
+}
+\`\`\`
+
+### platform 取值
+
+| 值 | 适用场景 |
+|----|---------|
+| \`web\` | 网页应用（React/Vue/Angular/纯HTML） |
+| \`desktop\` | Windows桌面应用（需额外填 \`window_title\`） |
+| \`android\` | Android应用（需额外填 \`app_package\`、\`app_activity\`） |
+| \`ios\` | iOS应用（需额外填 \`bundle_id\`） |
+| \`miniprogram\` | 微信小程序（\`base_url\` 格式为 \`miniprogram://项目路径\`） |
+
+### 步骤动作
+
+| 动作 | 必填参数 | 说明 |
+|------|---------|------|
+| \`click\` | \`target\`, \`description\` | 点击元素 |
+| \`fill\` | \`target\`, \`value\`, \`description\` | 输入文本 |
+| \`screenshot\` | \`description\` | 截图 |
+| \`assert_text\` | \`expected\`, \`description\` | 断言页面包含文本 |
+| \`wait\` | \`description\` | 等待（可用 \`value\` 指定毫秒） |
+| \`navigate\` | \`value\`(URL), \`description\` | 页面跳转 |
+
+### target 写法
+
+- **Web/小程序**：CSS选择器，如 \`#loginBtn\`、\`.submit-btn\`、\`input[name="email"]\`
+- **桌面应用**：\`name:屏幕上可见的原文\`，如 \`name:Login\`、\`name:确定\`
+  - ⚠️ **禁止**在 \`name:\` 后加中文后缀（按钮/输入框/列表项等）
+- **Android/iOS**：\`accessibility_id:xxx\` 或 \`id:xxx\`
+
+### description 最佳实践
+
+每个步骤的 \`description\` 应包含：
+1. **位置**：上方/下方/左侧/右侧
+2. **预期变化**：点击后页面会发生什么（编程AI已读过源码，应预测页面变化）
+
+\`\`\`
+✅ "点击提交按钮，点击后表单数据提交到后端，页面显示'提交成功'提示"
+❌ "点击按钮"
+\`\`\`
+
+---
+
+## 四、边写代码边维护蓝本（推荐工作流）
+
+**最佳实践**：不要等项目全部写完才生成蓝本，而是**每实现一个功能就追加一个场景**。
+
+\`\`\`
+实现登录功能 → 立即在蓝本中添加"登录成功"和"登录失败"两个场景
+实现商品列表 → 立即添加"商品展示"和"搜索过滤"场景
+实现购物车   → 立即添加"加入购物车"和"修改数量"和"删除商品"场景
+\`\`\`
+
+这样做的好处：
+- 功能不会遗漏（写一个测一个）
+- 蓝本的选择器和预期值一定是准确的（刚写完代码，记忆最清晰）
+- 用户随时可以跑测试验证
+
+---
+
+## 五、蓝本自检清单
+
+生成或修改蓝本后，逐项检查：
+
+- [ ] 所有用户可交互功能都有对应场景
+- [ ] \`target\` 选择器与实际代码中的 id/class 一致
+- [ ] \`expected\` 是界面上实际会显示的文字（不是描述性文字）
+- [ ] 页面切换后有 \`wait\` 步骤
+- [ ] 场景间状态连贯（后一个场景基于前一个的结束状态）
+- [ ] \`platform\` 字段正确
+- [ ] \`base_url\` 和 \`start_command\` 填写正确
+`;
+}
+
+/**
+ * 在指定工作区目录中注入规则文件
+ * @param workspaceRoot 工作区根目录绝对路径
+ * @param outputChannel 日志输出通道
+ * @returns 注入结果 { created: string[], skipped: string[] }
+ */
+export function injectRules(
+  workspaceRoot: string,
+  outputChannel?: vscode.OutputChannel,
+): { created: string[]; skipped: string[] } {
+  const template = getTemplateContent();
+  const created: string[] = [];
+  const skipped: string[] = [];
+
+  for (const relPath of RULES_FILES) {
+    const fullPath = path.join(workspaceRoot, relPath);
+
+    // 已存在则跳过（不覆盖用户自定义内容）
+    if (fs.existsSync(fullPath)) {
+      skipped.push(relPath);
+      continue;
+    }
+
+    try {
+      // 创建目录
+      const dir = path.dirname(fullPath);
+      if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+      }
+
+      // 写入规则文件
+      fs.writeFileSync(fullPath, template, "utf-8");
+      created.push(relPath);
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      outputChannel?.appendLine(`[TestPilot AI] ⚠️ 创建 ${relPath} 失败: ${msg}`);
+    }
+  }
+
+  if (created.length > 0) {
+    outputChannel?.appendLine(
+      `[TestPilot AI] ✅ 已注入 ${created.length} 个规则文件: ${created.join(", ")}`,
+    );
+  }
+  if (skipped.length > 0) {
+    outputChannel?.appendLine(
+      `[TestPilot AI] ⏭️ 跳过 ${skipped.length} 个已存在文件: ${skipped.join(", ")}`,
+    );
+  }
+
+  return { created, skipped };
+}
+
+/**
+ * 检查工作区是否需要注入规则（没有任何规则文件时才需要）
+ */
+export function needsInjection(workspaceRoot: string): boolean {
+  return !RULES_FILES.some((relPath) =>
+    fs.existsSync(path.join(workspaceRoot, relPath)),
+  );
+}
+
+/**
+ * 插件激活时自动注入规则到所有工作区
+ * 只在首次（工作区没有任何规则文件时）自动注入，不会重复打扰用户
+ */
+export async function autoInjectOnActivate(
+  outputChannel?: vscode.OutputChannel,
+): Promise<void> {
+  const folders = vscode.workspace.workspaceFolders;
+  if (!folders) { return; }
+
+  for (const folder of folders) {
+    const root = folder.uri.fsPath;
+
+    // 跳过 TestPilotAI 项目本身（我们自己的项目已有规则）
+    if (fs.existsSync(path.join(root, "cli.py")) && fs.existsSync(path.join(root, "src", "app.py"))) {
+      outputChannel?.appendLine(`[TestPilot AI] 跳过 TestPilotAI 项目本身: ${root}`);
+      continue;
+    }
+
+    // 检查是否需要注入
+    if (needsInjection(root)) {
+      outputChannel?.appendLine(`[TestPilot AI] 检测到 ${folder.name} 没有蓝本规则，自动注入中...`);
+      const result = injectRules(root, outputChannel);
+
+      if (result.created.length > 0) {
+        vscode.window.showInformationMessage(
+          `TestPilot AI: 已为 ${folder.name} 注入 ${result.created.length} 个编程AI规则文件，编程AI将自动生成测试蓝本`,
+          "查看详情",
+        ).then((action) => {
+          if (action === "查看详情") {
+            // 打开 AGENTS.md 让用户看看注入了什么
+            const agentsPath = path.join(root, "AGENTS.md");
+            if (fs.existsSync(agentsPath)) {
+              vscode.workspace.openTextDocument(agentsPath).then((doc) => {
+                vscode.window.showTextDocument(doc, { preview: true });
+              });
+            }
+          }
+        });
+      }
+    } else {
+      outputChannel?.appendLine(`[TestPilot AI] ${folder.name} 已有规则文件，跳过注入`);
+    }
+  }
+}