@ai-dev-tools/csharp-copilot-core 0.0.19

package/out/codebk/prompts/xap/xapGuideline-o.liquid

@@ -0,0 +1,45 @@
+These guideline ONLY apply when the source code is a XAP class. Ignore all these rules for non-XAP classes.
+### XAP code unit test Guideline
+- XAP class introduction:
+  - inherits from one of the following interfaces: IPlugin, IConditionPlugin, Workflow, IExperiment, IPluginDataStore, IAsyncPlugin.
+  - implements an Execute method.
+- Use ExecutionServices.CreateInstance<I{className}>() to create an instance of the XAP class, where the interface name is the class name prefixed with "I".
+  - For example, for a XAP class named SampleWorkflow, create the instance: **var sampleWorkflow = ExecutionServices.CreateInstance<ISampleWorkflow>()**.
+  - ExecutionServices.CreateInstance can only be used to create XAP class instances, MUST not be used to create non-XAP class instance. 
+- The parameters of the Execute method in XAP class instance is different with source code.
+  - The parameters with types: **PluginServices**, **PluginOutput<T>** or decorated with the **[ConfigFile("{configFileName}")]** attribute are deleted, try to mock these types is invalid and not allowed.
+  - For example, for a Execute definition of SampleWorkflow is: **PluginResult Execute(PluginServices ps, PluginOutput<Resp> resp, [ConfigFile("config.ini")] EplantConfig eplantConfig, OsResponse osResp = null, EplantRequest req = null)**, call the Execute with SampleWorkflow instance: **sampleWorkflow.Execute(Task.FromResult(osResp), Task.FromResult(req)).Result**.
+  - For the remaining parameters
+    - MUST ensure that the argument types in test code match the method's declared parameter types exactly.
+    - MUST keep the original order and be wrapped using Task.FromResult({value}).
+- The parameters of type **PluginOutput<T>** in Execute method must be treated as method outputs
+  - MUST access the value of type PluginOutput<T> using XAP class instance with the **parameter name**.
+  - For example, one XAP class named SampleWorkflow, its Execute method signature: **Execute(PluginServices pluginServices, PluginOutput<Resp> pluginResp, Request req = null)**, SampleWorkflow instance is sampleWorkflow，use **sampleWorkflow.pluginResult** can get the output value.
+  - MUST get the fields of type **T** of **PluginOutput<T>** using **Result** property.
+  - For example, there's one field named **Key** of **Resp**, for PluginOutput<Resp> pluginOutput, **MUST** get **Key** using: **sampleWorkflow.pluginResp.Result.Key**.
+  - **MUST** assert the value of type **PluginOutput<T>**, example:
+    ```
+    Assert.NotNull(sampleWorkflow.pluginResult.Result.Key);
+    Assert.AreEqual(sampleWorkflow.pluginResult.Result.Key,  "key");
+    ```
+- Bond stucture will be used in XAP class, treat these structures as regular C# classes.
+  - MUST import bond namespaces when using bond structures in test code.
+  - Bond structure can import other bond structures using namespaces, get dependent structures definition from user input by name and treat all as regular C# class.
+  - MUST use ExecutionServices.CreateInstance<{bond structure name}>() to create an instance of bond structure.
+  - Bond structure fields without **optional** modifier is **required** and MUST ensure that all required fields are assigned valid non-null values in test code.
+    - For following bond structure, after creating the instance, MUST assign valid non-null value to all fields except **WalletContent**
+    ```
+    struct EplantUser
+    {
+        1: uint64 Score;
+        2: int64 Created;
+        3: string Market;
+        4: vector<string> Trees;
+        5: EplantUserWeatherContent WeatherContent;
+        6: bool MergeNotifyWeather;
+        7: optional EplantUserWalletContent WalletContent;
+    }
+    ```
+    - One confusing example, for the required fields, there's still null checks in the source code, like **if (eplantUser?.Market == null)**. In this case, just test the case when eplantUser is null.
+- MUST take care type convertion
+  - To convert IList type to IEnumerable type, must use AsEnumerable(), e.g. IEnumerable osResponses = osResponseList.AsEnumerable()

package/out/codebk/testbuildkill.d.ts

@@ -0,0 +1 @@
+export {};

package/out/codebk/testbuildkill.js

@@ -0,0 +1,37 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const child_process_1 = require("child_process");
+const tree_kill_1 = __importDefault(require("tree-kill"));
+const msbuildCommand = `"C:\\Program Files\\Microsoft Visual Studio\\2022\\Enterprise\\MSBuild\\Current\\Bin\\amd64\\MSBuild.exe" "D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Tests\\SharedSegments.Tests.csproj" /t:Build /m /nr:true /v:minimal`;
+const child = (0, child_process_1.exec)(msbuildCommand);
+const pid = child.pid;
+// setTimeout(() => {
+//   console.log('Trying to kill...');
+// //   child.kill(); // 默认发送 SIGTERM（Windows 下为 TerminateProcess）
+// treeKill(pid,  (err) => {
+//             if (err) {
+//                 console.error(`Failed to kill pid=${pid}:`, err.message);
+//             } else {
+//                 console.log(`Killed pid=${pid}`);
+//             }
+//         });
+// }, 30000);
+// setTimeout(() => {
+//           console.log('exit...');
+//         }, 40000);
+process.on('SIGINT', async () => {
+    console.log('🛑 SIGINT received (Ctrl+C)');
+    (0, tree_kill_1.default)(pid, (err) => {
+        if (err) {
+            console.error(`Failed to kill pid=${pid}:`, err.message);
+        }
+        else {
+            console.log(`Killed pid=${pid}`);
+        }
+    });
+    // process.exit(0);
+});
+//# sourceMappingURL=testbuildkill.js.map

