@bytebase/dbhub 0.11.4 → 0.11.5

package/README.md

@@ -281,6 +281,40 @@ In read-only mode, only [readonly SQL operations](https://github.com/bytebase/db
 
 This provides an additional layer of security when connecting to production databases.
 
+### Suffix Tool Names with ID
+
+You can suffix tool names with a custom ID using the `--id` flag or `ID` environment variable. This is useful when running multiple DBHub instances (e.g., in Cursor) to allow the client to route queries to the correct database.
+
+**Example configuration for multiple databases in Cursor:**
+
+```json
+{
+  "mcpServers": {
+    "dbhub-prod": {
+      "command": "npx",
+      "args": ["-y", "@bytebase/dbhub"],
+      "env": {
+        "ID": "prod",
+        "DSN": "postgres://user:password@prod-host:5432/dbname"
+      }
+    },
+    "dbhub-staging": {
+      "command": "npx",
+      "args": ["-y", "@bytebase/dbhub"],
+      "env": {
+        "ID": "staging",
+        "DSN": "mysql://user:password@staging-host:3306/dbname"
+      }
+    }
+  }
+}
+```
+
+With this configuration:
+
+- Production database tools: `execute_sql_prod`
+- Staging database tools: `execute_sql_staging`
+
 ### Row Limiting
 
 You can limit the number of rows returned from SELECT queries using the `--max-rows` parameter. This helps prevent accidentally retrieving too much data from large tables:
@@ -517,6 +551,7 @@ Extra query parameters:
 | readonly       | `READONLY`           | Restrict SQL execution to read-only operations                        | `false`                      |
 | max-rows       | N/A                  | Limit the number of rows returned from SELECT queries                 | No limit                     |
 | demo           | N/A                  | Run in demo mode with sample employee database                        | `false`                      |
+| id             | `ID`                 | Instance identifier to suffix tool names (for multi-instance)         | N/A                          |
 | ssh-host       | `SSH_HOST`           | SSH server hostname for tunnel connection                             | N/A                          |
 | ssh-port       | `SSH_PORT`           | SSH server port                                                       | `22`                         |
 | ssh-user       | `SSH_USER`           | SSH username                                                          | N/A                          |

package/dist/index.js

@@ -2455,6 +2455,16 @@ function redactDSN(dsn) {
     return dsn.replace(/\/\/([^:]+):([^@]+)@/, "//$1:***@");
   }
 }
+function resolveId() {
+  const args = parseCommandLineArgs();
+  if (args.id) {
+    return { id: args.id, source: "command line argument" };
+  }
+  if (process.env.ID) {
+    return { id: process.env.ID, source: "environment variable" };
+  }
+  return null;
+}
 function resolveSSHConfig() {
   const args = parseCommandLineArgs();
   const hasSSHArgs = args["ssh-host"] || process.env.SSH_HOST;
@@ -3129,9 +3139,10 @@ async function executeSqlToolHandler({ sql: sql2 }, _extra) {
 }
 
 // src/tools/index.ts
-function registerTools(server) {
+function registerTools(server, id) {
+  const toolName = id ? `execute_sql_${id}` : "execute_sql";
   server.tool(
-    "execute_sql",
+    toolName,
     "Execute a SQL query on the current database",
     executeSqlSchema,
     executeSqlToolHandler
@@ -3575,10 +3586,12 @@ v${version}${modeText} - Universal Database MCP Server
 }
 async function main() {
   try {
+    const idData = resolveId();
+    const id = idData?.id;
     const dsnData = resolveDSN();
     if (!dsnData) {
       const samples = ConnectorRegistry.getAllSampleDSNs();
-      const sampleFormats = Object.entries(samples).map(([id, dsn]) => `  - ${id}: ${dsn}`).join("\n");
+      const sampleFormats = Object.entries(samples).map(([id2, dsn]) => `  - ${id2}: ${dsn}`).join("\n");
       console.error(`
 ERROR: Database connection string (DSN) is required.
 Please provide the DSN in one of these ways (in order of priority):
@@ -3601,13 +3614,16 @@ See documentation for more details on configuring database connections.
         version: SERVER_VERSION
       });
       registerResources(server);
-      registerTools(server);
+      registerTools(server, id);
       registerPrompts(server);
       return server;
     };
     const connectorManager = new ConnectorManager();
     console.error(`Connecting with DSN: ${redactDSN(dsnData.dsn)}`);
     console.error(`DSN source: ${dsnData.source}`);
+    if (idData) {
+      console.error(`ID: ${idData.id} (from ${idData.source})`);
+    }
     if (dsnData.isDemo) {
       const initScript = getSqliteInMemorySetupSql();
       await connectorManager.connectWithDSN(dsnData.dsn, initScript);

package/package.json

@@ -1,7 +1,11 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "0.11.4",
+  "version": "0.11.5",
   "description": "Universal Database MCP Server",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/bytebase/dbhub.git"
+  },
   "main": "dist/index.js",
   "type": "module",
   "bin": {