@ai-dev-tools/csharp-copilot-core 0.0.31 → 0.0.32

package/out/gen/autoFix.js

@@ -0,0 +1,129 @@
+"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");
+const writeFailedVerifyResult_1 = require("../utils/writeFailedVerifyResult");
+const getCodeStructurePath_1 = require("../utils/getCodeStructurePath");
+async function autoFixTestCode(prompts, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, retryTimes = 3, isBenchmark = false) {
+    const startTime = Date.now();
+    let generatedCode = testCode;
+    let testResult;
+    let buildResult;
+    let errorMessage;
+    let isBuildFailure = true;
+    try {
+        const sourceCodeProjectPath = (0, getCodeStructurePath_1.getSourceFileCsprojFilePath)(sourceCodePath);
+        for (let i = 0; i <= retryTimes; i++) {
+            // write generated code, to review and improve code quality
+            isBenchmark && (0, writeGenCode_1.writeGenCode)(sourceCodePath, generatedCode, i, sourceCodeProjectPath);
+            // 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, testFramework, isXapCode);
+                // extract valid code from LLM response
+                generatedCode = (0, postGenProcess_1.postGenProcess)(generatedCode, testFramework, 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);
+            }
+            isBenchmark && (0, writeFailedVerifyResult_1.writeFailedVerifyResult)(sourceCodePath, i, sourceCodeProjectPath, errorMessage, !testResult);
+            // 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, testFramework, 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, testFramework, 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, testFramework, 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, testFramework, 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, testFramework, 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,0CA8FC;AAkBD,kDAmBC;AAID,8CA8BC;AAjLD,gEAA6D;AAC7D,qEAAkE;AAClE,6DAA0D;AAC1D,wDAA8D;AAC9D,oFAAiF;AACjF,4DAAyD;AACzD,iFAA+E;AAC/E,wDAAqD;AACrD,kDAAmD;AACnD,8EAA2E;AAC3E,wEAA4E;AAErE,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,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,MAAM,qBAAqB,GAAG,IAAA,kDAA2B,EAAC,cAAc,CAAC,CAAC;QAC1E,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,EAAE,qBAAqB,CAAC,CAAC;YAErF,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,aAAa,EAAE,SAAS,CAAC,CAAC;gBAChF,uCAAuC;gBACvC,aAAa,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;gBACxE,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,WAAW,IAAI,IAAA,iDAAuB,EAAC,cAAc,EAAE,CAAC,EAAE,qBAAqB,EAAE,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC;YAC5G,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,aAAa,EAAE,SAAS,CAAC,CAAC;QAC5I,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,aAAqB,EACrB,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,aAAa,EAAE,SAAS,CAAC,CAAC;IAE/F,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,aAAqB,EACrB,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,aAAa,EAAE,SAAS,CAAC,CAAC;QACtF,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 { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { prepareUtAutoFixPrompt } from \"../llm/preparePrompt\";\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\nimport { writeFailedVerifyResult } from \"../utils/writeFailedVerifyResult\";\r\nimport { getSourceFileCsprojFilePath } from \"../utils/getCodeStructurePath\";\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    testFramework: 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        const sourceCodeProjectPath = getSourceFileCsprojFilePath(sourceCodePath);\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, sourceCodeProjectPath);\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, testFramework, isXapCode);\r\n                // extract valid code from LLM response\r\n                generatedCode = postGenProcess(generatedCode, testFramework, 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            isBenchmark && writeFailedVerifyResult(sourceCodePath, i, sourceCodeProjectPath, errorMessage, !testResult);\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, testFramework, 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    testFramework: 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, testFramework, 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    testFramework: 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, testFramework, 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, nugetPackagePath?: string): Promise<GenResult>;

package/out/gen/csharpUtGen.js

@@ -0,0 +1,125 @@
+"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.generateCsharpUtCode = generateCsharpUtCode;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const constants_1 = require("../types/constants");
+const fileUtils_1 = require("../utils/fileUtils");
+const autoFix_1 = require("./autoFix");
+const checkXapCode_1 = require("../utils/checkXapCode");
+const detectTestFramework_1 = require("../utils/detectTestFramework");
+const ensureValidLLMResponse_1 = require("./ensureValidLLMResponse");
+const generateCodeDependency_1 = require("../analyze/generateCodeDependency");
+const getTestFile_1 = require("../utils/getTestFile");
+const installAnalyzerTool_1 = require("../analyze/installAnalyzerTool");
+const preGenProcess_1 = require("./preGen/preGenProcess");
+const preparePrompt_1 = require("../llm/preparePrompt");
+const removeFailedTestMethods_1 = require("../utils/removeFailedTestMethods");
+const appInsightTelemetry_1 = require("../metrics/appInsightTelemetry");
+const RETRYTIMES = 3;
+async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix, keepFailedTests, source = constants_1.CSHARPUTGEN, isBenchmark = false, nugetPackagePath = null) {
+    const startTime = Date.now();
+    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);
+    console.log(`Current work dir: ${process.cwd()}`);
+    // send usage metrics
+    (0, appInsightTelemetry_1.sendUserUsageEvent)(source);
+    sourceCodePath = path.normalize(sourceCodePath);
+    if (!fs.existsSync(sourceCodePath)) {
+        console.error(`Source code file not found: ${sourceCodePath}`);
+        return;
+    }
+    // install nuget cli tool to analyze code
+    (0, installAnalyzerTool_1.installOrUpdateAnalyzerTool)();
+    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');
+    // check if source code is xap code, if so, use xap code generation prompt
+    const isXapCode = (0, checkXapCode_1.checkXapCode)(sourceCode);
+    sourceCode = await (0, preGenProcess_1.preGenProcess)(sourceCode, sourceCodePath, isXapCode);
+    if (!sourceCode) {
+        console.error(`Skip generate test for source code: ${sourceCodePath}`);
+        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    }
+    // get code dependencies
+    let codeDependency = await (0, generateCodeDependency_1.generateCodeDependency)(sourceCodePath, nugetPackagePath);
+    // detect existing test file or create a new one by source code path
+    const { testFilePath, testProjectPath, testFileExist } = (0, getTestFile_1.getOrCreatTestFile)(sourceCodePath, testCodePath, isXapCode);
+    if (!testFilePath || !testProjectPath) {
+        console.error(`Failed to get or create test file for source code: ${sourceCodePath}. Please check if the test project exists.`);
+        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    }
+    // get test framework
+    const testFramework = (0, detectTestFramework_1.detectTestFramework)(testProjectPath);
+    // prepare prompt
+    let prompt = (0, preparePrompt_1.prepareUtGenPrompt)(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);
+    // call LLM service to gen code with retry
+    const testCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompt, testFramework, isXapCode);
+    // write test code to test project
+    // To run vstest, need to write the test code to test project first, then build the test project and run the tests
+    (0, fileUtils_1.writeTestCode)(testCode, testFilePath);
+    // autofix
+    let autoFixResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    if (enableAutoFix) {
+        const isVsPlugin = source === constants_1.CSVSPLUGIN;
+        autoFixResult = await (0, autoFix_1.autoFixTestCode)(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);
+        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);
+    }
+    if (autoFixResult.success) {
+        console.log(`Generate UT success time: ${(Date.now() - startTime) / 1000} seconds`);
+        return autoFixResult;
+    }
+    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods
+    if (!keepFailedTests) {
+        // keep succeed test code, or delete test file
+        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {
+            !testFileExist && (0, fileUtils_1.deleteTestFileIfExists)(testFilePath);
+            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);
+        }
+        else {
+            (0, removeFailedTestMethods_1.removeFailedTestCasesInTestFile)(autoFixResult.testResult, testFilePath);
+            console.log(`write succeed test code to file: ${testFilePath}`);
+        }
+    }
+    else {
+        console.log(`keep failed test file: ${testFilePath}`);
+    }
+    console.log(`Generate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);
+    return autoFixResult;
+}
+// generateCsharpUtCode("D:\\code\\IndexServe\\private\\indexserve\\RecommendationEngine\\RecoPS\\PSTableClient\\PSTableClient.cs", undefined, false, true, false);
+// generateCsharpUtCode("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelper.cs", undefined, true, true, false);
+// generateCsharpUtCode("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\ResponseBuilders\\EplantMigrateResponseBuilder.cs", undefined, true, true, true);
+// generateCsharpUtCode("D:\\code\\UCMDynamics\\private\\DynamicCode\\Plugin\\UCMDynamics.AppointmentPlugin\\AppointmentOperation.cs", undefined, false, true, false);
+// generateCsharpUtCode("D:\\code\\AIUnitTestExample\\src\\ToListinator.Analyzers\\ToListCountAnalyzer.cs", undefined, true, true, false);
+//# sourceMappingURL=csharpUtGen.js.map

package/out/gen/csharpUtGen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"csharpUtGen.js","sourceRoot":"","sources":["../../src/gen/csharpUtGen.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBA,oDAwFC;AA7GD,uCAAyB;AACzB,2CAA6B;AAE7B,kDAA6D;AAC7D,kDAA2E;AAG3E,uCAA4C;AAC5C,wDAAqD;AACrD,sEAAmE;AACnE,qEAAkE;AAClE,8EAA2E;AAC3E,sDAA0D;AAC1D,wEAA6E;AAC7E,0DAAuD;AACvD,wDAA0D;AAC1D,8EAAmF;AACnF,wEAAoE;AAEpE,MAAM,UAAU,GAAG,CAAC,CAAC;AAEd,KAAK,UAAU,oBAAoB,CACtC,cAAsB,EACtB,YAAoB,EACpB,aAAsB,EACtB,eAAwB,EACxB,SAAiB,uBAAW,EAC5B,cAAuB,KAAK,EAC5B,mBAA2B,IAAI;IAE/B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,sDAAsD,cAAc,mBAAmB,CAAC,CAAC;IACrG,OAAO,CAAC,GAAG,CAAC,qBAAqB,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAElD,qBAAqB;IACrB,IAAA,wCAAkB,EAAC,MAAM,CAAC,CAAC;IAE3B,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,+BAA+B,cAAc,EAAE,CAAC,CAAC;QAC/D,OAAO;IACX,CAAC;IAED,yCAAyC;IACzC,IAAA,iDAA2B,GAAE,CAAC;IAE9B,IAAI,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAE1D,0EAA0E;IAC1E,MAAM,SAAS,GAAG,IAAA,2BAAY,EAAC,UAAU,CAAC,CAAC;IAE3C,UAAU,GAAG,MAAM,IAAA,6BAAa,EAAC,UAAU,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IACxE,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,uCAAuC,cAAc,EAAE,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,GAAG,MAAM,IAAA,+CAAsB,EAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;IAEpF,oEAAoE;IACpE,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,gCAAkB,EAAC,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACrH,IAAI,CAAC,YAAY,IAAI,CAAC,eAAe,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,cAAc,4CAA4C,CAAC,CAAC;QAChI,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,qBAAqB;IACrB,MAAM,aAAa,GAAG,IAAA,yCAAmB,EAAC,eAAe,CAAC,CAAC;IAE3D,iBAAiB;IACjB,IAAI,MAAM,GAAG,IAAA,kCAAkB,EAAC,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IAEtG,0CAA0C;IAC1C,MAAM,QAAQ,GAAG,MAAM,IAAA,+CAAsB,EAAC,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;IAEhF,kCAAkC;IAClC,kHAAkH;IAClH,IAAA,yBAAa,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAEtC,UAAU;IACV,IAAI,aAAa,GAAc,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACnJ,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,MAAM,KAAK,sBAAU,CAAC;QACzC,aAAa,GAAG,MAAM,IAAA,yBAAe,EAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACtK,mHAAmH;IACvH,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QACpF,OAAO,aAAa,CAAC;IACzB,CAAC;IAED,oIAAoI;IACpI,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,8CAA8C;QAC9C,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrE,CAAC,aAAa,IAAI,IAAA,kCAAsB,EAAC,YAAY,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,2DAA2D,YAAY,EAAE,CAAC,CAAC;QAC3F,CAAC;aAAM,CAAC;YACJ,IAAA,yDAA+B,EAAC,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,oCAAoC,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,CAAC,GAAG,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACnF,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,mKAAmK;AACnK,iNAAiN;AACjN,uNAAuN;AACvN,sKAAsK;AACtK,0IAA0I","sourcesContent":["import * as fs from 'fs';\r\nimport * as path from 'path';\r\n\r\nimport { CSHARPUTGEN, CSVSPLUGIN } from '../types/constants';\r\nimport { deleteTestFileIfExists, writeTestCode } from \"../utils/fileUtils\";\r\n\r\nimport { GenResult } from '../types/genResult';\r\nimport { autoFixTestCode } from './autoFix';\r\nimport { checkXapCode } from '../utils/checkXapCode';\r\nimport { detectTestFramework } from '../utils/detectTestFramework';\r\nimport { ensureValidLLMResponse } from './ensureValidLLMResponse';\r\nimport { generateCodeDependency } from '../analyze/generateCodeDependency';\r\nimport { getOrCreatTestFile } from '../utils/getTestFile';\r\nimport { installOrUpdateAnalyzerTool } from '../analyze/installAnalyzerTool';\r\nimport { preGenProcess } from './preGen/preGenProcess';\r\nimport { prepareUtGenPrompt } from '../llm/preparePrompt';\r\nimport { removeFailedTestCasesInTestFile } from '../utils/removeFailedTestMethods';\r\nimport { sendUserUsageEvent } from '../metrics/appInsightTelemetry';\r\n\r\nconst RETRYTIMES = 3;\r\n\r\nexport async function generateCsharpUtCode(\r\n    sourceCodePath: string,\r\n    testCodePath: string,\r\n    enableAutoFix: boolean,\r\n    keepFailedTests: boolean,\r\n    source: string = CSHARPUTGEN,\r\n    isBenchmark: boolean = false,\r\n    nugetPackagePath: string = null\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);\r\n    console.log(`Current work dir: ${process.cwd()}`);\r\n\r\n    // send usage metrics\r\n    sendUserUsageEvent(source);\r\n\r\n    sourceCodePath = path.normalize(sourceCodePath);\r\n    if (!fs.existsSync(sourceCodePath)) {\r\n        console.error(`Source code file not found: ${sourceCodePath}`);\r\n        return;\r\n    }\r\n\r\n    // install nuget cli tool to analyze code\r\n    installOrUpdateAnalyzerTool();\r\n\r\n    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');\r\n\r\n    // check if source code is xap code, if so, use xap code generation prompt\r\n    const isXapCode = checkXapCode(sourceCode);\r\n\r\n    sourceCode = await preGenProcess(sourceCode, sourceCodePath, isXapCode);\r\n    if (!sourceCode) {\r\n        console.error(`Skip generate test for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get code dependencies\r\n    let codeDependency = await generateCodeDependency(sourceCodePath, nugetPackagePath);\r\n\r\n    // detect existing test file or create a new one by source code path\r\n    const { testFilePath, testProjectPath, testFileExist } = getOrCreatTestFile(sourceCodePath, testCodePath, isXapCode);\r\n    if (!testFilePath || !testProjectPath) {\r\n        console.error(`Failed to get or create test file for source code: ${sourceCodePath}. Please check if the test project exists.`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get test framework\r\n    const testFramework = detectTestFramework(testProjectPath);\r\n\r\n    // prepare prompt\r\n    let prompt = prepareUtGenPrompt(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);\r\n\r\n    // call LLM service to gen code with retry\r\n    const testCode = await ensureValidLLMResponse(prompt, testFramework, isXapCode);\r\n\r\n    // write test code to test project\r\n    // To run vstest, need to write the test code to test project first, then build the test project and run the tests\r\n    writeTestCode(testCode, testFilePath);\r\n\r\n    // autofix\r\n    let autoFixResult: GenResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    if (enableAutoFix) {\r\n        const isVsPlugin = source === CSVSPLUGIN;\r\n        autoFixResult = await autoFixTestCode(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, testFramework, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);\r\n        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);\r\n    }\r\n\r\n    if (autoFixResult.success) {\r\n        console.log(`Generate UT success time: ${(Date.now() - startTime) / 1000} seconds`);\r\n        return autoFixResult;\r\n    }\r\n\r\n    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods\r\n    if (!keepFailedTests) {\r\n        // keep succeed test code, or delete test file\r\n        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {\r\n            !testFileExist && deleteTestFileIfExists(testFilePath);\r\n            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);\r\n        } else {\r\n            removeFailedTestCasesInTestFile(autoFixResult.testResult, testFilePath);\r\n            console.log(`write succeed test code to file: ${testFilePath}`);\r\n        }\r\n    } else {\r\n        console.log(`keep failed test file: ${testFilePath}`);\r\n    }\r\n    console.log(`Generate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    return autoFixResult;\r\n}\r\n\r\n// generateCsharpUtCode(\"D:\\\\code\\\\IndexServe\\\\private\\\\indexserve\\\\RecommendationEngine\\\\RecoPS\\\\PSTableClient\\\\PSTableClient.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", undefined, true, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\ResponseBuilders\\\\EplantMigrateResponseBuilder.cs\", undefined, true, true, true);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\UCMDynamics\\\\private\\\\DynamicCode\\\\Plugin\\\\UCMDynamics.AppointmentPlugin\\\\AppointmentOperation.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\AIUnitTestExample\\\\src\\\\ToListinator.Analyzers\\\\ToListCountAnalyzer.cs\", undefined, true, true, false);\r\n"]}

package/out/gen/ensureValidLLMResponse.d.ts

@@ -0,0 +1 @@
+export declare function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, retryTimes?: number): Promise<string>;