package/out/codebk/testbuildkill.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testbuildkill.js","sourceRoot":"","sources":["../../src/codebk/testbuildkill.ts"],"names":[],"mappings":";;;;;AAAA,iDAA4C;AAE5C,0DAAiC;AAEjC,MAAM,cAAc,GAAG,iQAAiQ,CAAC;AACzR,MAAM,KAAK,GAAG,IAAA,oBAAI,EAAC,cAAc,CAAC,CAAC;AACnC,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC;AACtB,qBAAqB;AACrB,sCAAsC;AACtC,kEAAkE;AAClE,4BAA4B;AAC5B,yBAAyB;AACzB,4EAA4E;AAC5E,uBAAuB;AACvB,oDAAoD;AACpD,gBAAgB;AAChB,cAAc;AAGd,aAAa;AAEb,qBAAqB;AACrB,oCAAoC;AACpC,qBAAqB;AAGrB,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;IAC5B,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC3C,IAAA,mBAAQ,EAAC,GAAG,EAAG,CAAC,GAAG,EAAE,EAAE;QACf,IAAI,GAAG,EAAE,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,sBAAsB,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7D,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,EAAE,CAAC,CAAC;QACrC,CAAC;IACL,CAAC,CAAC,CAAC;IACP,mBAAmB;AACvB,CAAC,CAAC,CAAC","sourcesContent":["import { exec, spawn } from 'child_process';\r\n\r\nimport treeKill from 'tree-kill';\r\n\r\nconst msbuildCommand = `\"C:\\\\Program Files\\\\Microsoft Visual Studio\\\\2022\\\\Enterprise\\\\MSBuild\\\\Current\\\\Bin\\\\amd64\\\\MSBuild.exe\" \"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Tests\\\\SharedSegments.Tests.csproj\" /t:Build /m /nr:true /v:minimal`;\r\nconst child = exec(msbuildCommand);\r\nconst pid = child.pid;\r\n// setTimeout(() => {\r\n//   console.log('Trying to kill...');\r\n// //   child.kill(); // 默认发送 SIGTERM（Windows 下为 TerminateProcess）\r\n// treeKill(pid,  (err) => {\r\n//             if (err) {\r\n//                 console.error(`Failed to kill pid=${pid}:`, err.message);\r\n//             } else {\r\n//                 console.log(`Killed pid=${pid}`);\r\n//             }\r\n//         });\r\n        \r\n        \r\n// }, 30000);\r\n\r\n// setTimeout(() => {\r\n//           console.log('exit...');\r\n//         }, 40000);\r\n\r\n\r\nprocess.on('SIGINT', async () => {\r\n    console.log('🛑 SIGINT received (Ctrl+C)');\r\n    treeKill(pid,  (err) => {\r\n            if (err) {\r\n                console.error(`Failed to kill pid=${pid}:`, err.message);\r\n            } else {\r\n                console.log(`Killed pid=${pid}`);\r\n            }\r\n        });\r\n    // process.exit(0);\r\n});\r\n"]}

package/out/command/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/out/command/index.js

@@ -0,0 +1,85 @@
+#!/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 utGenWrapper_1 = require("./utGenWrapper");
+const yargs_1 = __importDefault(require("yargs/yargs"));
+const helpers_1 = require("yargs/helpers");
+const setupProcessSignalHandler_1 = require("../exit/setupProcessSignalHandler");
+// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelper.cs', '--keepFailedTest', "1", "--af", "1"];
+// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\Conditions\\EplantRequestCategoryCondition.cs', '--keepFailedTest', "1", "--af", "1"];
+// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\ResponseBuilders\\EplantMigrateResponseBuilder.cs', '--keepFailedTest', "1", "--af", "1"];
+// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Overview\\HttpRequestParser\\Utils\\WeatherParamsHttpConverter.cs', '--keepFailedTest', "1", "--af", "1"];
+// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\RequestBuilders\\EplantRankRequestBuilder.cs', '--keepFailedTest', "1", "--af", "1"];
+function main() {
+    (0, yargs_1.default)((0, helpers_1.hideBin)(process.argv))
+        // Yargs(hideBin(hardcodedArgs))
+        .scriptName('xap-ut')
+        .option('source', {
+        describe: 'where the test copilot is running',
+        choices: ['local', 'vsplugin'],
+        default: 'local',
+        type: 'string',
+    })
+        .option('mode', {
+        alias: 'm',
+        describe: 'test copilot work mode.', // Note: only generate is supported now
+        choices: ['generate', 'repair'],
+        default: 'generate',
+        type: 'string',
+    })
+        .option('keepFailedTest', {
+        describe: 'whether to keep failed test cases',
+        choices: [0, 1], // 0 means delete failed test cases, 1 means keep failed test cases
+        default: 1,
+        type: 'number',
+    })
+        .option('enableCov', {
+        describe: 'whether to run coverage',
+        choices: [0, 1], // 0 means not run coverage, 1 means run coverage
+        default: 0,
+        type: 'number',
+    })
+        .option('LLMModel', {
+        describe: 'use which LLM model', // Note: only gpt4o is supported now
+        default: "gpt4o",
+        type: 'string',
+    })
+        .option('sourceCodePath', {
+        alias: 's',
+        describe: 'source code file path / project path or solution path for batch generation',
+        type: 'string',
+    })
+        .option('testCodePath', {
+        alias: 't',
+        describe: 'Target file or directory',
+        type: 'string',
+    })
+        .option('autofix', {
+        alias: 'af',
+        describe: 'whether to auto fix generated test code',
+        choices: [0, 1], // 0 means not auto fix, 1 means auto fix
+        default: 0,
+        type: 'number',
+    })
+        .command('utcmd', 'Generate unit test using command line', () => { }, (c) => {
+        console.log('Command executed with options:', c);
+        (0, utGenWrapper_1.generateUtCode)(c);
+    })
+        .command('utbenchmark', 'Generate unit test report for all project/solution files using command line', () => { }, (c) => {
+        console.log('Command executed with options:', c);
+        (0, utGenWrapper_1.generateBatchUtCode)(c);
+    })
+        .command('utvsplugin', 'Generate unit test by interact with Visual Studio plugin', () => { }, (c) => {
+        console.log('Command executed with options:', c);
+        (0, utGenWrapper_1.generateUtWithVsPlugin)(c);
+    })
+        .help()
+        .alias('help', 'h')
+        .parse();
+}
+(0, setupProcessSignalHandler_1.setupSignalHandlers)();
+main();
+//# sourceMappingURL=index.js.map

