ai-worktool 1.0.0

package/dist/tools/git.js

@@ -0,0 +1,204 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.clone = clone;
+exports.add = add;
+exports.commit = commit;
+exports.pull = pull;
+exports.push = push;
+exports.createBranch = createBranch;
+exports.switchBranch = switchBranch;
+exports.getStatus = getStatus;
+exports.addRemote = addRemote;
+exports.setUserInfo = setUserInfo;
+const simple_git_1 = __importDefault(require("simple-git"));
+const path_1 = require("path");
+/**
+ * 创建 Git 实例
+ * @param repoPath 仓库路径，默认为当前工作目录
+ */
+function createGitInstance(repoPath) {
+    const options = {
+        baseDir: repoPath,
+        binary: 'git',
+        maxConcurrentProcesses: 6
+    };
+    return (0, simple_git_1.default)(options);
+}
+/**
+ * 克隆远程仓库到本地
+ * @param remoteUrl 远程仓库URL
+ * @param localPath 本地存储路径
+ * @param branch 要克隆的分支
+ * @param repoPath 仓库根路径
+ */
+async function clone(repoPath, remoteUrl, localPath, branch) {
+    const git = createGitInstance(repoPath);
+    try {
+        const clonePath = localPath ? (0, path_1.join)(repoPath, localPath) : repoPath;
+        const options = branch ? ['-b', branch] : [];
+        await git.clone(remoteUrl, clonePath, options);
+        console.log(`成功克隆仓库: ${remoteUrl} 到 ${clonePath}`);
+    }
+    catch (error) {
+        console.log(`克隆仓库失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 添加文件到暂存区
+ * @param files 文件路径，可以是单个文件或文件数组
+ * @param repoPath 仓库路径
+ */
+async function add(files = '.', repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        await git.add(files);
+        console.log(`成功添加文件到暂存区: ${Array.isArray(files) ? files.join(', ') : files}`);
+    }
+    catch (error) {
+        console.log(`添加文件失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 提交代码
+ * @param message 提交信息
+ * @param files 可选的文件路径，指定要提交的文件
+ * @param repoPath 仓库路径
+ */
+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}`);
+    }
+    catch (error) {
+        console.log(`提交失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 拉取远程分支
+ * @param remote 远程仓库名称，默认为 'origin'
+ * @param branch 分支名称，默认为当前分支
+ * @param repoPath 仓库路径
+ */
+async function pull(repoPath, remote = 'origin', branch) {
+    const git = createGitInstance(repoPath);
+    try {
+        const currentBranch = branch || (await git.branchLocal()).current;
+        const result = await git.pull(remote, currentBranch);
+        console.log(`拉取成功: ${result?.summary}`);
+    }
+    catch (error) {
+        console.log(`拉取失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 推送代码到远程仓库
+ * @param remote 远程仓库名称，默认为 'origin'
+ * @param branch 分支名称，默认为当前分支
+ * @param repoPath 仓库路径
+ */
+async function push(repoPath, remote = 'origin', branch) {
+    const git = createGitInstance(repoPath);
+    try {
+        const currentBranch = branch || (await git.branchLocal()).current;
+        await git.push(remote, currentBranch);
+        console.log(`推送成功: ${remote}/${currentBranch}`);
+    }
+    catch (error) {
+        console.log(`推送失败: ${error}`);
+        // 处理常见的推送失败情况（如远程分支有新提交）
+        if (error.message.includes('non-fast-forward')) {
+            console.log('提示: 远程分支有新提交，建议先拉取再推送');
+        }
+        throw error;
+    }
+}
+/**
+ * 创建并切换到新分支
+ * @param branchName 新分支名称
+ * @param repoPath 仓库路径
+ */
+async function createBranch(branchName, repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        await git.checkoutLocalBranch(branchName);
+        console.log(`成功创建并切换到分支: ${branchName}`);
+    }
+    catch (error) {
+        console.log(`创建分支失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 切换分支
+ * @param branchName 目标分支名称
+ * @param repoPath 仓库路径
+ */
+async function switchBranch(branchName, repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        await git.checkout(branchName);
+        console.log(`成功切换到分支: ${branchName}`);
+    }
+    catch (error) {
+        console.log(`切换分支失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 获取当前仓库状态
+ * @param repoPath 仓库路径
+ */
+async function getStatus(repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        return await git.status();
+    }
+    catch (error) {
+        console.log(`获取仓库状态失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 添加远程仓库
+ * @param remoteName 远程仓库名称
+ * @param remoteUrl 远程仓库URL
+ * @param repoPath 仓库路径
+ */
+async function addRemote(remoteName, remoteUrl, repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        await git.addRemote(remoteName, remoteUrl);
+        console.log(`成功添加远程仓库: ${remoteName} -> ${remoteUrl}`);
+    }
+    catch (error) {
+        console.log(`添加远程仓库失败: ${error}`);
+        throw error;
+    }
+}
+/**
+ * 设置提交用户信息
+ * @param username 用户名
+ * @param email 邮箱
+ * @param repoPath 仓库路径
+ */
+async function setUserInfo(username, email, repoPath) {
+    const git = createGitInstance(repoPath);
+    try {
+        await git.addConfig('user.name', username);
+        await git.addConfig('user.email', email);
+        console.log(`已设置用户信息: ${username} <${email}>`);
+    }
+    catch (error) {
+        console.log(`设置用户信息失败: ${error}`);
+        throw error;
+    }
+}
+//# sourceMappingURL=git.js.map

package/dist/tools/index.js

@@ -0,0 +1,71 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.modifyTools = void 0;
+exports.wait = wait;
+exports.runTests = runTests;
+const jest_1 = require("./jest");
+const vitest_1 = require("./vitest");
+const mocha_1 = require("./mocha");
+__exportStar(require("./interface"), exports);
+__exportStar(require("./file"), exports);
+// export * from './javascript';
+// export * from './typescript';
+__exportStar(require("./jest"), exports);
+__exportStar(require("./vitest"), exports);
+__exportStar(require("./mocha"), exports);
+__exportStar(require("./git"), exports);
+__exportStar(require("./package"), exports);
+__exportStar(require("./project"), exports);
+exports.modifyTools = [
+    'writeFile',
+    'batchModifyLines',
+    'insertLine',
+    'deleteLine',
+    'modifyLine',
+    'renameFile',
+    'initProject',
+    'installDependencies',
+    'createTsConfigFile',
+    'createGitIgnore',
+    'createPackageJson',
+    'createTsConfigJson',
+    'createJestConfig',
+    'createVitestConfig',
+    'createMochaConfig'
+];
+async function wait(timerout = 1000) {
+    return new Promise((resolve) => {
+        setTimeout(() => {
+            resolve(true);
+        }, timerout);
+    });
+}
+async function runTests(projectRoot, testFramework, coverageDir, configPath) {
+    if (testFramework === 'jest') {
+        return (0, jest_1.runJestTests)(projectRoot, coverageDir, configPath);
+    }
+    else if (testFramework === 'vitest') {
+        return (0, vitest_1.runVitestTests)(projectRoot, coverageDir, configPath);
+    }
+    else if (testFramework === 'mocha') {
+        return (0, mocha_1.runMochaTests)(projectRoot, coverageDir, configPath);
+    }
+    else {
+        throw new Error(`尚未支持 ${testFramework} 测试框架`);
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/tools/interface.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=interface.js.map

package/dist/tools/javascript.js

@@ -0,0 +1,183 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.parseJavaScriptFile = parseJavaScriptFile;
+exports.extractExportedFunctions = extractExportedFunctions;
+exports.extractExportedClasses = extractExportedClasses;
+exports.extractImportedModules = extractImportedModules;
+exports.loadCodeLine = loadCodeLine;
+exports.loadCodeLines = loadCodeLines;
+const fs = __importStar(require("fs"));
+const acorn = __importStar(require("acorn"));
+const walk = __importStar(require("acorn-walk"));
+/**
+ * 解析 JavaScript 文件为 AST
+ * @param filePath 文件路径
+ * @returns AST 对象
+ */
+function parseJavaScriptFile(filePath) {
+    if (!fs.existsSync(filePath)) {
+        throw new Error(`文件不存在: ${filePath}`);
+    }
+    const sourceCode = fs.readFileSync(filePath, 'utf-8');
+    try {
+        return acorn.parse(sourceCode, {
+            ecmaVersion: 2020,
+            sourceType: 'module'
+        });
+    }
+    catch (error) {
+        throw new Error(`解析文件失败: ${error.message}`);
+    }
+}
+/**
+ * 从 AST 中提取导出的函数
+ * @param ast AST 对象
+ * @returns 导出函数名称数组
+ */
+function extractExportedFunctions(filePath) {
+    const ast = typeof filePath === 'string' ? parseJavaScriptFile(filePath) : filePath;
+    const exportedFunctions = [];
+    walk.simple(ast, {
+        // 处理命名函数导出
+        ExportNamedDeclaration(node) {
+            if (node.declaration && node.declaration.type === 'FunctionDeclaration') {
+                exportedFunctions.push(node.declaration.id.name);
+            }
+            // 处理变量导出 (可能是函数)
+            if (node.specifiers) {
+                node.specifiers.forEach((specifier) => {
+                    if (specifier.exported.type === 'Identifier') {
+                        exportedFunctions.push(specifier.exported.name);
+                    }
+                });
+            }
+        },
+        // 处理默认导出的函数
+        ExportDefaultDeclaration(node) {
+            if (node.declaration.type === 'FunctionDeclaration') {
+                exportedFunctions.push(node.declaration.id?.name || 'default');
+            }
+            else if (node.declaration.type === 'Identifier') {
+                exportedFunctions.push(`default (${node.declaration.name})`);
+            }
+        }
+    });
+    return exportedFunctions;
+}
+/**
+ * 从 AST 中提取导出的类
+ * @param ast AST 对象
+ * @returns 导出类名称数组
+ */
+function extractExportedClasses(filePath) {
+    const ast = typeof filePath === 'string' ? parseJavaScriptFile(filePath) : filePath;
+    const exportedClasses = [];
+    walk.simple(ast, {
+        // 处理命名类导出
+        ExportNamedDeclaration(node) {
+            if (node.declaration && node.declaration.type === 'ClassDeclaration') {
+                exportedClasses.push(node.declaration.id.name);
+            }
+            // 处理变量导出 (可能是类)
+            if (node.specifiers) {
+                node.specifiers.forEach((specifier) => {
+                    if (specifier.exported.type === 'Identifier') {
+                        exportedClasses.push(specifier.exported.name);
+                    }
+                });
+            }
+        },
+        // 处理默认导出的类
+        ExportDefaultDeclaration(node) {
+            if (node.declaration.type === 'ClassDeclaration') {
+                exportedClasses.push(node.declaration.id?.name || 'default');
+            }
+            else if (node.declaration.type === 'Identifier') {
+                exportedClasses.push(`default (${node.declaration.name})`);
+            }
+        }
+    });
+    return exportedClasses;
+}
+/**
+ * 从 AST 中提取导入的模块
+ * @param ast AST 对象
+ * @returns 导入模块路径数组
+ */
+function extractImportedModules(filePath) {
+    const ast = typeof filePath === 'string' ? parseJavaScriptFile(filePath) : filePath;
+    const importedModules = [];
+    walk.simple(ast, {
+        ImportDeclaration(node) {
+            importedModules.push(node.source.value);
+        }
+    });
+    return importedModules;
+}
+/**
+ * 从文件中加载指定行的代码
+ * @param filePath 文件路径
+ * @param lineNumber 行号 (从1开始)
+ * @returns 指定行的代码
+ */
+function loadCodeLine(filePath, lineNumber) {
+    if (!fs.existsSync(filePath)) {
+        throw new Error(`文件不存在: ${filePath}`);
+    }
+    const lines = fs.readFileSync(filePath, 'utf-8').split('\n');
+    if (lineNumber < 1 || lineNumber > lines.length) {
+        throw new Error(`行号超出范围 (1-${lines.length})`);
+    }
+    return lines[lineNumber - 1];
+}
+/**
+ * 从文件中加载指定范围的代码行
+ * @param filePath 文件路径
+ * @param startLine 起始行号 (从1开始)
+ * @param endLine 结束行号 (从1开始)
+ * @returns 指定范围的代码
+ */
+function loadCodeLines(filePath, startLine, endLine) {
+    if (!fs.existsSync(filePath)) {
+        throw new Error(`文件不存在: ${filePath}`);
+    }
+    const lines = fs.readFileSync(filePath, 'utf-8').split('\n');
+    if (startLine < 1 || endLine > lines.length || startLine > endLine) {
+        throw new Error(`行号范围无效 (1-${lines.length})`);
+    }
+    return lines.slice(startLine - 1, endLine).join('\n');
+}
+//# sourceMappingURL=javascript.js.map