package/out/gen/ensureValidLLMResponse.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ensureValidLLMResponse = ensureValidLLMResponse;
+const postGenProcess_1 = require("./postGen/postGenProcess");
+const sendRequestToLLM_1 = require("../llm/sendRequestToLLM");
+async function ensureValidLLMResponse(prompt, testFramework, isXapCode, retryTimes = 3) {
+    for (let i = 0; i < retryTimes; i++) {
+        // call LLM service to gen code
+        const { content: generatedCode } = await (0, sendRequestToLLM_1.sendRequestToLLM)(prompt);
+        // extract valid code from LLM response
+        const testCode = (0, postGenProcess_1.postGenProcess)(generatedCode, testFramework, isXapCode);
+        if (!!testCode) {
+            return testCode;
+        }
+        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);
+    }
+    return undefined;
+}
+//# sourceMappingURL=ensureValidLLMResponse.js.map

package/out/gen/ensureValidLLMResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AAGA,wDAcC;AAjBD,6DAA0D;AAC1D,8DAA2D;AAEpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,aAAqB,EAAE,SAAkB,EAAE,aAAqB,CAAC;IACzH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QACzE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, retryTimes: number = 3): Promise<string> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        const testCode = postGenProcess(generatedCode, testFramework, isXapCode);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}