package/out/command/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/command/index.ts"],"names":[],"mappings":";;;;;;AAEA,iDAA6F;AAE7F,wDAAgC;AAChC,2CAAwC;AACxC,iFAAwE;AAExE,mPAAmP;AACnP,sPAAsP;AACtP,0PAA0P;AAC1P,kQAAkQ;AAClQ,qPAAqP;AACrP,SAAS,IAAI;IACT,IAAA,eAAK,EAAC,IAAA,iBAAO,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC5B,gCAAgC;SAC3B,UAAU,CAAC,QAAQ,CAAC;SACpB,MAAM,CACH,QAAQ,EACR;QACI,QAAQ,EAAE,mCAAmC;QAC7C,OAAO,EAAE,CAAC,OAAO,EAAE,UAAU,CAAC;QAC9B,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,MAAM,EACN;QACI,KAAK,EAAE,GAAG;QACV,QAAQ,EAAE,yBAAyB,EAAE,uCAAuC;QAC5E,OAAO,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC;QAC/B,OAAO,EAAE,UAAU;QACnB,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,gBAAgB,EAChB;QACI,QAAQ,EAAE,mCAAmC;QAC7C,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,mEAAmE;QACpF,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,WAAW,EACX;QACI,QAAQ,EAAE,yBAAyB;QACnC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,iDAAiD;QAClE,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,UAAU,EACV;QACI,QAAQ,EAAE,qBAAqB,EAAE,oCAAoC;QACrE,OAAO,EAAE,OAAO;QAChB,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,gBAAgB,EAChB;QACI,KAAK,EAAE,GAAG;QACV,QAAQ,EAAE,4EAA4E;QACtF,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,cAAc,EACd;QACI,KAAK,EAAE,GAAG;QACV,QAAQ,EAAE,0BAA0B;QACpC,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,MAAM,CACH,SAAS,EACT;QACI,KAAK,EAAE,IAAI;QACX,QAAQ,EAAE,yCAAyC;QACnD,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,yCAAyC;QAC1D,OAAO,EAAE,CAAC;QACV,IAAI,EAAE,QAAQ;KACjB,CACJ;SACA,OAAO,CACJ,OAAO,EACP,uCAAuC,EACvC,GAAG,EAAE,GAAE,CAAC,EACR,CAAC,CAAC,EAAE,EAAE;QACF,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACjD,IAAA,6BAAc,EAAC,CAAC,CAAC,CAAC;IACtB,CAAC,CACJ;SACA,OAAO,CACJ,aAAa,EACb,6EAA6E,EAC7E,GAAG,EAAE,GAAE,CAAC,EACR,CAAC,CAAC,EAAE,EAAE;QACF,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACjD,IAAA,kCAAmB,EAAC,CAAC,CAAC,CAAC;IAC3B,CAAC,CACJ;SACA,OAAO,CACJ,YAAY,EACZ,0DAA0D,EAC1D,GAAG,EAAE,GAAE,CAAC,EACR,CAAC,CAAC,EAAE,EAAE;QACF,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACjD,IAAA,qCAAsB,EAAC,CAAC,CAAC,CAAC;IAC9B,CAAC,CACJ;SACA,IAAI,EAAE;SACN,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC;SAClB,KAAK,EAAE,CAAC;AACjB,CAAC;AAED,IAAA,+CAAmB,GAAE,CAAC;AACtB,IAAI,EAAE,CAAC","sourcesContent":["#!/usr/bin/env node\r\n\r\nimport { generateBatchUtCode, generateUtCode, generateUtWithVsPlugin } from './utGenWrapper';\r\n\r\nimport Yargs from 'yargs/yargs';\r\nimport { hideBin } from 'yargs/helpers';\r\nimport { setupSignalHandlers } from '../exit/setupProcessSignalHandler';\r\n\r\n// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs', '--keepFailedTest', \"1\", \"--af\", \"1\"];\r\n// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\Conditions\\\\EplantRequestCategoryCondition.cs', '--keepFailedTest', \"1\", \"--af\", \"1\"];\r\n// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\ResponseBuilders\\\\EplantMigrateResponseBuilder.cs', '--keepFailedTest', \"1\", \"--af\", \"1\"];\r\n// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\WeatherParamsHttpConverter.cs', '--keepFailedTest', \"1\", \"--af\", \"1\"];\r\n// const hardcodedArgs = ['', '', 'utcmd', '--s', 'D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\RequestBuilders\\\\EplantRankRequestBuilder.cs', '--keepFailedTest', \"1\", \"--af\", \"1\"];\r\nfunction main() {\r\n    Yargs(hideBin(process.argv))\r\n    // Yargs(hideBin(hardcodedArgs))\r\n        .scriptName('xap-ut')\r\n        .option(\r\n            'source',\r\n            {\r\n                describe: 'where the test copilot is running',\r\n                choices: ['local', 'vsplugin'],\r\n                default: 'local',\r\n                type: 'string',\r\n            }\r\n        )\r\n        .option(\r\n            'mode',\r\n            {\r\n                alias: 'm',\r\n                describe: 'test copilot work mode.', // Note: only generate is supported now\r\n                choices: ['generate', 'repair'],\r\n                default: 'generate',\r\n                type: 'string',\r\n            }\r\n        )\r\n        .option(\r\n            'keepFailedTest',\r\n            {\r\n                describe: 'whether to keep failed test cases',\r\n                choices: [0, 1], // 0 means delete failed test cases, 1 means keep failed test cases\r\n                default: 1,\r\n                type: 'number',\r\n            }\r\n        )\r\n        .option(\r\n            'enableCov',\r\n            {\r\n                describe: 'whether to run coverage',\r\n                choices: [0, 1], // 0 means not run coverage, 1 means run coverage\r\n                default: 0,\r\n                type: 'number',\r\n            }\r\n        )\r\n        .option(\r\n            'LLMModel',\r\n            {\r\n                describe: 'use which LLM model', // Note: only gpt4o is supported now\r\n                default: \"gpt4o\",\r\n                type: 'string',\r\n            }\r\n        )\r\n        .option(\r\n            'sourceCodePath',\r\n            {\r\n                alias: 's',\r\n                describe: 'source code file path / project path or solution path for batch generation',\r\n                type: 'string',\r\n            }\r\n        )\r\n        .option(\r\n            'testCodePath',\r\n            {\r\n                alias: 't',\r\n                describe: 'Target file or directory',\r\n                type: 'string',\r\n            }\r\n        )\r\n        .option(\r\n            'autofix',\r\n            {\r\n                alias: 'af',\r\n                describe: 'whether to auto fix generated test code',\r\n                choices: [0, 1], // 0 means not auto fix, 1 means auto fix\r\n                default: 0,\r\n                type: 'number',\r\n            }\r\n        )\r\n        .command(\r\n            'utcmd',\r\n            'Generate unit test using command line',\r\n            () => {},\r\n            (c) => {\r\n                console.log('Command executed with options:', c);\r\n                generateUtCode(c);\r\n            }\r\n        ) \r\n        .command(\r\n            'utbenchmark',\r\n            'Generate unit test report for all project/solution files using command line',\r\n            () => {},\r\n            (c) => {\r\n                console.log('Command executed with options:', c);\r\n                generateBatchUtCode(c);\r\n            }\r\n        )\r\n        .command(\r\n            'utvsplugin',\r\n            'Generate unit test by interact with Visual Studio plugin',\r\n            () => {},\r\n            (c) => {\r\n                console.log('Command executed with options:', c);\r\n                generateUtWithVsPlugin(c);\r\n            }\r\n        )\r\n        .help()\r\n        .alias('help', 'h')\r\n        .parse();\r\n}\r\n\r\nsetupSignalHandlers();\r\nmain();"]}

