ai-worktool 1.0.12 → 1.0.14

package/CHANGELOG.md

@@ -1,4 +1,4 @@
-## 1.0.12 (2025-08-03)
+## 1.0.14 (2025-08-05)
 
 
 ### Bug Fixes
@@ -142,6 +142,7 @@
 * **jianguoke:** 优化工具调用信息格式化及消息 ID 生成 ([1b240e7](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/1b240e7fbcabb2112789c116b7f483308c184fce))
 * **log:** 增加日志容器拖动调整功能 ([cfdaf9d](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/cfdaf9d7bec56687a2d49ca92785463dc6713a77))
 * **plugin:** 更新插件版本和功能描述 ([f3773d6](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/f3773d675fcb274080339cd0274a1a7aff8415a9))
+* **program:** 增加代码覆盖率报告展示功能 ([f693e27](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/f693e272ef0fe7e58083d2589339af17e45cff95))
 * **project:** 优化项目类型检测逻辑 ([9bc024d](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/9bc024deeed239c05c9cab810f9f12a6a3306aab))
 * Refactor CLI and program structure, update commands for testing and fixing code ([8dcbb19](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/8dcbb191fc8a95a40e07647de337452935926b10))
 * **test/demo:** 添加用户成绩 API 路由 ([6f838c7](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/6f838c7fda9924b1e5a7556d0a429427a652b985))
@@ -152,6 +153,8 @@
 * **toolbox:** 重构并增强项目配置检测功能 ([6cc4e33](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/6cc4e33a58c7ebb622d27bc643c1e65d1e8af577))
 * **tools:** 为简咕客插件生成 zod 工具 ([42dfcd9](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/42dfcd9eb337603ecb0c5fa381ad89784345c9dc))
 * **tools:** 优化工具插件支持连续多行操作并添加新功能 ([734ca52](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/734ca5287785d84be0f8fc98b074476a3cc643f2))