package/out/gen/postGen/addAiAnnotation.d.ts

@@ -0,0 +1 @@
+export declare function addAiAnnotation(code: string): string;

package/out/gen/postGen/addAiAnnotation.js

@@ -0,0 +1,57 @@
+"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.addAiAnnotation = addAiAnnotation;
+const constants_1 = require("../../types/constants");
+const util = __importStar(require("util"));
+function addAiAnnotation(code) {
+    const lines = code.split(constants_1.LINESEPARATOR);
+    if (code.includes(constants_1.AICONTRIBUTED)) {
+        const aiAnnotationTagIndex = lines.findIndex(line => line.startsWith(constants_1.AICONTRIBUTED));
+        lines.splice(aiAnnotationTagIndex, 1);
+    }
+    const currentDate = new Date().toISOString().split('T')[0];
+    const aiAnnotationTemplate = '/* %s-%s */';
+    const aiAnnotation = util.format(aiAnnotationTemplate, constants_1.AICONTRIBUTED, currentDate);
+    const namespaceIndex = lines.findIndex(line => line.startsWith(constants_1.NAMESPACE));
+    if (namespaceIndex !== -1) {
+        lines.splice(namespaceIndex, 0, aiAnnotation);
+        return lines.join('\n');
+    }
+    else {
+        return aiAnnotation + '\n' + code;
+    }
+}
+//# sourceMappingURL=addAiAnnotation.js.map