package/out/command/utGenWrapper.d.ts

@@ -0,0 +1,3 @@
+export declare function generateUtCode(c: any): Promise<void>;
+export declare function generateBatchUtCode(c: any): Promise<void>;
+export declare function generateUtWithVsPlugin(c: any): Promise<void>;

package/out/command/utGenWrapper.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateUtCode = generateUtCode;
+exports.generateBatchUtCode = generateBatchUtCode;
+exports.generateUtWithVsPlugin = generateUtWithVsPlugin;
+const stdListener_1 = require("../vsPlugin/stdListener");
+const constants_1 = require("../types/constants");
+const generateCodeTests_1 = require("../batch/generateCodeTests");
+const csharpUtGen_1 = require("../gen/csharpUtGen");
+async function generateUtCode(c) {
+    const autofix = c.autofix === 1;
+    const keepFailedTest = c.keepFailedTest === 1;
+    await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSVSPLUGIN);
+}
+async function generateBatchUtCode(c) {
+    const autofix = c.autofix === 1;
+    const keepFailedTest = c.keepFailedTest === 1;
+    await (0, generateCodeTests_1.generateBatchTests)(c.sourceCodePath, autofix, keepFailedTest);
+}
+async function generateUtWithVsPlugin(c) {
+    try {
+        const autofix = c.autofix === 1;
+        if (autofix) {
+            (0, stdListener_1.startStdListener)();
+        }
+        const keepFailedTest = c.keepFailedTest === 1;
+        await (0, csharpUtGen_1.generateCsharpUtCode)(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, constants_1.CSVSPLUGIN, true);
+    }
+    catch (error) {
+        console.error("Error generateUtWithVsPlugin:", error);
+    }
+    finally {
+        // Ensure the listener is closed after processing
+        if (c.autofix === 1 && stdListener_1.readLineInterface) {
+            stdListener_1.readLineInterface.close();
+        }
+    }
+}
+// generateUtCode({sourceCodePath: "D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelper.cs", keepFailedTest: 1})
+//# sourceMappingURL=utGenWrapper.js.map

package/out/command/utGenWrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utGenWrapper.js","sourceRoot":"","sources":["../../src/command/utGenWrapper.ts"],"names":[],"mappings":";;AAMA,wCAIC;AAED,kDAIC;AAED,wDAgBC;AAlCD,yDAA8E;AAE9E,kDAAgD;AAChD,kEAAgE;AAChE,oDAA0D;AAEnD,KAAK,UAAU,cAAc,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,sBAAU,CAAC,CAAC;AACtG,CAAC;AAEM,KAAK,UAAU,mBAAmB,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;IAChC,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;IAC9C,MAAM,IAAA,sCAAkB,EAAC,CAAC,CAAC,cAAc,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;AACxE,CAAC;AAEM,KAAK,UAAU,sBAAsB,CAAC,CAAC;IAC1C,IAAI,CAAC;QACD,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,CAAC;YACV,IAAA,8BAAgB,GAAE,CAAC;QACvB,CAAC;QACD,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC;QAC9C,MAAM,IAAA,kCAAoB,EAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,sBAAU,EAAE,IAAI,CAAC,CAAC;IAC5G,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;IAC1D,CAAC;YAAS,CAAC;QACP,iDAAiD;QACjD,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,IAAI,+BAAiB,EAAE,CAAC;YACvC,+BAAiB,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;IACL,CAAC;AACL,CAAC;AAGD,iNAAiN","sourcesContent":["import { readLineInterface, startStdListener } from \"../vsPlugin/stdListener\";\r\n\r\nimport { CSVSPLUGIN } from \"../types/constants\";\r\nimport { generateBatchTests } from \"../batch/generateCodeTests\";\r\nimport { generateCsharpUtCode } from \"../gen/csharpUtGen\";\r\n\r\nexport async function generateUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSVSPLUGIN);\r\n}\r\n\r\nexport async function generateBatchUtCode(c) {\r\n    const autofix = c.autofix === 1;\r\n    const keepFailedTest = c.keepFailedTest === 1;\r\n    await generateBatchTests(c.sourceCodePath, autofix, keepFailedTest);\r\n}\r\n\r\nexport async function generateUtWithVsPlugin(c) {\r\n    try {\r\n        const autofix = c.autofix === 1;\r\n        if (autofix) {\r\n            startStdListener();\r\n        }\r\n        const keepFailedTest = c.keepFailedTest === 1;\r\n        await generateCsharpUtCode(c.sourceCodePath, c.testCodePath, autofix, keepFailedTest, CSVSPLUGIN, true);\r\n    } catch (error) {\r\n        console.error(\"Error generateUtWithVsPlugin:\", error);\r\n    } finally {\r\n        // Ensure the listener is closed after processing\r\n        if (c.autofix === 1 && readLineInterface) {\r\n            readLineInterface.close();\r\n        }\r\n    }\r\n}\r\n\r\n\r\n// generateUtCode({sourceCodePath: \"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", keepFailedTest: 1})"]}

