feat: mysql mcp

Author: Areo-Joe

mcp/src/server.ts

@@ -1,5 +1,6 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
-import { registerDatabaseTools } from "./tools/database.js";
+import { registerDatabaseTools } from "./tools/databaseNoSQL.js";
+import { registerSQLDatabaseTools } from "./tools/databaseSQL.js";
 import { registerDownloadTools } from "./tools/download.js";
 import { registerEnvTools } from "./tools/env.js";
 import { registerFunctionTools } from "./tools/functions.js";
@@ -30,6 +31,7 @@ const DEFAULT_PLUGINS = ['env', 'database', 'functions', 'hosting', 'storage', '
 
 function registerDatabase(server: ExtendedMcpServer) {
   registerDatabaseTools(server);
+  registerSQLDatabaseTools(server);
   registerDataModelTools(server);
 }
 
@@ -176,3 +178,4 @@ export type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 export { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 export { error, info, warn } from "./utils/logger.js";
 export { reportToolCall, reportToolkitLifecycle, telemetryReporter } from "./utils/telemetry.js";
+

mcp/src/tools/database.ts renamed to mcp/src/tools/databaseNoSQL.ts

@@ -0,0 +1,165 @@
+import { z } from "zod";
+import { getCloudBaseManager, getEnvId } from "../cloudbase-manager.js";
+import { ExtendedMcpServer } from "../server.js";
+
+const CATEGORY = "SQL database";
+
+export function registerSQLDatabaseTools(server: ExtendedMcpServer) {
+  // Get cloudBaseOptions, if not available then undefined
+  const cloudBaseOptions = server.cloudBaseOptions;
+
+  // Create closure function to get CloudBase Manager
+  const getManager = () => getCloudBaseManager({ cloudBaseOptions });
+
+  // executeReadOnlySQL
+  server.registerTool?.(
+    "executeReadOnlySQL",
+    {
+      title: "Execute read-only SQL query",
+      description: "Execute a read-only SQL query on the SQL database",
+      inputSchema: {
+        sql: z.string().describe("SQL query statement (SELECT queries only)"),
+        instanceId: z
+          .string()
+          .optional()
+          .describe("Database instance ID (defaults to 'default')"),
+      },
+      annotations: {
+        readOnlyHint: true,
+        openWorldHint: true,
+        category: CATEGORY,
+      },
+    },
+    async ({ sql, instanceId = "default" }) => {
+      try {
+        const cloudbase = await getManager();
+        const envId = await getEnvId(cloudBaseOptions);
+
+        const result = await cloudbase.commonService("tcb").call({
+          Action: "RunSql",
+          Param: {
+            EnvId: envId,
+            Sql: sql,
+            DbInstance: {
+              EnvId: envId,
+              InstanceId: instanceId,
+            },
+          },
+        });
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: true,
+                  message: "SQL query executed successfully",
+                  result,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error: any) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: false,
+                  error: error.message,
+                  message: "SQL query execution failed",
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      }
+    }
+  );
+
+  // executeWriteSQL
+  server.registerTool?.(
+    "executeWriteSQL",
+    {
+      title: "Execute write SQL statement",
+      description:
+        "Execute a write SQL statement on the SQL database (INSERT, UPDATE, DELETE, etc.)",
+      inputSchema: {
+        sql: z
+          .string()
+          .describe(
+            "SQL statement (INSERT, UPDATE, DELETE, CREATE, ALTER, etc.)"
+          ),
+        instanceId: z
+          .string()
+          .optional()
+          .describe("Database instance ID (defaults to 'default')"),
+      },
+      annotations: {
+        readOnlyHint: false,
+        destructiveHint: true,
+        idempotentHint: false,
+        openWorldHint: true,
+        category: CATEGORY,
+      },
+    },
+    async ({ sql, instanceId = "default" }) => {
+      try {
+        const cloudbase = await getManager();
+        const envId = await getEnvId(cloudBaseOptions);
+
+        const result = await cloudbase.commonService("tcb").call({
+          Action: "RunSql",
+          Param: {
+            EnvId: envId,
+            Sql: sql,
+            DbInstance: {
+              EnvId: envId,
+              InstanceId: instanceId,
+            },
+          },
+        });
+
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: true,
+                  message: "SQL statement executed successfully",
+                  result,
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      } catch (error: any) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  success: false,
+                  error: error.message,
+                  message: "SQL statement execution failed",
+                },
+                null,
+                2
+              ),
+            },
+          ],
+        };
+      }
+    }
+  );
+}