@tng-sh/mcp-server 1.0.0

package/dist/tools/xray.js

@@ -0,0 +1,187 @@
+import { exec, execFile } from 'child_process';
+import { promisify } from 'util';
+import { resolve } from 'path';
+import { findProjectRoot, getTngCommandConfig, getShellEnv, shellEscape } from './utils.js';
+const execAsync = promisify(exec);
+const execFileAsync = promisify(execFile);
+export async function xrayMethod(args) {
+    const { file_path, method_name, project_root } = args;
+    try {
+        const absoluteFilePath = resolve(file_path);
+        const cwd = project_root || findProjectRoot(absoluteFilePath) || process.cwd();
+        const tngCommand = getTngCommandConfig(cwd, file_path);
+        // Build xray command with JSON output
+        // For JS: tng xray --json -f "file" -m "method"
+        // For Python: tng xray --json -f "file" -m "method"
+        // For Ruby: tng --file "file" --method "method" --xray --json
+        const ext = file_path.split('.').pop()?.toLowerCase();
+        let command;
+        let baseArgs = [];
+        if (ext === 'rb') {
+            baseArgs = ['--json', '-f', file_path, '-m', method_name, '--xray'];
+        }
+        else {
+            // JS/Python have dedicated xray command
+            baseArgs = ['xray', '--json', '-f', file_path, '-m', method_name];
+        }
+        command = tngCommand.shell
+            ? `${tngCommand.cmd} ${baseArgs.map(shellEscape).join(' ')}`
+            : `${tngCommand.cmd} ${[...tngCommand.args, ...baseArgs].map(shellEscape).join(' ')}`;
+        console.error(`[TNG MCP] Running: ${command}`);
+        console.error(`[TNG MCP] CWD: ${cwd}`);
+        const { stdout, stderr } = tngCommand.shell
+            ? await execAsync(command, {
+                cwd,
+                env: getShellEnv(),
+                timeout: 120000, // 2 minutes
+                maxBuffer: 10 * 1024 * 1024,
+            })
+            : await execFileAsync(tngCommand.cmd, [...tngCommand.args, ...baseArgs], {
+                cwd,
+                env: getShellEnv(),
+                timeout: 120000,
+                maxBuffer: 10 * 1024 * 1024,
+            });
+        if (stderr && !stderr.includes('TNG')) {
+            console.error('[TNG MCP] stderr:', stderr);
+        }
+        // Parse JSON output
+        const lines = stdout.trim().split('\n');
+        let result = null;
+        for (const line of lines) {
+            if (!line.trim())
+                continue;
+            try {
+                const event = JSON.parse(line);
+                if (event.type === 'result' || event.mermaid || event.diagram) {
+                    result = event;
+                    break;
+                }
+                if (event.type === 'error' || event.type === 'auth_error') {
+                    throw new Error(event.message || 'X-Ray generation failed');
+                }
+            }
+            catch (e) {
+                // Try parsing as single JSON
+                if (line.includes('{') && (line.includes('mermaid') || line.includes('diagram'))) {
+                    result = JSON.parse(line);
+                }
+            }
+        }
+        if (!result) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `No X-Ray results returned. Raw output:\n${stdout.substring(0, 500)}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        const formattedOutput = formatXRayResult(result, file_path, method_name);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: formattedOutput,
+                },
+            ],
+        };
+    }
+    catch (error) {
+        if (error.code === 'ENOENT') {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Error: TNG CLI not found in project. Please install TNG first.`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        if (error.killed) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: `Error: X-Ray generation timed out after 2 minutes.`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        let errorMessage = error.message;
+        try {
+            const errorOutput = error.stdout || error.stderr || '';
+            const lines = errorOutput.split('\n');
+            for (const line of lines) {
+                if (!line.trim())
+                    continue;
+                try {
+                    const event = JSON.parse(line);
+                    if (event.type === 'error' || event.type === 'auth_error') {
+                        errorMessage = event.message || errorMessage;
+                        break;
+                    }
+                }
+                catch { }
+            }
+        }
+        catch { }
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Error generating X-Ray for ${file_path}#${method_name}: ${errorMessage}`,
+                },
+            ],
+            isError: true,
+        };
+    }
+}
+function formatXRayResult(result, filePath, methodName) {
+    let mermaidCode = result.mermaid || result.diagram || result.visualization || '';
+    const className = result.class_name || result.className || '';
+    const fullName = className ? `${className}#${methodName}` : methodName;
+    // Fallback: If no mermaid code but we have a timeline (Ruby trace), generate a flowchart
+    if (!mermaidCode && Array.isArray(result.timeline)) {
+        mermaidCode = 'graph TD\n';
+        mermaidCode += `  Start([Start: ${methodName}])\n`;
+        let prevNode = 'Start';
+        result.timeline.forEach((step, idx) => {
+            const nodeId = `step${idx}`;
+            const label = step.code.replace(/"/g, "'");
+            const action = step.action || 'STEP';
+            // Map actions to shapes
+            if (action === 'IF' || action === 'SWITCH' || action === 'GUARD_CLAUSE') {
+                mermaidCode += `  ${nodeId}{"${label}"}\n`;
+            }
+            else if (action === 'CALL' || action === 'ASSIGNMENT') {
+                mermaidCode += `  ${nodeId}["${label}"]\n`;
+            }
+            else if (action === 'EXIT' || action === 'FAILURE') {
+                mermaidCode += `  ${nodeId}(("${label}"))\n`;
+            }
+            else {
+                mermaidCode += `  ${nodeId}["${label}"]\n`;
+            }
+            mermaidCode += `  ${prevNode} --> ${nodeId}\n`;
+            prevNode = nodeId;
+        });
+        mermaidCode += `  ${prevNode} --> End([End])\n`;
+    }
+    let output = `# X-Ray Visualization for ${fullName}\n\n`;
+    output += `**File:** ${filePath}\n\n`;
+    if (mermaidCode) {
+        output += `## Mermaid Flowchart\n\n`;
+        output += `\`\`\`mermaid\n${mermaidCode}\n\`\`\`\n\n`;
+        output += `> **Tip:** Copy the Mermaid code above and paste it into [mermaid.live](https://mermaid.live) to view the interactive diagram.\n`;
+    }
+    else {
+        output += `⚠️ No Mermaid diagram was generated.\n`;
+    }
+    return output;
+}
+//# sourceMappingURL=xray.js.map

package/dist/tools/xray.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"xray.js","sourceRoot":"","sources":["../../src/tools/xray.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,MAAM,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC/B,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE5F,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;AAClC,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAQ1C,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,IAAc;IAC3C,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC;IAEtD,IAAI,CAAC;QACD,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,GAAG,GAAG,YAAY,IAAI,eAAe,CAAC,gBAAgB,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAC/E,MAAM,UAAU,GAAG,mBAAmB,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QAEvD,sCAAsC;QACtC,gDAAgD;QAChD,oDAAoD;QACpD,8DAA8D;QAC9D,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,CAAC;QACtD,IAAI,OAAe,CAAC;QACpB,IAAI,QAAQ,GAAa,EAAE,CAAC;QAE5B,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;YACf,QAAQ,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;QACxE,CAAC;aAAM,CAAC;YACJ,wCAAwC;YACxC,QAAQ,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QACtE,CAAC;QAED,OAAO,GAAG,UAAU,CAAC,KAAK;YACtB,CAAC,CAAC,GAAG,UAAU,CAAC,GAAG,IAAI,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;YAC5D,CAAC,CAAC,GAAG,UAAU,CAAC,GAAG,IAAI,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QAE1F,OAAO,CAAC,KAAK,CAAC,sBAAsB,OAAO,EAAE,CAAC,CAAC;QAC/C,OAAO,CAAC,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC,CAAC;QAEvC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,KAAK;YACvC,CAAC,CAAC,MAAM,SAAS,CAAC,OAAO,EAAE;gBACvB,GAAG;gBACH,GAAG,EAAE,WAAW,EAAE;gBAClB,OAAO,EAAE,MAAM,EAAE,YAAY;gBAC7B,SAAS,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;aAC9B,CAAC;YACF,CAAC,CAAC,MAAM,aAAa,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,GAAG,QAAQ,CAAC,EAAE;gBACrE,GAAG;gBACH,GAAG,EAAE,WAAW,EAAE;gBAClB,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;aAC9B,CAAC,CAAC;QAEP,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,CAAC,KAAK,CAAC,mBAAmB,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC;QAED,oBAAoB;QACpB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,MAAM,GAAG,IAAI,CAAC;QAElB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAC3B,IAAI,CAAC;gBACD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC/B,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAC5D,MAAM,GAAG,KAAK,CAAC;oBACf,MAAM;gBACV,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACxD,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,IAAI,yBAAyB,CAAC,CAAC;gBAChE,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,6BAA6B;gBAC7B,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;oBAC/E,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC9B,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,2CAA2C,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE;qBAC9E;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;QAED,MAAM,eAAe,GAAG,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAEzE,OAAO;YACH,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,eAAe;iBACxB;aACJ;SACJ,CAAC;IACN,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,gEAAgE;qBACzE;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACf,OAAO;gBACH,OAAO,EAAE;oBACL;wBACI,IAAI,EAAE,MAAe;wBACrB,IAAI,EAAE,oDAAoD;qBAC7D;iBACJ;gBACD,OAAO,EAAE,IAAI;aAChB,CAAC;QACN,CAAC;QAED,IAAI,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;QACjC,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC;YACvD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oBAAE,SAAS;gBAC3B,IAAI,CAAC;oBACD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC/B,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBACxD,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,YAAY,CAAC;wBAC7C,MAAM;oBACV,CAAC;gBACL,CAAC;gBAAC,MAAM,CAAC,CAAC,CAAC;YACf,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,CAAC;QAEX,OAAO;YACH,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,8BAA8B,SAAS,IAAI,WAAW,KAAK,YAAY,EAAE;iBAClF;aACJ;YACD,OAAO,EAAE,IAAI;SAChB,CAAC;IACN,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAW,EAAE,QAAgB,EAAE,UAAkB;IACvE,IAAI,WAAW,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC;IACjF,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;IAC9D,MAAM,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,IAAI,UAAU,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;IAEvE,yFAAyF;IACzF,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjD,WAAW,GAAG,YAAY,CAAC;QAC3B,WAAW,IAAI,mBAAmB,UAAU,MAAM,CAAC;QACnD,IAAI,QAAQ,GAAG,OAAO,CAAC;QAEvB,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,GAAW,EAAE,EAAE;YAC/C,MAAM,MAAM,GAAG,OAAO,GAAG,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC;YAErC,wBAAwB;YACxB,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,cAAc,EAAE,CAAC;gBACtE,WAAW,IAAI,KAAK,MAAM,KAAK,KAAK,MAAM,CAAC;YAC/C,CAAC;iBAAM,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;gBACtD,WAAW,IAAI,KAAK,MAAM,KAAK,KAAK,MAAM,CAAC;YAC/C,CAAC;iBAAM,IAAI,MAAM,KAAK,MAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACnD,WAAW,IAAI,KAAK,MAAM,MAAM,KAAK,OAAO,CAAC;YACjD,CAAC;iBAAM,CAAC;gBACJ,WAAW,IAAI,KAAK,MAAM,KAAK,KAAK,MAAM,CAAC;YAC/C,CAAC;YAED,WAAW,IAAI,KAAK,QAAQ,QAAQ,MAAM,IAAI,CAAC;YAC/C,QAAQ,GAAG,MAAM,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,WAAW,IAAI,KAAK,QAAQ,mBAAmB,CAAC;IACpD,CAAC;IAED,IAAI,MAAM,GAAG,6BAA6B,QAAQ,MAAM,CAAC;IACzD,MAAM,IAAI,aAAa,QAAQ,MAAM,CAAC;IAEtC,IAAI,WAAW,EAAE,CAAC;QACd,MAAM,IAAI,0BAA0B,CAAC;QACrC,MAAM,IAAI,kBAAkB,WAAW,cAAc,CAAC;QACtD,MAAM,IAAI,kIAAkI,CAAC;IACjJ,CAAC;SAAM,CAAC;QACJ,MAAM,IAAI,wCAAwC,CAAC;IACvD,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/package.json

@@ -0,0 +1,44 @@
+{
+    "name": "@tng-sh/mcp-server",
+    "version": "1.0.0",
+    "description": "MCP server for TNG test generation and code auditing",
+    "type": "module",
+    "main": "dist/index.js",
+    "bin": {
+        "tng-mcp": "dist/index.js"
+    },
+    "files": [
+        "dist"
+    ],
+    "scripts": {
+        "build": "tsc",
+        "dev": "tsc --watch",
+        "start": "node dist/index.js",
+        "prepublishOnly": "npm run build"
+    },
+    "keywords": [
+        "mcp",
+        "tng",
+        "testing",
+        "code-audit",
+        "ai",
+        "assistant"
+    ],
+    "author": "Binary Dreams LLC",
+    "license": "MIT",
+    "dependencies": {
+        "@modelcontextprotocol/sdk": "^1.0.4",
+        "zod": "^3.25.0"
+    },
+    "devDependencies": {
+        "@types/node": "^20.10.0",
+        "typescript": "^5.3.3"
+    },
+    "repository": {
+        "type": "git",
+        "url": "https://github.com/tng-sh/mcp-server"
+    },
+    "publishConfig": {
+        "access": "public"
+    }
+}