package/out/exit/childProcessManager.d.ts

@@ -0,0 +1,2 @@
+export declare const childProcessSet: Set<number>;
+export declare function killAllChildProcesses(): Promise<void>;

package/out/exit/childProcessManager.js

@@ -0,0 +1,43 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.childProcessSet = void 0;
+exports.killAllChildProcesses = killAllChildProcesses;
+const tree_kill_1 = __importDefault(require("tree-kill"));
+exports.childProcessSet = new Set();
+function isProcessAlive(pid) {
+    try {
+        process.kill(pid, 0);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+function killPidAsync(pid) {
+    return new Promise(async (resolve) => {
+        if (!isProcessAlive(pid)) {
+            console.log(`PID ${pid} already exited`);
+            return resolve();
+        }
+        console.log(`Killing child process with PID: ${pid}`);
+        (0, tree_kill_1.default)(pid, "SIGKILL", (err) => {
+            if (err) {
+                console.error(`Failed to kill pid=${pid}:`, err.message);
+            }
+            else {
+                console.log(`Killed pid=${pid}`);
+            }
+            resolve();
+        });
+    });
+}
+async function killAllChildProcesses() {
+    const pids = Array.from(exports.childProcessSet);
+    console.log('Killing all child processes: ' + pids.join(', '));
+    await Promise.all(pids.map((pid) => killPidAsync(pid)));
+    console.log('All child processes have been killed.');
+}
+//# sourceMappingURL=childProcessManager.js.map

package/out/exit/childProcessManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"childProcessManager.js","sourceRoot":"","sources":["../../src/exit/childProcessManager.ts"],"names":[],"mappings":";;;;;;AAgCA,sDAKC;AArCD,0DAAiC;AAEpB,QAAA,eAAe,GAAG,IAAI,GAAG,EAAU,CAAC;AAEjD,SAAS,cAAc,CAAC,GAAW;IAC/B,IAAI,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACrB,OAAO,IAAI,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACL,OAAO,KAAK,CAAC;IACjB,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,GAAW;IAC7B,OAAO,IAAI,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;QACjC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CAAC,OAAO,GAAG,iBAAiB,CAAC,CAAC;YACzC,OAAO,OAAO,EAAE,CAAC;QACrB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,mCAAmC,GAAG,EAAE,CAAC,CAAC;QACtD,IAAA,mBAAQ,EAAC,GAAG,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;YAC7B,IAAI,GAAG,EAAE,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,sBAAsB,GAAG,GAAG,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7D,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,EAAE,CAAC,CAAC;YACrC,CAAC;YACD,OAAO,EAAE,CAAC;QACd,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC;AAEM,KAAK,UAAU,qBAAqB;IACvC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,uBAAe,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,+BAA+B,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC/D,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;AACzD,CAAC","sourcesContent":["import treeKill from \"tree-kill\";\r\n\r\nexport const childProcessSet = new Set<number>();\r\n\r\nfunction isProcessAlive(pid: number): boolean {\r\n    try {\r\n        process.kill(pid, 0);\r\n        return true;\r\n    } catch {\r\n        return false;\r\n    }\r\n}\r\n\r\nfunction killPidAsync(pid: number): Promise<void> {\r\n    return new Promise(async (resolve) => {\r\n        if (!isProcessAlive(pid)) {\r\n            console.log(`PID ${pid} already exited`);\r\n            return resolve();\r\n        }\r\n        \r\n        console.log(`Killing child process with PID: ${pid}`);\r\n        treeKill(pid, \"SIGKILL\", (err) => {\r\n            if (err) {\r\n                console.error(`Failed to kill pid=${pid}:`, err.message);\r\n            } else {\r\n                console.log(`Killed pid=${pid}`);\r\n            }\r\n            resolve();\r\n        });\r\n    });\r\n}\r\n\r\nexport async function killAllChildProcesses(): Promise<void> {\r\n    const pids = Array.from(childProcessSet);\r\n    console.log('Killing all child processes: ' + pids.join(', '));\r\n    await Promise.all(pids.map((pid) => killPidAsync(pid)));\r\n    console.log('All child processes have been killed.');\r\n}"]}

package/out/exit/setupProcessSignalHandler.d.ts

@@ -0,0 +1 @@
+export declare function setupSignalHandlers(): void;

package/out/exit/setupProcessSignalHandler.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setupSignalHandlers = setupSignalHandlers;
+const childProcessManager_1 = require("./childProcessManager");
+const stdListener_1 = require("../vsPlugin/stdListener");
+function setupSignalHandlers() {
+    process.on('SIGINT', async () => {
+        console.log('SIGINT received (Ctrl+C)');
+        await (0, childProcessManager_1.killAllChildProcesses)();
+        stdListener_1.readLineInterface?.close();
+        process.exit(0);
+    });
+    process.on('exit', () => {
+        childProcessManager_1.childProcessSet.clear();
+        console.log('Process exiting...');
+    });
+}
+//# sourceMappingURL=setupProcessSignalHandler.js.map

package/out/exit/setupProcessSignalHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setupProcessSignalHandler.js","sourceRoot":"","sources":["../../src/exit/setupProcessSignalHandler.ts"],"names":[],"mappings":";;AAIA,kDAYC;AAhBD,+DAA+E;AAE/E,yDAA4D;AAE5D,SAAgB,mBAAmB;IAC/B,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC5B,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACxC,MAAM,IAAA,2CAAqB,GAAE,CAAC;QAC9B,+BAAiB,EAAE,KAAK,EAAE,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,qCAAe,CAAC,KAAK,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACP,CAAC","sourcesContent":["import { childProcessSet, killAllChildProcesses } from \"./childProcessManager\";\r\n\r\nimport { readLineInterface } from \"../vsPlugin/stdListener\";\r\n\r\nexport function setupSignalHandlers() {\r\n    process.on('SIGINT', async () => {\r\n        console.log('SIGINT received (Ctrl+C)');\r\n        await killAllChildProcesses();\r\n        readLineInterface?.close();\r\n        process.exit(0);\r\n    });\r\n\r\n    process.on('exit', () => {\r\n        childProcessSet.clear();\r\n        console.log('Process exiting...');\r\n    });\r\n}"]}

