@bytebase/dbhub 1.2.0 → 1.2.2

package/dist/public/index.html

@@ -5,7 +5,7 @@
     <link rel="icon" type="image/svg+xml" href="/favicon.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>DBHub - Minimal Database MCP Server</title>
-    <script type="module" crossorigin src="/assets/index-DOZ5HOZ9.js"></script>
+    <script type="module" crossorigin src="/assets/index-BdBaFSHd.js"></script>
     <link rel="stylesheet" crossorigin href="/assets/index-BJ-1UrcV.css">
   </head>
   <body>

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

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

package/dist/{sqlite-CZE6Z36Y.js → sqlite-ASYMU5BI.js}

@@ -4,13 +4,13 @@ import {
 import {
   SQLRowLimiter,
   closeQuietly
-} from "./chunk-RSB7DPO2.js";
+} from "./chunk-6QETXS44.js";
 import {
   ConnectorRegistry,
   SafeURL,
   obfuscateDSNPassword,
   splitSQLStatements
-} from "./chunk-WIQ4DWGY.js";
+} from "./chunk-MF6LSVWK.js";
 
 // src/connectors/sqlite/suppress-experimental-warning.ts
 var installed = false;
@@ -334,17 +334,20 @@ var SQLiteConnector = class _SQLiteConnector {
         let processedStatement = statements[0];
         const trimmedStatement = statements[0].toLowerCase().trim();
         const isReadStatement = trimmedStatement.startsWith("select") || trimmedStatement.startsWith("with") || trimmedStatement.startsWith("explain") || trimmedStatement.startsWith("analyze") || trimmedStatement.startsWith("pragma") && (trimmedStatement.includes("table_info") || trimmedStatement.includes("index_info") || trimmedStatement.includes("index_list") || trimmedStatement.includes("foreign_key_list"));
+        let probeApplied = false;
         if (options.maxRows) {
-          processedStatement = SQLRowLimiter.applyMaxRows(processedStatement, options.maxRows);
+          const rewrite = SQLRowLimiter.applyMaxRowsWithTruncationProbe(
+            processedStatement,
+            options.maxRows
+          );
+          processedStatement = rewrite.sql;
+          probeApplied = rewrite.probeApplied;
         }
         if (isReadStatement) {
-          if (parameters && parameters.length > 0) {
-            const rows = this.prepare(processedStatement).all(...parameters);
-            return { resultSets: [{ sql: statements[0], rows, rowCount: rows.length }] };
-          } else {
-            const rows = this.prepare(processedStatement).all();
-            return { resultSets: [{ sql: statements[0], rows, rowCount: rows.length }] };
-          }
+          const rows = parameters && parameters.length > 0 ? this.prepare(processedStatement).all(...parameters) : this.prepare(processedStatement).all();
+          const resultSet = { sql: statements[0], rows, rowCount: rows.length };
+          SQLRowLimiter.flagTruncation(resultSet, options.maxRows, probeApplied);
+          return { resultSets: [resultSet] };
         } else {
           let result;
           if (parameters && parameters.length > 0) {
@@ -382,9 +385,14 @@ var SQLiteConnector = class _SQLiteConnector {
           if (options.readonly) {
             this.db.exec("PRAGMA query_only = ON");
           }
-          const processedStatement = SQLRowLimiter.applyMaxRows(statement, options.maxRows);
+          const { sql: processedStatement, probeApplied } = SQLRowLimiter.applyMaxRowsWithTruncationProbe(
+            statement,
+            options.maxRows
+          );
           const rows = this.prepare(processedStatement).all();
-          resultSets.push({ sql: statement, rows, rowCount: rows.length });
+          const resultSet = { sql: statement, rows, rowCount: rows.length };
+          SQLRowLimiter.flagTruncation(resultSet, options.maxRows, probeApplied);
+          resultSets.push(resultSet);
         }
         return { resultSets };
       }

package/dist/{sqlserver-IIGRXGYK.js → sqlserver-WNVSBFJ3.js}

@@ -3,7 +3,7 @@ import {
   sqlServerDynamicSqlPattern,
   sqlServerPassThroughKeywords,
   sqlServerPassThroughPattern
-} from "./chunk-PBN2UXB5.js";
+} from "./chunk-IQRBLAB4.js";
 import {
   isDriverNotInstalled
 } from "./chunk-WVVMH6FJ.js";
@@ -14,14 +14,14 @@ import {
 import {
   SQLRowLimiter,
   closeQuietly
-} from "./chunk-RSB7DPO2.js";
+} from "./chunk-6QETXS44.js";
 import {
   ConnectorRegistry,
   SafeURL,
   obfuscateDSNPassword,
   splitSQLStatements,
   stripCommentsAndStrings
-} from "./chunk-WIQ4DWGY.js";
+} from "./chunk-MF6LSVWK.js";
 
 // src/connectors/sqlserver/index.ts
 import sql from "mssql";