package/out/gen/postGen/addAiAnnotation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"addAiAnnotation.js","sourceRoot":"","sources":["../../../src/gen/postGen/addAiAnnotation.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAGA,0CAmBC;AAtBD,qDAAgF;AAChF,2CAA6B;AAE7B,SAAgB,eAAe,CAAC,IAAY;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,yBAAa,CAAC,CAAC;IACxC,IAAI,IAAI,CAAC,QAAQ,CAAC,yBAAa,CAAC,EAAE,CAAC;QAC/B,MAAM,oBAAoB,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAa,CAAC,CAAC,CAAC;QACrF,KAAK,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC;IAC1C,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3D,MAAM,oBAAoB,GAAG,aAAa,CAAC;IAC3C,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,yBAAa,EAAE,WAAW,CAAC,CAAC;IAEnF,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,qBAAS,CAAC,CAAC,CAAC;IAC3E,IAAI,cAAc,KAAK,CAAC,CAAC,EAAE,CAAC;QACxB,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,EAAE,YAAY,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;SACI,CAAC;QACF,OAAO,YAAY,GAAG,IAAI,GAAG,IAAI,CAAC;IACtC,CAAC;AACL,CAAC","sourcesContent":["import { LINESEPARATOR, AICONTRIBUTED, NAMESPACE } from '../../types/constants';\r\nimport * as util from 'util';\r\n\r\nexport function addAiAnnotation(code: string): string {\r\n    const lines = code.split(LINESEPARATOR);\r\n    if (code.includes(AICONTRIBUTED)) {\r\n        const aiAnnotationTagIndex = lines.findIndex(line => line.startsWith(AICONTRIBUTED));\r\n        lines.splice(aiAnnotationTagIndex, 1);\r\n    }\r\n\r\n    const currentDate = new Date().toISOString().split('T')[0];\r\n    const aiAnnotationTemplate = '/* %s-%s */';\r\n    const aiAnnotation = util.format(aiAnnotationTemplate, AICONTRIBUTED, currentDate);\r\n\r\n    const namespaceIndex = lines.findIndex(line => line.startsWith(NAMESPACE));\r\n    if (namespaceIndex !== -1) {\r\n        lines.splice(namespaceIndex, 0, aiAnnotation);\r\n        return lines.join('\\n');\r\n    }\r\n    else {\r\n        return aiAnnotation + '\\n' + code;\r\n    }\r\n}"]}

