@bytebase/dbhub 1.2.4 → 1.2.5

package/dist/{chunk-F5TPYAVO.js → chunk-VDA5BVNE.js}

@@ -1832,7 +1832,7 @@ var ConnectorManager = class {
     if (!lazySource) {
       if (sourceId) {
         throw new Error(
-          `Source '${sourceId}' not found. Available sources: ${this.sourceIds.join(", ")}`
+          `Source '${sourceId}' not found. Available sources: ${this.getAvailableSourceIds().join(", ")}`
         );
       } else {
         throw new Error("No sources configured. Call connectWithSources() first.");
@@ -1871,6 +1871,7 @@ var ConnectorManager = class {
     const dsn = await this.buildConnectionDSN(source);
     console.error(`  - ${sourceId}: ${redactDSN(dsn)}`);
     let actualDSN = dsn;
+    let tunnel;
     if (source.ssh_host) {
       const sshConfigPath = getDefaultSSHConfigPath();
       let resolvedSSHConfig = null;
@@ -1914,7 +1915,7 @@ var ConnectorManager = class {
       const url = new URL(dsn);
       const targetHost = url.hostname;
       const targetPort = parseInt(url.port) || this.getDefaultPort(dsn);
-      const tunnel = new SSHTunnel();
+      tunnel = new SSHTunnel();
       let tunnelInfo;
       try {
         tunnelInfo = await tunnel.establish(sshConfig, {
@@ -1962,7 +1963,19 @@ var ConnectorManager = class {
     if (source.collation) {
       config.collation = source.collation;
     }
-    await connector.connect(actualDSN, source.init_script, config);
+    try {
+      await connector.connect(actualDSN, source.init_script, config);
+    } catch (error) {
+      if (tunnel) {
+        this.sshTunnels.delete(sourceId);
+        try {
+          await tunnel.close();
+        } catch (closeError) {
+          console.error(`Error closing SSH tunnel for source '${sourceId}':`, closeError);
+        }
+      }
+      throw error;
+    }
     this.connectors.set(sourceId, connector);
     if (!this.sourceIds.includes(sourceId)) {
       this.sourceIds.push(sourceId);
@@ -2012,7 +2025,7 @@ var ConnectorManager = class {
     if (!connector) {
       if (sourceId) {
         throw new Error(
-          `Source '${sourceId}' not found. Available sources: ${this.sourceIds.join(", ")}`
+          `Source '${sourceId}' not found. Available sources: ${this.getAvailableSourceIds().join(", ")}`
         );
       } else {
         throw new Error("No sources connected. Call connectWithSources() first.");
@@ -2049,6 +2062,14 @@ var ConnectorManager = class {
   getSourceIds() {
     return [...this.sourceIds];
   }
+  /**
+   * Source IDs that can actually serve a request: connected, or registered for
+   * lazy (re)connection on first use. Used for error messages so a source that
+   * is neither is not reported as available.
+   */
+  getAvailableSourceIds() {
+    return this.sourceIds.filter((id) => this.connectors.has(id) || this.lazySources.has(id));
+  }
   /** Get all available source IDs */
   static getAvailableSourceIds() {
     if (!managerInstance) {
@@ -2126,7 +2147,7 @@ var ConnectorManager = class {
           error
         );
       } finally {
-        if (!this.isDisconnecting && this.sourceConfigs.has(sourceId)) {
+        if (!this.isDisconnecting && this.connectors.has(sourceId)) {
           this.scheduleIamRefresh(source);
         }
       }
@@ -2153,7 +2174,14 @@ var ConnectorManager = class {
     if (this.isDisconnecting) {
       return;
     }
-    await this.connectSource(source);
+    try {
+      await this.connectSource(source);
+    } catch (error) {
+      if (!this.isDisconnecting && this.sourceConfigs.has(sourceId)) {
+        this.lazySources.set(sourceId, source);
+      }
+      throw error;
+    }
   }
   /**
    * Build a connection DSN, optionally replacing password with

package/dist/index.js

@@ -18,7 +18,7 @@ import {
   resolveSourceConfigs,
   resolveTomlConfigPath,
   resolveTransport
-} from "./chunk-F5TPYAVO.js";
+} from "./chunk-VDA5BVNE.js";
 import {
   allowedKeywords,
   getFirstKeyword,
@@ -1913,7 +1913,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: initializeToolRegistry2 } = await import("./registry-6ABK3M6V.js");
+    const { initializeToolRegistry: initializeToolRegistry2 } = await import("./registry-UF3N2H4G.js");
     initializeToolRegistry2({
       sources: sourceConfigsData.sources,
       tools: sourceConfigsData.tools

package/dist/{registry-6ABK3M6V.js → registry-UF3N2H4G.js}

@@ -2,7 +2,7 @@ import {
   ToolRegistry,
   getToolRegistry,
   initializeToolRegistry
-} from "./chunk-F5TPYAVO.js";
+} from "./chunk-VDA5BVNE.js";
 import "./chunk-WVVMH6FJ.js";
 import "./chunk-MF6LSVWK.js";
 export {

package/package.json

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