@comate/zulu 0.0.9 → 0.0.11

package/comate-engine/plugins/bfc/dist/index.js

@@ -0,0 +1,452 @@
+import { SkillProvider, StringChunkStream, ElementChunkStream } from '@comate/plugin-host';
+import { readFileSync } from 'fs';
+import { fileURLToPath } from 'node:url';
+import { dirname } from 'path';
+
+class HelpSkillProvider extends SkillProvider {
+    static{
+        this.skillName = 'bfc-help';
+    }
+    static{
+        this.description = '插件介绍';
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {}
+        };
+    }
+    async *execute() {
+        const stream = new StringChunkStream();
+        const filename = fileURLToPath(import.meta.url);
+        const content = readFileSync(`${dirname(filename)}/help.md`, 'utf8');
+        yield stream.flush(content);
+    }
+}
+
+const AI_SERVER_HOST = 'http://10.11.87.193:8710';
+const abGitcaseId = '13882351';
+const caseGitcaseId = '13794791';
+const instructModeId = '13793177';
+
+var ResponseCode = /*#__PURE__*/ function(ResponseCode) {
+    ResponseCode[ResponseCode["SUCCESS"] = 200] = "SUCCESS";
+    ResponseCode[ResponseCode["FAILURE"] = 300] = "FAILURE";
+    return ResponseCode;
+}({});
+async function generateService(_this, params) {
+    try {
+        const response = await fetch(`${AI_SERVER_HOST}/mockAPICaseGenerator`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({
+                ...params
+            }),
+            signal: AbortSignal.timeout(300000)
+        });
+        return await response.json();
+    } catch (ex) {
+        if (ex.name === 'TimeoutError') {
+            return {
+                code: 300,
+                message: '网络请求超时。'
+            };
+        }
+        return {
+            code: 300,
+            message: `执行失败 ${ex instanceof Error ? `: ${ex.message}` : ''}`
+        };
+    }
+}
+
+class GenNewCaseSkillProvider extends SkillProvider {
+    static{
+        this.skillName = 'generateBfcCode';
+    }
+    static{
+        this.displayName = 'bfc用例生成';
+    }
+    static{
+        this.description = 'bfc用例生成';
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {}
+        };
+    }
+    async *execute() {
+        const stream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield stream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield stream.flushReplaceLast('请输入需求描述。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield stream.flushReplaceLast('请确认打开的文件为用例文件。');
+                return;
+            }
+            const userName = userDetail.name;
+            const realQuery = `修改点：${query}`;
+            yield stream.flush('开始生成服务代码...\n\n');
+            yield stream.flush('\n\n正在生成服务代码...\n\n');
+            const retServerCode = await generateService(this, {
+                function_name: 'gen_case_by_demo',
+                user_demand: realQuery,
+                username: userName,
+                gitcase_ids: [
+                    caseGitcaseId
+                ],
+                case_path: filePath,
+                case_demo: codeContent
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(`生成服务代码出错：${msg}，错误码：${code}\n\n`);
+                return;
+            }
+            if (Array.isArray(data[caseGitcaseId]) && data[caseGitcaseId].length > 0) {
+                const case_data = data[caseGitcaseId];
+                const lens = case_data.length;
+                for(let i = 0; i < lens; i++){
+                    const content = case_data[i].content;
+                    if (case_data[i].filename && case_data[i].filename !== '') {
+                        yield stream.flush(`- 参考文件名：${case_data[i].filename}\n\n`);
+                    }
+                    yield stream.flush([
+                        '```python',
+                        content,
+                        '```',
+                        '\n\n'
+                    ].join('\n'));
+                }
+            } else {
+                yield stream.flush(`未生成服务代码，请联系负责人或稍后再试。${data}\n\n`);
+            }
+        } catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield stream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}
+
+class GenAbnormalCaseSkillProvider extends SkillProvider {
+    static{
+        this.skillName = 'generateAbnormalCode';
+    }
+    static{
+        this.displayName = '异常用例生成';
+    }
+    static{
+        this.description = '异常用例生成';
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {}
+        };
+    }
+    async *execute() {
+        const stream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield stream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield stream.flushReplaceLast('请输入需求描述。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield stream.flushReplaceLast('请确认打开的文件为用例文件。');
+                return;
+            }
+            const userName = userDetail.name;
+            yield stream.flush('异常用例生成请求已发送...\n\n');
+            yield stream.flush('异常用例正在生成中...\n\n');
+            const retServerCode = await generateService(this, {
+                function_name: 'gen_abnormal_case',
+                user_demand: query,
+                username: userName,
+                gitcase_ids: [
+                    abGitcaseId
+                ],
+                case_path: filePath,
+                case_demo: codeContent
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(`生成服务代码出错：${msg}，错误码：${code}\n\n`);
+                return;
+            }
+            if (Array.isArray(data[abGitcaseId]) && data[abGitcaseId].length > 0) {
+                const case_data = data[abGitcaseId];
+                const lens = case_data.length;
+                for(let i = 0; i < lens; i++){
+                    const content = case_data[i].content;
+                    yield stream.flush(`- 异常用例文件名：${case_data[i].filename}\n\n`);
+                    yield stream.flush([
+                        '```python',
+                        content,
+                        '```',
+                        '\n\n'
+                    ].join('\n'));
+                }
+            } else {
+                yield stream.flush(`未生成服务代码，请联系负责人或稍后再试。${data}\n\n`);
+            }
+        } catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield stream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}
+
+class InstructModeSkillProvider extends SkillProvider {
+    static{
+        this.skillName = 'bfcInstructMode';
+    }
+    static{
+        this.displayName = 'bfc用例指令';
+    }
+    static{
+        this.description = 'bfc用例指令';
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {}
+        };
+    }
+    async *execute() {
+        const stream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield stream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield stream.flushReplaceLast('请输入需求描述。');
+                return;
+            }
+            if (!query.includes('：')) {
+                yield stream.flushReplaceLast('请使用中文冒号分割业务线和指令内容，eg： 打车：续写协议一致性。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield stream.flushReplaceLast('请确认打开的文件为用例文件。');
+                return;
+            }
+            const userName = userDetail.name;
+            yield stream.flush('开始生成服务代码...\n\n');
+            yield stream.flush('\n\n正在生成服务代码...\n\n');
+            const retServerCode = await generateService(this, {
+                function_name: 'gen_case_by_instruct',
+                user_demand: query,
+                username: userName,
+                gitcase_ids: [
+                    instructModeId
+                ],
+                case_path: filePath,
+                case_demo: codeContent
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(`生成服务代码出错：${msg}，错误码：${code}\n\n`);
+                return;
+            }
+            if (Array.isArray(data[instructModeId]) && data[instructModeId].length > 0) {
+                const case_data = data[instructModeId];
+                const lens = case_data.length;
+                for(let i = 0; i < lens; i++){
+                    const content = case_data[i].content;
+                    if (case_data[i].filename && case_data[i].filename !== '') {
+                        yield stream.flush(`- 参考文件名：${case_data[i].filename}\n\n`);
+                    }
+                    yield stream.flush([
+                        '```python',
+                        content,
+                        '```',
+                        '\n\n'
+                    ].join('\n'));
+                }
+            } else {
+                yield stream.flush(`未生成服务代码，请联系负责人或稍后再试。${data}\n\n`);
+            }
+        } catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield stream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}
+
+function jsx(type, props) {
+    // 按照React的逻辑，`boolean`、`null`、`undefined`是不会显示出来的
+    const children = Array.isArray(props.children) ? props.children.filter((v)=>v != null && typeof v !== 'boolean') : props.children;
+    return {
+        type,
+        ...props,
+        children
+    };
+}
+
+class GetBatchCases extends SkillProvider {
+    static{
+        this.skillName = 'getBatchCases';
+    }
+    static{
+        this.displayName = '获取自动化用例列表';
+    }
+    static{
+        this.description = '获取自动化用例列表';
+    }
+    constructor(init){
+        super(init);
+        this.cwd = init.cwd;
+    }
+    static{
+        this.parameters = {
+            type: 'object',
+            properties: {},
+            required: []
+        };
+    }
+    async *execute() {
+        const stream = new ElementChunkStream();
+        const strStream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield strStream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield strStream.flushReplaceLast('请输入 gitcase 项目名称。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield strStream.flushReplaceLast('请确认当前打开的文件为自动化用例文件，采纳文件将与该文件保存至同一目录下。');
+                return;
+            }
+            const userName = userDetail.name;
+            const retServerCode = await generateService(this, {
+                function_name: 'get_batch_cases',
+                user_demand: query,
+                username: userName,
+                gitcase_ids: [
+                    "0"
+                ],
+                case_path: this.cwd + '/' + filePath,
+                case_demo: ""
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(/*#__PURE__*/ jsx("p", {
+                    children: [
+                        "查询生成用例失败，错误信息：",
+                        msg
+                    ]
+                }));
+                return;
+            }
+            yield stream.flush(/*#__PURE__*/ jsx("table", {
+                children: [
+                    /*#__PURE__*/ jsx("thead", {
+                        children: /*#__PURE__*/ jsx("tr", {
+                            children: [
+                                /*#__PURE__*/ jsx("th", {
+                                    children: "GitCase ID"
+                                }),
+                                /*#__PURE__*/ jsx("th", {
+                                    children: "描述"
+                                }),
+                                /*#__PURE__*/ jsx("th", {
+                                    children: "用例文件名"
+                                }),
+                                /*#__PURE__*/ jsx("th", {
+                                    children: /*#__PURE__*/ jsx("flex", {
+                                        horizontalCentered: true,
+                                        children: "操作"
+                                    })
+                                })
+                            ]
+                        })
+                    }),
+                    /*#__PURE__*/ jsx("tbody", {
+                        children: (()=>{
+                            const rows = [];
+                            for(let i = 0; i < data.length; i++){
+                                const item = data[i];
+                                rows.push(/*#__PURE__*/ jsx("tr", {
+                                    children: [
+                                        /*#__PURE__*/ jsx("td", {
+                                            children: item.gitcase_id
+                                        }),
+                                        /*#__PURE__*/ jsx("td", {
+                                            children: item.case_desc
+                                        }),
+                                        /*#__PURE__*/ jsx("td", {
+                                            children: item.case_name
+                                        }),
+                                        /*#__PURE__*/ jsx("td", {
+                                            children: /*#__PURE__*/ jsx("flex", {
+                                                horizontalCentered: true,
+                                                children: /*#__PURE__*/ jsx("command-button", {
+                                                    commandName: "commitMessage:confirmIssue",
+                                                    data: "yes",
+                                                    replyText: "单独Button replyText",
+                                                    // variant="text"
+                                                    action: "insertToFile",
+                                                    actionData: {
+                                                        filePath: item.case_path,
+                                                        position: {
+                                                            line: 1,
+                                                            character: 0
+                                                        },
+                                                        newText: item.case_content
+                                                    },
+                                                    children: "采纳"
+                                                })
+                                            })
+                                        })
+                                    ]
+                                }));
+                            }
+                            return rows;
+                        })()
+                    })
+                ]
+            }));
+        } catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield strStream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}
+
+function setup({ registry }) {
+    registry.registerSkillProvider('bfc-help', HelpSkillProvider);
+    registry.registerSkillProvider('genBfcCode', GenNewCaseSkillProvider);
+    registry.registerSkillProvider('genAbnormalCode', GenAbnormalCaseSkillProvider);
+    registry.registerSkillProvider('bfcInstructMode', InstructModeSkillProvider);
+    registry.registerSkillProvider('getBatchCases', GetBatchCases);
+}
+
+export { setup };

package/comate-engine/plugins/bfc/dist/providers/bfcInstructMode.d.ts

@@ -0,0 +1,8 @@
+import { SkillProvider, FunctionParameterDefinition, TaskProgressChunk } from '@comate/plugin-host';
+export declare class InstructModeSkillProvider extends SkillProvider {
+    static skillName: string;
+    static displayName: string;
+    static description: string;
+    static parameters: FunctionParameterDefinition;
+    execute(): AsyncIterableIterator<TaskProgressChunk>;
+}

package/comate-engine/plugins/bfc/dist/providers/bfcInstructMode.js

@@ -0,0 +1,77 @@
+/* eslint-disable camelcase, complexity, max-depth */
+import { SkillProvider, StringChunkStream, } from '@comate/plugin-host';
+import { generateService, ResponseCode } from '../service/generateService.js';
+import { instructModeId } from '../config/config.js';
+export class InstructModeSkillProvider extends SkillProvider {
+    static skillName = 'bfcInstructMode';
+    static displayName = 'bfc用例指令';
+    static description = 'bfc用例指令';
+    static parameters = {
+        type: 'object',
+        properties: {},
+    };
+    async *execute() {
+        const stream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield stream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield stream.flushReplaceLast('请输入需求描述。');
+                return;
+            }
+            if (!query.includes('：')) {
+                yield stream.flushReplaceLast('请使用中文冒号分割业务线和指令内容，eg： 打车：续写协议一致性。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield stream.flushReplaceLast('请确认打开的文件为用例文件。');
+                return;
+            }
+            const userName = userDetail.name;
+            yield stream.flush('开始生成服务代码...\n\n');
+            yield stream.flush('\n\n正在生成服务代码...\n\n');
+            const retServerCode = await generateService(this, {
+                function_name: 'gen_case_by_instruct',
+                user_demand: query,
+                username: userName,
+                gitcase_ids: [instructModeId],
+                case_path: filePath,
+                case_demo: codeContent,
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(`生成服务代码出错：${msg}，错误码：${code}\n\n`);
+                return;
+            }
+            if (Array.isArray(data[instructModeId]) && data[instructModeId].length > 0) {
+                const case_data = data[instructModeId];
+                const lens = case_data.length;
+                for (let i = 0; i < lens; i++) {
+                    const content = case_data[i].content;
+                    if (case_data[i].filename && case_data[i].filename !== '') {
+                        yield stream.flush(`- 参考文件名：${case_data[i].filename}\n\n`);
+                    }
+                    yield stream.flush([
+                        '```python',
+                        content,
+                        '```',
+                        '\n\n',
+                    ]
+                        .join('\n'));
+                }
+            }
+            else {
+                yield stream.flush(`未生成服务代码，请联系负责人或稍后再试。${data}\n\n`);
+            }
+        }
+        catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield stream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}

package/comate-engine/plugins/bfc/dist/providers/genAbnormalCode.d.ts

@@ -0,0 +1,8 @@
+import { SkillProvider, FunctionParameterDefinition, TaskProgressChunk } from '@comate/plugin-host';
+export declare class GenAbnormalCaseSkillProvider extends SkillProvider {
+    static skillName: string;
+    static displayName: string;
+    static description: string;
+    static parameters: FunctionParameterDefinition;
+    execute(): AsyncIterableIterator<TaskProgressChunk>;
+}

package/comate-engine/plugins/bfc/dist/providers/genAbnormalCode.js

@@ -0,0 +1,71 @@
+/* eslint-disable camelcase, complexity */
+import { SkillProvider, StringChunkStream, } from '@comate/plugin-host';
+import { generateService, ResponseCode } from '../service/generateService.js';
+import { abGitcaseId } from '../config/config.js';
+export class GenAbnormalCaseSkillProvider extends SkillProvider {
+    static skillName = 'generateAbnormalCode';
+    static displayName = '异常用例生成';
+    static description = '异常用例生成';
+    static parameters = {
+        type: 'object',
+        properties: {},
+    };
+    async *execute() {
+        const stream = new StringChunkStream();
+        try {
+            const userDetail = await this.currentUser.requestDetail();
+            if (!userDetail) {
+                yield stream.flushReplaceLast('请先完成授权');
+                return;
+            }
+            const { query, activeFilePath: filePath, activeFileContent: codeContent } = this.currentContext;
+            if (!query) {
+                yield stream.flushReplaceLast('请输入需求描述。');
+                return;
+            }
+            const case_full_name = filePath.split('/').pop() || '';
+            if (!filePath.endsWith('.py') || !case_full_name.startsWith('test_')) {
+                yield stream.flushReplaceLast('请确认打开的文件为用例文件。');
+                return;
+            }
+            const userName = userDetail.name;
+            yield stream.flush('异常用例生成请求已发送...\n\n');
+            yield stream.flush('异常用例正在生成中...\n\n');
+            const retServerCode = await generateService(this, {
+                function_name: 'gen_abnormal_case',
+                user_demand: query,
+                username: userName,
+                gitcase_ids: [abGitcaseId],
+                case_path: filePath,
+                case_demo: codeContent,
+            });
+            const { code, message: msg, data } = retServerCode;
+            if (code !== ResponseCode.SUCCESS || !data) {
+                yield stream.flush(`生成服务代码出错：${msg}，错误码：${code}\n\n`);
+                return;
+            }
+            if (Array.isArray(data[abGitcaseId]) && data[abGitcaseId].length > 0) {
+                const case_data = data[abGitcaseId];
+                const lens = case_data.length;
+                for (let i = 0; i < lens; i++) {
+                    const content = case_data[i].content;
+                    yield stream.flush(`- 异常用例文件名：${case_data[i].filename}\n\n`);
+                    yield stream.flush([
+                        '```python',
+                        content,
+                        '```',
+                        '\n\n',
+                    ]
+                        .join('\n'));
+                }
+            }
+            else {
+                yield stream.flush(`未生成服务代码，请联系负责人或稍后再试。${data}\n\n`);
+            }
+        }
+        catch (error) {
+            this.logger.error(`Error: ${error}`);
+            yield stream.flushReplaceLast(`请求出错，请联系负责人或稍后再试。${error}`);
+        }
+    }
+}

package/comate-engine/plugins/bfc/dist/providers/genBfcCode.d.ts

@@ -0,0 +1,8 @@
+import { SkillProvider, FunctionParameterDefinition, TaskProgressChunk } from '@comate/plugin-host';
+export declare class GenNewCaseSkillProvider extends SkillProvider {
+    static skillName: string;
+    static displayName: string;
+    static description: string;
+    static parameters: FunctionParameterDefinition;
+    execute(): AsyncIterableIterator<TaskProgressChunk>;
+}