package/out/gen/postGen/extractCodeFromResponse.d.ts

@@ -0,0 +1 @@
+export declare function extractCodeFromResponse(response: string): string;

package/out/gen/postGen/extractCodeFromResponse.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractCodeFromResponse = extractCodeFromResponse;
+const constants_1 = require("../../types/constants");
+const regexList = [
+    /```(?:C#|csharp|CSharp|json|mermaid)?([\s\S]*?)(\n?```\n*|$)/g,
+    /```([\s\S]*?)(\n?```\n*|$)/g,
+    /```C#\s*([\s\S]*?)```/gi
+];
+function extractCodeFromResponse(response) {
+    if (!response) {
+        return '';
+    }
+    const text = response.trim();
+    for (const regex of regexList) {
+        let testCode = getRegexResult(text, regex);
+        if (!!testCode) {
+            return testCode;
+        }
+    }
+    const lines = text.split(constants_1.LINESEPARATOR);
+    const hasMarkdown = lines.some(line => line.startsWith('#'));
+    if (hasMarkdown) {
+        return '';
+    }
+    return text;
+}
+function getRegexResult(response, regex) {
+    const match = regex.exec(response);
+    if (!match || match.length < 2) {
+        return null;
+    }
+    return match[1].trim();
+}
+// const input = '```C#\nusing Microsoft.VisualStudio.TestTools.UnitTesting;\nusing Moq;\nusing System;\nusing System.Collections.Generic;\nusing SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\n\nnamespace SharedSegments.Tests\n{\n    [TestClass]\n    public class SqmidHelperTests\n    {\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithValidBlisId_ReturnsDecodedSqmid()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=QlJfU01JRA==\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(\"SMID\", result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithoutBlisId_ReturnsNull()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"ANOTHERDATA=123\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.IsNull(result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithEmptyValues_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_InvalidBase64BlisId_ThrowsArgumentException_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=InvalidBase64\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_NoLocationCookie_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>();\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\n        {\n            var result = SqmidHelper.GetSqmidFromCookie(null);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_ValidBase64_ReturnsExpectedValue()\n        {\n            var blisId = \"QlJfU01JRA==\";\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(\"SMID\", result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_InvalidBase64_ReturnsEmptyString()\n        {\n            var blisId = \"InvalidBase64\";\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_EmptyBlisId_ReturnsEmptyString()\n        {\n            var blisId = string.Empty;\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_NullBlisId_ReturnsEmptyString()\n        {\n            var result = InvokeDecodeBlisId(null);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        private string InvokeDecodeBlisId(string blisId)\n        {\n            var method = typeof(SqmidHelper).GetMethod(\"DecodeBlisId\", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static);\n            return (string)method.Invoke(null, new object[] { blisId });\n        }\n    }\n}\n```';
+// const code = extractCode(input);
+// console.log(code);
+//# sourceMappingURL=extractCodeFromResponse.js.map

