@bytebase/dbhub 0.15.0 → 0.16.1

package/README.md

@@ -23,20 +23,20 @@
             |                  |    |              |    |                  |
             |  VS Code         +--->+              +--->+    MySQL         |
             |                  |    |              |    |                  |
-            |  Other Clients   +--->+              +--->+    MariaDB       |
+            |  Copilot CLI     +--->+              +--->+    MariaDB       |
             |                  |    |              |    |                  |
             |                  |    |              |    |                  |
             +------------------+    +--------------+    +------------------+
                  MCP Clients           MCP Server             Databases
 ```
 
-DBHub is a zero-dependency, minimal database MCP server implementing the Model Context Protocol (MCP) server interface. This lightweight gateway allows MCP-compatible clients to connect to and explore different databases:
+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:
 
-- **Token Efficient**: Just two general MCP tools (execute_sql, search_objects) to minimize context window usage, plus support for custom tools
-- **Multi-Database**: Single interface for PostgreSQL, MySQL, MariaDB, SQL Server, and SQLite
-- **Secure Access**: Read-only mode, SSH tunneling, and SSL/TLS encryption support
-- **Multiple Connections**: Connect to multiple databases simultaneously with TOML configuration
-- **Production-Ready**: Row limiting, lock timeout control, and connection pooling
+- **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-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
 

package/dist/{chunk-TPHNNFR5.js → chunk-IBBG4PSO.js}

@@ -1451,7 +1451,7 @@ function buildDSNFromSource(source) {
 // src/connectors/manager.ts
 var managerInstance = null;
 var ConnectorManager = class {
-  // Ordered list of source IDs (first is default)
+  // Prevent race conditions
   constructor() {
     // Maps for multi-source support
     this.connectors = /* @__PURE__ */ new Map();
@@ -1459,6 +1459,11 @@ var ConnectorManager = class {
     this.sourceConfigs = /* @__PURE__ */ new Map();
     // Store original source configs
     this.sourceIds = [];
+    // Ordered list of source IDs (first is default)
+    // Lazy connection support
+    this.lazySources = /* @__PURE__ */ new Map();
+    // Sources pending lazy connection
+    this.pendingConnections = /* @__PURE__ */ new Map();
     if (!managerInstance) {
       managerInstance = this;
     }
@@ -1471,10 +1476,73 @@ var ConnectorManager = class {
     if (sources.length === 0) {
       throw new Error("No sources provided");
     }
-    console.error(`Connecting to ${sources.length} database source(s)...`);
-    for (const source of sources) {
+    const eagerSources = sources.filter((s) => !s.lazy);
+    const lazySources = sources.filter((s) => s.lazy);
+    if (eagerSources.length > 0) {
+      console.error(`Connecting to ${eagerSources.length} database source(s)...`);
+    }
+    for (const source of eagerSources) {
       await this.connectSource(source);
     }
+    for (const source of lazySources) {
+      this.registerLazySource(source);
+    }
+  }
+  /**
+   * Register a lazy source without establishing connection
+   * Connection will be established on first use via ensureConnected()
+   */
+  registerLazySource(source) {
+    const sourceId = source.id;
+    const dsn = buildDSNFromSource(source);
+    console.error(`  - ${sourceId}: ${redactDSN(dsn)} (lazy, will connect on first use)`);
+    this.lazySources.set(sourceId, source);
+    this.sourceConfigs.set(sourceId, source);
+    this.sourceIds.push(sourceId);
+  }
+  /**
+   * Ensure a source is connected (handles lazy connection on demand)
+   * Safe to call multiple times - uses promise-based deduplication so concurrent calls share the same connection attempt
+   */
+  async ensureConnected(sourceId) {
+    const id = sourceId || this.sourceIds[0];
+    if (this.connectors.has(id)) {
+      return;
+    }
+    const lazySource = this.lazySources.get(id);
+    if (!lazySource) {
+      if (sourceId) {
+        throw new Error(
+          `Source '${sourceId}' not found. Available sources: ${this.sourceIds.join(", ")}`
+        );
+      } else {
+        throw new Error("No sources configured. Call connectWithSources() first.");
+      }
+    }
+    const pending = this.pendingConnections.get(id);
+    if (pending) {
+      return pending;
+    }
+    const connectionPromise = (async () => {
+      try {
+        console.error(`Lazy connecting to source '${id}'...`);
+        await this.connectSource(lazySource);
+        this.lazySources.delete(id);
+      } finally {
+        this.pendingConnections.delete(id);
+      }
+    })();
+    this.pendingConnections.set(id, connectionPromise);
+    return connectionPromise;
+  }
+  /**
+   * Static method to ensure a source is connected (for tool handlers)
+   */
+  static async ensureConnected(sourceId) {
+    if (!managerInstance) {
+      throw new Error("ConnectorManager not initialized");
+    }
+    return managerInstance.ensureConnected(sourceId);
   }
   /**
    * Connect to a single source (helper for connectWithSources)
@@ -1540,7 +1608,9 @@ var ConnectorManager = class {
     }
     await connector.connect(actualDSN, source.init_script, config);
     this.connectors.set(sourceId, connector);
-    this.sourceIds.push(sourceId);
+    if (!this.sourceIds.includes(sourceId)) {
+      this.sourceIds.push(sourceId);
+    }
     this.sourceConfigs.set(sourceId, source);
   }
   /**
@@ -1565,6 +1635,8 @@ var ConnectorManager = class {
     this.connectors.clear();
     this.sshTunnels.clear();
     this.sourceConfigs.clear();
+    this.lazySources.clear();
+    this.pendingConnections.clear();
     this.sourceIds = [];
   }
   /**
@@ -1626,7 +1698,7 @@ var ConnectorManager = class {
    * @param sourceId - Source ID. If not provided, returns default (first) source config
    */
   getSourceConfig(sourceId) {
-    if (this.connectors.size === 0) {
+    if (this.sourceIds.length === 0) {
       return null;
     }
     const id = sourceId || this.sourceIds[0];

package/dist/index.js

@@ -16,7 +16,7 @@ import {
   resolveSourceConfigs,
   resolveTransport,
   stripCommentsAndStrings
-} from "./chunk-TPHNNFR5.js";
+} from "./chunk-IBBG4PSO.js";
 
 // src/connectors/postgres/index.ts
 import pg from "pg";
@@ -1757,7 +1757,7 @@ var mysqlConnector = new MySQLConnector();
 ConnectorRegistry.register(mysqlConnector);
 
 // src/connectors/mariadb/index.ts
-import mariadb from "mariadb";
+import * as mariadb from "mariadb";
 var MariadbDSNParser = class {
   async parse(dsn, config) {
     const connectionTimeoutSeconds = config?.connectionTimeoutSeconds;
@@ -2344,6 +2344,7 @@ function createExecuteSqlToolHandler(sourceId) {
     let errorMessage;
     let result;
     try {
+      await ConnectorManager.ensureConnected(sourceId);
       const connector = ConnectorManager.getCurrentConnector(sourceId);
       const actualSourceId = connector.getId();
       const registry = getToolRegistry();
@@ -2681,6 +2682,7 @@ function createSearchDatabaseObjectsToolHandler(sourceId) {
     let success = true;
     let errorMessage;
     try {
+      await ConnectorManager.ensureConnected(sourceId);
       const connector = ConnectorManager.getCurrentConnector(sourceId);
       if (table) {
         if (!schema) {
@@ -2993,6 +2995,7 @@ function createCustomToolHandler(toolConfig) {
     let paramValues = [];
     try {
       const validatedArgs = zodSchema.parse(args);
+      await ConnectorManager.ensureConnected(toolConfig.source);
       const connector = ConnectorManager.getCurrentConnector(toolConfig.source);
       const executeOptions = {
         readonly: toolConfig.readonly,
@@ -3135,6 +3138,9 @@ function transformSourceConfig(source) {
     id: source.id,
     type: source.type
   };
+  if (source.description) {
+    dataSource.description = source.description;
+  }
   if (source.host) {
     dataSource.host = source.host;
   }
@@ -3386,7 +3392,7 @@ See documentation for more details on configuring database connections.
     const sources = sourceConfigsData.sources;
     console.error(`Configuration source: ${sourceConfigsData.source}`);
     await connectorManager.connectWithSources(sources);
-    const { initializeToolRegistry } = await import("./registry-XXEL5IXH.js");
+    const { initializeToolRegistry } = await import("./registry-D77Y4CIA.js");
     initializeToolRegistry({
       sources: sourceConfigsData.sources,
       tools: sourceConfigsData.tools
@@ -3424,9 +3430,6 @@ See documentation for more details on configuring database connections.
       app.use(express.json());
       app.use((req, res, next) => {
         const origin = req.headers.origin;
-        if (origin && !origin.startsWith("http://localhost") && !origin.startsWith("https://localhost")) {
-          return res.status(403).json({ error: "Forbidden origin" });
-        }
         res.header("Access-Control-Allow-Origin", origin || "http://localhost");
         res.header("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
         res.header("Access-Control-Allow-Headers", "Content-Type, Mcp-Session-Id");

package/dist/{registry-XXEL5IXH.js → registry-D77Y4CIA.js}

@@ -2,7 +2,7 @@ import {
   ToolRegistry,
   getToolRegistry,
   initializeToolRegistry
-} from "./chunk-TPHNNFR5.js";
+} from "./chunk-IBBG4PSO.js";
 export {
   ToolRegistry,
   getToolRegistry,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "0.15.0",
+  "version": "0.16.1",
   "mcpName": "io.github.bytebase/dbhub",
   "description": "Minimal, token-efficient Database MCP Server for PostgreSQL, MySQL, SQL Server, SQLite, MariaDB",
   "repository": {