@bytebase/dbhub 1.2.3 → 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
@@ -2068,11 +2068,11 @@ See documentation for more details on configuring database connections.
 
 // src/index.ts
 var connectorModules = [
-  { load: () => import("./postgres-OVI2PNIJ.js"), name: "PostgreSQL", driver: "pg" },
+  { load: () => import("./postgres-M3W3Y6F5.js"), name: "PostgreSQL", driver: "pg" },
   { load: () => import("./sqlserver-WNVSBFJ3.js"), name: "SQL Server", driver: "mssql" },
   { load: () => import("./sqlite-ASYMU5BI.js"), name: "SQLite", driver: "node:sqlite" },
   { load: () => import("./mysql-GXDUNLJ6.js"), name: "MySQL", driver: "mysql2" },
-  { load: () => import("./mariadb-DQ6P57IX.js"), name: "MariaDB", driver: "mariadb" }
+  { load: () => import("./mariadb-EVJ5HLYH.js"), name: "MariaDB", driver: "mariadb" }
 ];
 loadConnectors(connectorModules).then(() => main()).catch((error) => {
   console.error("Fatal error:", error);

package/dist/{mariadb-DQ6P57IX.js → mariadb-EVJ5HLYH.js}

@@ -125,6 +125,12 @@ var MariaDBConnector = class _MariaDBConnector {
     try {
       const connectionConfig = await this.dsnParser.parse(dsn, config);
       this.pool = mariadb.createPool(connectionConfig);
+      this.pool.on("error", (err) => {
+        console.error(
+          `MariaDB pool (source "${this.sourceId}"): background connection error, pool will retry:`,
+          err.message
+        );
+      });
       const rows = await this.pool.query("SELECT VERSION() AS version");
       this.supportsReadOnlyTransaction = !isTiDBVersion(rows?.[0]?.version);
     } catch (err) {

package/dist/{postgres-OVI2PNIJ.js → postgres-M3W3Y6F5.js}

@@ -179,6 +179,12 @@ var PostgresConnector = class _PostgresConnector {
         }
       }
       this.pool = new Pool(poolConfig);
+      this.pool.on("error", (err) => {
+        console.error(
+          `PostgreSQL pool (source "${this.sourceId}"): idle connection dropped, will reconnect on next query:`,
+          err.message
+        );
+      });
       const client = await this.pool.connect();
       client.release();
     } catch (err) {
@@ -226,7 +232,7 @@ var PostgresConnector = class _PostgresConnector {
         SELECT table_name
         FROM information_schema.tables
         WHERE table_schema = $1
-        AND table_type = 'BASE TABLE'
+        AND table_type IN ('BASE TABLE', 'FOREIGN')
         ORDER BY table_name
       `,
         [schemaToUse]
@@ -305,7 +311,7 @@ var PostgresConnector = class _PostgresConnector {
           AND i.oid = ix.indexrelid
           AND a.attrelid = t.oid
           AND a.attnum = ANY(ix.indkey)
-          AND t.relkind = 'r'
+          AND t.relkind IN ('r','p')
           AND t.relname = $1
           AND ns.oid = t.relnamespace
           AND ns.nspname = $2

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.3",
+  "version": "1.2.5",
   "mcpName": "io.github.bytebase/dbhub",
   "description": "Minimal, token-efficient Database MCP Server for PostgreSQL, MySQL, SQL Server, SQLite, MariaDB",
   "repository": {