package/out/gen/autoFix.d.ts

@@ -0,0 +1,4 @@
+import { GenResult } from "../types/genResult";
+export declare function autoFixTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, isXapCode: boolean, isVsPlugin: boolean, retryTimes?: number, isBenchmark?: boolean): Promise<GenResult>;
+export declare function generateAutoFixCode(prompts: any[], generatedCode: string, errorMessage: string, isBuildFailure: boolean, testFilePath: string, isXapCode: boolean): Promise<string>;
+export declare function autoRetryTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, isXapCode: boolean, retryTimes?: number): Promise<GenResult>;

package/out/gen/autoFix.js

@@ -0,0 +1,125 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.autoFixTestCode = autoFixTestCode;
+exports.generateAutoFixCode = generateAutoFixCode;
+exports.autoRetryTestCode = autoRetryTestCode;
+const checkCodeSyntax_1 = require("../analyze/checkCodeSyntax");
+const ensureValidLLMResponse_1 = require("./ensureValidLLMResponse");
+const postGenProcess_1 = require("./postGen/postGenProcess");
+const preparePrompt_1 = require("../llm/preparePrompt");
+const verifyBuildErrorCausedFile_1 = require("../utils/verifyBuildErrorCausedFile");
+const verifyTestCode_1 = require("../utils/verifyTestCode");
+const getVsPluginVerifyResult_1 = require("../vsPlugin/getVsPluginVerifyResult");
+const writeGenCode_1 = require("../utils/writeGenCode");
+const fileUtils_1 = require("../utils/fileUtils");
+async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, retryTimes = 3, isBenchmark = false) {
+    const startTime = Date.now();
+    let generatedCode = testCode;
+    let testResult;
+    let buildResult;
+    let errorMessage;
+    let isBuildFailure = true;
+    try {
+        for (let i = 0; i <= retryTimes; i++) {
+            // write generated code, to review and improve code quality
+            isBenchmark && (0, writeGenCode_1.writeGenCode)(sourceCodePath, generatedCode, i);
+            // for case: generatedCode is invalid, just retry instead of auto fix process
+            if (!generatedCode) {
+                console.error(`Generated code is invalid, try to regenerate. \nGenerated code: \n${generatedCode}`);
+                generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompts, isXapCode);
+                // extract valid code from LLM response
+                generatedCode = (0, postGenProcess_1.postGenProcess)(generatedCode, isXapCode);
+                continue;
+            }
+            // analyze generated code issue before verify: build, because build will cost much time.
+            const syntaxResult = await (0, checkCodeSyntax_1.checkCodeSyntax)(testFilePath);
+            if (syntaxResult.success) {
+                // if generated code passed the syntax check, verify the generated test code
+                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);
+                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);
+                errorMessage = verifyResult.message;
+                testResult = verifyResult.testResult;
+                buildResult = verifyResult.buildErrors;
+                isBuildFailure = !testResult;
+                // if build errors caused by other code files instead of generate test code, abort generation
+                const validBuildError = (0, verifyBuildErrorCausedFile_1.verifyBuildErrorCausedFile)(testFilePath, buildResult);
+                if (!validBuildError) {
+                    console.error("Due to build errors caused by other code files, abort generation, please fix it first.");
+                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };
+                }
+                if (verifyResult.success) {
+                    console.log("XapUt code generated successfully");
+                    return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };
+                }
+                console.error("Failed to verifyTestCode, error:", errorMessage);
+            }
+            else {
+                buildResult = syntaxResult.buildErrors;
+                errorMessage = syntaxResult.message;
+                console.error("Failed to check the generated UT code syntax, error:", errorMessage);
+            }
+            // keep test failed tast cases in test file automatically
+            // if (!isBuildFailure && vsTestRes.passed !== 0) {
+            //     console.log("remove failed test cases in test file");
+            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);
+            //     return { success: true, testCode: generatedCode };
+            // }
+            if (i === retryTimes) {
+                console.error(`Failed to generate ut code after ${retryTimes} attempts.`);
+                break;
+            }
+            // case: vs plugin no response or msbuild/vstest failed.
+            if (!buildResult && !testResult) {
+                console.error("!!!Verify process failed, retrying...");
+                continue;
+            }
+            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode);
+        }
+        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);
+        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };
+    }
+    catch (error) {
+        console.error("Error during auto-fix process:", error);
+        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    }
+}
+async function verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin = false) {
+    if (isVsPlugin) {
+        const vsVerifyResult = await (0, getVsPluginVerifyResult_1.verifyTestCodeByVsPlugin)(testFilePath);
+        return vsVerifyResult;
+    }
+    else {
+        // verifyResult: { success, message, buildErrors, vsTestResult }
+        const verifyResult = await (0, verifyTestCode_1.verifyTestCode)(generatedCode, testProjectPath);
+        return verifyResult;
+    }
+}
+async function generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode) {
+    console.log("start auto-fix process...");
+    const autofixPrompt = (0, preparePrompt_1.prepareUtAutoFixPrompt)(prompts, generatedCode, errorMessage, isBuildFailure);
+    // get valid code from LLM response with retry
+    const newGeneratedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, isXapCode);
+    // if generate code is valid, write test code to test project
+    !!newGeneratedCode && (0, fileUtils_1.writeTestCode)(newGeneratedCode, testFilePath);
+    return newGeneratedCode;
+}
+// use for quick dev test
+async function autoRetryTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, retryTimes = 3) {
+    const startTime = Date.now();
+    let generatedCode = testCode;
+    let vsTestRes;
+    for (let i = 0; i <= retryTimes; i++) {
+        if (i === retryTimes) {
+            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);
+            break;
+        }
+        const autofixPrompt = prompts;
+        // get valid code from LLM response
+        generatedCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, isXapCode);
+        // write test code to test project
+        (0, fileUtils_1.writeTestCode)(generatedCode, testFilePath);
+    }
+    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);
+    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };
+}
+//# sourceMappingURL=autoFix.js.map

