ai-worktool 1.0.0

package/dist/view.js

@@ -0,0 +1,225 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.COVERAGE_FILE_PATH = void 0;
+exports.showCoverageView = showCoverageView;
+exports.stopMonitoring = stopMonitoring;
+exports.loadAndDisplayCoverage = loadAndDisplayCoverage;
+exports.disposeTimer = disposeTimer;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const table_1 = require("table");
+const cli_color_1 = __importDefault(require("cli-color"));
+let REFRESH_INTERVAL = 3000;
+// 存储上一次的覆盖率数据
+let previousCoverage = null;
+// 存储变化的单元格位置
+let changedCells = [];
+// 文件监听对象
+let fileWatcher = null;
+// 定时器ID
+let refreshTimer = null;
+const output = console.log;
+exports.COVERAGE_FILE_PATH = 'coverage-summary.json';
+// 主函数
+async function showCoverageView(rootDir, coverageFilePath = exports.COVERAGE_FILE_PATH) {
+    const coveragePath = path_1.default.join(rootDir, coverageFilePath);
+    console.log(cli_color_1.default.cyan('开始监控覆盖率数据...' + coveragePath));
+    // 初始加载
+    await loadAndDisplayCoverage(coveragePath);
+    // 设置文件监听
+    fileWatcher = fs_1.default.watch(coveragePath, async (eventType, filename) => {
+        if (filename && eventType === 'change') {
+            await loadAndDisplayCoverage(coveragePath);
+        }
+    });
+    // 定时刷新，确保捕获所有变化
+    // refreshTimer = setInterval(() => loadAndDisplayCoverage(coveragePath), refershTime) as any;
+}
+// 停止监控函数
+function stopMonitoring() {
+    // 清理文件监听器
+    if (fileWatcher) {
+        fileWatcher.close();
+        fileWatcher = null;
+    }
+    // 清理定时器
+    if (refreshTimer) {
+        clearInterval(refreshTimer);
+        refreshTimer = null;
+    }
+}
+let timer = null;
+// 加载并显示覆盖率数据
+async function loadAndDisplayCoverage(coveragePath, refreshTimeout = REFRESH_INTERVAL) {
+    try {
+        const coverage = await readCoverageFile(coveragePath);
+        // 生成表格数据
+        const tableData = generateTableData(coverage);
+        // 比较并标记变化
+        if (previousCoverage && refreshTimeout > 0) {
+            changedCells = compareCoverage(previousCoverage, coverage);
+            // 如果有变化，3秒后清除标记
+            if (changedCells.length > 0) {
+                timer = setTimeout(() => {
+                    timer = null;
+                    changedCells = [];
+                    loadAndDisplayCoverage(coveragePath);
+                }, refreshTimeout);
+            }
+        }
+        // 存储当前覆盖率数据供下次比较
+        previousCoverage = coverage;
+        // 渲染表格
+        renderTable(tableData);
+    }
+    catch (error) {
+        console.error(cli_color_1.default.red(`加载覆盖率数据时出错: ${error.message}`));
+    }
+}
+// 读取覆盖率文件
+async function readCoverageFile(coveragePath) {
+    return new Promise((resolve, reject) => {
+        fs_1.default.readFile(coveragePath, 'utf8', (err, data) => {
+            if (err) {
+                reject(new Error(`无法读取覆盖率文件: ${err.message}`));
+                return;
+            }
+            try {
+                const coverage = JSON.parse(data);
+                resolve(coverage);
+            }
+            catch (parseError) {
+                reject(new Error(`解析覆盖率文件失败: ${parseError.message}`));
+            }
+        });
+    });
+}
+// 生成表格数据
+function generateTableData(coverage) {
+    const tableData = [['文件路径', '行覆盖率(%)', '语句覆盖率(%)', '函数覆盖率(%)', '分支覆盖率(%)']];
+    // 提取并排序文件路径
+    const filePaths = Object.keys(coverage).filter((key) => key !== 'total');
+    filePaths.sort();
+    // 添加总计行
+    tableData.push([
+        cli_color_1.default.bold('总计'),
+        formatPercentage(coverage.total.lines.pct),
+        formatPercentage(coverage.total.statements.pct),
+        formatPercentage(coverage.total.functions.pct),
+        formatPercentage(coverage.total.branches.pct)
+    ]);
+    // 添加分隔线
+    tableData.push(['---', '---', '---', '---', '---']);
+    // 添加每个文件的覆盖率数据
+    filePaths.forEach((filePath) => {
+        const fileCoverage = coverage[filePath];
+        tableData.push([
+            path_1.default.basename(filePath),
+            formatPercentage(fileCoverage.lines.pct),
+            formatPercentage(fileCoverage.statements.pct),
+            formatPercentage(fileCoverage.functions.pct),
+            formatPercentage(fileCoverage.branches.pct)
+        ]);
+    });
+    return tableData;
+}
+// 格式化百分比并应用颜色
+function formatPercentage(pct) {
+    if (pct === 'Unknown')
+        return cli_color_1.default.italic('Unknown');
+    let formatted = pct.toFixed(2) + '%';
+    // 根据覆盖率设置颜色
+    if (pct < 50) {
+        formatted = cli_color_1.default.red(formatted);
+    }
+    else if (pct < 80) {
+        formatted = cli_color_1.default.yellow(formatted);
+    }
+    else {
+        formatted = cli_color_1.default.green(formatted);
+    }
+    return formatted;
+}
+// 比较覆盖率数据，找出变化的单元格
+function compareCoverage(oldData, newData) {
+    const changes = [];
+    // 比较总计
+    const compareMetrics = (oldMetrics, newMetrics, rowIndex) => {
+        if (oldMetrics.lines.pct !== newMetrics.lines.pct) {
+            changes.push([rowIndex, 1]);
+        }
+        if (oldMetrics.statements.pct !== newMetrics.statements.pct) {
+            changes.push([rowIndex, 2]);
+        }
+        if (oldMetrics.functions.pct !== newMetrics.functions.pct) {
+            changes.push([rowIndex, 3]);
+        }
+        if (oldMetrics.branches.pct !== newMetrics.branches.pct) {
+            changes.push([rowIndex, 4]);
+        }
+    };
+    // 比较总计行
+    compareMetrics(oldData.total, newData.total, 1);
+    // 比较文件行
+    const filePaths = Object.keys(oldData).filter((key) => key !== 'total');
+    filePaths.forEach((filePath, index) => {
+        if (newData[filePath]) {
+            compareMetrics(oldData[filePath], newData[filePath], index + 3);
+        }
+    });
+    return changes;
+}
+// 渲染表格
+function renderTable(tableData) {
+    // 应用变化标记
+    const markedData = tableData.map((row, rowIndex) => {
+        return row.map((cell, colIndex) => {
+            // 检查是否是变化的单元格
+            if (changedCells.some(([r, c]) => r === rowIndex && c === colIndex)) {
+                return cli_color_1.default.bgRed.white(cell);
+            }
+            return cell;
+        });
+    });
+    // 配置表格样式
+    const tableConfig = {
+        columns: {
+            0: { width: 30 },
+            1: { width: 15, alignment: 'right' },
+            2: { width: 15, alignment: 'right' },
+            3: { width: 15, alignment: 'right' },
+            4: { width: 15, alignment: 'right' }
+        },
+        border: {
+            topBody: `─`,
+            topJoin: `┬`,
+            topLeft: `┌`,
+            topRight: `┐`,
+            bottomBody: `─`,
+            bottomJoin: `┴`,
+            bottomLeft: `└`,
+            bottomRight: `┘`,
+            bodyLeft: `│`,
+            bodyRight: `│`,
+            bodyJoin: `│`,
+            joinBody: `─`,
+            joinLeft: `├`,
+            joinRight: `┤`,
+            joinJoin: `┼`
+        }
+    };
+    // 清除控制台并打印表格
+    console.clear();
+    console.log(cli_color_1.default.cyan('单测覆盖率监控工具 - 最后更新: ' + new Date().toLocaleString()));
+    output((0, table_1.table)(markedData, tableConfig));
+}
+function disposeTimer() {
+    if (timer) {
+        clearTimeout(timer);
+        timer = null;
+    }
+}
+//# sourceMappingURL=view.js.map