package/out/gen/postGen/extractCodeFromResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractCodeFromResponse.js","sourceRoot":"","sources":["../../../src/gen/postGen/extractCodeFromResponse.ts"],"names":[],"mappings":";;AAQA,0DAkBC;AA1BD,qDAAsD;AAEtD,MAAM,SAAS,GAAG;IACd,+DAA+D;IAC/D,6BAA6B;IAC7B,yBAAyB;CAC5B,CAAC;AAEF,SAAgB,uBAAuB,CAAC,QAAgB;IACpD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;IACd,CAAC;IACD,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC7B,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;QAC5B,IAAI,QAAQ,GAAG,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;IACL,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,yBAAa,CAAC,CAAC;IACxC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7D,IAAI,WAAW,EAAE,CAAC;QACd,OAAO,EAAE,CAAC;IACd,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,QAAgB,EAAE,KAAa;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;AAC3B,CAAC;AAED,+qHAA+qH;AAC/qH,mCAAmC;AACnC,qBAAqB","sourcesContent":["import { LINESEPARATOR } from \"../../types/constants\";\r\n\r\nconst regexList = [\r\n    /```(?:C#|csharp|CSharp|json|mermaid)?([\\s\\S]*?)(\\n?```\\n*|$)/g,\r\n    /```([\\s\\S]*?)(\\n?```\\n*|$)/g,\r\n    /```C#\\s*([\\s\\S]*?)```/gi\r\n];\r\n\r\nexport function extractCodeFromResponse(response: string): string {\r\n    if (!response) {\r\n        return '';\r\n    }\r\n    const text = response.trim();\r\n    for (const regex of regexList) {\r\n        let testCode = getRegexResult(text, regex);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n    }\r\n\r\n    const lines = text.split(LINESEPARATOR);\r\n    const hasMarkdown = lines.some(line => line.startsWith('#'));\r\n    if (hasMarkdown) {\r\n        return '';\r\n    }\r\n    return text;\r\n}\r\n\r\nfunction getRegexResult(response: string, regex: RegExp): string | null {\r\n    const match = regex.exec(response);\r\n    if (!match || match.length < 2) {\r\n        return null;\r\n    }\r\n    return match[1].trim();\r\n}\r\n\r\n// const input = '```C#\\nusing Microsoft.VisualStudio.TestTools.UnitTesting;\\nusing Moq;\\nusing System;\\nusing System.Collections.Generic;\\nusing SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\\n\\nnamespace SharedSegments.Tests\\n{\\n    [TestClass]\\n    public class SqmidHelperTests\\n    {\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithValidBlisId_ReturnsDecodedSqmid()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=QlJfU01JRA==\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(\\\"SMID\\\", result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithoutBlisId_ReturnsNull()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"ANOTHERDATA=123\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.IsNull(result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithEmptyValues_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_InvalidBase64BlisId_ThrowsArgumentException_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=InvalidBase64\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_NoLocationCookie_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>();\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\\n        {\\n            var result = SqmidHelper.GetSqmidFromCookie(null);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_ValidBase64_ReturnsExpectedValue()\\n        {\\n            var blisId = \\\"QlJfU01JRA==\\\";\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(\\\"SMID\\\", result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_InvalidBase64_ReturnsEmptyString()\\n        {\\n            var blisId = \\\"InvalidBase64\\\";\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_EmptyBlisId_ReturnsEmptyString()\\n        {\\n            var blisId = string.Empty;\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_NullBlisId_ReturnsEmptyString()\\n        {\\n            var result = InvokeDecodeBlisId(null);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        private string InvokeDecodeBlisId(string blisId)\\n        {\\n            var method = typeof(SqmidHelper).GetMethod(\\\"DecodeBlisId\\\", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static);\\n            return (string)method.Invoke(null, new object[] { blisId });\\n        }\\n    }\\n}\\n```';\r\n// const code = extractCode(input);\r\n// console.log(code);\r\n"]}

package/out/gen/postGen/postGenProcess.d.ts

@@ -0,0 +1 @@
+export declare function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean): string;

