@performance-agent/mcp-server 1.0.1 → 1.0.6

package/dist/tools/jmx-generator/index.js

@@ -0,0 +1,115 @@
+/**
+ * JMX Generator - Generate JMeter scripts from Swagger/OpenAPI specifications using Azure OpenAI
+ *
+ * This module provides direct OpenAI integration for JMX generation, eliminating the need
+ * for the Flask backend API.
+ */
+import { readFileSync } from 'fs';
+import { join, dirname } from 'path';
+import { fileURLToPath } from 'url';
+import { sanitizeJMX } from './sanitizer.js';
+// ES module equivalent of __dirname
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+// Azure OpenAI configuration from environment variables
+const OPENAI_ENDPOINT = process.env.OPENAI_ENDPOINT || process.env.AZURE_OPENAI_ENDPOINT || '';
+const OPENAI_API_KEY = process.env.OPENAI_API_KEY || process.env.AZURE_OPENAI_KEY || '';
+const OPENAI_DEPLOYMENT = process.env.OPENAI_DEPLOYMENT || process.env.AZURE_OPENAI_DEPLOYMENT || 'gpt-4';
+const OPENAI_API_VERSION = process.env.OPENAI_API_VERSION || '2024-02-15-preview';
+/**
+ * Generate JMeter JMX script from Swagger/OpenAPI specification
+ * @param swaggerContent Swagger/OpenAPI JSON object
+ * @param repoName Optional repository name for logging
+ * @returns Result object with success status and JMX script or error
+ */
+export async function generateJMX(swaggerContent, repoName) {
+    try {
+        console.log(`[JMX Generator] Starting JMX generation${repoName ? ` for repo: ${repoName}` : ''}...`);
+        // Validate Azure OpenAI configuration
+        if (!OPENAI_ENDPOINT || !OPENAI_API_KEY) {
+            const error = 'Azure OpenAI configuration missing. Set OPENAI_ENDPOINT and OPENAI_API_KEY environment variables.';
+            console.error(`[JMX Generator] ${error}`);
+            return { success: false, error };
+        }
+        // Load JMX prompt template
+        const promptPath = join(__dirname, '../../prompts/jmx_prompt.txt');
+        let promptTemplate;
+        try {
+            promptTemplate = readFileSync(promptPath, 'utf-8');
+            console.log('[JMX Generator] Loaded JMX prompt template');
+        }
+        catch (err) {
+            const error = `Failed to load JMX prompt template from ${promptPath}: ${err}`;
+            console.error(`[JMX Generator] ${error}`);
+            return { success: false, error };
+        }
+        // Format prompt with Swagger content
+        const swaggerText = JSON.stringify(swaggerContent, null, 2);
+        const formattedPrompt = promptTemplate.replace('{swagger_text}', swaggerText);
+        console.log(`[JMX Generator] Formatted prompt (${formattedPrompt.length} chars, Swagger: ${swaggerText.length} chars)`);
+        // Call Azure OpenAI API using fetch
+        console.log(`[JMX Generator] Calling Azure OpenAI at ${OPENAI_ENDPOINT}...`);
+        const url = `${OPENAI_ENDPOINT}/openai/deployments/${OPENAI_DEPLOYMENT}/chat/completions?api-version=${OPENAI_API_VERSION}`;
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'api-key': OPENAI_API_KEY,
+            },
+            body: JSON.stringify({
+                messages: [
+                    {
+                        role: 'user',
+                        content: formattedPrompt,
+                    },
+                ]
+            }),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            const error = `Azure OpenAI API error (${response.status}): ${errorText}`;
+            console.error(`[JMX Generator] ${error}`);
+            return { success: false, error };
+        }
+        const data = await response.json();
+        if (!data.choices || data.choices.length === 0) {
+            const error = 'Azure OpenAI returned no choices';
+            console.error(`[JMX Generator] ${error}`);
+            return { success: false, error };
+        }
+        let jmxScript = data.choices[0].message?.content?.trim() || '';
+        console.log(`[JMX Generator] Received response from OpenAI (${jmxScript.length} chars)`);
+        // Remove markdown code blocks if present
+        if (jmxScript.startsWith('```')) {
+            console.log('[JMX Generator] Removing markdown code blocks...');
+            jmxScript = jmxScript
+                .replace(/^```xml\n?/i, '')
+                .replace(/^```\n?/, '')
+                .replace(/\n?```$/, '')
+                .trim();
+        }
+        // Sanitize the JMX script
+        console.log('[JMX Generator] Sanitizing JMX script...');
+        jmxScript = sanitizeJMX(jmxScript);
+        // Validate that we have XML content
+        if (!jmxScript.startsWith('<?xml')) {
+            const error = 'Generated content is not valid XML';
+            console.error(`[JMX Generator] ${error}. Content preview: ${jmxScript.substring(0, 200)}`);
+            return { success: false, error };
+        }
+        console.log(`[JMX Generator] Successfully generated JMX script (${jmxScript.length} chars)`);
+        return {
+            success: true,
+            jmx_script: jmxScript,
+        };
+    }
+    catch (error) {
+        const errorMsg = `JMX generation failed: ${error}`;
+        console.error(`[JMX Generator] ${errorMsg}`);
+        return {
+            success: false,
+            error: errorMsg,
+        };
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/tools/jmx-generator/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/jmx-generator/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,oCAAoC;AACpC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,wDAAwD;AACxD,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,EAAE,CAAC;AAC/F,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,CAAC;AACxF,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,OAAO,CAAC;AAC1G,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,oBAAoB,CAAC;AAQlF;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,cAAmC,EACnC,QAAiB;IAEjB,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,0CAA0C,QAAQ,CAAC,CAAC,CAAC,cAAc,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAErG,sCAAsC;QACtC,IAAI,CAAC,eAAe,IAAI,CAAC,cAAc,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,mGAAmG,CAAC;YAClH,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,2BAA2B;QAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,8BAA8B,CAAC,CAAC;QACnE,IAAI,cAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,cAAc,GAAG,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;QAC5D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,KAAK,GAAG,2CAA2C,UAAU,KAAK,GAAG,EAAE,CAAC;YAC9E,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,qCAAqC;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC5D,MAAM,eAAe,GAAG,cAAc,CAAC,OAAO,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QAC9E,OAAO,CAAC,GAAG,CAAC,qCAAqC,eAAe,CAAC,MAAM,oBAAoB,WAAW,CAAC,MAAM,SAAS,CAAC,CAAC;QAExH,oCAAoC;QACpC,OAAO,CAAC,GAAG,CAAC,2CAA2C,eAAe,KAAK,CAAC,CAAC;QAC7E,MAAM,GAAG,GAAG,GAAG,eAAe,uBAAuB,iBAAiB,iCAAiC,kBAAkB,EAAE,CAAC;QAE5H,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,SAAS,EAAE,cAAc;aAC1B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,eAAe;qBACzB;iBACF;aACF,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,2BAA2B,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC;YAC1E,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,MAAM,IAAI,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAExC,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,kCAAkC,CAAC;YACjD,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,EAAE,CAAC,CAAC;YAC1C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,IAAI,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,kDAAkD,SAAS,CAAC,MAAM,SAAS,CAAC,CAAC;QAEzF,yCAAyC;QACzC,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,kDAAkD,CAAC,CAAC;YAChE,SAAS,GAAG,SAAS;iBAClB,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;iBAC1B,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;iBACtB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;iBACtB,IAAI,EAAE,CAAC;QACZ,CAAC;QAED,0BAA0B;QAC1B,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QACxD,SAAS,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC;QAEnC,oCAAoC;QACpC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,oCAAoC,CAAC;YACnD,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,sBAAsB,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;YAC3F,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACnC,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,sDAAsD,SAAS,CAAC,MAAM,SAAS,CAAC,CAAC;QAC7F,OAAO;YACL,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,SAAS;SACtB,CAAC;IAEJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,QAAQ,GAAG,0BAA0B,KAAK,EAAE,CAAC;QACnD,OAAO,CAAC,KAAK,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;QAC7C,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,QAAQ;SAChB,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/tools/jmx-generator/sanitizer.d.ts

@@ -0,0 +1,19 @@
+/**
+ * JMX Sanitizer - Cleans up common invalid JMeter XML patterns in AI output
+ *
+ * This module sanitizes JMX (JMeter XML) files to fix common issues:
+ * - Adds missing guiclass attributes for test elements
+ * - Replaces ArrayList-based HTTP file args with proper HTTPFileArgs container
+ * - Removes ArrayList elementProp entries that cause casting errors
+ * - Fixes invalid <argument> tags into proper <elementProp elementType="Argument">
+ * - Ensures Arguments blocks have <collectionProp> wrapper
+ * - Normalizes enabled="true" attributes
+ * - Ensures every major element is followed by <hashTree/>
+ */
+/**
+ * Sanitize JMeter XML to fix common AI-generated errors
+ * @param xmlText Raw XML text from AI model
+ * @returns Sanitized XML text that's loadable in JMeter
+ */
+export declare function sanitizeJMX(xmlText: string): string;
+//# sourceMappingURL=sanitizer.d.ts.map

package/dist/tools/jmx-generator/sanitizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sanitizer.d.ts","sourceRoot":"","sources":["../../../src/tools/jmx-generator/sanitizer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH;;;;GAIG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CA4PnD"}

package/dist/tools/jmx-generator/sanitizer.js

@@ -0,0 +1,166 @@
+/**
+ * JMX Sanitizer - Cleans up common invalid JMeter XML patterns in AI output
+ *
+ * This module sanitizes JMX (JMeter XML) files to fix common issues:
+ * - Adds missing guiclass attributes for test elements
+ * - Replaces ArrayList-based HTTP file args with proper HTTPFileArgs container
+ * - Removes ArrayList elementProp entries that cause casting errors
+ * - Fixes invalid <argument> tags into proper <elementProp elementType="Argument">
+ * - Ensures Arguments blocks have <collectionProp> wrapper
+ * - Normalizes enabled="true" attributes
+ * - Ensures every major element is followed by <hashTree/>
+ */
+/**
+ * Sanitize JMeter XML to fix common AI-generated errors
+ * @param xmlText Raw XML text from AI model
+ * @returns Sanitized XML text that's loadable in JMeter
+ */
+export function sanitizeJMX(xmlText) {
+    if (!xmlText) {
+        return xmlText;
+    }
+    try {
+        console.log('[JMX Sanitizer] Starting sanitization...');
+        // Fix invalid root structure: jmeterTestPlan should not have testclass/guiclass
+        xmlText = xmlText.replace(/<jmeterTestPlan\s+guiclass="[^"]*"\s+testclass="[^"]*"\s+testname="[^"]*"\s+enabled="[^"]*"/g, '<jmeterTestPlan version="1.2" properties="5.0" jmeter="5.6.3"');
+        // Fix: Extract user-defined variables from TestPlan property and create separate Arguments element
+        const userVarsMatch = xmlText.match(/<stringProp name="TestPlan\.user_defined_variables">\s*((?:<elementProp[^>]*>.*?<\/elementProp>\s*)+)<\/stringProp>/s);
+        if (userVarsMatch) {
+            console.log('[JMX Sanitizer] Extracting user-defined variables to separate Arguments element...');
+            const variablesContent = userVarsMatch[1];
+            // Create proper Arguments element with collectionProp wrapper
+            const argumentsElement = `<Arguments guiclass="ArgumentsPanel" testclass="Arguments" testname="User Defined Variables" enabled="true">
+          <collectionProp name="Arguments.arguments">
+            ${variablesContent.trim()}
+          </collectionProp>
+        </Arguments>
+        <hashTree/>`;
+            // Remove the TestPlan.user_defined_variables property
+            xmlText = xmlText.replace(/<stringProp name="TestPlan\.user_defined_variables">.*?<\/stringProp>\s*/s, '');
+            // Insert the Arguments element as first child in TestPlan's hashTree
+            xmlText = xmlText.replace(/(<\/TestPlan>\s*<\/hashTree>)/, `</TestPlan>\n        <hashTree>\n        ${argumentsElement}\n        `);
+            const varCount = (variablesContent.match(/elementProp/g) || []).length;
+            console.log(`[JMX Sanitizer] Created separate User Defined Variables with ${varCount} variables`);
+        }
+        // Fix missing TestPlan wrapper after jmeterTestPlan
+        if (/<jmeterTestPlan[^>]*>\s*<hashTree\/>\s*<stringProp/.test(xmlText)) {
+            console.log('[JMX Sanitizer] Fixing missing TestPlan wrapper...');
+            xmlText = xmlText.replace(/(<jmeterTestPlan[^>]*>)\s*<hashTree\/>\s*(<stringProp)/, '$1\n  <hashTree>\n    <TestPlan guiclass="TestPlanGui" testclass="TestPlan" testname="API Test Plan" enabled="true">\n      $2');
+        }
+        // Ensure TestPlan properties are wrapped correctly
+        if (/<\/TestPlan>\s*<hashTree\/>\s*<elementProp name="User Defined Variables"/.test(xmlText)) {
+            console.log('[JMX Sanitizer] Moving User Defined Variables inside TestPlan...');
+            xmlText = xmlText.replace(/(<\/TestPlan>)\s*<hashTree\/>\s*(<elementProp name="User Defined Variables"[^>]*>.*?<\/elementProp>)/s, '$2\n      <boolProp name="TestPlan.functional_mode">false</boolProp>\n      <boolProp name="TestPlan.serialize_threadgroups">false</boolProp>\n    $1');
+        }
+        // Fix invalid elementType values
+        xmlText = xmlText.replace(/elementType="StringParameter"/g, 'elementType="Argument"');
+        // Fix invalid property type names
+        if (xmlText.includes('booleanProp')) {
+            console.log('[JMX Sanitizer] Fixing booleanProp -> boolProp...');
+            xmlText = xmlText.replace(/<booleanProp/g, '<boolProp');
+            xmlText = xmlText.replace(/<\/booleanProp>/g, '</boolProp>');
+        }
+        // Fix common XML nesting errors in ResponseAssertion
+        xmlText = xmlText.replace(/(<collectionProp[^>]*>)\s*(<stringProp[^>]*>[^<]*<\/stringProp>)\s*<\/stringProp>/g, '$1\n          $2');
+        // Fix: <collectionProp> closed with </stringProp>
+        xmlText = xmlText.replace(/(<collectionProp[^>]*>.*?)<\/stringProp>(\s*<\/ResponseAssertion>)/gs, '$1</collectionProp>$2');
+        // Fix: <stringProp> closed with </collectionProp> in ResponseAssertion
+        if (xmlText.includes('Assertion.test_field')) {
+            xmlText = xmlText.replace(/(<stringProp name="Assertion\.test_field">[^<]*)<\/collectionProp>/g, '$1</stringProp>');
+            console.log('[JMX Sanitizer] Fixed mismatched closing tag for Assertion.test_field');
+        }
+        // General fix: <stringProp> incorrectly closed with </collectionProp>
+        xmlText = xmlText.replace(/(<stringProp[^>]*>[^<]*)<\/collectionProp>/g, '$1</stringProp>');
+        // Fix: Remove invalid elementProp with elementType="Assertion" inside ResponseAssertion
+        if (xmlText.includes('elementType="Assertion"')) {
+            console.log('[JMX Sanitizer] Fixing invalid elementType="Assertion" in ResponseAssertion...');
+            xmlText = xmlText.replace(/<elementProp[^>]*elementType="Assertion"[^>]*>\s*(<stringProp[^>]*>[^<]*<\/stringProp>)\s*<\/elementProp>/g, '$1');
+        }
+        // Fix stray <argument> tags into proper elementProp Argument
+        if (xmlText.includes('<argument>')) {
+            console.log('[JMX Sanitizer] Fixing <argument> tags to proper elementProp...');
+            // Pattern 1: <argument><name>...</name><value>...</value></argument>
+            xmlText = xmlText.replace(/<argument>\s*<name>([^<]+)<\/name>\s*<value>([^<]*)<\/value>\s*<\/argument>/g, `<elementProp name="$1" elementType="Argument">
+        <stringProp name="Argument.name">$1</stringProp>
+        <stringProp name="Argument.value">$2</stringProp>
+        <stringProp name="Argument.metadata">=</stringProp>
+      </elementProp>`);
+            // Pattern 2: Simple <argument>value</argument>
+            xmlText = xmlText.replace(/<argument>([^<]*)<\/argument>/g, `<elementProp name="arg" elementType="Argument">
+        <stringProp name="Argument.name">arg</stringProp>
+        <stringProp name="Argument.value">$1</stringProp>
+        <stringProp name="Argument.metadata">=</stringProp>
+      </elementProp>`);
+        }
+        // Fix duplicate/malformed collectionProp on same line
+        xmlText = xmlText.replace(/<collectionProp name="Arguments\.arguments">\s*<collectionProp name="Arguments\.arguments">/g, '<collectionProp name="Arguments.arguments">');
+        console.log('[JMX Sanitizer] Checked for duplicate collectionProp tags');
+        // Fix Argument -> HTTPArgument inside HTTPsampler.Arguments
+        xmlText = xmlText.replace(/<elementProp name="HTTPsampler\.Arguments"[^>]*>.*?<\/elementProp>/gs, (match) => {
+            let content = match;
+            if (content.includes('elementType="Argument"') && content.includes('HTTPsampler.Arguments')) {
+                console.log('[JMX Sanitizer] Fixing Argument -> HTTPArgument in HTTPsampler.Arguments...');
+                content = content.replace(/<elementProp\s+name="[^"]*"\s+elementType="Argument">/g, (m) => m.replace('elementType="Argument"', 'elementType="HTTPArgument"'));
+                if (!content.includes('HTTPArgument.always_encode')) {
+                    content = content.replace(/(<elementProp[^>]*elementType="HTTPArgument">)/, '$1\n        <boolProp name="HTTPArgument.always_encode">false</boolProp>');
+                }
+                if (!content.includes('HTTPArgument.use_equals')) {
+                    content = content.replace(/(<\/elementProp>)(?=.*HTTPArgument)/, '        <boolProp name="HTTPArgument.use_equals">true</boolProp>\n      $1');
+                }
+            }
+            return content;
+        });
+        // Fix common hashTree structure issues
+        xmlText = xmlText.replace(/<hashTree>\s*<hashTree\/>\s*<\/hashTree>/g, '<hashTree/>');
+        xmlText = xmlText.replace(/<hashTree>\s*<hashTree>\s*<\/hashTree>\s*<\/hashTree>/g, '<hashTree/>');
+        // Ensure every major test element is followed by hashTree
+        const elements = ['HTTPSamplerProxy', 'HeaderManager', 'ResponseAssertion', 'ThreadGroup', 'Arguments'];
+        for (const element of elements) {
+            const pattern = new RegExp(`(</${element}>)(?!\\s*<hashTree)`, 'g');
+            xmlText = xmlText.replace(pattern, '$1\n        <hashTree/>');
+        }
+        // Fix missing guiclass attributes
+        const guiclassMappings = {
+            '<TestPlan\\s+testclass="TestPlan"(?!\\s+guiclass=)': '<TestPlan testclass="TestPlan" guiclass="TestPlanGui"',
+            '<ThreadGroup\\s+testclass="ThreadGroup"(?!\\s+guiclass=)': '<ThreadGroup testclass="ThreadGroup" guiclass="ThreadGroupGui"',
+            '<HTTPSamplerProxy\\s+testclass="HTTPSamplerProxy"(?!\\s+guiclass=)': '<HTTPSamplerProxy testclass="HTTPSamplerProxy" guiclass="HttpTestSampleGui"',
+            '<Arguments\\s+testclass="Arguments"(?!\\s+guiclass=)': '<Arguments testclass="Arguments" guiclass="ArgumentsPanel"',
+            '<ResponseAssertion\\s+testclass="ResponseAssertion"(?!\\s+guiclass=)': '<ResponseAssertion testclass="ResponseAssertion" guiclass="AssertionGui"',
+            '<HeaderManager\\s+testclass="HeaderManager"(?!\\s+guiclass=)': '<HeaderManager testclass="HeaderManager" guiclass="HeaderPanel"',
+            '<LoopController\\s+testclass="LoopController"(?!\\s+guiclass=)': '<LoopController testclass="LoopController" guiclass="LoopControlPanel"',
+        };
+        for (const [pattern, replacement] of Object.entries(guiclassMappings)) {
+            xmlText = xmlText.replace(new RegExp(pattern, 'g'), replacement);
+        }
+        // Normalize enabled="true" attributes
+        const elementsToNormalize = ['TestPlan', 'ThreadGroup', 'HTTPSamplerProxy', 'Arguments', 'ResponseAssertion', 'HeaderManager'];
+        for (const element of elementsToNormalize) {
+            xmlText = xmlText.replace(new RegExp(`(<${element}[^>]*)(?<!enabled="true")>`, 'g'), '$1 enabled="true">');
+        }
+        // Fix ArrayList issues - comprehensive check
+        if (xmlText.includes('ArrayList')) {
+            console.log('[JMX Sanitizer] Found ArrayList references, fixing...');
+            // Fix HTTPsampler.Files with ArrayList
+            xmlText = xmlText.replace(/<elementProp name="HTTPsampler\.Files" elementType="ArrayList"\/>/g, '<elementProp name="HTTPsampler.Files" elementType="HTTPFileArgs">\n          <collectionProp name="HTTPFileArgs.files"/>\n        </elementProp>');
+            xmlText = xmlText.replace(/<elementProp name="HTTPsampler\.Files" elementType="ArrayList"[^>]*>.*?<\/elementProp>/gs, '<elementProp name="HTTPsampler.Files" elementType="HTTPFileArgs">\n          <collectionProp name="HTTPFileArgs.files"/>\n        </elementProp>');
+            // Remove any other ArrayList elementProp (self-closing or with content)
+            xmlText = xmlText.replace(/<elementProp[^>]*elementType="ArrayList"[^>]*\/>/g, '');
+            xmlText = xmlText.replace(/<elementProp[^>]*elementType="ArrayList"[^>]*>.*?<\/elementProp>/gs, '');
+            // Fix any collectionProp that might contain ArrayList references
+            xmlText = xmlText.replace(/<collectionProp name="Arguments\.arguments">\s*<ArrayList\/>/g, '<collectionProp name="Arguments.arguments">');
+            xmlText = xmlText.replace(/<collectionProp name="Arguments\.arguments">\s*<ArrayList>.*?<\/ArrayList>/gs, '<collectionProp name="Arguments.arguments">');
+            // Remove any standalone <ArrayList> tags
+            xmlText = xmlText.replace(/<ArrayList\s*\/>/g, '');
+            xmlText = xmlText.replace(/<ArrayList>.*?<\/ArrayList>/gs, '');
+            console.log('[JMX Sanitizer] ArrayList references removed');
+        }
+        console.log('[JMX Sanitizer] Sanitization complete');
+        // Note: XML validation is skipped in TypeScript version
+        // JMeter will validate when loading the file
+    }
+    catch (error) {
+        console.error(`[JMX Sanitizer] Error during sanitization: ${error}`);
+    }
+    return xmlText;
+}
+//# sourceMappingURL=sanitizer.js.map

package/dist/tools/jmx-generator/sanitizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sanitizer.js","sourceRoot":"","sources":["../../../src/tools/jmx-generator/sanitizer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH;;;;GAIG;AACH,MAAM,UAAU,WAAW,CAAC,OAAe;IACzC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;QAExD,gFAAgF;QAChF,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,8FAA8F,EAC9F,+DAA+D,CAChE,CAAC;QAEF,mGAAmG;QACnG,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CACjC,sHAAsH,CACvH,CAAC;QAEF,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CAAC,oFAAoF,CAAC,CAAC;YAClG,MAAM,gBAAgB,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAE1C,8DAA8D;YAC9D,MAAM,gBAAgB,GAAG;;cAEjB,gBAAgB,CAAC,IAAI,EAAE;;;oBAGjB,CAAC;YAEf,sDAAsD;YACtD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,2EAA2E,EAC3E,EAAE,CACH,CAAC;YAEF,qEAAqE;YACrE,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,+BAA+B,EAC/B,4CAA4C,gBAAgB,YAAY,CAEzE,CAAC;YAEF,MAAM,QAAQ,GAAG,CAAC,gBAAgB,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YACvE,OAAO,CAAC,GAAG,CAAC,gEAAgE,QAAQ,YAAY,CAAC,CAAC;QACpG,CAAC;QAED,oDAAoD;QACpD,IAAI,oDAAoD,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACvE,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,CAAC;YAClE,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,wDAAwD,EACxD,gIAAgI,CACjI,CAAC;QACJ,CAAC;QAED,mDAAmD;QACnD,IAAI,0EAA0E,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YAC7F,OAAO,CAAC,GAAG,CAAC,kEAAkE,CAAC,CAAC;YAChF,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,uGAAuG,EACvG,uJAAuJ,CACxJ,CAAC;QACJ,CAAC;QAED,iCAAiC;QACjC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,gCAAgC,EAAE,wBAAwB,CAAC,CAAC;QAEtF,kCAAkC;QAClC,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;YACjE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;YACxD,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,kBAAkB,EAAE,aAAa,CAAC,CAAC;QAC/D,CAAC;QAED,qDAAqD;QACrD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,oFAAoF,EACpF,kBAAkB,CACnB,CAAC;QAEF,kDAAkD;QAClD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,sEAAsE,EACtE,uBAAuB,CACxB,CAAC;QAEF,uEAAuE;QACvE,IAAI,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,CAAC;YAC7C,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,qEAAqE,EACrE,iBAAiB,CAClB,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,uEAAuE,CAAC,CAAC;QACvF,CAAC;QAED,sEAAsE;QACtE,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,6CAA6C,EAC7C,iBAAiB,CAClB,CAAC;QAEF,wFAAwF;QACxF,IAAI,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,gFAAgF,CAAC,CAAC;YAC9F,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,4GAA4G,EAC5G,IAAI,CACL,CAAC;QACJ,CAAC;QAED,6DAA6D;QAC7D,IAAI,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,iEAAiE,CAAC,CAAC;YAE/E,qEAAqE;YACrE,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,8EAA8E,EAC9E;;;;qBAIa,CACd,CAAC;YAEF,+CAA+C;YAC/C,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,gCAAgC,EAChC;;;;qBAIa,CACd,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,8FAA8F,EAC9F,6CAA6C,CAC9C,CAAC;QACF,OAAO,CAAC,GAAG,CAAC,2DAA2D,CAAC,CAAC;QAEzE,4DAA4D;QAC5D,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,sEAAsE,EACtE,CAAC,KAAK,EAAE,EAAE;YACR,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,IAAI,OAAO,CAAC,QAAQ,CAAC,wBAAwB,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAC,EAAE,CAAC;gBAC5F,OAAO,CAAC,GAAG,CAAC,6EAA6E,CAAC,CAAC;gBAC3F,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,wDAAwD,EACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,wBAAwB,EAAE,4BAA4B,CAAC,CACzE,CAAC;gBACF,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,CAAC;oBACpD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,gDAAgD,EAChD,0EAA0E,CAC3E,CAAC;gBACJ,CAAC;gBACD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,CAAC;oBACjD,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,qCAAqC,EACrC,4EAA4E,CAC7E,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CACF,CAAC;QAEF,uCAAuC;QACvC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,2CAA2C,EAAE,aAAa,CAAC,CAAC;QACtF,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,wDAAwD,EAAE,aAAa,CAAC,CAAC;QAEnG,0DAA0D;QAC1D,MAAM,QAAQ,GAAG,CAAC,kBAAkB,EAAE,eAAe,EAAE,mBAAmB,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC;QACxG,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,IAAI,MAAM,CAAC,MAAM,OAAO,qBAAqB,EAAE,GAAG,CAAC,CAAC;YACpE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;QAChE,CAAC;QAED,kCAAkC;QAClC,MAAM,gBAAgB,GAA2B;YAC/C,oDAAoD,EAAE,uDAAuD;YAC7G,0DAA0D,EAAE,gEAAgE;YAC5H,oEAAoE,EAAE,6EAA6E;YACnJ,sDAAsD,EAAE,4DAA4D;YACpH,sEAAsE,EAAE,0EAA0E;YAClJ,8DAA8D,EAAE,iEAAiE;YACjI,gEAAgE,EAAE,wEAAwE;SAC3I,CAAC;QAEF,KAAK,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACtE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,WAAW,CAAC,CAAC;QACnE,CAAC;QAED,sCAAsC;QACtC,MAAM,mBAAmB,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,kBAAkB,EAAE,WAAW,EAAE,mBAAmB,EAAE,eAAe,CAAC,CAAC;QAC/H,KAAK,MAAM,OAAO,IAAI,mBAAmB,EAAE,CAAC;YAC1C,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,IAAI,MAAM,CAAC,KAAK,OAAO,4BAA4B,EAAE,GAAG,CAAC,EACzD,oBAAoB,CACrB,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,uDAAuD,CAAC,CAAC;YAErE,uCAAuC;YACvC,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,oEAAoE,EACpE,kJAAkJ,CACnJ,CAAC;YACF,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,0FAA0F,EAC1F,kJAAkJ,CACnJ,CAAC;YAEF,wEAAwE;YACxE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,mDAAmD,EAAE,EAAE,CAAC,CAAC;YACnF,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,oEAAoE,EAAE,EAAE,CAAC,CAAC;YAEpG,iEAAiE;YACjE,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,+DAA+D,EAC/D,6CAA6C,CAC9C,CAAC;YACF,OAAO,GAAG,OAAO,CAAC,OAAO,CACvB,8EAA8E,EAC9E,6CAA6C,CAC9C,CAAC;YAEF,yCAAyC;YACzC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;YACnD,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,+BAA+B,EAAE,EAAE,CAAC,CAAC;YAE/D,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;QAErD,wDAAwD;QACxD,6CAA6C;IAE/C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,8CAA8C,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@performance-agent/mcp-server",
-  "version": "1.0.1",
+  "version": "1.0.6",
   "description": "MCP Server for Performance Testing Agent - JMX generation, test execution, and result analysis",
   "main": "dist/index.js",
   "type": "module",
@@ -8,7 +8,7 @@
     "performance-agent-mcp": "./dist/index.js"
   },
   "scripts": {
-    "build": "tsc && chmod +x dist/index.js",
+    "build": "tsc && node scripts/copy-prompts.js",
     "dev": "tsx src/index.ts",
     "prepare": "npm run build",
     "test": "jest"
@@ -34,11 +34,14 @@
     "escomplex": "^2.0.0-alpha",
     "fs-extra": "^11.2.0",
     "glob": "^10.3.0",
+    "openai": "^6.16.0",
     "simple-git": "^3.21.0"
   },
   "devDependencies": {
     "@types/fs-extra": "^11.0.4",
     "@types/node": "^20.10.0",
+    "cpy-cli": "^6.0.0",
+    "cross-env": "^10.1.0",
     "tsx": "^4.7.0",
     "typescript": "^5.3.0"
   },

package/scripts/copy-prompts.js

@@ -0,0 +1,46 @@
+#!/usr/bin/env node
+
+/**
+ * Cross-platform script to copy prompts directory
+ * Works on Windows, macOS, and Linux
+ */
+
+import fs from 'fs-extra';
+import path from 'path';
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const projectRoot = path.resolve(__dirname, '..');
+const srcPrompts = path.join(projectRoot, 'src', 'prompts');
+const distPrompts = path.join(projectRoot, 'dist', 'prompts');
+const distIndex = path.join(projectRoot, 'dist', 'index.js');
+
+async function copyPrompts() {
+  try {
+    // Copy prompts directory
+    console.log('📁 Copying prompts directory...');
+    await fs.copy(srcPrompts, distPrompts, { overwrite: true });
+    console.log('✅ Prompts copied successfully');
+
+    // Make index.js executable (Unix/Linux/macOS only, no-op on Windows)
+    if (process.platform !== 'win32') {
+      try {
+        await fs.chmod(distIndex, 0o755);
+        console.log('✅ Made index.js executable');
+      } catch (err) {
+        console.warn('⚠️  Could not make index.js executable (not critical):', err.message);
+      }
+    } else {
+      console.log('ℹ️  Skipping chmod on Windows (not needed)');
+    }
+
+    console.log('🎉 Build post-processing complete!');
+  } catch (error) {
+    console.error('❌ Error during build post-processing:', error);
+    process.exit(1);
+  }
+}
+
+copyPrompts();