ai-worktool 1.0.7

package/dist/agents/index.js

@@ -0,0 +1,73 @@
+"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;
+    };
+})();
+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.setConversationId = void 0;
+exports.getPlatform = getPlatform;
+exports.setPlatform = setPlatform;
+exports.startAgent = startAgent;
+const jianguoke = __importStar(require("./jianguoke"));
+const chanjet = __importStar(require("./chanjet"));
+__exportStar(require("./prompt"), exports);
+const platforms = {
+    chanjet,
+    jianguoke
+};
+let currentPlatform;
+function getPlatform() {
+    return platforms[currentPlatform || 'jianguoke'];
+}
+function setPlatform(target) {
+    currentPlatform = target;
+}
+const setConversationId = (id) => {
+    const platform = getPlatform();
+    if (!platform) {
+        throw new Error(`智能体 "${currentPlatform}" 不支持`);
+    }
+    platform.setConversationId(id);
+};
+exports.setConversationId = setConversationId;
+async function startAgent(currentPlatform, name, question, vars, withConversation, ctrl, onMessage, onToolCall) {
+    const platform = platforms[currentPlatform] || getPlatform();
+    if (!platform) {
+        throw new Error(`智能体 "${currentPlatform}" 不支持`);
+    }
+    await platform.startAgent.apply(null, [name, question, vars, withConversation, ctrl, onMessage, onToolCall]);
+    console.log(`智能体 "${currentPlatform}" 调用完成`);
+}
+//# sourceMappingURL=index.js.map

package/dist/agents/jianguoke.js

@@ -0,0 +1,182 @@
+"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.setConversationId = void 0;
+exports.toolCallback = toolCallback;
+exports.startAgent = startAgent;
+const fetch_sse_1 = require("fetch-sse");
+const tools = __importStar(require("../tools"));
+const toolCall_1 = require("./toolCall");
+const user_1 = require("../user");
+const DEFAULT_SERVER = 'https://ai.jianguoke.cn';
+let conversationId = ''; // 会话ID
+const setConversationId = (id) => {
+    conversationId = id;
+};
+exports.setConversationId = setConversationId;
+async function postData(url, data) {
+    const token = (0, user_1.getTokenStorage)().get();
+    try {
+        // 发送POST请求
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: {
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                'Content-Type': 'application/json',
+                // eslint-disable-next-line @typescript-eslint/naming-convention
+                Authorization: token ? `Bearer ${token}` : ''
+            },
+            body: JSON.stringify(data) // 将数据转换为JSON字符串
+        });
+        // 检查响应状态
+        if (!response.ok) {
+            throw new Error(`HTTP error! status: ${response.status}`);
+        }
+        // 解析响应数据（根据实际响应格式选择解析方式）
+        const result = await response.text();
+        return {
+            status: response.status,
+            statusText: response.statusText,
+            data: result,
+            headers: response.headers
+        };
+    }
+    catch (error) {
+        console.error('请求失败:', error);
+        throw error; // 重新抛出错误以便调用者处理
+    }
+}
+async function toolCallback(toolName, callId, data) {
+    const baseURL = process.env.AI_SERVER || DEFAULT_SERVER;
+    await postData(`${baseURL}/api/tool/${toolName}`, { callId, data });
+}
+async function handleToolCall(toolCall) {
+    let data = null;
+    let message;
+    let success;
+    try {
+        const tool = tools[toolCall.toolName];
+        if (!tool) {
+            throw new Error(`Tool ${toolCall.toolName} not found`);
+        }
+        data = (await (0, toolCall_1.callWithObject)(tool, toolCall.params)) || '执行成功';
+        if (toolCall.toolName === 'readFile') {
+            data = `\n\`\`\`${(0, toolCall_1.inferCodeTypeFromExtension)(toolCall.params.filePath)}\n${data}\n\`\`\``;
+        }
+        success = true;
+    }
+    catch (err) {
+        success = false;
+        message = err.message || '执行失败';
+    }
+    await toolCallback(toolCall.toolName, toolCall.callId, data || message).catch(console.error);
+    return { data, message, success };
+}
+async function startAgent(name, question, vars, withConversation, ctrl, onMessage, onToolCall) {
+    let err = null;
+    let msg = '';
+    let lastMsgNo = 0;
+    const baseURL = process.env.AI_SERVER || DEFAULT_SERVER;
+    const token = (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',
+        signal: ctrl?.signal,
+        headers: {
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            Authorization: token ? `Bearer ${token}` : '',
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            'Content-Type': 'application/json',
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            'Cache-Control': 'no-cache',
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            Pragma: 'no-cache',
+            // 添加一个唯一的自定义头部，包含当前时间戳
+            // eslint-disable-next-line @typescript-eslint/naming-convention
+            'X-Custom-No-Cache': new Date().getTime().toString()
+        },
+        data: {
+            name,
+            vars,
+            question,
+            sid: withConversation ? conversationId : undefined,
+            tagEnabled: false
+        },
+        onError(error) {
+            err = error;
+        },
+        async onMessage(ev) {
+            const data = JSON.parse(ev.data);
+            // console.log(ev?.event, data);
+            if (ev?.event === 'MESSAGE') {
+                if (data.no !== lastMsgNo) {
+                    lastMsgNo = data.no;
+                    msg = '';
+                }
+                msg += data.content;
+                await onMessage?.(msg, 'msg-' + data.id + data.no);
+            }
+            if (ev?.event === 'TOOLCALL') {
+                const id = 'tool-' + data.callId;
+                await onMessage?.((0, toolCall_1.formatToolCallInfo)({
+                    // eslint-disable-next-line @typescript-eslint/naming-convention
+                    function_name: data.toolName,
+                    params: data.params
+                }, vars.projectRoot), id);
+                const result = await handleToolCall(data);
+                await onToolCall?.({
+                    functionName: data.toolName,
+                    params: data.params,
+                    result
+                });
+                await onMessage?.((0, toolCall_1.formatToolCallInfo)({
+                    // eslint-disable-next-line @typescript-eslint/naming-convention
+                    function_name: data.toolName,
+                    params: data.params
+                }, vars.projectRoot, result), id);
+            }
+            if (ev?.event === 'DONE') {
+                console.log(msg);
+            }
+            if (ev?.event === 'ERROR') {
+                err = data.message;
+            }
+        }
+    });
+    if (err) {
+        throw new Error(err);
+    }
+}
+//# sourceMappingURL=jianguoke.js.map