+* **user:** 实现登录、登出功能并优化 token 存储 ([c2e046c](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/c2e046ce3068254f9b245d2b39c4c3bf3e7be44b))
+* **user:** 实现登录、登出功能并优化 token 存储 ([1514408](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/151440887e90d90d0522303009f6940f0507258e))
 * **user:** 实现登录取消功能并优化登录流程 ([50d124e](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/50d124e720c778e537488ca436855af21dcbcdeb))
 * **user:** 为认证回调页面添加立即关闭按钮 ([4328b48](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/4328b48e06002fa9d70427b21992f1022ffa73a0))
 * **views:** 优化代码覆盖率显示 ([ef98b11](https://codeup.aliyun.com/666cf9ed29ecbe23053513a3/JianGuoKe/ai-worktools/commits/ef98b115e57650483fae65ddec5ded085839ddd0))

package/README.md

@@ -1,67 +1,67 @@
-# 🎮 吃豆豆:单测智能体
-
-<p align="center">
-  <img src="https://aicoder.jianguoke.cn/assets/images/logo.png" alt="吃豆豆logo" width="120" height="120">
-</p>
-
-<p align="center">
-  <strong>自动生成高质量单元测试，让代码覆盖率轻松达到100% 🚀</strong>
-</p>
-
-<p align="center">
-  <em>你负责编写源代码，我负责编写测试代码，解放双手，专注核心业务逻辑开发 💻</em>
-</p>
-
----
-
-## 目录
-- [🎮 吃豆豆:单测智能体](#-吃豆豆单测智能体)
-  - [目录](#目录)
-  - [🌟 特性亮点](#-特性亮点)
-  - [🚀 核心功能](#-核心功能)
-  - [💎 核心价值](#-核心价值)
-  - [📝 使用指南](#-使用指南)
-  - [🔧 适用范围](#-适用范围)
-  - [🚀 开始使用](#-开始使用)
-  - [💬 联系我们](#-联系我们)
-
-## 🌟 特性亮点
-- 🔍 **智能代码分析**：深入理解代码结构和逻辑关系
-- 📝 **精准测试生成**：针对不同代码场景生成最佳测试策略
-- 🔄 **持续优化**：根据测试结果不断完善测试用例
-- 💡 **AI驱动**：利用先进的大语言模型提升测试质量
-
-## 🚀 核心功能
-- 🤖 **自动化测试生成**：自动识别未覆盖代码，生成完整测试用例，直到100%代码覆盖通过
-- 🛠️ **智能错误修复**：检测并修复单元测试中的语法错误、逻辑错误及断言失败
-- 📊 **覆盖率可视化**：实时展示行/语句/函数/分支覆盖率，通过颜色编码直观呈现测试状态
-- 🔄 **全流程自动化**：从环境初始化、依赖安装到持续测试优化，全程无需人工干预
-
-## 💎 核心价值
-- ⚡ **提升开发效率**：再也**不用手写**单测代码，专注核心业务逻辑开发
-- 🛡️ **保障代码质量**：确保测试覆盖率达标，降低线上bug风险
-- 🧩 **简化测试流程**：无需深入掌握测试框架细节，一键启动全自动化测试流程
-- 👁️ **可视化反馈**：通过直观的覆盖率报表和进度展示，让测试状态一目了然
-
-## 📝 使用指南
-1. 📂 打开项目文件夹
-2. ▶️ 启动吃豆豆智能体
-3. 📈 查看实时测试进度和覆盖率报告
-4. 🧠 智能体自动编写代码
-
-## 🔧 适用范围
-- 🔌 支持多种调用方式：VSCode插件、CLI命令
-- 💻 支持多种编程语言：TypeScript、JavaScript等
-- 🧪 支持主流测试框架：Jest、Vitest等
-
----
-
-## 🚀 开始使用
-准备好提升你的开发效率了吗？立即安装吃豆豆单测智能体，让测试工作变得轻松简单！
-
-## 💬 联系我们
-如有任何问题或建议，可以给我发邮件 9160294@qq.com。
-
-<p align="center">
-  <strong>© 2025 吃豆豆单测智能体 - 让单元测试自动化、智能化</strong>
+# 🎮 吃豆豆:单测智能体
+
+<p align="center">
+  <img src="https://aicoder.jianguoke.cn/assets/images/logo.png" alt="吃豆豆logo" width="120" height="120">
+</p>
+
+<p align="center">
+  <strong>自动生成高质量单元测试，让代码覆盖率轻松达到100% 🚀</strong>
+</p>
+
+<p align="center">
+  <em>你负责编写源代码，我负责编写测试代码，解放双手，专注核心业务逻辑开发 💻</em>
+</p>
+
+---
+
+## 目录
+- [🎮 吃豆豆:单测智能体](#-吃豆豆单测智能体)
+  - [目录](#目录)
+  - [🌟 特性亮点](#-特性亮点)
+  - [🚀 核心功能](#-核心功能)
+  - [💎 核心价值](#-核心价值)
+  - [📝 使用指南](#-使用指南)
+  - [🔧 适用范围](#-适用范围)
+  - [🚀 开始使用](#-开始使用)
+  - [💬 联系我们](#-联系我们)
+
+## 🌟 特性亮点
+- 🔍 **智能代码分析**：深入理解代码结构和逻辑关系
+- 📝 **精准测试生成**：针对不同代码场景生成最佳测试策略
+- 🔄 **持续优化**：根据测试结果不断完善测试用例
+- 💡 **AI驱动**：利用先进的大语言模型提升测试质量
+
+## 🚀 核心功能
+- 🤖 **自动化测试生成**：自动识别未覆盖代码，生成完整测试用例，直到100%代码覆盖通过
+- 🛠️ **智能错误修复**：检测并修复单元测试中的语法错误、逻辑错误及断言失败
+- 📊 **覆盖率可视化**：实时展示行/语句/函数/分支覆盖率，通过颜色编码直观呈现测试状态
+- 🔄 **全流程自动化**：从环境初始化、依赖安装到持续测试优化，全程无需人工干预
+
+## 💎 核心价值
+- ⚡ **提升开发效率**：再也**不用手写**单测代码，专注核心业务逻辑开发
+- 🛡️ **保障代码质量**：确保测试覆盖率达标，降低线上bug风险
+- 🧩 **简化测试流程**：无需深入掌握测试框架细节，一键启动全自动化测试流程
+- 👁️ **可视化反馈**：通过直观的覆盖率报表和进度展示，让测试状态一目了然
+
+## 📝 使用指南
+1. 📂 打开项目文件夹
+2. ▶️ 启动吃豆豆智能体
+3. 📈 查看实时测试进度和覆盖率报告
+4. 🧠 智能体自动编写代码
+
+## 🔧 适用范围
+- 🔌 支持多种调用方式：VSCode插件、CLI命令
+- 💻 支持多种编程语言：TypeScript、JavaScript等
+- 🧪 支持主流测试框架：Jest、Vitest等
+
+---
+
+## 🚀 开始使用
+准备好提升你的开发效率了吗？立即安装吃豆豆单测智能体，让测试工作变得轻松简单！
+
+## 💬 联系我们
+如有任何问题或建议，可以给我发邮件 9160294@qq.com。
+
+<p align="center">
+  <strong>© 2025 吃豆豆单测智能体 - 让单元测试自动化、智能化</strong>
 </p>

package/dist/agents/jianguoke.js

@@ -47,7 +47,7 @@ const setConversationId = (id) => {
 };
 exports.setConversationId = setConversationId;
 async function postData(url, data) {
-    const token = (0, user_1.getTokenStorage)().get();
+    const token = await (0, user_1.getTokenStorage)().get();
     try {
         // 发送POST请求
         const response = await fetch(url, {
@@ -109,7 +109,7 @@ async function startAgent(name, question, vars, withConversation, ctrl, onMessag
     let msg = '';
     let lastMsgNo = 0;
     const baseURL = process.env.AI_SERVER || DEFAULT_SERVER;
-    const token = (0, user_1.getTokenStorage)().get();
+    const token = await (0, user_1.getTokenStorage)().get();
     console.log(`Starting agent: ${baseURL}, token: ${token}`);
     await (0, fetch_sse_1.fetchEventData)(`${baseURL}/api/agent/exec/aicoder`, {
         method: 'POST',

package/dist/agents/prompt.js

@@ -10,32 +10,32 @@ const path_1 = __importDefault(require("path"));
 const tools_1 = require("../tools");
 const MAXLEN = 10000;
 function fixErrorPrompt(error) {
-    return `单测执行失败,日志如下:
-\`\`\`
-${error.slice(0, MAXLEN)}
-\`\`\`
+    return `单测执行失败,日志如下:
+\`\`\`
+${error.slice(0, MAXLEN)}
+\`\`\`
 `;
 }
 // 提示词管理，根据任务合成特定提示词
 async function fixCodePrompt(failedTest) {
-    return `测试文件"${path_1.default.resolve(failedTest.testFilePath)}"${failedTest.testName ? '里的"' + failedTest.testName + '"' : ''}执行失败.
-
-错误信息为:
-\`\`\`
-${failedTest.errorMessages.join('\n')}
-\`\`\`
-
-测试文件内容如下:
-\`\`\`
-${await (0, tools_1.readFile)(failedTest.testFilePath, 'utf8', true)}
+    return `测试文件"${path_1.default.resolve(failedTest.testFilePath)}"${failedTest.testName ? '里的"' + failedTest.testName + '"' : ''}执行失败.
+
+错误信息为:
+\`\`\`
+${failedTest.errorMessages.join('\n')}
+\`\`\`
+
+测试文件内容如下:
+\`\`\`
+${await (0, tools_1.readFile)(failedTest.testFilePath, 'utf8', true)}
 \`\`\``;
 }
 async function addUncoveredLinePrompt(srcPath, report) {
-    return `源文件"${path_1.default.resolve(srcPath)}"的单元测试覆盖率是${report.coverage.lines.pct}%, 还有"${report.uncoveredLines.join(',')}"行未覆盖单测。
-
-源文件内容如下:
-\`\`\`
-${await (0, tools_1.readFile)(srcPath, 'utf8', true)}
+    return `源文件"${path_1.default.resolve(srcPath)}"的单元测试覆盖率是${report.coverage.lines.pct}%, 还有"${report.uncoveredLines.join(',')}"行未覆盖单测。
+
+源文件内容如下:
+\`\`\`
+${await (0, tools_1.readFile)(srcPath, 'utf8', true)}
 \`\`\``;
 }
 //# sourceMappingURL=prompt.js.map

package/dist/program.js

@@ -6,15 +6,54 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const commander_1 = require("commander");
 const path_1 = __importDefault(require("path"));
+const promises_1 = __importDefault(require("fs/promises"));
+const os_1 = __importDefault(require("os"));
 const testAgent_1 = require("./testAgent");
 const tools_1 = require("./tools");
 const view_1 = require("./view");
-// import { showCoverageView } from './view';
+const user_1 = require("./user");
+const view_2 = require("./view");
 const program = new commander_1.Command();
 program.name('testAgent').description('吃豆豆:单测智能体').version('1.0.1');
 // 屏蔽调试日志
 const logHandler = console.log;
 console.log = () => { };
+const tokenFile = path_1.default.join(os_1.default.homedir(), '.aiworktools');
+// token存储在系统用户文件夹的.ai-worktools
+(0, user_1.setTokenStorage)({
+    get: async () => {
+        let exists = false;
+        try {
+            await promises_1.default.access(tokenFile);
+            exists = true;
+        }
+        catch (err) {
+            return false;
+        }
+        if (exists) {
+            return JSON.parse(await promises_1.default.readFile(tokenFile, 'utf-8')).token;
+        }
+        return null;
+    },
+    set: async (token) => {
+        await promises_1.default.writeFile(tokenFile, JSON.stringify({ token }), 'utf-8');
+    },
+    clear: async () => {
+        await promises_1.default.unlink(tokenFile);
+    }
+});
+program
+    .command('login')
+    .description('登录')
+    .action(async () => {
+    await (0, user_1.login)('ai-worktools-cli');
+});
+program
+    .command('logout')
+    .description('登出')
+    .action(async () => {
+    await (0, user_1.logout)();
+});
 program
     .command('start [projectRoot]')
     .description('启动单测智能体,根据源代码文件开始补全单测代码,直到覆盖率100%')
@@ -27,7 +66,7 @@ program
     .option('--fullCoverageEnd <boolean>', '是否在达到100%覆盖率后结束', (txt) => txt === 'true', true)
     .action(async (projectRoot, options) => {
     let projectConfig;
-    (0, testAgent_1.startTestAgent)({
+    await (0, testAgent_1.startTestAgent)({
         ...options,
         projectRoot: projectRoot || options.projectRoot,
         onReady: async (project) => {
@@ -40,6 +79,7 @@ program
             }
         }
     });
+    (0, view_1.disposeTimer)();
 });
 program
     .command('showCoverage [projectRoot]')
@@ -48,7 +88,7 @@ program
     .option('-c, --coverageDir <string>', '覆盖率报告文件夹', 'coverage')
     .action(async (projectRoot, options) => {
     await (0, tools_1.runJestTests)(projectRoot || options.projectRoot);
-    // await showCoverageView(path.join(options.projectRoot, options.coverageDir));
+    await (0, view_2.showCoverageView)(path_1.default.join(options.projectRoot, options.coverageDir));
 });
 exports.default = program;
 //# sourceMappingURL=program.js.map

package/dist/tools/git.js

@@ -43,7 +43,7 @@ async function clone(repoPath, remoteUrl, localPath, branch) {
         console.log(`成功克隆仓库: ${remoteUrl} 到 ${clonePath}`);
     }
     catch (error) {
-        console.error(`克隆仓库失败: ${error}`);
+        console.log(`克隆仓库失败: ${error}`);
         throw error;
     }
 }
@@ -59,7 +59,7 @@ async function add(files = '.', repoPath) {
         console.log(`成功添加文件到暂存区: ${Array.isArray(files) ? files.join(', ') : files}`);
     }
     catch (error) {
-        console.error(`添加文件失败: ${error}`);
+        console.log(`添加文件失败: ${error}`);
         throw error;
     }
 }
@@ -73,10 +73,10 @@ async function commit(repoPath, message, files) {
     const git = createGitInstance(repoPath);
     try {
         const result = files ? await git.commit(message, files) : await git.commit(message);
-        console.log(`提交成功: ${result.summary}`);
+        console.log(`提交成功: ${result?.summary}`);
     }
     catch (error) {
-        console.error(`提交失败: ${error}`);
+        console.log(`提交失败: ${error}`);
         throw error;
     }
 }
@@ -91,10 +91,10 @@ async function pull(repoPath, remote = 'origin', branch) {
     try {
         const currentBranch = branch || (await git.branchLocal()).current;
         const result = await git.pull(remote, currentBranch);
-        console.log(`拉取成功: ${result.summary}`);
+        console.log(`拉取成功: ${result?.summary}`);
     }
     catch (error) {
-        console.error(`拉取失败: ${error}`);
+        console.log(`拉取失败: ${error}`);
         throw error;
     }
 }
@@ -112,7 +112,7 @@ async function push(repoPath, remote = 'origin', branch) {
         console.log(`推送成功: ${remote}/${currentBranch}`);
     }
     catch (error) {
-        console.error(`推送失败: ${error}`);
+        console.log(`推送失败: ${error}`);
         // 处理常见的推送失败情况（如远程分支有新提交）
         if (error.message.includes('non-fast-forward')) {
             console.log('提示: 远程分支有新提交，建议先拉取再推送');
@@ -132,7 +132,7 @@ async function createBranch(branchName, repoPath) {
         console.log(`成功创建并切换到分支: ${branchName}`);
     }
     catch (error) {
-        console.error(`创建分支失败: ${error}`);
+        console.log(`创建分支失败: ${error}`);
         throw error;
     }
 }
@@ -148,7 +148,7 @@ async function switchBranch(branchName, repoPath) {
         console.log(`成功切换到分支: ${branchName}`);
     }
     catch (error) {
-        console.error(`切换分支失败: ${error}`);
+        console.log(`切换分支失败: ${error}`);
         throw error;
     }
 }
@@ -162,7 +162,7 @@ async function getStatus(repoPath) {
         return await git.status();
     }
     catch (error) {
-        console.error(`获取仓库状态失败: ${error}`);
+        console.log(`获取仓库状态失败: ${error}`);
         throw error;
     }
 }
@@ -179,7 +179,7 @@ async function addRemote(remoteName, remoteUrl, repoPath) {
         console.log(`成功添加远程仓库: ${remoteName} -> ${remoteUrl}`);
     }
     catch (error) {
-        console.error(`添加远程仓库失败: ${error}`);
+        console.log(`添加远程仓库失败: ${error}`);
         throw error;
     }
 }
@@ -197,7 +197,7 @@ async function setUserInfo(username, email, repoPath) {
         console.log(`已设置用户信息: ${username} <${email}>`);
     }
     catch (error) {
-        console.error(`设置用户信息失败: ${error}`);
+        console.log(`设置用户信息失败: ${error}`);
         throw error;
     }
 }

package/dist/tools/index.js

@@ -54,7 +54,7 @@ async function wait(timerout = 1000) {
         }, timerout);
     });
 }
-function runTests(projectRoot, testFramework, coverageDir, configPath) {
+async function runTests(projectRoot, testFramework, coverageDir, configPath) {
     if (testFramework === 'jest') {
         return (0, jest_1.runJestTests)(projectRoot, coverageDir, configPath);
     }

package/dist/tools/package.js

@@ -279,7 +279,8 @@ async function checkDependency(projectRoot, packageName, isDev = false) {
 async function setupTestEnvironment(projectRoot, packageManager, devLanguage, testFramework) {
     // 验证参数
     validateParameters(packageManager, devLanguage, testFramework);
-    const packageJsonPath = path_1.default.join(projectRoot, 'package.json');
+    const resolvedPath = path_1.default.resolve(projectRoot);
+    const packageJsonPath = path_1.default.join(resolvedPath, 'package.json');
     // 异步检查并初始化package.json
     try {
         await promises_1.default.access(packageJsonPath);
@@ -287,29 +288,29 @@ async function setupTestEnvironment(projectRoot, packageManager, devLanguage, te
     catch {
         console.log('未找到package.json，初始化项目...');
         const initCommand = packageManager === 'yarn' ? 'yarn init -y' : `${packageManager} init -y`;
-        await executeCommand(projectRoot, initCommand, '初始化项目');
+        await executeCommand(resolvedPath, initCommand, '初始化项目');
     }
     // 如果是TypeScript项目，检查并安装TypeScript
     if (devLanguage === 'typescript') {
         console.log('检查TypeScript是否安装...');
-        const hasTypeScript = await checkDependency(projectRoot, 'typescript', true);
+        const hasTypeScript = await checkDependency(resolvedPath, 'typescript', true);
         if (!hasTypeScript) {
             const installCmd = getInstallCommand(['typescript', '@types/node'], true, packageManager);
-            await executeCommand(projectRoot, installCmd, '安装TypeScript及类型定义');
+            await executeCommand(resolvedPath, installCmd, '安装TypeScript及类型定义');
             // 异步检查并创建tsconfig.json
-            const tsconfigPath = path_1.default.join(projectRoot, 'tsconfig.json');
+            const tsconfigPath = path_1.default.join(resolvedPath, 'tsconfig.json');
             try {
                 await promises_1.default.access(tsconfigPath);
             }
             catch {
-                await executeCommand(projectRoot, 'npx tsc --init', '创建tsconfig.json');
+                await executeCommand(resolvedPath, 'npx tsc --init', '创建tsconfig.json');
             }
         }
     }
     // 检查并安装测试框架
     console.log(`检查${testFramework}是否安装...`);
-    const hasTestFramework = await checkDependency(projectRoot, testFramework, true);
-    const hasTestTypes = devLanguage === 'typescript' ? await checkDependency(projectRoot, `@types/${testFramework}`, true) : true;
+    const hasTestFramework = await checkDependency(resolvedPath, testFramework, true);
+    const hasTestTypes = devLanguage === 'typescript' ? await checkDependency(resolvedPath, `@types/${testFramework}`, true) : true;
     const packagesToInstall = [];
     if (!hasTestFramework) {
         packagesToInstall.push(testFramework);

package/dist/tools/project.js

@@ -4,6 +4,15 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.detectProjectConfig = detectProjectConfig;
+exports.checkProjectInstalled = checkProjectInstalled;
+exports.detectPackageManager = detectPackageManager;
+exports.detectLanguages = detectLanguages;
+exports.detectSrcDir = detectSrcDir;
+exports.detectTestConfig = detectTestConfig;
+exports.detectExistingTestPatterns = detectExistingTestPatterns;
+exports.getDependencies = getDependencies;
+exports.detectProjectType = detectProjectType;
+exports.detectFramework = detectFramework;
 const promises_1 = __importDefault(require("fs/promises"));
 const path_1 = __importDefault(require("path"));
 /**
@@ -207,13 +216,13 @@ async function detectTestConfig(projectRoot, config) {
     ];
     // 根据项目语言生成对应扩展名的模式
     const extensions = new Set();
-    if (config.languages.includes('javascript')) {
+    if (config.languages?.includes('javascript')) {
         extensions.add('js');
     }
-    if (config.languages.includes('typescript')) {
+    if (config.languages?.includes('typescript')) {
         extensions.add('ts');
     }
-    if (config.languages.includes('coffeescript')) {
+    if (config.languages?.includes('coffeescript')) {
         extensions.add('coffee');
     }
     // 默认为js（如果没有检测到语言）
@@ -404,11 +413,11 @@ async function calculateScore(projectRoot, type) {
         }
     });
     // 检查配置文件匹配（并行处理）
-    const configFileChecks = features.configFiles.map((file) => fileExists(path_1.default.join(process.cwd(), file)));
+    const configFileChecks = features.configFiles.map((file) => fileExists(path_1.default.join(projectRoot, file)));
     const configFileResults = await Promise.all(configFileChecks);
     score += configFileResults.filter(Boolean).length * 2; // 配置文件权重中等
     // 检查目录结构匹配（并行处理）
-    const directoryChecks = features.directories.map((dir) => directoryExists(path_1.default.join(process.cwd(), dir)));
+    const directoryChecks = features.directories.map((dir) => directoryExists(path_1.default.join(projectRoot, dir)));
     const directoryResults = await Promise.all(directoryChecks);
     score += directoryResults.filter(Boolean).length * 1; // 目录权重较低
     return score;
@@ -430,11 +439,6 @@ async function detectProjectType(projectRoot, config) {
         backend: backendScore,
         cli: cliScore
     };
-    // 判断是否为全栈项目
-    const isFullstack = frontendScore > 0 && backendScore > 0 && Math.abs(frontendScore - backendScore) < frontendScore * 0.5;
-    if (isFullstack) {
-        config.projectType = 'fullstack';
-    }
     // 确定最高分数的项目类型
     const maxScore = Math.max(frontendScore, backendScore, cliScore);
     if (maxScore === 0) {
@@ -449,6 +453,11 @@ async function detectProjectType(projectRoot, config) {
     else {
         config.projectType = 'cli';
     }
+    // 判断是否为全栈项目
+    const isFullstack = frontendScore > 0 && backendScore > 0 && Math.abs(frontendScore - backendScore) < frontendScore * 0.5;
+    if (isFullstack) {
+        config.projectType = 'fullstack';
+    }
 }
 /**
  * 检测框架信息

package/dist/user.js

@@ -56,8 +56,25 @@ const util_1 = __importDefault(require("util"));
 const child_process_1 = require("child_process");
 // 转换exec为Promise形式，便于async/await使用
 const execAsync = util_1.default.promisify(child_process_1.exec);
-let tokenStorage;
-let environment;
+let tokenStorage = {
+    get: async () => {
+        return localStorage.getItem('token');
+    },
+    set: async (token) => {
+        localStorage.setItem('token', token);
+    },
+    clear: async () => {
+        localStorage.removeItem('token');
+    }
+};
+let environment = {
+    showInfoMessage: async (message) => {
+        console.log(message);
+    },
+    showErrorMessage: async (message) => {
+        console.error(message);
+    }
+};
 function setEnvironment(env) {
     environment = env;
 }
@@ -98,54 +115,54 @@ function startCallbackServer(state, port) {
             if (callbackState !== state) {
                 // eslint-disable-next-line @typescript-eslint/naming-convention
                 res.writeHead(403, { 'Content-Type': 'text/html; charset=utf-8' });
-                res.end(`
-                    <html>
-                        <head>
-                            <style>
-                                body {
-                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-                                    display: flex;
-                                    flex-direction: column;
-                                    align-items: center;
-                                    justify-content: center;
-                                    height: 100vh;
-                                    margin: 0;
-                                    background-color: #f8f9fa;
-                                    color: #333;
-                                }
-                                .card {
-                                    background-color: white;
-                                    border-radius: 12px;
-                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-                                    padding: 2rem;
-                                    text-align: center;
-                                    max-width: 400px;
-                                    width: 90%;
-                                    border-left: 4px solid #dc3545;
-                                }
-                                h1 {
-                                    color: #dc3545;
-                                    margin-bottom: 1rem;
-                                    font-size: 1.8rem;
-                                }
-                                p {
-                                    margin: 0.5rem 0;
-                                    color: #6c757d;
-                                }
-                                .countdown {
-                                    font-weight: bold;
-                                    color: #dc3545;
-                                    font-size: 1.2rem;
-                                }
-                            </style>
-                        </head>
-                        <body>
-                            <div class="card">
-                                <h1>登录验证失败</h1>
-                                <p>无效的状态参数，可能存在安全风险</p>
-                            </div>
-                        </body>
-                    </html>
+                res.end(`
+                    <html>
+                        <head>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #dc3545;
+                                }
+                                h1 {
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #dc3545;
+                                    font-size: 1.2rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <h1>登录验证失败</h1>
+                                <p>无效的状态参数，可能存在安全风险</p>
+                            </div>
+                        </body>
+                    </html>
                 `);
                 reject(new Error('登录验证失败：无效的状态参数'));
                 server.close();
@@ -160,121 +177,121 @@ function startCallbackServer(state, port) {
                 // 返回包含自动刷新和关闭逻辑的HTML
                 // eslint-disable-next-line @typescript-eslint/naming-convention
                 res.writeHead(200, { 'Content-Type': 'text/html; charset=utf-8' });
-                res.end(`
-                    <html>
-                        <head>
-                            <title>登录成功</title>
-                            <style>
-                                body {
-                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-                                    display: flex;
-                                    flex-direction: column;
-                                    align-items: center;
-                                    justify-content: center;
-                                    height: 100vh;
-                                    margin: 0;
-                                    background-color: #f8f9fa;
-                                    color: #333;
-                                }
-                                .card {
-                                    background-color: white;
-                                    border-radius: 12px;
-                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-                                    padding: 2rem;
-                                    text-align: center;
-                                    max-width: 400px;
-                                    width: 90%;
-                                    border-left: 4px solid #28a745;
-                                }
-                                h1 {
-                                    color: #28a745;
-                                    margin-bottom: 1rem;
-                                    font-size: 1.8rem;
-                                }
-                                p {
-                                    margin: 0.5rem 0;
-                                    color: #6c757d;
-                                }
-                                .countdown {
-                                    font-weight: bold;
-                                    color: #28a745;
-                                    font-size: 1.2rem;
-                                }
-                                .checkmark {
-                                    font-size: 3rem;
-                                    color: #28a745;
-                                    margin-bottom: 1rem;
-                                }
-                            </style>
-                        </head>
-                        <body>
-                            <div class="card">
-                                <div class="checkmark">✓</div>
-                                <h1>登录成功！</h1>
-                                <p>您可以关闭此窗口返回应用。</p>
-                            </div>
-                        </body>
-                    </html>
+                res.end(`
+                    <html>
+                        <head>
+                            <title>登录成功</title>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #28a745;
+                                }
+                                h1 {
+                                    color: #28a745;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #28a745;
+                                    font-size: 1.2rem;
+                                }
+                                .checkmark {
+                                    font-size: 3rem;
+                                    color: #28a745;
+                                    margin-bottom: 1rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <div class="checkmark">✓</div>
+                                <h1>登录成功！</h1>
+                                <p>您可以关闭此窗口返回应用。</p>
+                            </div>
+                        </body>
+                    </html>
                 `);
                 resolve(token);
             }
             else {
                 // eslint-disable-next-line @typescript-eslint/naming-convention
                 res.writeHead(400, { 'Content-Type': 'text/html; charset=utf-8' });
-                res.end(`
-                    <html>
-                        <head>
-                            <style>
-                                body {
-                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-                                    display: flex;
-                                    flex-direction: column;
-                                    align-items: center;
-                                    justify-content: center;
-                                    height: 100vh;
-                                    margin: 0;
-                                    background-color: #f8f9fa;
-                                    color: #333;
-                                }
-                                .card {
-                                    background-color: white;
-                                    border-radius: 12px;
-                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-                                    padding: 2rem;
-                                    text-align: center;
-                                    max-width: 400px;
-                                    width: 90%;
-                                    border-left: 4px solid #dc3545;
-                                }
-                                h1 {
-                                    color: #dc3545;
-                                    margin-bottom: 1rem;
-                                    font-size: 1.8rem;
-                                }
-                                p {
-                                    margin: 0.5rem 0;
-                                    color: #6c757d;
-                                }
-                                .countdown {
-                                    font-weight: bold;
-                                    color: #dc3545;
-                                    font-size: 1.2rem;
-                                }
-                                .error-icon {
-                                    font-size: 3rem;
-                                    color: #dc3545;
-                                    margin-bottom: 1rem;
-                                }
-                            </style>
-                        </head>
-                        <body>
-                            <div class="card">
-                                <div class="error-icon">✗</div>
-                                <h1>登录失败</h1>
-                                <p>未收到有效的认证信息</p>
-                            </div>
-                        </body>
-                    </html>
+                res.end(`
+                    <html>
+                        <head>
+                            <style>
+                                body {
+                                    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                                    display: flex;
+                                    flex-direction: column;
+                                    align-items: center;
+                                    justify-content: center;
+                                    height: 100vh;
+                                    margin: 0;
+                                    background-color: #f8f9fa;
+                                    color: #333;
+                                }
+                                .card {
+                                    background-color: white;
+                                    border-radius: 12px;
+                                    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+                                    padding: 2rem;
+                                    text-align: center;
+                                    max-width: 400px;
+                                    width: 90%;
+                                    border-left: 4px solid #dc3545;
+                                }
+                                h1 {
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                    font-size: 1.8rem;
+                                }
+                                p {
+                                    margin: 0.5rem 0;
+                                    color: #6c757d;
+                                }
+                                .countdown {
+                                    font-weight: bold;
+                                    color: #dc3545;
+                                    font-size: 1.2rem;
+                                }
+                                .error-icon {
+                                    font-size: 3rem;
+                                    color: #dc3545;
+                                    margin-bottom: 1rem;
+                                }
+                            </style>
+                        </head>
+                        <body>
+                            <div class="card">
+                                <div class="error-icon">✗</div>
+                                <h1>登录失败</h1>
+                                <p>未收到有效的认证信息</p>
+                            </div>
+                        </body>
+                    </html>
                 `);
                 reject(new Error('登录失败：未收到token'));
             }
@@ -335,6 +352,7 @@ async function openExternalUrl(url) {
         return false;
     }
 }
+const AUTH_SERVER_URL = 'https://login.jianguoke.cn/login';
 /**
  * 执行登录流程
  * @param env 环境接口
@@ -343,7 +361,8 @@ async function openExternalUrl(url) {
  * @param clientId 客户端ID
  * @returns 登录成功后的token
  */
-async function login(authServerUrl, clientId) {
+async function login(clientId) {
+    const authServerUrl = AUTH_SERVER_URL;
     const env = getEnvironment();
     const tokenStorage = getTokenStorage();
     const state = generateState();
@@ -407,8 +426,8 @@ function isLoggedIn() {
     }
     return true;
 }
-function getUserData() {
-    const token = getTokenStorage().get();
+async function getUserData() {
+    const token = await getTokenStorage().get();
     if (!token) {
         return null;
     }

package/dist/view.js

@@ -7,10 +7,11 @@ 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 chalk_1 = __importDefault(require("chalk"));
+const cli_color_1 = __importDefault(require("cli-color"));
 let REFRESH_INTERVAL = 3000;
 // 存储上一次的覆盖率数据
 let previousCoverage = null;
@@ -25,7 +26,7 @@ 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(chalk_1.default.cyan('开始监控覆盖率数据...' + coveragePath));
+    console.log(cli_color_1.default.cyan('开始监控覆盖率数据...' + coveragePath));
     // 初始加载
     await loadAndDisplayCoverage(coveragePath);
     // 设置文件监听
@@ -50,6 +51,7 @@ function stopMonitoring() {
         refreshTimer = null;
     }
 }
+let timer = null;
 // 加载并显示覆盖率数据
 async function loadAndDisplayCoverage(coveragePath, refreshTimeout = REFRESH_INTERVAL) {
     try {
@@ -61,7 +63,8 @@ async function loadAndDisplayCoverage(coveragePath, refreshTimeout = REFRESH_INT
             changedCells = compareCoverage(previousCoverage, coverage);
             // 如果有变化，3秒后清除标记
             if (changedCells.length > 0) {
-                setTimeout(() => {
+                timer = setTimeout(() => {
+                    timer = null;
                     changedCells = [];
                     loadAndDisplayCoverage(coveragePath);
                 }, refreshTimeout);
@@ -73,7 +76,7 @@ async function loadAndDisplayCoverage(coveragePath, refreshTimeout = REFRESH_INT
         renderTable(tableData);
     }
     catch (error) {
-        console.error(chalk_1.default.red(`加载覆盖率数据时出错: ${error.message}`));
+        console.error(cli_color_1.default.red(`加载覆盖率数据时出错: ${error.message}`));
     }
 }
 // 读取覆盖率文件
@@ -102,7 +105,7 @@ function generateTableData(coverage) {
     filePaths.sort();
     // 添加总计行
     tableData.push([
-        chalk_1.default.bold('总计'),
+        cli_color_1.default.bold('总计'),
         formatPercentage(coverage.total.lines.pct),
         formatPercentage(coverage.total.statements.pct),
         formatPercentage(coverage.total.functions.pct),
@@ -126,17 +129,17 @@ function generateTableData(coverage) {
 // 格式化百分比并应用颜色
 function formatPercentage(pct) {
     if (pct === 'Unknown')
-        return chalk_1.default.gray('Unknown');
+        return cli_color_1.default.italic('Unknown');
     let formatted = pct.toFixed(2) + '%';
     // 根据覆盖率设置颜色
     if (pct < 50) {
-        formatted = chalk_1.default.red(formatted);
+        formatted = cli_color_1.default.red(formatted);
     }
     else if (pct < 80) {
-        formatted = chalk_1.default.yellow(formatted);
+        formatted = cli_color_1.default.yellow(formatted);
     }
     else {
-        formatted = chalk_1.default.green(formatted);
+        formatted = cli_color_1.default.green(formatted);
     }
     return formatted;
 }
@@ -176,7 +179,7 @@ function renderTable(tableData) {
         return row.map((cell, colIndex) => {
             // 检查是否是变化的单元格
             if (changedCells.some(([r, c]) => r === rowIndex && c === colIndex)) {
-                return chalk_1.default.bgRed.white(cell);
+                return cli_color_1.default.bgRed.white(cell);
             }
             return cell;
         });
@@ -210,7 +213,13 @@ function renderTable(tableData) {
     };
     // 清除控制台并打印表格
     console.clear();
-    console.log(chalk_1.default.cyan('单测覆盖率监控工具 - 最后更新: ' + new Date().toLocaleString()));
+    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

@@ -1,7 +1,7 @@
 {
   "name": "ai-worktool",
   "displayName": "吃豆豆:单测智能体",
-  "version": "1.0.12",
+  "version": "1.0.14",
   "description": "单元测试智能体,帮你开发单元测试用例代码直到100%单测覆盖通过。",
   "author": "jianguoke.cn",
   "license": "MIT",
@@ -13,6 +13,7 @@
     "watch": "tsc -w",
     "build": "tsc",
     "test": "jest test/tools",
+    "coverage": "jest --coverage test/tools",
     "plugin": "ts-node ./tools.ts && 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",
@@ -256,11 +257,12 @@
     "acorn": "^8.15.0",
     "acorn-walk": "^8.3.4",
     "typescript": "^5.9.2",
-    "chalk": "^5.4.1",
+    "cli-color": "^2.0.4",
     "commander": "^14.0.0",
     "table": "^6.9.0"
   },
   "devDependencies": {
+    "@types/cli-color": "^2.0.6",
     "@types/jest": "^30.0.0",
     "@types/jsonwebtoken": "^9.0.10",
     "@types/node": "^24.1.0",