ai-worktool 1.0.12 → 1.0.13

package/CHANGELOG.md

@@ -1,4 +1,4 @@
-## 1.0.12 (2025-08-03)
+## 1.0.13 (2025-08-04)
 
 
 ### Bug Fixes
@@ -152,6 +152,7 @@
 * **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 存储 ([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/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");
+const user_1 = require("./user");
 // import { showCoverageView } from './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) => {

package/dist/tools/git.js

@@ -73,7 +73,7 @@ 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}`);
@@ -91,7 +91,7 @@ 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}`);

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/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;
 }
@@ -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/package.json

@@ -1,7 +1,7 @@
 {
   "name": "ai-worktool",
   "displayName": "吃豆豆:单测智能体",
-  "version": "1.0.12",
+  "version": "1.0.13",
   "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",