package/dist/agents/prompt.js

@@ -0,0 +1,41 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fixErrorPrompt = fixErrorPrompt;
+exports.fixCodePrompt = fixCodePrompt;
+exports.addUncoveredLinePrompt = addUncoveredLinePrompt;
+const path_1 = __importDefault(require("path"));
+const tools_1 = require("../tools");
+const MAXLEN = 10000;
+function fixErrorPrompt(error) {
+    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)}
+\`\`\``;
+}
+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)}
+\`\`\``;
+}
+//# sourceMappingURL=prompt.js.map

package/dist/agents/toolCall.js

@@ -0,0 +1,135 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.inferCodeTypeFromExtension = inferCodeTypeFromExtension;
+exports.callWithObject = callWithObject;
+exports.getParamNames = getParamNames;
+exports.formatToolCallInfo = formatToolCallInfo;
+const path_1 = __importDefault(require("path"));
+/**
+ * 根据文件路径的扩展名推断Markdown代码块应使用的类型名
+ * @param filePath - 文件路径
+ * @returns 代码块类型名，如果无法识别则返回空字符串
+ */
+function inferCodeTypeFromExtension(filePath) {
+    // 获取文件扩展名（包含点号）并转为小写
+    const extname = path_1.default.extname(filePath).toLowerCase();
+    // 移除点号，获取纯扩展名
+    const extension = extname.slice(1);
+    // 常见文件扩展名与Markdown代码块类型名的映射
+    const extensionMap = {
+        // 前端相关
+        html: 'html',
+        htm: 'html',
+        css: 'css',
+        scss: 'scss',
+        sass: 'sass',
+        less: 'less',
+        js: 'javascript',
+        jsx: 'jsx',
+        ts: 'typescript',
+        tsx: 'tsx',
+        // 后端相关
+        java: 'java',
+        py: 'python',
+        php: 'php',
+        rb: 'ruby',
+        go: 'go',
+        rs: 'rust',
+        c: 'c',
+        cpp: 'cpp',
+        cs: 'csharp',
+        swift: 'swift',
+        kt: 'kotlin',
+        // 脚本与配置
+        sh: 'bash',
+        bash: 'bash',
+        json: 'json',
+        yml: 'yaml',
+        yaml: 'yaml',
+        xml: 'xml',
+        md: 'markdown',
+        markdown: 'markdown',
+        // 数据库
+        sql: 'sql',
+        mysql: 'sql',
+        // 其他
+        txt: 'text',
+        diff: 'diff',
+        patch: 'diff'
+    };
+    // 返回对应的类型名，如果没有匹配则返回空字符串
+    return extensionMap[extension] || '';
+}
+/**
+ * 从对象中提取函数参数并调用函数
+ * @param fn 目标函数
+ * @param argsObj 包含参数的对象
+ * @returns 函数执行结果
+ */
+function callWithObject(fn, argsObj) {
+    // 获取函数参数名
+    const paramNames = getParamNames(fn);
+    // 从对象中提取参数值
+    const args = paramNames.map((name) => argsObj[name]);
+    // 调用函数
+    return fn(...args);
+}
+/**
+ * 获取函数参数名
+ * @param fn 目标函数
+ * @returns 参数名数组
+ */
+function getParamNames(fn) {
+    // 匹配函数参数列表的正则表达式
+    const paramRegex = /\(([^)]*)\)/;
+    // 匹配参数名的正则表达式（处理默认值等情况）
+    const nameRegex = /([^\s,=]+)/g;
+    // 获取函数源码
+    const fnStr = fn.toString();
+    // 提取参数列表字符串
+    const match = paramRegex.exec(fnStr);
+    if (!match) {
+        return [];
+    }
+    const params = match[1].trim();
+    if (!params) {
+        return [];
+    }
+    // 提取参数名（处理有默认值的情况）
+    const paramNames = [];
+    let nameMatch;
+    while ((nameMatch = nameRegex.exec(params)) !== null) {
+        paramNames.push(nameMatch[1]);
+    }
+    return paramNames;
+}
+function formatToolCallInfo(
+// eslint-disable-next-line @typescript-eslint/naming-convention
+toolCall, root, result) {
+    root = root || '/';
+    if (toolCall.function_name === 'readFile') {
+        const outputResult = result ? (result?.success ? ` ✅` : result?.message || '❌') : '';
+        return `读取文件：${path_1.default.relative(root, toolCall.params[0] || toolCall.params.filePath)}${outputResult}`;
+    }
+    if (toolCall.function_name === 'writeFile') {
+        const outputResult = result ? (result?.success ? ` ✅` : result?.message || '❌') : '';
+        return `修改文件：${path_1.default.relative(root, toolCall.params[0] || toolCall.params.filePath)}${outputResult}`;
+    }
+    if (toolCall.function_name === 'searchFilesByExtension') {
+        const outputResult = result
+            ? result?.success
+                ? ` => ${result.data
+                    .map((filePath) => path_1.default.relative(root, filePath))
+                    .join(', ')
+                    .slice(0, 50)} ✅`
+                : result?.message || '❌'
+            : '';
+        return `检索文件：${path_1.default.relative(root, toolCall.params[0] || toolCall.params.directory)}/*${toolCall.params[1] || toolCall.params.extensions}${outputResult}`;
+    }
+    const outputResult = result ? (result?.success ? ` ✅` : result?.message || '❌') : '';
+    return toolCall.function_name + outputResult;
+}
+//# sourceMappingURL=toolCall.js.map