package/out/gen/postGen/postGenProcess.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.postGenProcess = postGenProcess;
+const extractCodeFromResponse_1 = require("./extractCodeFromResponse");
+const removeSingleLines_1 = require("./removeSingleLines");
+const removeComments_1 = require("./removeComments");
+const repairRequiredNameSpaces_1 = require("./repairRequiredNameSpaces");
+const validateTestCode_1 = require("./validateTestCode");
+const addAiAnnotation_1 = require("./addAiAnnotation");
+function postGenProcess(generatedCode, testFramework, isXapCode) {
+    if (!isValidTestCode(generatedCode)) {
+        return undefined;
+    }
+    let testCode = (0, extractCodeFromResponse_1.extractCodeFromResponse)(generatedCode);
+    testCode = (0, removeComments_1.removeTestCodeComments)(testCode);
+    testCode = (0, removeSingleLines_1.removeCSharpOnlyLines)(testCode);
+    testCode = (0, repairRequiredNameSpaces_1.repairRequiredNameSpaces)(testCode, testFramework);
+    if (isXapCode) {
+        testCode = xapCodePostGenProcess(testCode);
+    }
+    testCode = (0, addAiAnnotation_1.addAiAnnotation)(testCode);
+    console.log(`Post generation process completed, code length: ${testCode.length}`);
+    return testCode;
+}
+function xapCodePostGenProcess(testCode) {
+    const processedCode = (0, repairRequiredNameSpaces_1.repairXapRequiredNameSpaces)(testCode);
+    return processedCode;
+}
+function isValidTestCode(testCode) {
+    if (!testCode || testCode.trim() === '') {
+        return false; // Empty code is not valid
+    }
+    // Check if the test code contains any test attributes or using statements
+    const isValid = (0, validateTestCode_1.validateTestCode)(testCode);
+    return isValid;
+}
+// const code = fs.readFileSync("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Tests\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelperTests.cs", 'utf-8');
+// postGenProcess(code, false);
+//# sourceMappingURL=postGenProcess.js.map