@@ -562,12 +562,24 @@ var _SQLServerConnector = class _SQLServerConnector {
     }
     try {
       let processedSQL = sqlQuery;
+      let probeApplied = false;
       if (options.maxRows) {
-        processedSQL = SQLRowLimiter.applyMaxRowsForSQLServer(sqlQuery, options.maxRows);
+        const rewrite = SQLRowLimiter.applyMaxRowsForSQLServerWithTruncationProbe(
+          sqlQuery,
+          options.maxRows
+        );
+        processedSQL = rewrite.sql;
+        probeApplied = rewrite.probeApplied;
       }
       const isSingleStatement = splitSQLStatements(processedSQL, "sqlserver").length === 1;
       if (options.readonly) {
-        return await this.executeReadOnly(processedSQL, parameters, isSingleStatement);
+        return await this.executeReadOnly(
+          processedSQL,
+          parameters,
+          isSingleStatement ? sqlQuery : void 0,
+          options.maxRows,
+          probeApplied
+        );
       }
       const request = this.connection.request();
       const messages = [];
@@ -585,12 +597,16 @@ var _SQLServerConnector = class _SQLServerConnector {
       );
       _SQLServerConnector.bindParameters(request, parameters);
       const result = await request.query(processedSQL);
+      const resultSets = _SQLServerConnector.buildResultSets(
+        result.recordsets,
+        result.rowsAffected,
+        isSingleStatement ? sqlQuery : void 0
+      );
+      if (resultSets.length > 0) {
+        SQLRowLimiter.flagTruncation(resultSets[0], options.maxRows, probeApplied);
+      }
       return {
-        resultSets: _SQLServerConnector.buildResultSets(
-          result.recordsets,
-          result.rowsAffected,
-          isSingleStatement ? processedSQL : void 0
-        ),
+        resultSets,
         ...messages.length > 0 ? { messages } : {}
       };
     } catch (error) {
@@ -718,7 +734,7 @@ var _SQLServerConnector = class _SQLServerConnector {
    * Dangerous constructs are rejected before the transaction opens; see
    * assertNoReadOnlyEscapes.
    */
-  async executeReadOnly(processedSQL, parameters, isSingleStatement) {
+  async executeReadOnly(processedSQL, parameters, sourceSql, maxRows, probeApplied) {
     this.assertNoReadOnlyEscapes(processedSQL, { transactionControl: true });
     const transaction = new sql.Transaction(this.connection);
     await transaction.begin();
@@ -754,12 +770,16 @@ var _SQLServerConnector = class _SQLServerConnector {
         }
       }
     }
+    const resultSets = _SQLServerConnector.buildResultSets(
+      result.recordsets,
+      result.rowsAffected,
+      sourceSql
+    );
+    if (resultSets.length > 0) {
+      SQLRowLimiter.flagTruncation(resultSets[0], maxRows, probeApplied);
+    }
     return {
-      resultSets: _SQLServerConnector.buildResultSets(
-        result.recordsets,
-        result.rowsAffected,
-        isSingleStatement ? processedSQL : void 0
-      ),
+      resultSets,
       ...messages.length > 0 ? { messages } : {}
     };
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bytebase/dbhub",
-  "version": "1.2.0",
+  "version": "1.2.2",
   "mcpName": "io.github.bytebase/dbhub",
   "description": "Minimal, token-efficient Database MCP Server for PostgreSQL, MySQL, SQL Server, SQLite, MariaDB",
   "repository": {
@@ -55,6 +55,7 @@
     "zod": "^4.2.0"
   },
   "optionalDependencies": {
+    "@aws-sdk/credential-providers": "^3.1001.0",
     "@aws-sdk/rds-signer": "^3.1001.0",
     "@azure/identity": "^4.8.0",
     "mariadb": "^3.4.0",