package/dist/cli.js

@@ -0,0 +1,9 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const program_1 = __importDefault(require("./program"));
+program_1.default.parse(process.argv);
+//# sourceMappingURL=cli.js.map

package/dist/loging.js

@@ -0,0 +1,110 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LogLevel = void 0;
+exports.showLogs = showLogs;
+exports.setSpy = setSpy;
+exports.addLog = addLog;
+exports.updateLog = updateLog;
+exports.applyAspectToModule = applyAspectToModule;
+exports.spyConsole = spyConsole;
+const logs = [];
+const output = console.log;
+/**
+ * 日志级别枚举
+ */
+var LogLevel;
+(function (LogLevel) {
+    LogLevel["DEBUG"] = "debug";
+    LogLevel["INFO"] = "info";
+    LogLevel["WARN"] = "warn";
+    LogLevel["ERROR"] = "error";
+})(LogLevel || (exports.LogLevel = LogLevel = {}));
+/**
+ * 默认配置
+ */
+const defaultConfig = {
+    level: LogLevel.INFO,
+    logArgs: true,
+    logResult: true,
+    logExecutionTime: true,
+    logger: (txt) => addLog(txt)
+};
+/**
+ * 合并配置
+ */
+function mergeConfig(customConfig) {
+    return { ...defaultConfig, ...customConfig };
+}
+function showLogs() {
+    console.log(logs.join('\n'));
+    logs.length = 0;
+}
+let logSpy = undefined;
+function setSpy(fn) {
+    logSpy = fn;
+}
+function addLog(...logs) {
+    logs.push(...logs);
+    if (logs.length > 20) {
+        logs.splice(0, logs.length - 20);
+    }
+    for (const log of logs) {
+        logSpy?.(log, 'add');
+    }
+}
+function updateLog(log) {
+    logs[logs.length - 1] = log;
+    logSpy?.(log, 'update');
+}
+/**
+ * 为模块中的所有导出函数应用切面 - 用于普通函数模块
+ */
+function applyAspectToModule(moduleExports, config) {
+    const mergedConfig = mergeConfig(config);
+    for (const key in moduleExports) {
+        if (typeof moduleExports[key] === 'function') {
+            const originalFunction = moduleExports[key];
+            moduleExports[key] = async function (...args) {
+                const startTime = Date.now();
+                const functionName = key;
+                // 记录函数调用
+                if (mergedConfig.logArgs) {
+                    mergedConfig.logger?.(`${functionName}, 参数: ${JSON.stringify(args)}`);
+                }
+                else {
+                    mergedConfig.logger?.(`${functionName}`);
+                }
+                // 执行原函数
+                try {
+                    const result = await originalFunction.apply(this, args);
+                    // 处理同步返回
+                    const endTime = Date.now();
+                    const executionTime = endTime - startTime;
+                    if (mergedConfig.logResult) {
+                        mergedConfig.logger?.(`${functionName} 返回: ${JSON.stringify(result)} 执行时间: ${executionTime}ms`);
+                    }
+                    return result;
+                }
+                catch (error) {
+                    // 处理异常
+                    const endTime = Date.now();
+                    const executionTime = endTime - startTime;
+                    mergedConfig.logger?.(`${functionName} 抛出异常: ${error.message} 执行时间: ${executionTime}ms`);
+                    throw error;
+                }
+            };
+        }
+    }
+    return moduleExports;
+}
+function spyConsole(logCount = 1, format = 'compact' // 紧凑还是宽松
+) {
+    console.log = async (...txts) => {
+        addLog(...txts.map((txt) => (format === 'compact' && typeof txt === 'string' ? txt.replace(/\n/g, '\\n') : txt)));
+        if (logs.length > logCount) {
+            logs.shift();
+        }
+        logs.forEach((txts) => output(...txts));
+    };
+}
+//# sourceMappingURL=loging.js.map

