@devzhou/mcp-mysql 1.0.0

package/bin/mcp-mysql.js

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+
+import("../index.js");

package/index.js

@@ -0,0 +1,258 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import mysql from "mysql2/promise";
+import fs from "fs";
+import path from "path";
+import { fileURLToPath } from "url";
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+// ─── 命令行参数解析 ───
+
+let connectionString = "";
+let skipConfirm = false;
+
+function parseArgs() {
+  const args = process.argv.slice(2);
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    const argLower = arg.toLowerCase();
+
+    // 支持 ConnectionString=xxx 和 ConnectionString xxx（忽略大小写和前导 -）
+    if (argLower.replace(/^-+/, "").startsWith("connectionstring")) {
+      const stripped = arg.replace(/^-+/, "");
+      const idx = stripped.indexOf("=");
+      if (idx !== -1) {
+        connectionString = stripped.substring(idx + 1);
+      } else if (i + 1 < args.length) {
+        connectionString = args[++i];
+      }
+    }
+
+    // 支持 SkipConfirm 和 SkipConfirm=true（忽略大小写和前导 -）
+    if (argLower.replace(/^-+/, "").startsWith("skipconfirm")) {
+      const stripped = arg.replace(/^-+/, "");
+      const idx = stripped.indexOf("=");
+      if (idx !== -1) {
+        skipConfirm = stripped.substring(idx + 1).toLowerCase() === "true";
+      } else {
+        skipConfirm = true;
+      }
+    }
+  }
+}
+
+// ─── 配置与连接字符串 ───
+
+function getConfigPath(fileName) {
+  // 优先查找当前工作目录，再查找可执行文件目录
+  const cwdPath = path.join(process.cwd(), fileName);
+  if (fs.existsSync(cwdPath)) return cwdPath;
+  return path.join(__dirname, "..", fileName);
+}
+
+function readConfig(configPath) {
+  const data = fs.readFileSync(configPath, "utf8");
+  const cfg = JSON.parse(data);
+  return cfg.database;
+}
+
+function getADOParam(params, ...keys) {
+  for (const key of keys) {
+    if (params[key.toLowerCase()] !== undefined) {
+      return params[key.toLowerCase()];
+    }
+  }
+  return undefined;
+}
+
+/**
+ * 将 ADO.NET 格式连接字符串转换为 mysql2 兼容的连接配置对象
+ * 格式: Server=host;Port=3306;Database=dbname;User=root;Password=123456
+ */
+function parseAdoNetCS(cs) {
+  cs = cs.trim();
+  const params = {};
+  cs.split(";").forEach((pair) => {
+    pair = pair.trim();
+    if (!pair) return;
+    const eqIdx = pair.indexOf("=");
+    if (eqIdx === -1) return;
+    const key = pair.substring(0, eqIdx).trim().toLowerCase();
+    const value = pair.substring(eqIdx + 1).trim();
+    params[key] = value;
+  });
+
+  const config = {};
+  config.user = getADOParam(params, "user", "uid", "username") || undefined;
+  config.password = getADOParam(params, "password", "pwd") || undefined;
+  config.host = getADOParam(params, "server", "host", "data source", "datasource") || "127.0.0.1";
+  const portStr = getADOParam(params, "port");
+  config.port = portStr ? parseInt(portStr, 10) : 3306;
+  config.database = getADOParam(params, "database", "initial catalog") || undefined;
+
+  // 额外参数
+  const charset = getADOParam(params, "charset");
+  if (charset) config.charset = charset;
+
+  return config;
+}
+
+function buildDBConfig() {
+  if (connectionString) {
+    return parseAdoNetCS(connectionString);
+  }
+
+  const configPath = getConfigPath("config.json");
+  if (fs.existsSync(configPath)) {
+    const dbCfg = readConfig(configPath);
+    return {
+      user: dbCfg.user,
+      password: dbCfg.password,
+      host: dbCfg.host,
+      port: dbCfg.port,
+      database: dbCfg.dbname,
+      charset: "utf8mb4",
+    };
+  }
+
+  throw new Error("未提供数据库连接信息：请通过 -ConnectionString 参数或 config.json 配置文件提供");
+}
+
+// ─── 数据库操作 ───
+
+async function createConnection() {
+  const config = buildDBConfig();
+  const connection = await mysql.createConnection(config);
+  await connection.ping();
+  return connection;
+}
+
+async function executeSelect(sql) {
+  const connection = await createConnection();
+  try {
+    const [rows] = await connection.query(sql);
+
+    // 处理 Buffer 类型，转换为字符串
+    const results = (rows || []).map((row) => {
+      const newRow = {};
+      for (const [key, value] of Object.entries(row)) {
+        newRow[key] = Buffer.isBuffer(value) ? value.toString("utf8") : value;
+      }
+      return newRow;
+    });
+
+    if (results.length === 0) return "[]";
+    return JSON.stringify(results);
+  } finally {
+    await connection.end();
+  }
+}
+
+async function executeExec(sql) {
+  const connection = await createConnection();
+  try {
+    const [result] = await connection.execute(sql);
+    const rowsAffected = result.affectedRows || 0;
+    return `${rowsAffected} rows affected`;
+  } finally {
+    await connection.end();
+  }
+}
+
+// ─── MCP 服务器 ───
+
+async function main() {
+  parseArgs();
+
+  const mcpServer = new McpServer({
+    name: "mcp-mysql",
+    version: "1.0.0",
+  });
+
+  // 注册 select 工具
+  mcpServer.tool(
+    "select",
+    "执行查询sql",
+    {
+      sql: z.string().describe("执行的sql语句"),
+    },
+    async ({ sql }) => {
+      try {
+        const result = await executeSelect(sql);
+        return { content: [{ type: "text", text: result }] };
+      } catch (err) {
+        return { content: [{ type: "text", text: `查询执行失败: ${err.message}` }] };
+      }
+    }
+  );
+
+  // 注册 exec 工具（带确认机制）
+  mcpServer.tool(
+    "exec",
+    "执行sql",
+    {
+      sql: z.string().describe("执行的sql语句"),
+    },
+    async ({ sql }) => {
+      try {
+        // 如果跳过确认，直接执行
+        if (skipConfirm) {
+          const result = await executeExec(sql);
+          return { content: [{ type: "text", text: result }] };
+        }
+
+        // 尝试向客户端发送确认请求（Elicitation）
+        try {
+          const elicitationResult = await mcpServer.server.elicitInput({
+            mode: "form",
+            message: `即将执行以下SQL语句，请确认是否继续执行：\n\n${sql}`,
+            requestedSchema: {
+              type: "object",
+              properties: {
+                confirm: {
+                  type: "boolean",
+                  description: "是否确认执行此SQL语句",
+                  default: false,
+                },
+              },
+              required: ["confirm"],
+            },
+          });
+
+          switch (elicitationResult.action) {
+            case "accept":
+              if (elicitationResult.content?.confirm === true) {
+                const result = await executeExec(sql);
+                return { content: [{ type: "text", text: result }] };
+              }
+              return { content: [{ type: "text", text: "用户取消了执行" }] };
+            case "decline":
+              return { content: [{ type: "text", text: "用户拒绝了执行" }] };
+            case "cancel":
+              return { content: [{ type: "text", text: "用户取消了执行" }] };
+            default:
+              return { content: [{ type: "text", text: `未知的用户响应: ${elicitationResult.action}` }] };
+          }
+        } catch (err) {
+          // 客户端不支持 elicitation，直接执行
+          const result = await executeExec(sql);
+          return { content: [{ type: "text", text: result }] };
+        }
+      } catch (err) {
+        return { content: [{ type: "text", text: err.message }] };
+      }
+    }
+  );
+
+  // 启动 STDIO 传输
+  const transport = new StdioServerTransport();
+  await mcpServer.connect(transport);
+}
+
+main().catch((err) => {
+  console.error("服务器启动失败:", err);
+  process.exit(1);
+});

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@devzhou/mcp-mysql",
+  "version": "1.0.0",
+  "description": "MCP MySQL Server - 通过MCP协议连接MySQL数据库",
+  "type": "module",
+  "main": "index.js",
+  "bin": {
+    "mcp-mysql": "./bin/mcp-mysql.js"
+  },
+  "files": [
+    "index.js",
+    "bin/"
+  ],
+  "scripts": {
+    "start": "node bin/mcp-mysql.js"
+  },
+  "keywords": [
+    "mcp",
+    "mysql",
+    "ai",
+    "database",
+    "model-context-protocol"
+  ],
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/rainy99/mcp-mysql-node.git"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.1",
+    "mysql2": "^3.11.0",
+    "zod": "^3.25.0"
+  }
+}