@zz1996/dbhub-dameng 0.1.0 → 0.1.2

package/README.md

@@ -5,6 +5,10 @@
 > This package is a local Dameng/DM8 connector fork of Bytebase DBHub. Use
 > `npx -y @zz1996/dbhub-dameng --transport stdio --config ./dbhub.dameng.toml`
 > for Dameng MCP usage.
+>
+> Dameng DSN format: `dameng://user:password@host:5236/schema`. Prefer a
+> least-privilege read-only Dameng account for production or shared databases,
+> even when the `execute_sql` tool is configured with `readonly = true`.
 
 <p align="center">
 <a href="https://dbhub.ai/" target="_blank">
@@ -30,6 +34,7 @@
             |                  |    |              |    |                  |
             |  Copilot CLI     +--->+              +--->+    MariaDB       |
             |                  |    |              |    |                  |
+            |                  +--->+              +--->+    Dameng / DM8  |
             |                  |    |              |    |                  |
             +------------------+    +--------------+    +------------------+
                  MCP Clients           MCP Server             Databases
@@ -38,14 +43,14 @@
 DBHub is a zero-dependency, token efficient MCP server implementing the Model Context Protocol (MCP) server interface. This lightweight gateway allows MCP-compatible clients to connect to and explore different databases:
 
 - **Local Development First**: Zero dependency, token efficient with just two MCP tools to maximize context window
-- **Multi-Database**: PostgreSQL, MySQL, MariaDB, SQL Server, and SQLite through a single interface
+- **Multi-Database**: PostgreSQL, MySQL, MariaDB, SQL Server, SQLite, and Dameng/DM8 through a single interface
 - **Multi-Connection**: Connect to multiple databases simultaneously with TOML configuration
 - **Guardrails**: Read-only mode, row limiting, and query timeout to prevent runaway operations
 - **Secure Access**: SSH tunneling and SSL/TLS encryption
 
 ## Supported Databases
 
-PostgreSQL, MySQL, SQL Server, MariaDB, and SQLite.
+PostgreSQL, MySQL, SQL Server, MariaDB, SQLite, and Dameng/DM8.
 
 ## MCP Tools
 
@@ -85,6 +90,41 @@ docker run --rm --init \
 npx @bytebase/dbhub@latest --transport http --port 8080 --dsn "postgres://user:password@localhost:5432/dbname?sslmode=disable"
 ```
 
+**Dameng / DM8 fork:**
+
+```bash
+npx -y @zz1996/dbhub-dameng --transport stdio --config ./dbhub.dameng.toml
+```
+
+Minimal Dameng TOML configuration:
+
+```toml
+[[sources]]
+id = "dm8_readonly"
+description = "Dameng DM8 database for schema discovery and read-only SQL queries"
+dsn = "dameng://${DM8_USER}:${DM8_PASSWORD}@${DM8_HOST}:5236/${DM8_SCHEMA}"
+query_timeout = 30
+lazy = true
+
+[[tools]]
+name = "search_objects"
+source = "dm8_readonly"
+
+[[tools]]
+name = "execute_sql"
+source = "dm8_readonly"
+readonly = true
+max_rows = 100
+```
+
+To verify the local Dameng connector against a real DM8 instance without
+committing credentials, set `DAMENG_DSN` or point to a local TOML file:
+
+```bash
+DAMENG_DSN="dameng://user:password@host:5236/schema" pnpm verify:dameng
+pnpm verify:dameng -- --config ./dbhub.dameng.toml --source dm8_readonly
+```
+
 **Demo Mode:**
 
 ```bash

package/dbhub.dameng.toml.example

@@ -5,6 +5,7 @@
 id = "dm8_dev"
 description = "Dameng DM8 development database. Read-only access for agent schema discovery and SQL queries."
 dsn = "dameng://${DM8_USER}:${DM8_PASSWORD}@${DM8_HOST}:5236/${DM8_SCHEMA}"
+connection_timeout = 5
 query_timeout = 30
 lazy = true
 

package/dist/{dameng-DC7OP4VV.js → dameng-UJELBZ5R.js}