package/dist/program.js

@@ -0,0 +1,35 @@
+#!/usr/bin/env node
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
+const path_1 = __importDefault(require("path"));
+const testAgent_1 = require("./testAgent");
+const tools_1 = require("./tools");
+const view_1 = require("./view");
+const program = new commander_1.Command();
+program.name('aicoder').description('你的AI编程搭子').version('1.0.0');
+program
+    .command('startTestAgent')
+    .description('启动单测智能体,根据源代码文件开始补全单测代码,直到覆盖率100%')
+    .option('-p, --platform <string>', '智能体平台', 'jianguoke')
+    .option('-r, --projectRoot <directory>', '源代码文件夹', (txt) => path_1.default.resolve(txt), process.cwd())
+    .option('-t, --testDir <string>', '测试用例文件夹', 'test')
+    .option('-s, --srcDir <string>', '测试用例文件夹', 'src')
+    .option('-c, --coverageDir <string>', '覆盖率报告文件夹', 'coverage')
+    .option('--autoCommit <boolean>', '覆盖率报告文件夹', (txt) => txt === 'true', true)
+    .option('--fullCoverageEnd <boolean>', '是否在达到100%覆盖率后结束', (txt) => txt === 'true', true)
+    .action((options) => (0, testAgent_1.startTestAgent)(options));
+program
+    .command('showCoverage')
+    .description('监控代码覆盖率报告')
+    .option('-r, --projectRoot <directory>', '源代码文件夹', (txt) => path_1.default.resolve(txt), process.cwd())
+    .option('-c, --coverageDir <string>', '覆盖率报告文件夹', 'coverage')
+    .action(async (options) => {
+    await (0, tools_1.runJestTests)(options.projectRoot);
+    await (0, view_1.showCoverageView)(path_1.default.join(options.projectRoot, options.coverageDir));
+});
+exports.default = program;
+//# sourceMappingURL=program.js.map