package/out/gen/autoFix.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"autoFix.js","sourceRoot":"","sources":["../../src/gen/autoFix.ts"],"names":[],"mappings":";;AAcA,0CA2FC;AAkBD,kDAkBC;AAID,8CA6BC;AA5KD,gEAA6D;AAC7D,qEAAkE;AAElE,6DAA0D;AAC1D,wDAA8D;AAE9D,oFAAiF;AACjF,4DAAyD;AACzD,iFAA+E;AAC/E,wDAAqD;AACrD,kDAAmD;AAE5C,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,SAAkB,EAClB,UAAmB,EACnB,aAAqB,CAAC,EACtB,cAAuB,KAAK;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,UAAU,CAAC;IACf,IAAI,WAAW,CAAC;IAChB,IAAI,YAAY,CAAC;IACjB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,2DAA2D;YAC3D,WAAW,IAAI,IAAA,2BAAY,EAAC,cAAc,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;YAE9D,6EAA6E;YAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qEAAqE,aAAa,EAAE,CAAC,CAAC;gBACpG,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,OAAO,EAAE,SAAS,CAAC,CAAC;gBACjE,uCAAuC;gBACvC,aAAa,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;gBACzD,SAAS;YACb,CAAC;YAED,wFAAwF;YACxF,MAAM,YAAY,GAAG,MAAM,IAAA,iCAAe,EAAC,YAAY,CAAC,CAAC;YACzD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACvB,4EAA4E;gBAC5E,gHAAgH;gBAChH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBACzG,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACrC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,cAAc,GAAG,CAAC,UAAU,CAAC;gBAE7B,6FAA6F;gBAC7F,MAAM,eAAe,GAAG,IAAA,uDAA0B,EAAC,YAAY,EAAE,WAAW,CAAC,CAAC;gBAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;oBACnB,OAAO,CAAC,KAAK,CAAC,wFAAwF,CAAC,CAAC;oBACxG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACpH,CAAC;gBAED,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;oBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACnH,CAAC;gBAED,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,YAAY,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,EAAE,YAAY,CAAC,CAAC;YACxF,CAAC;YAED,yDAAyD;YACzD,mDAAmD;YACnD,4DAA4D;YAC5D,gEAAgE;YAChE,yDAAyD;YACzD,IAAI;YAEJ,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,oCAAoC,UAAU,YAAY,CAAC,CAAC;gBAC1E,MAAM;YACV,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,SAAS;YACb,CAAC;YAED,aAAa,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;QAC7H,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QAExE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;IACpH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAEhI,CAAC;AACL,CAAC;AAED,KAAK,UAAU,mBAAmB,CAC9B,aAAqB,EACrB,eAAuB,EACvB,YAAoB,EACpB,aAAsB,KAAK;IAE3B,IAAI,UAAU,EAAE,CAAC;QACb,MAAM,cAAc,GAAG,MAAM,IAAA,kDAAwB,EAAC,YAAY,CAAC,CAAC;QACpE,OAAO,cAAc,CAAC;IAC1B,CAAC;SAAM,CAAC;QACJ,gEAAgE;QAChE,MAAM,YAAY,GAAG,MAAM,IAAA,+BAAc,EAAC,aAAa,EAAE,eAAe,CAAC,CAAC;QAC1E,OAAO,YAAY,CAAC;IACxB,CAAC;AACL,CAAC;AAEM,KAAK,UAAU,mBAAmB,CACrC,OAAc,EACd,aAAqB,EACrB,YAAoB,EACpB,cAAuB,EACvB,YAAoB,EACpB,SAAkB;IAElB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,IAAA,sCAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;IAEnG,8CAA8C;IAC9C,MAAM,gBAAgB,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IAEhF,6DAA6D;IAC7D,CAAC,CAAC,gBAAgB,IAAI,IAAA,yBAAa,EAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAGD,yBAAyB;AAClB,KAAK,UAAU,iBAAiB,CACnC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,SAAkB,EAClB,aAAqB,CAAC;IAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,SAAS,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,uCAAuC,UAAU,YAAY,CAAC,CAAC;YAC7E,MAAM;QACV,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC;QAC9B,mCAAmC;QACnC,aAAa,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACvE,kCAAkC;QAClC,IAAA,yBAAa,EAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAExE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAClJ,CAAC","sourcesContent":["import { GenResult } from \"../types/genResult\";\r\nimport { VerifyResult } from \"../types/verifyResult\";\r\nimport { checkCodeSyntax } from \"../analyze/checkCodeSyntax\";\r\nimport { ensureValidLLMResponse } from \"./ensureValidLLMResponse\";\r\nimport { parseSyntaxCheckErrors } from \"../utils/parseSyntaxCheckResult\";\r\nimport { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { prepareUtAutoFixPrompt } from \"../llm/preparePrompt\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\nimport { verifyBuildErrorCausedFile } from \"../utils/verifyBuildErrorCausedFile\";\r\nimport { verifyTestCode } from \"../utils/verifyTestCode\";\r\nimport { verifyTestCodeByVsPlugin } from \"../vsPlugin/getVsPluginVerifyResult\";\r\nimport { writeGenCode } from \"../utils/writeGenCode\";\r\nimport { writeTestCode } from \"../utils/fileUtils\";\r\n\r\nexport async function autoFixTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isXapCode: boolean,\r\n    isVsPlugin: boolean,\r\n    retryTimes: number = 3,\r\n    isBenchmark: boolean = false\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let testResult;\r\n    let buildResult;\r\n    let errorMessage;\r\n    let isBuildFailure = true;\r\n    try {\r\n        for (let i = 0; i <= retryTimes; i++) {\r\n            // write generated code, to review and improve code quality\r\n            isBenchmark && writeGenCode(sourceCodePath, generatedCode, i);\r\n\r\n            // for case: generatedCode is invalid, just retry instead of auto fix process\r\n            if (!generatedCode) {\r\n                console.error(`Generated code is invalid, try to regenerate. \\nGenerated code: \\n${generatedCode}`);\r\n                generatedCode = await ensureValidLLMResponse(prompts, isXapCode);\r\n                // extract valid code from LLM response\r\n                generatedCode = postGenProcess(generatedCode, isXapCode);\r\n                continue;\r\n            }\r\n\r\n            // analyze generated code issue before verify: build, because build will cost much time.\r\n            const syntaxResult = await checkCodeSyntax(testFilePath);\r\n            if (syntaxResult.success) {\r\n                // if generated code passed the syntax check, verify the generated test code\r\n                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);\r\n                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);\r\n                errorMessage = verifyResult.message;\r\n                testResult = verifyResult.testResult;\r\n                buildResult = verifyResult.buildErrors;\r\n                isBuildFailure = !testResult;\r\n                \r\n                // if build errors caused by other code files instead of generate test code, abort generation\r\n                const validBuildError = verifyBuildErrorCausedFile(testFilePath, buildResult);\r\n                if (!validBuildError) {\r\n                    console.error(\"Due to build errors caused by other code files, abort generation, please fix it first.\");\r\n                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n                \r\n                if (verifyResult.success) {\r\n                    console.log(\"XapUt code generated successfully\");\r\n                    return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n\r\n                console.error(\"Failed to verifyTestCode, error:\", errorMessage);\r\n            } else {\r\n                buildResult = syntaxResult.buildErrors;\r\n                errorMessage = syntaxResult.message;\r\n                console.error(\"Failed to check the generated UT code syntax, error:\", errorMessage);\r\n            }\r\n\r\n            // keep test failed tast cases in test file automatically\r\n            // if (!isBuildFailure && vsTestRes.passed !== 0) {\r\n            //     console.log(\"remove failed test cases in test file\");\r\n            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);\r\n            //     return { success: true, testCode: generatedCode };\r\n            // }\r\n\r\n            if (i === retryTimes) {\r\n                console.error(`Failed to generate ut code after ${retryTimes} attempts.`);\r\n                break;\r\n            }\r\n\r\n            // case: vs plugin no response or msbuild/vstest failed.\r\n            if (!buildResult && !testResult) {\r\n                console.error(\"!!!Verify process failed, retrying...\");\r\n                continue;\r\n            }\r\n\r\n            generatedCode = await generateAutoFixCode(prompts, generatedCode, errorMessage, isBuildFailure, testFilePath, isXapCode);\r\n        }\r\n\r\n        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n    } catch (error) {\r\n        console.error(\"Error during auto-fix process:\", error);\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n\r\n    }\r\n}\r\n\r\nasync function verifyGeneratedCode(\r\n    generatedCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isVsPlugin: boolean = false\r\n): Promise<VerifyResult> {\r\n    if (isVsPlugin) {\r\n        const vsVerifyResult = await verifyTestCodeByVsPlugin(testFilePath);\r\n        return vsVerifyResult;\r\n    } else {\r\n        // verifyResult: { success, message, buildErrors, vsTestResult }\r\n        const verifyResult = await verifyTestCode(generatedCode, testProjectPath);\r\n        return verifyResult;\r\n    }\r\n}\r\n\r\nexport async function generateAutoFixCode(\r\n    prompts: any[],\r\n    generatedCode: string,\r\n    errorMessage: string,\r\n    isBuildFailure: boolean,\r\n    testFilePath: string,\r\n    isXapCode: boolean\r\n): Promise<string> {\r\n    console.log(\"start auto-fix process...\");\r\n    const autofixPrompt = prepareUtAutoFixPrompt(prompts, generatedCode, errorMessage, isBuildFailure);\r\n\r\n    // get valid code from LLM response with retry\r\n    const newGeneratedCode = await ensureValidLLMResponse(autofixPrompt, isXapCode);\r\n\r\n    // if generate code is valid, write test code to test project\r\n    !!newGeneratedCode && writeTestCode(newGeneratedCode, testFilePath);\r\n\r\n    return newGeneratedCode;\r\n}\r\n\r\n\r\n// use for quick dev test\r\nexport async function autoRetryTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    isXapCode: boolean,\r\n    retryTimes: number = 3\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let vsTestRes;\r\n    for (let i = 0; i <= retryTimes; i++) {\r\n        if (i === retryTimes) {\r\n            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);\r\n            break;\r\n        }\r\n\r\n        const autofixPrompt = prompts;\r\n        // get valid code from LLM response\r\n        generatedCode = await ensureValidLLMResponse(autofixPrompt, isXapCode);\r\n        // write test code to test project\r\n        writeTestCode(generatedCode, testFilePath);\r\n    }\r\n\r\n    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };\r\n}"]}

package/out/gen/csharpUtGen.d.ts

@@ -0,0 +1,2 @@
+import { GenResult } from '../types/genResult';
+export declare function generateCsharpUtCode(sourceCodePath: string, testCodePath: string, enableAutoFix: boolean, keepFailedTests: boolean, source?: string, isBenchmark?: boolean): Promise<GenResult>;