@neverinfamous/mysql-mcp 2.1.0 → 2.3.0

package/dist/adapters/mysql/tools/codemode/index.js

@@ -0,0 +1,203 @@
+/**
+ * mysql-mcp - Code Mode Tool: mysql_execute_code
+ *
+ * MCP tool that executes LLM-generated code in a sandboxed environment
+ * with access to all MySQL tools via the mysql.* API.
+ */
+import { z } from "zod";
+import { createSandboxPool, } from "../../../../codemode/sandbox-factory.js";
+import { CodeModeSecurityManager } from "../../../../codemode/security.js";
+import { createMysqlApi } from "../../../../codemode/api.js";
+// Schema for mysql_execute_code input
+export const ExecuteCodeSchema = z.object({
+    code: z
+        .string()
+        .describe("TypeScript/JavaScript code to execute. Use mysql.{group}.{method}() for database operations."),
+    timeout: z
+        .number()
+        .optional()
+        .describe("Execution timeout in milliseconds (max 30000, default 30000)"),
+    readonly: z
+        .boolean()
+        .optional()
+        .describe("If true, restricts to read-only operations"),
+});
+// Schema for mysql_execute_code output
+export const ExecuteCodeOutputSchema = z.object({
+    success: z.boolean().describe("Whether the code executed successfully"),
+    result: z
+        .unknown()
+        .optional()
+        .describe("Return value from the executed code"),
+    error: z.string().optional().describe("Error message if execution failed"),
+    metrics: z
+        .object({
+        wallTimeMs: z
+            .number()
+            .describe("Wall clock execution time in milliseconds"),
+        cpuTimeMs: z.number().describe("CPU time used in milliseconds"),
+        memoryUsedMb: z.number().describe("Memory used in megabytes"),
+    })
+        .optional()
+        .describe("Execution performance metrics"),
+    hint: z.string().optional().describe("Helpful tip or additional information"),
+});
+// Singleton instances (initialized on first use)
+let sandboxPool = null;
+let securityManager = null;
+/**
+ * Get isolation mode from environment variable
+ */
+function getIsolationMode() {
+    const envMode = process.env["CODEMODE_ISOLATION"];
+    if (envMode === "worker")
+        return "worker";
+    return "vm"; // Default
+}
+/**
+ * Initialize Code Mode infrastructure
+ */
+function ensureInitialized() {
+    sandboxPool ??= createSandboxPool(getIsolationMode());
+    sandboxPool.initialize();
+    securityManager ??= new CodeModeSecurityManager();
+    return { pool: sandboxPool, security: securityManager };
+}
+/**
+ * Create the mysql_execute_code tool
+ */
+export function createExecuteCodeTool(adapter) {
+    return {
+        name: "mysql_execute_code",
+        title: "MySQL Execute Code",
+        description: `Execute TypeScript/JavaScript code in a sandboxed environment with access to all MySQL tools via the mysql.* API.
+
+Available API groups:
+- mysql.core: readQuery, writeQuery, listTables, describeTable, createTable, createIndex (8 methods)
+- mysql.transactions: begin, commit, rollback, savepoint, execute (7 methods)
+- mysql.json: extract, set, insert, remove, contains, keys, merge, diff, stats (17 methods)
+- mysql.text: regexpMatch, likeSearch, soundex, substring, concat, collationConvert (6 methods)
+- mysql.fulltext: fulltextSearch, fulltextCreate, fulltextBoolean, fulltextExpand (5 methods)
+- mysql.performance: explain, explainAnalyze, slowQueries, bufferPoolStats, tableStats (8 methods)
+- mysql.optimization: indexRecommendation, queryRewrite, forceIndex, optimizerTrace (4 methods)
+- mysql.admin: optimizeTable, analyzeTable, checkTable, repairTable, flushTables, killQuery (6 methods)
+- mysql.monitoring: showProcesslist, showStatus, showVariables, innodbStatus, poolStats (7 methods)
+- mysql.backup: createDump, exportTable, importData, restoreDump (4 methods)
+- mysql.replication: masterStatus, slaveStatus, binlogEvents, gtidStatus, replicationLag (5 methods)
+- mysql.partitioning: partitionInfo, addPartition, dropPartition, reorganizePartition (4 methods)
+- mysql.schema: listSchemas, createView, listFunctions, listTriggers (10 methods)
+- mysql.events: eventCreate, eventAlter, eventDrop, eventList, schedulerStatus (6 methods)
+- mysql.sysschema: sysSchemaStats, sysStatementSummary, sysIoSummary (8 methods)
+- mysql.stats: descriptive, percentiles, correlation, regression, timeSeries, histogram (8 methods)
+- mysql.spatial: distance, distanceSphere, point, polygon, buffer (12 methods)
+- mysql.security: sslStatus, userPrivileges, audit, sensitiveTables (9 methods)
+- mysql.cluster: clusterStatus, grStatus, grMembers, clusterTopology (10 methods)
+- mysql.roles: roleCreate, roleGrant, roleAssign, roleList (8 methods)
+- mysql.docstore: docCreateCollection, docFind, docAdd, docModify (9 methods)
+- mysql.router: routerStatus, routerRoutes, routerRouteHealth (9 methods)
+
+Example:
+\`\`\`javascript
+const tables = await mysql.core.listTables();
+const results = [];
+for (const t of tables.tables) {
+    const count = await mysql.core.readQuery(\`SELECT COUNT(*) as n FROM \\\`\${t.name}\\\`\`);
+    results.push({ table: t.name, rows: count.rows[0].n });
+}
+return results;
+\`\`\``,
+        group: "codemode",
+        inputSchema: ExecuteCodeSchema,
+        requiredScopes: ["admin"],
+        annotations: {
+            readOnlyHint: false,
+            destructiveHint: true,
+            idempotentHint: false,
+            openWorldHint: false,
+        },
+        handler: async (params, _context) => {
+            const { code, readonly } = params;
+            // Initialize infrastructure
+            const { pool, security } = ensureInitialized();
+            // Validate code
+            const validation = security.validateCode(code);
+            if (!validation.valid) {
+                return {
+                    success: false,
+                    error: `Code validation failed: ${validation.errors.join("; ")}`,
+                    metrics: { wallTimeMs: 0, cpuTimeMs: 0, memoryUsedMb: 0 },
+                };
+            }
+            // Check rate limit
+            const clientId = "default";
+            if (!security.checkRateLimit(clientId)) {
+                return {
+                    success: false,
+                    error: "Rate limit exceeded. Please wait before executing more code.",
+                    metrics: { wallTimeMs: 0, cpuTimeMs: 0, memoryUsedMb: 0 },
+                };
+            }
+            // Create mysql API bindings
+            const mysqlApi = createMysqlApi(adapter);
+            const bindings = mysqlApi.createSandboxBindings();
+            // Validate bindings are populated
+            const totalMethods = Object.values(bindings).reduce((sum, group) => {
+                if (typeof group === "object" && group !== null) {
+                    return sum + Object.keys(group).length;
+                }
+                return sum;
+            }, 0);
+            if (totalMethods === 0) {
+                return {
+                    success: false,
+                    error: "mysql.* API not available: no tool bindings were created. Ensure adapter.getToolDefinitions() returns valid tools.",
+                    metrics: { wallTimeMs: 0, cpuTimeMs: 0, memoryUsedMb: 0 },
+                };
+            }
+            // Capture active transactions before execution for cleanup on error
+            const transactionsBefore = new Set(adapter.getActiveTransactionIds());
+            // Execute in sandbox
+            const result = await pool.execute(code, bindings);
+            // Always cleanup orphaned transactions (uncommitted txns from any execution)
+            const transactionsAfter = adapter.getActiveTransactionIds();
+            const orphanedTransactions = transactionsAfter.filter((txId) => !transactionsBefore.has(txId));
+            for (const txId of orphanedTransactions) {
+                try {
+                    await adapter.rollbackTransaction(txId);
+                }
+                catch {
+                    // Best-effort cleanup
+                }
+            }
+            // Sanitize result
+            if (result.success && result.result !== undefined) {
+                result.result = security.sanitizeResult(result.result);
+            }
+            // Audit log
+            const record = security.createExecutionRecord(code, result, readonly ?? false, clientId);
+            security.auditLog(record);
+            // Add help hint for discoverability
+            const helpHint = "Tip: Use mysql.help() to list all groups, or mysql.core.help() for group-specific methods.";
+            return {
+                ...result,
+                hint: helpHint,
+            };
+        },
+    };
+}
+/**
+ * Get all Code Mode tools
+ */
+export function getCodeModeTools(adapter) {
+    return [createExecuteCodeTool(adapter)];
+}
+/**
+ * Cleanup Code Mode resources (call on server shutdown)
+ */
+export function cleanupCodeMode() {
+    if (sandboxPool) {
+        sandboxPool.dispose();
+        sandboxPool = null;
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/adapters/mysql/tools/codemode/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/adapters/mysql/tools/codemode/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAMxB,OAAO,EACL,iBAAiB,GAGlB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,kCAAkC,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAG7D,sCAAsC;AACtC,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,IAAI,EAAE,CAAC;SACJ,MAAM,EAAE;SACR,QAAQ,CACP,8FAA8F,CAC/F;IACH,OAAO,EAAE,CAAC;SACP,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,8DAA8D,CAAC;IAC3E,QAAQ,EAAE,CAAC;SACR,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,QAAQ,CAAC,4CAA4C,CAAC;CAC1D,CAAC,CAAC;AAEH,uCAAuC;AACvC,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IACvE,MAAM,EAAE,CAAC;SACN,OAAO,EAAE;SACT,QAAQ,EAAE;SACV,QAAQ,CAAC,qCAAqC,CAAC;IAClD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;IAC1E,OAAO,EAAE,CAAC;SACP,MAAM,CAAC;QACN,UAAU,EAAE,CAAC;aACV,MAAM,EAAE;aACR,QAAQ,CAAC,2CAA2C,CAAC;QACxD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,+BAA+B,CAAC;QAC/D,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;KAC9D,CAAC;SACD,QAAQ,EAAE;SACV,QAAQ,CAAC,+BAA+B,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;CAC9E,CAAC,CAAC;AAEH,iDAAiD;AACjD,IAAI,WAAW,GAAwB,IAAI,CAAC;AAC5C,IAAI,eAAe,GAAmC,IAAI,CAAC;AAE3D;;GAEG;AACH,SAAS,gBAAgB;IACvB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAClD,IAAI,OAAO,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC1C,OAAO,IAAI,CAAC,CAAC,UAAU;AACzB,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB;IAIxB,WAAW,KAAK,iBAAiB,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACtD,WAAW,CAAC,UAAU,EAAE,CAAC;IACzB,eAAe,KAAK,IAAI,uBAAuB,EAAE,CAAC;IAClD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC;AAC1D,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,OAAqB;IACzD,OAAO;QACL,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE,oBAAoB;QAC3B,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmCV;QACH,KAAK,EAAE,UAAU;QACjB,WAAW,EAAE,iBAAiB;QAC9B,cAAc,EAAE,CAAC,OAAO,CAAC;QACzB,WAAW,EAAE;YACX,YAAY,EAAE,KAAK;YACnB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,KAAK;YACrB,aAAa,EAAE,KAAK;SACrB;QACD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,MAA4B,CAAC;YAExD,4BAA4B;YAC5B,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,iBAAiB,EAAE,CAAC;YAE/C,gBAAgB;YAChB,MAAM,UAAU,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC/C,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACtB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,2BAA2B,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;oBAChE,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;iBAC1D,CAAC;YACJ,CAAC;YAED,mBAAmB;YACnB,MAAM,QAAQ,GAAG,SAAS,CAAC;YAC3B,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8DAA8D;oBACrE,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;iBAC1D,CAAC;YACJ,CAAC;YAED,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;YACzC,MAAM,QAAQ,GAAG,QAAQ,CAAC,qBAAqB,EAAE,CAAC;YAElD,kCAAkC;YAClC,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CACjD,CAAC,GAAW,EAAE,KAAK,EAAE,EAAE;gBACrB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;oBAChD,OAAO,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;gBACzC,CAAC;gBACD,OAAO,GAAG,CAAC;YACb,CAAC,EACD,CAAC,CACF,CAAC;YACF,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EACH,oHAAoH;oBACtH,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE;iBAC1D,CAAC;YACJ,CAAC;YAED,oEAAoE;YACpE,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,uBAAuB,EAAE,CAAC,CAAC;YAEtE,qBAAqB;YACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAElD,6EAA6E;YAC7E,MAAM,iBAAiB,GAAG,OAAO,CAAC,uBAAuB,EAAE,CAAC;YAC5D,MAAM,oBAAoB,GAAG,iBAAiB,CAAC,MAAM,CACnD,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAChD,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,oBAAoB,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;gBAC1C,CAAC;gBAAC,MAAM,CAAC;oBACP,sBAAsB;gBACxB,CAAC;YACH,CAAC;YAED,kBAAkB;YAClB,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAClD,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACzD,CAAC;YAED,YAAY;YACZ,MAAM,MAAM,GAAG,QAAQ,CAAC,qBAAqB,CAC3C,IAAI,EACJ,MAAM,EACN,QAAQ,IAAI,KAAK,EACjB,QAAQ,CACT,CAAC;YACF,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAE1B,oCAAoC;YACpC,MAAM,QAAQ,GACZ,4FAA4F,CAAC;YAE/F,OAAO;gBACL,GAAG,MAAM;gBACT,IAAI,EAAE,QAAQ;aACf,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,OAAqB;IACpD,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;AAC1C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,IAAI,WAAW,EAAE,CAAC;QAChB,WAAW,CAAC,OAAO,EAAE,CAAC;QACtB,WAAW,GAAG,IAAI,CAAC;IACrB,CAAC;AACH,CAAC"}

package/dist/adapters/mysql/tools/core.d.ts

@@ -4,8 +4,8 @@
  * Fundamental database operations: read, write, table management, indexes.
  * 8 tools total.
  */
-import type { MySQLAdapter } from '../MySQLAdapter.js';
-import type { ToolDefinition } from '../../../types/index.js';
+import type { MySQLAdapter } from "../MySQLAdapter.js";
+import type { ToolDefinition } from "../../../types/index.js";
 /**
  * Get all core database tools
  */

package/dist/adapters/mysql/tools/core.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"core.d.ts","sourceRoot":"","sources":["../../../../src/adapters/mysql/tools/core.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,yBAAyB,CAAC;AA8B9E;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,YAAY,GAAG,cAAc,EAAE,CAWpE"}
+{"version":3,"file":"core.d.ts","sourceRoot":"","sources":["../../../../src/adapters/mysql/tools/core.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,yBAAyB,CAAC;AA4C9E;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,YAAY,GAAG,cAAc,EAAE,CAWpE"}