package/out/gen/postGen/postGenProcess.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAOA,wCAiBC;AAxBD,uEAAoE;AACpE,2DAA4D;AAC5D,qDAA0D;AAC1D,yEAAmG;AACnG,yDAAsD;AACtD,uDAAoD;AAEpD,SAAgB,cAAc,CAAC,aAAqB,EAAE,aAAqB,EAAE,SAAkB;IAC3F,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ,GAAG,IAAA,iDAAuB,EAAC,aAAa,CAAC,CAAC;IACtD,QAAQ,GAAG,IAAA,uCAAsB,EAAC,QAAQ,CAAC,CAAC;IAC5C,QAAQ,GAAG,IAAA,yCAAqB,EAAC,QAAQ,CAAC,CAAC;IAC3C,QAAQ,GAAG,IAAA,mDAAwB,EAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC7D,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,GAAG,IAAA,iCAAe,EAAC,QAAQ,CAAC,CAAC;IAErC,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,sDAA2B,EAAC,QAAQ,CAAC,CAAC;IAC5D,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACrC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,CAAC,0BAA0B;IAC5C,CAAC;IACD,0EAA0E;IAC1E,MAAM,OAAO,GAAG,IAAA,mCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,uMAAuM;AACvM,+BAA+B","sourcesContent":["import { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { removeTestCodeComments } from \"./removeComments\";\r\nimport { repairRequiredNameSpaces, repairXapRequiredNameSpaces } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\nimport { addAiAnnotation } from \"./addAiAnnotation\";\r\n\r\nexport function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean): string {\r\n    if (!isValidTestCode(generatedCode)) {\r\n        return undefined;\r\n    }\r\n\r\n    let testCode = extractCodeFromResponse(generatedCode);\r\n    testCode = removeTestCodeComments(testCode);\r\n    testCode = removeCSharpOnlyLines(testCode);\r\n    testCode = repairRequiredNameSpaces(testCode, testFramework);\r\n    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n\r\n    testCode = addAiAnnotation(testCode);\r\n\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairXapRequiredNameSpaces(testCode);\r\n    return processedCode;\r\n}\r\n\r\nfunction isValidTestCode(testCode: string): boolean {\r\n    if (!testCode || testCode.trim() === '') {\r\n        return false; // Empty code is not valid\r\n    }\r\n    // Check if the test code contains any test attributes or using statements\r\n    const isValid = validateTestCode(testCode);\r\n    return isValid;\r\n}\r\n\r\n// const code = fs.readFileSync(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Tests\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelperTests.cs\", 'utf-8');\r\n// postGenProcess(code, false);\r\n"]}

package/out/gen/postGen/removeComments.d.ts

@@ -0,0 +1 @@
+export declare function removeTestCodeComments(code: string): string;

package/out/gen/postGen/removeComments.js

@@ -0,0 +1,92 @@
+"use strict";
+// todo: remove line：
+// C#(just contains C# in a line)
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.removeTestCodeComments = removeTestCodeComments;
+function removeTestCodeComments(code) {
+    // Remove multi-line comments (/* ... */)
+    let cleanedCode = code.replace(/\/\*[\s\S]*?\*\//g, '');
+    // Remove single-line comments (// ...)
+    cleanedCode = cleanedCode.replace(/\/\/.*$/gm, '');
+    // Remove extra blank lines that might result from comment removal, only left at most one blank line
+    // cleanedCode = cleanedCode.replace(/\n\s*\n/g, '\n');
+    cleanedCode = cleanedCode.replace(/(\r?\n){3,}/g, '\n\n');
+    return cleanedCode.trim();
+}
+// const code = `
+// using Microsoft.VisualStudio.TestTools.UnitTesting;
+// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;
+// using System;
+// using System.Collections.Generic;
+// using System.Text;
+// using Moq;
+// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils
+// {
+//     [TestClass]
+//     public class SqmidHelperTests
+//     {
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "BID=" + Convert.ToBase64String(Encoding.UTF8.GetBytes("test_decodedBlisId_test")) }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual("decodedBlisId", result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "SomeOtherValue=123" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.IsNull(result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>();
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             IDictionary<string, string> cookies = null;
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//     }
+// }
+// `;
+// const cleanedCode = removeTestCodeComments(code);
+// console.log(cleanedCode);
+//# sourceMappingURL=removeComments.js.map