package/package.json

@@ -0,0 +1,302 @@
+{
+  "name": "ai-worktool",
+  "displayName": "吃豆豆:单测智能体",
+  "version": "1.0.0",
+  "description": "单元测试智能体,帮你开发单元测试用例代码直到100%单测覆盖通过。",
+  "author": "jianguoke.cn",
+  "license": "MIT",
+  "repository": "https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools",
+  "bin": {
+    "testAgent": "dist/cli.js"
+  },
+  "scripts": {
+    "watch": "tsc -w",
+    "build": "tsc",
+    "test": "jest test/tools",
+    "coverage": "jest --coverage test/tools",
+    "tools": "ts-node ./tools.ts",
+    "plugin": "yarn tools && dify plugin package plugins/dify-plugin",
+    "package": "cross-env PKG_CACHE_PATH=./binaries pkg . --targets node20-win-x64 --out-path aicoder ",
+    "changelog": "conventional-changelog -p angular -i CHANGELOG.md -s -r 0",
+    "ver": "node ./updatever.js",
+    "remove-deps": "node ./remove-deps.js",
+    "restore-deps": "node ./restore-deps.js",
+    "pub": "yarn remove-deps && yarn changelog && vsce publish --yarn --baseContentUrl https://aicoder.jianguoke.cn/assets && yarn restore-deps",
+    "pubovsx": "yarn remove-deps && ovsx publish -p 47621ff6-be56-4814-865e-d2a8e8a76f86 --yarn --baseContentUrl https://aicoder.jianguoke.cn/assets && yarn restore-deps",
+    "puball": "yarn ver && yarn pub && yarn pubovsx && yarn puboss && npm publish --registry=https://registry.npmjs.org",
+    "patch": "yarn remove-deps && vsce publish patch --yarn && yarn restore-deps",
+    "vspack": "yarn remove-deps && vsce package -o aicoder-1.0.19.vsix --yarn --baseContentUrl https://aicoder.jianguoke.cn/assets && yarn restore-deps",
+    "uposs": "node ./uposs.js",
+    "puboss": "yarn vspack && yarn uposs",
+    "prepare": "husky"
+  },
+  "publisher": "JianGuoKe",
+  "icon": "resources/logo_white.png",
+  "engines": {
+    "vscode": "^1.54.0"
+  },
+  "pkg": {
+    "targets": [
+      "node18-win-x64",
+      "node18-macos-x64",
+      "node18-linux-x64"
+    ],
+    "outputPath": "dist"
+  },
+  "categories": [
+    "Other"
+  ],
+  "activationEvents": [
+    "onView:ai-worktool-apps"
+  ],
+  "main": "./dist/extension.js",
+  "contributes": {
+    "commands": [
+      {
+        "command": "ai-worktool.startTestAgent",
+        "category": "单测智能体",
+        "title": "启用/停用单测智能体"
+      },
+      {
+        "command": "ai-worktool.loginCommand",
+        "category": "单测智能体",
+        "title": "登录"
+      },
+      {
+        "command": "ai-worktool.logoutCommand",
+        "category": "单测智能体",
+        "title": "退出"
+      },
+      {
+        "command": "ai-worktool.cancelLoginCommand",
+        "category": "单测智能体",
+        "title": "取消登录"
+      },
+      {
+        "command": "ai-worktool.orderCommand",
+        "category": "单测智能体",
+        "title": "购买"
+      },
+      {
+        "command": "ai-worktool.usageCommand",
+        "category": "单测智能体",
+        "title": "用量"
+      },
+      {
+        "command": "ai-worktool.homePageCommand",
+        "category": "单测智能体",
+        "title": "首页"
+      }
+    ],
+    "submenus": [
+      {
+        "id": "ai-worktool.submenu",
+        "label": "更多操作",
+        "icon": "$(ellipsis)"
+      }
+    ],
+    "menus": {
+      "view/title": [
+        {
+          "when": "view == ai-worktool-webview && ai-worktool.isLoggedIn == true",
+          "submenu": "ai-worktool.submenu",
+          "group": "navigation"
+        }
+      ],
+      "ai-worktool.submenu": [
+        {
+          "command": "ai-worktool.homePageCommand",
+          "when": "ai-worktool.isLoggedIn == true",
+          "group": "navigation@1"
+        },
+        {
+          "command": "ai-worktool.loginCommand",
+          "when": "ai-worktool.isLoggedIn == false",
+          "group": "navigation@2"
+        },
+        {
+          "command": "ai-worktool.usageCommand",
+          "when": "ai-worktool.isLoggedIn == true",
+          "group": "navigation@5"
+        },
+        {
+          "command": "ai-worktool.logoutCommand",
+          "when": "ai-worktool.isLoggedIn == true",
+          "group": "navigation@10"
+        }
+      ]
+    },
+    "viewsContainers": {
+      "activitybar": [
+        {
+          "id": "ai-worktool",
+          "title": "吃豆豆:单测智能体",
+          "icon": "resources/logo.svg"
+        }
+      ]
+    },
+    "views": {
+      "ai-worktool": [
+        {
+          "type": "webview",
+          "id": "ai-worktool-webview",
+          "icon": "resources/logo.svg",
+          "name": "吃豆豆:单测智能体"
+        }
+      ]
+    },
+    "configuration": {
+      "type": "object",
+      "title": "吃豆豆:单测智能体",
+      "properties": {
+        "ai-worktool.platform": {
+          "type": "string",
+          "default": "jianguoke",
+          "enum": [
+            "jianguoke",
+            "chanjet"
+          ],
+          "description": "智能体平台选择"
+        },
+        "ai-worktool.packageManager": {
+          "type": "string",
+          "default": "yarn",
+          "description": "包管理器选择",
+          "enum": [
+            "npm",
+            "yarn",
+            "pnpm",
+            "bun"
+          ],
+          "scope": "resource"
+        },
+        "ai-worktool.testFramework": {
+          "type": "string",
+          "default": "jest",
+          "description": "测试框架选择",
+          "enum": [
+            "jest",
+            "vitest",
+            "mocha",
+            "jasmine",
+            "junit",
+            "pytest",
+            "cypress",
+            "cucumber",
+            "protractor",
+            "testcafe"
+          ],
+          "scope": "resource"
+        },
+        "ai-worktool.language": {
+          "type": "string",
+          "default": "typescript",
+          "description": "开发语言选择",
+          "enum": [
+            "typescript",
+            "javascript",
+            "python",
+            "java",
+            "c++",
+            "ruby",
+            "php",
+            "go",
+            "rust",
+            "swift",
+            "kotlin",
+            "scala",
+            "objective-c",
+            "c#",
+            "fortran",
+            "lisp",
+            "commonlisp",
+            "scheme",
+            "cobol",
+            "apl",
+            "c"
+          ],
+          "scope": "resource"
+        },
+        "ai-worktool.coverageDir": {
+          "type": "string",
+          "default": "coverage",
+          "description": "单测报告文件夹",
+          "scope": "resource"
+        },
+        "ai-worktool.testDir": {
+          "type": "string",
+          "default": "test",
+          "description": "单测文件夹",
+          "scope": "resource"
+        },
+        "ai-worktool.srcDir": {
+          "type": "string",
+          "default": "src",
+          "description": "源代码文件夹",
+          "scope": "resource"
+        },
+        "ai-worktool.autoCommit": {
+          "type": "boolean",
+          "default": true,
+          "description": "单测通过修改文件自动提交代码到远程仓库"
+        },
+        "ai-worktool.withConversation": {
+          "type": "boolean",
+          "default": false,
+          "description": "每次执行单测时是否继续与智能体进行会话对话"
+        },
+        "ai-worktool.fullCoverageEnd": {
+          "type": "boolean",
+          "default": true,
+          "description": "单测覆盖100%时结束执行"
+        }
+      }
+    }
+  },
+  "dependencies": {
+    "fetch-sse": "^1.1.2",
+    "jsonwebtoken": "^9.0.2",
+    "simple-git": "^3.28.0",
+    "acorn": "^8.15.0",
+    "acorn-walk": "^8.3.4",
+    "typescript": "^5.9.2",
+    "cli-color": "^2.0.4",
+    "commander": "^14.0.0",
+    "table": "^6.9.0"
+  },
+  "devDependencies": {
+    "@types/adm-zip": "^0.5.7",
+    "@types/cli-color": "^2.0.6",
+    "@types/jest": "^30.0.0",
+    "@types/jsonwebtoken": "^9.0.10",
+    "@types/node": "^24.1.0",
+    "@types/vscode": "^1.54.0",
+    "@typescript-eslint/eslint-plugin": "^4.31.1",
+    "@typescript-eslint/parser": "^4.31.1",
+    "@vscode/test-electron": "^1.6.2",
+    "@yao-pkg/pkg": "^6.6.0",
+    "@yao-pkg/pkg-fetch": "^3.5.24",
+    "adm-zip": "^0.5.16",
+    "ali-oss": "^6.23.0",
+    "conventional-changelog-cli": "^2.2.2",
+    "cross-env": "^10.0.0",
+    "dotenv": "^17.2.1",
+    "eslint": "^7.32.0",
+    "husky": "^9.1.7",
+    "jest": "^29.7.0",
+    "lint-staged": "^16.1.2",
+    "nexe": "^5.0.0-beta.4",
+    "ovsx": "^0.10.5",
+    "ts-jest": "^29.4.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.9.2",
+    "vsce": "^2.6.4"
+  },
+  "jest": {
+    "setupFiles": [
+      "dotenv/config"
+    ],
+    "preset": "ts-jest",
+    "testEnvironment": "node"
+  }
+}