@@ -65,6 +65,7 @@ var DamengConnector = class _DamengConnector {
     this.pool = null;
     this.sourceId = "default";
     this.defaultSchema = null;
+    this.poolAlias = null;
   }
   getId() {
     return this.sourceId;
@@ -73,13 +74,20 @@ var DamengConnector = class _DamengConnector {
     return new _DamengConnector();
   }
   async connect(dsn, initScript, config) {
+    let createdPool = null;
     try {
       const connectionConfig = await this.dsnParser.parse(dsn, config);
+      connectionConfig.poolAlias = this.buildPoolAlias();
       this.defaultSchema = connectionConfig.schema ?? null;
+      this.poolAlias = connectionConfig.poolAlias;
+      this.queryTimeoutMs = void 0;
       if (config?.queryTimeoutSeconds !== void 0) {
         this.queryTimeoutMs = config.queryTimeoutSeconds * 1e3;
       }
-      this.pool = await dmdb.createPool(connectionConfig);
+      await this.closeRegisteredPool(connectionConfig.poolAlias);
+      await this.validateDirectConnection(connectionConfig);
+      createdPool = await dmdb.createPool(connectionConfig);
+      this.pool = createdPool;
       await this.withConnection(async (conn) => {
         await conn.execute("SELECT 1 AS OK", [], this.executeOptions());
         if (!this.defaultSchema) {
@@ -97,15 +105,26 @@ var DamengConnector = class _DamengConnector {
         }
       });
     } catch (error) {
+      if (createdPool) {
+        await this.closePoolQuietly(createdPool);
+      } else if (this.poolAlias) {
+        await this.closeRegisteredPool(this.poolAlias);
+      }
+      this.pool = null;
+      this.poolAlias = null;
       console.error("Failed to connect to Dameng database:", error);
       throw error;
     }
   }
   async disconnect() {
     if (this.pool) {
-      await this.pool.close(0);
+      await this.closePoolQuietly(this.pool);
       this.pool = null;
     }
+    if (this.poolAlias) {
+      await this.closeRegisteredPool(this.poolAlias);
+      this.poolAlias = null;
+    }
   }
   async getSchemas() {
     const rows = await this.queryRows(`
@@ -366,6 +385,38 @@ var DamengConnector = class _DamengConnector {
       ...extra
     };
   }
+  async validateDirectConnection(config) {
+    let conn = null;
+    try {
+      conn = await dmdb.getConnection(config.connectString);
+      await conn.execute("SELECT 1 AS OK", [], this.executeOptions());
+    } finally {
+      if (conn) {
+        await conn.close();
+      }
+    }
+  }
+  buildPoolAlias() {
+    const safeSourceId = this.sourceId.replace(/[^a-zA-Z0-9_-]/g, "_") || "default";
+    return `dbhub_dameng_${safeSourceId}`;
+  }
+  async closeRegisteredPool(poolAlias) {
+    if (!dmdb.pools?.has?.(poolAlias)) {
+      return;
+    }
+    const pool = dmdb.pools.get(poolAlias);
+    await this.closePoolQuietly(pool);
+    dmdb.pools?.delete?.(poolAlias);
+  }
+  async closePoolQuietly(pool) {
+    try {
+      await pool.close(0);
+    } catch {
+      if (pool.poolAlias) {
+        dmdb.pools?.delete?.(pool.poolAlias);
+      }
+    }
+  }
   async resolveSchema(schema) {
     const resolved = schema || this.defaultSchema || await this.getDefaultSchema();
     if (!resolved) {

package/dist/index.js

@@ -1833,7 +1833,7 @@ var connectorModules = [
   { load: () => import("./sqlite-IOUAYHGE.js"), name: "SQLite", driver: "node:sqlite" },
   { load: () => import("./mysql-A43SL7UM.js"), name: "MySQL", driver: "mysql2" },
   { load: () => import("./mariadb-7F72IRB4.js"), name: "MariaDB", driver: "mariadb" },
-  { load: () => import("./dameng-DC7OP4VV.js"), name: "Dameng", driver: "dmdb" }
+  { load: () => import("./dameng-UJELBZ5R.js"), name: "Dameng", driver: "dmdb" }
 ];
 loadConnectors(connectorModules).then(() => main()).catch((error) => {
   console.error("Fatal error:", error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zz1996/dbhub-dameng",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "mcpName": "io.github.zuozh11/dbhub-dameng",
   "description": "Local fork of DBHub with Dameng/DM8 database connector support",
   "repository": {
@@ -41,6 +41,7 @@
     "test:watch": "vitest",
     "test:integration": "vitest run --project integration",
     "test:build": "node scripts/smoke-test-build.mjs",
+    "verify:dameng": "tsx scripts/verify-dameng.mjs",
     "prepublishOnly": "npm run build:backend"
   },
   "engines": {