cq-mcp-server 0.1.0 → 0.2.0

package/README.md

@@ -5,15 +5,30 @@
 ## 工具调用顺序
 
 ```
-list_databases → list_tables → get_table_columns → execute_sql
+connect（可选）→ list_databases → list_tables → get_table_columns → execute_sql
 ```
 
-**必须按顺序调用**：后续工具所需的 `connection_id`、`connection_type`、`database`、`schema` 参数，都只能从前面的工具返回结果中获取，不能凭空猜测。
+- **`connect`**：动态传入连接参数时需要先调用（如 Dify 等场景）。若已通过环境变量 `CQ_BASE_URL` / `CQ_USERNAME` / `CQ_PASSWORD` 配置，可跳过。
+- 后续工具所需的 `connection_id`、`connection_type`、`database`、`schema` 参数，都只能从前面的工具返回结果中获取，不能凭空猜测。
 
 ---
 
 ## 工具详解
 
+### 0. `connect` — 设置连接参数（动态场景必须首先调用）
+
+| 参数 | 说明 |
+|------|------|
+| `base_url` | CloudQuery 平台地址，如 `http://10.10.2.73` |
+| `username` | 登录账号 |
+| `password` | 登录密码（明文，服务端自动加密） |
+
+返回：连接配置成功确认。调用后原 Session 失效，下次请求自动重新登录。
+
+**Dify 配置示例**：在 workflow 开始节点调用 `connect`，将用户输入的 `base_url` / `username` / `password` 传入，后续工具调用无需再传。
+
+---
+
 ### 1. `list_databases` — 查询数据库清单
 
 **无需任何参数。**

package/dist/index.js

@@ -3,15 +3,18 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { z } from "zod";
 import forge from "node-forge";
-const BASE_URL = process.env.CQ_BASE_URL ?? "http://10.10.2.73";
-const USERNAME = process.env.CQ_USERNAME ?? "admin001";
-const PASSWORD = process.env.CQ_PASSWORD ?? "Hello123$";
 // MCP 协议通过 stdio 通信，日志必须输出到 stderr 避免污染 stdout
 const log = {
     info: (...args) => process.stderr.write(`[INFO]  ${args.join(" ")}\n`),
     warn: (...args) => process.stderr.write(`[WARN]  ${args.join(" ")}\n`),
     error: (...args) => process.stderr.write(`[ERROR] ${args.join(" ")}\n`),
 };
+// 运行时连接配置（可由 connect 工具覆盖，也可由环境变量提供默认值）
+let config = {
+    baseUrl: process.env.CQ_BASE_URL ?? "",
+    username: process.env.CQ_USERNAME ?? "",
+    password: process.env.CQ_PASSWORD ?? "",
+};
 let sessionCookie = null;
 // ── 认证 ──────────────────────────────────────────────────────────
 /** 用 RSA 公钥加密密码（与前端 JSEncrypt 行为一致） */
@@ -23,18 +26,24 @@ function encryptPassword(publicKeyB64, password) {
     return forge.util.encode64(encrypted);
 }
 async function login() {
-    log.info(`登录 ${BASE_URL}，用户: ${USERNAME}`);
-    const keyResp = await fetch(`${BASE_URL}/user/sys/transmission/publicKey`);
+    if (!config.baseUrl)
+        throw new Error("未配置 baseUrl，请先调用 connect 工具或设置 CQ_BASE_URL 环境变量");
+    if (!config.username)
+        throw new Error("未配置 username，请先调用 connect 工具或设置 CQ_USERNAME 环境变量");
+    if (!config.password)
+        throw new Error("未配置 password，请先调用 connect 工具或设置 CQ_PASSWORD 环境变量");
+    log.info(`登录 ${config.baseUrl}，用户: ${config.username}`);
+    const keyResp = await fetch(`${config.baseUrl}/user/sys/transmission/publicKey`);
     if (!keyResp.ok)
         throw new Error(`获取公钥失败: HTTP ${keyResp.status}`);
     const keyData = (await keyResp.json());
     if (keyData.resCode !== 10000)
         throw new Error(`获取公钥异常: ${keyData.resCode}`);
-    const encryptedPassword = encryptPassword(keyData.data.publicKey, PASSWORD);
-    const resp = await fetch(`${BASE_URL}/user/login`, {
+    const encryptedPassword = encryptPassword(keyData.data.publicKey, config.password);
+    const resp = await fetch(`${config.baseUrl}/user/login`, {
         method: "POST",
         headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({ userId: USERNAME, password: encryptedPassword }),
+        body: JSON.stringify({ userId: config.username, password: encryptedPassword }),
     });
     if (!resp.ok)
         throw new Error(`登录请求失败: HTTP ${resp.status}`);
@@ -62,7 +71,7 @@ async function login() {
 async function post(path, body) {
     if (!sessionCookie)
         await login();
-    const resp = await fetch(`${BASE_URL}${path}`, {
+    const resp = await fetch(`${config.baseUrl}${path}`, {
         method: "POST",
         headers: {
             "Content-Type": "application/json",
@@ -93,7 +102,28 @@ async function metaNode(payload) {
 // ── MCP Server ────────────────────────────────────────────────────
 const server = new McpServer({
     name: "cq-mcp-server",
-    version: "0.1.0",
+    version: "0.2.0",
+});
+// ── 工具零：连接配置 ──────────────────────────────────────────────
+server.tool("connect", `配置 CloudQuery 平台的连接信息（地址、账号、密码）。
+
+用途：在调用其他工具前，先通过此工具设置连接参数。适用于未配置环境变量的场景（如 Dify、动态多租户等）。
+参数说明：
+  - base_url：CloudQuery 平台地址，如 http://10.10.2.73
+  - username：登录账号
+  - password：登录密码（明文，服务端自动加密）
+返回：连接配置成功的确认信息。
+注意：每次调用会重置连接配置并清除已有 Session，强制重新登录。`, {
+    base_url: z.string().describe("CloudQuery 平台地址，如 http://10.10.2.73"),
+    username: z.string().describe("登录账号"),
+    password: z.string().describe("登录密码"),
+}, async ({ base_url, username, password }) => {
+    config = { baseUrl: base_url.replace(/\/$/, ""), username, password };
+    sessionCookie = null;
+    log.info(`connect: 配置已更新 baseUrl=${config.baseUrl} user=${username}`);
+    return {
+        content: [{ type: "text", text: `连接配置已设置：${config.baseUrl}，用户：${username}` }],
+    };
 });
 // ── 工具一：查询数据库清单 ─────────────────────────────────────────
 server.tool("list_databases", `查询所有数据库连接及其下的数据库实例列表。
@@ -274,6 +304,6 @@ server.tool("execute_sql", `在指定数据库上执行 SQL 语句，返回查
     return { content: [{ type: "text", text: lines.join("\n") || "无输出" }] };
 });
 // ── 启动 ──────────────────────────────────────────────────────────
-log.info(`CloudQuery MCP Server 启动  BASE_URL=${BASE_URL}  USER=${USERNAME}`);
+log.info(`CloudQuery MCP Server 启动  BASE_URL=${config.baseUrl || "(未配置)"}  USER=${config.username || "(未配置)"}`);
 const transport = new StdioServerTransport();
 await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cq-mcp-server",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "MCP Server for CloudQuery platform — list databases, list tables, inspect columns, and execute SQL via MCP tools.",
   "type": "module",
   "main": "dist/index.js",