@bytebase/dbhub 1.2.5 → 1.3.1

package/dist/chunk-YIGJHEU6.js

@@ -0,0 +1,343 @@
+import {
+  hasMutatingKeyword
+} from "./chunk-ABTVMHD6.js";
+import {
+  blankCommentsAndStrings
+} from "./chunk-IXC4NQGC.js";
+
+// src/utils/sql-row-limiter.ts
+var SQLRowLimiter = class {
+  /**
+   * Check if a SQL statement is a row-returning query that can benefit from row limiting.
+   *
+   * Classification runs on the comment-blanked text, so a query introduced by a
+   * `--` or `/* ... *\/` comment (a query tag, say) is classified by its real
+   * leading keyword. The SQL sent to the server is never rewritten by this
+   * check, so an attribution comment survives verbatim.
+   *
+   * `WITH` leads a CTE whose final statement is normally a SELECT, so it is
+   * limitable too, except for a data-modifying CTE
+   * (`WITH x AS (DELETE ... RETURNING *) SELECT ...`), where a LIMIT would cap
+   * the rows handed back while the write still runs in full. Detection there is
+   * the same keyword heuristic the read-only classifier uses, so a false
+   * positive only means the statement keeps the old behaviour of not being limited.
+   *
+   * `dialect` selects the connector's scanner so dialect-specific quoting
+   * (PostgreSQL dollar quotes, MySQL backticks, SQL Server brackets) is
+   * blanked the same way the read-only classifier blanks it; without it only
+   * ANSI syntax is recognized.
+   */
+  static isSelectQuery(sql, dialect) {
+    const blankedSQL = blankCommentsAndStrings(sql, dialect).trim().toLowerCase();
+    const firstKeyword = /^[(;\s]*([a-z_]+)/.exec(blankedSQL)?.[1] ?? "";
+    if (firstKeyword === "select") {
+      return true;
+    }
+    return firstKeyword === "with" && !hasMutatingKeyword(blankedSQL, dialect);
+  }
+  /**
+   * Check if a SQL statement has a LIMIT clause of its own (not one nested in
+   * a CTE body or subquery).
+   */
+  static hasLimitClause(sql, dialect) {
+    return this.findTopLevelLimit(sql, dialect) !== null;
+  }
+  /**
+   * Check if a SQL statement has a TOP clause of its own (SQL Server).
+   */
+  static hasTopClause(sql) {
+    return this.findTopLevelTop(sql) !== null;
+  }
+  /**
+   * Extract the statement's own LIMIT value. Null when it has no LIMIT of its
+   * own, or when that LIMIT is a parameter placeholder rather than a literal
+   * (see hasParameterizedLimit).
+   */
+  static extractLimitValue(sql, dialect) {
+    return this.findTopLevelLimit(sql, dialect)?.value ?? null;
+  }
+  /**
+   * Extract the statement's own TOP value (SQL Server), or null when it has none.
+   */
+  static extractTopValue(sql) {
+    return this.findTopLevelTop(sql)?.value ?? null;
+  }
+  /**
+   * Check if the statement's own LIMIT clause uses a parameter placeholder
+   * ($1, ?, @p1) instead of a literal number.
+   */
+  static hasParameterizedLimit(sql, dialect) {
+    const limit = this.findTopLevelLimit(sql, dialect);
+    return limit !== null && limit.value === null;
+  }
+  /**
+   * Add or tighten the LIMIT clause of a SQL statement
+   */
+  static applyLimitToQuery(sql, maxRows, dialect) {
+    const limit = this.findTopLevelLimit(sql, dialect);
+    if (limit !== null && limit.value !== null) {
+      const effectiveLimit = Math.min(limit.value, maxRows);
+      return `${sql.slice(0, limit.index)}LIMIT ${effectiveLimit}${sql.slice(limit.index + limit.length)}`;
+    }
+    const { sql: sqlWithoutSemicolon, semicolon } = trimSemicolon(sql);
+    return `${sqlWithoutSemicolon}
+LIMIT ${maxRows}${semicolon}`;
+  }
+  /**
+   * Scan blanked (comment/string-free, length-preserving) SQL for parenthesis
+   * depth, invoking onMatch for every regex hit at depth 0 (i.e. not nested
+   * inside a subquery, function call, or window OVER clause).
+   */
+  static scanTopLevel(blankedSQL, regex, onMatch) {
+    let depth = 0;
+    let m;
+    while ((m = regex.exec(blankedSQL)) !== null) {
+      const token = m[0];
+      if (token === "(") {
+        depth++;
+      } else if (token === ")") {
+        depth--;
+      } else if (depth === 0) {
+        onMatch(m);
+      }
+    }
+  }
+  /**
+   * The first or last regex match at parenthesis depth 0, i.e. one belonging
+   * to the statement itself rather than to a nested query. Match indices refer
+   * to positions in the original `sql`, since blankCommentsAndStrings
+   * preserves length. The regex must offer `\(` and `\)` alternatives so depth
+   * can be tracked, and must carry the `g` flag.
+   */
+  static findTopLevelMatch(sql, regex, pick, dialect) {
+    let found = null;
+    this.scanTopLevel(blankCommentsAndStrings(sql, dialect), regex, (m) => {
+      if (pick === "last" || found === null) {
+        found = m;
+      }
+    });
+    return found;
+  }
+  /** The statement's own LIMIT clause, literal or parameterized ($1, ?, @p1). */
+  static findTopLevelLimit(sql, dialect) {
+    const match = this.findTopLevelMatch(
+      sql,
+      /\(|\)|\blimit\s+(?:(\d+)|\$\d+|\?|@p\d+)/gi,
+      "last",
+      dialect
+    );
+    if (match === null) {
+      return null;
+    }
+    return {
+      index: match.index,
+      length: match[0].length,
+      value: match[1] !== void 0 ? parseInt(match[1], 10) : null
+    };
+  }
+  /** The statement's own `SELECT TOP n` clause (SQL Server). */
+  static findTopLevelTop(sql) {
+    const match = this.findTopLevelMatch(sql, /\(|\)|\bselect\s+top\s+(\d+)/gi, "first", "sqlserver");
+    if (match === null) {
+      return null;
+    }
+    return { index: match.index, length: match[0].length, value: parseInt(match[1], 10) };
+  }
+  /**
+   * The statement's own SELECT keyword (SQL Server). For a CTE this is the
+   * final SELECT, not the one inside a CTE body.
+   */
+  static findTopLevelSelect(sql) {
+    return this.findTopLevelMatch(sql, /\(|\)|\bselect\b/gi, "first", "sqlserver");
+  }
+  /**
+   * Check if a SQL statement combines multiple SELECTs with a set operator
+   * (UNION [ALL], INTERSECT, EXCEPT) at the top level — i.e. not nested
+   * inside a subquery already wrapped in parentheses.
+   */
+  static hasSetOperator(sql) {
+    return this.findTopLevelMatch(sql, /\(|\)|\bunion\b|\bintersect\b|\bexcept\b/gi, "first", "sqlserver") !== null;
+  }
+  /**
+   * Find the start index of a top-level trailing ORDER BY clause (not one
+   * nested inside a subquery or a window function's OVER (...) clause).
+   * Returns -1 if none exists.
+   */
+  static findTopLevelOrderByIndex(sql) {
+    return this.findTopLevelMatch(sql, /\(|\)|\border\s+by\b/gi, "last", "sqlserver")?.index ?? -1;
+  }
+  /**
+   * Add or modify TOP clause in a SQL statement (SQL Server)
+   */
+  static applyTopToQuery(sql, maxRows) {
+    if (this.hasSetOperator(sql)) {
+      const { sql: sqlWithoutSemicolon, semicolon } = trimSemicolon(sql);
+      const cteIndex = this.findTopLevelSelect(sqlWithoutSemicolon)?.index ?? 0;
+      const ctePrefix = sqlWithoutSemicolon.slice(0, cteIndex);
+      const body = sqlWithoutSemicolon.slice(cteIndex);
+      const orderByIndex = this.findTopLevelOrderByIndex(body);
+      if (orderByIndex !== -1) {
+        const innerSql = body.slice(0, orderByIndex).trimEnd();
+        const orderByClause = body.slice(orderByIndex).trim();
+        return `${ctePrefix}SELECT TOP ${maxRows} * FROM (${innerSql}
+) AS subq ${orderByClause}${semicolon}`;
+      }
+      return `${ctePrefix}SELECT TOP ${maxRows} * FROM (${body}
+) AS subq${semicolon}`;
+    }
+    const existingTop = this.findTopLevelTop(sql);
+    if (existingTop !== null) {
+      const effectiveTop = Math.min(existingTop.value, maxRows);
+      return `${sql.slice(0, existingTop.index)}SELECT TOP ${effectiveTop}${sql.slice(existingTop.index + existingTop.length)}`;
+    }
+    const selectMatch = this.findTopLevelSelect(sql);
+    if (selectMatch === null) {
+      return sql;
+    }
+    return `${sql.slice(0, selectMatch.index)}SELECT TOP ${maxRows}${sql.slice(selectMatch.index + selectMatch[0].length)}`;
+  }
+  /**
+   * Apply maxRows limit to a row-returning query only
+   *
+   * This method is used by PostgreSQL, MySQL, MariaDB, and SQLite connectors which all support
+   * the LIMIT clause syntax. SQL Server uses applyMaxRowsForSQLServer() instead with TOP syntax.
+   *
+   * For parameterized LIMIT clauses (e.g., LIMIT $1 or LIMIT ?), we wrap the query in a subquery
+   * to enforce max_rows as a hard cap, since the parameter value is not known until runtime.
+   *
+   * `dialect` selects the connector's comment/string scanner (see isSelectQuery).
+   */
+  static applyMaxRows(sql, maxRows, dialect) {
+    if (!maxRows || !this.isSelectQuery(sql, dialect)) {
+      return sql;
+    }
+    if (this.hasParameterizedLimit(sql, dialect)) {
+      const { sql: sqlWithoutSemicolon, semicolon } = trimSemicolon(sql);
+      return `SELECT * FROM (${sqlWithoutSemicolon}
+) AS subq LIMIT ${maxRows}${semicolon}`;
+    }
+    return this.applyLimitToQuery(sql, maxRows, dialect);
+  }
+  /**
+   * Apply maxRows limit to a row-returning query using SQL Server TOP syntax
+   */
+  static applyMaxRowsForSQLServer(sql, maxRows) {
+    if (!maxRows || !this.isSelectQuery(sql, "sqlserver")) {
+      return sql;
+    }
+    return this.applyTopToQuery(sql, maxRows);
+  }
+  /**
+   * Apply maxRows to a row-returning query using Oracle's row-limiting clause.
+   *
+   * The statement is always wrapped in an inline view and capped from the
+   * outside: `SELECT * FROM (<sql>) FETCH FIRST n ROWS ONLY`. Wrapping rather
+   * than splicing keeps every shape correct with one rule - a set operation
+   * (UNION & co.) is capped as a whole, a trailing ORDER BY stays inside the
+   * view where Oracle honours it, and a query that already has its own
+   * `FETCH FIRST` / `ROWNUM` cap simply returns the smaller of the two. Oracle
+   * also accepts a leading CTE (subquery factoring) inside an inline view, so
+   * `WITH ... SELECT` needs no special casing.
+   */
+  static applyMaxRowsForOracle(sql, maxRows) {
+    if (!maxRows || !this.isSelectQuery(sql, "oracle") || this.hasOracleForUpdate(sql)) {
+      return sql;
+    }
+    const { sql: sqlWithoutSemicolon } = trimSemicolon(sql);
+    return `SELECT * FROM (${sqlWithoutSemicolon}
+) FETCH FIRST ${maxRows} ROWS ONLY`;
+  }
+  /**
+   * A statement-level `FOR UPDATE` (locking read). Oracle allows neither
+   * `FOR UPDATE` inside an inline view nor a row-limiting clause alongside
+   * it, so such a statement cannot be capped and is left as written.
+   */
+  static hasOracleForUpdate(sql) {
+    return this.findTopLevelMatch(sql, /\(|\)|\bfor\s+update\b/gi, "first", "oracle") !== null;
+  }
+  /**
+   * Oracle variant of applyMaxRowsWithTruncationProbe. Because the cap always
+   * wraps (see applyMaxRowsForOracle), the probe is applied to every
+   * row-returning statement; a query whose own cap is tighter than maxRows
+   * returns fewer than maxRows + 1 rows and is therefore never flagged.
+   * A `FOR UPDATE` statement is never capped (see hasOracleForUpdate).
+   */
+  static applyMaxRowsForOracleWithTruncationProbe(sql, maxRows) {
+    if (!maxRows || !this.isSelectQuery(sql, "oracle") || this.hasOracleForUpdate(sql)) {
+      return { sql, probeApplied: false };
+    }
+    return { sql: this.applyMaxRowsForOracle(sql, maxRows + 1), probeApplied: true };
+  }
+  /**
+   * Like applyMaxRows, but requests one probe row (maxRows + 1) whenever the
+   * cap is the binding constraint, so callers can detect truncation exactly
+   * instead of guessing from `rowCount === maxRows` (which a table with
+   * exactly maxRows rows also produces).
+   *
+   * `probeApplied` is false when the cap cannot fire: no maxRows, not a
+   * SELECT, or the query's own literal LIMIT is already within the cap (the
+   * user asked for fewer rows than the cap — that is their limit, not
+   * truncation).
+   */
+  static applyMaxRowsWithTruncationProbe(sql, maxRows, dialect) {
+    if (!maxRows || !this.isSelectQuery(sql, dialect)) {
+      return { sql, probeApplied: false };
+    }
+    if (!this.hasParameterizedLimit(sql, dialect)) {
+      const existingLimit = this.extractLimitValue(sql, dialect);
+      if (existingLimit !== null && existingLimit <= maxRows) {
+        return { sql, probeApplied: false };
+      }
+    }
+    return { sql: this.applyMaxRows(sql, maxRows + 1, dialect), probeApplied: true };
+  }
+  /**
+   * SQL Server variant of applyMaxRowsWithTruncationProbe, using TOP syntax.
+   * A set-operator statement is always wrapped (any TOP on an individual
+   * branch caps only that branch, not the combined output), so the probe
+   * applies there regardless of branch-level TOPs.
+   */
+  static applyMaxRowsForSQLServerWithTruncationProbe(sql, maxRows) {
+    if (!maxRows || !this.isSelectQuery(sql, "sqlserver")) {
+      return { sql, probeApplied: false };
+    }
+    if (!this.hasSetOperator(sql)) {
+      const existingTop = this.extractTopValue(sql);
+      if (existingTop !== null && existingTop <= maxRows) {
+        return { sql, probeApplied: false };
+      }
+    }
+    return { sql: this.applyMaxRowsForSQLServer(sql, maxRows + 1), probeApplied: true };
+  }
+  /**
+   * Post-process a result set produced by a truncation-probe rewrite: when
+   * the probe row came back (more than maxRows rows), drop it, clamp the
+   * count to maxRows, and mark the set truncated so consumers can tell a
+   * capped result from a complete one.
+   */
+  static flagTruncation(resultSet, maxRows, probeApplied) {
+    if (!probeApplied || !maxRows || resultSet.rows.length <= maxRows) {
+      return;
+    }
+    resultSet.rows = resultSet.rows.slice(0, maxRows);
+    resultSet.rowCount = maxRows;
+    resultSet.truncated = true;
+  }
+};
+function trimSemicolon(sql) {
+  const trimmed = sql.trim();
+  return trimmed.endsWith(";") ? { sql: trimmed.slice(0, -1), semicolon: ";" } : { sql: trimmed, semicolon: "" };
+}
+
+// src/utils/resource-cleanup.ts
+async function closeQuietly(close) {
+  try {
+    await close();
+  } catch {
+  }
+}
+
+export {
+  SQLRowLimiter,
+  closeQuietly
+};

package/dist/index.js

@@ -18,20 +18,20 @@ import {
   resolveSourceConfigs,
   resolveTomlConfigPath,
   resolveTransport
-} from "./chunk-VDA5BVNE.js";
+} from "./chunk-5FL4XMWG.js";
+import {
+  loadConnectors
+} from "./chunk-WVVMH6FJ.js";
+import {
+  quoteQualifiedIdentifier
+} from "./chunk-X2EC47W4.js";
 import {
   allowedKeywords,
   getFirstKeyword,
   hasEscapeHatchFunction,
   isReadOnlySQL,
   sqlServerDynamicSqlPattern
-} from "./chunk-IQRBLAB4.js";
-import {
-  loadConnectors
-} from "./chunk-WVVMH6FJ.js";
-import {
-  quoteQualifiedIdentifier
-} from "./chunk-JFWX35TB.js";
+} from "./chunk-ABTVMHD6.js";
 import {
   ConnectorRegistry,
   getDatabaseTypeFromDSN,
@@ -39,7 +39,7 @@ import {
   parseConnectionInfoFromDSN,
   splitSQLStatements,
   stripCommentsAndStrings
-} from "./chunk-MF6LSVWK.js";
+} from "./chunk-IXC4NQGC.js";
 
 // src/server.ts
 import { McpServer, createMcpHandler } from "@modelcontextprotocol/server";
@@ -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-UF3N2H4G.js");
+    const { initializeToolRegistry: initializeToolRegistry2 } = await import("./registry-ZTHDPLKP.js");
     initializeToolRegistry2({
       sources: sourceConfigsData.sources,
       tools: sourceConfigsData.tools
@@ -2068,11 +2068,12 @@ See documentation for more details on configuring database connections.
 
 // src/index.ts
 var connectorModules = [
-  { 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-EVJ5HLYH.js"), name: "MariaDB", driver: "mariadb" }
+  { load: () => import("./postgres-6XJ7V4J4.js"), name: "PostgreSQL", driver: "pg" },
+  { load: () => import("./sqlserver-AJUK6L55.js"), name: "SQL Server", driver: "mssql" },
+  { load: () => import("./sqlite-AO2BKIC6.js"), name: "SQLite", driver: "node:sqlite" },
+  { load: () => import("./mysql-XXIHWQGR.js"), name: "MySQL", driver: "mysql2" },
+  { load: () => import("./mariadb-L7Z55ROY.js"), name: "MariaDB", driver: "mariadb" },
+  { load: () => import("./oracle-OUKBU52X.js"), name: "Oracle", driver: "oracledb" }
 ];
 loadConnectors(connectorModules).then(() => main()).catch((error) => {
   console.error("Fatal error:", error);

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

@@ -5,21 +5,22 @@ import {
   parseQueryResultSets,
   requireDatabaseInDSN,
   withReadOnlyTransaction
-} from "./chunk-A4TSDCPY.js";
+} from "./chunk-6SZKW6GD.js";
 import "./chunk-FU2ZJE4E.js";
 import {
   quoteIdentifier
-} from "./chunk-JFWX35TB.js";
+} from "./chunk-X2EC47W4.js";
 import {
   SQLRowLimiter,
   closeQuietly
-} from "./chunk-6QETXS44.js";
+} from "./chunk-YIGJHEU6.js";
+import "./chunk-ABTVMHD6.js";
 import {
   ConnectorRegistry,
   SafeURL,
   obfuscateDSNPassword,
   splitSQLStatements
-} from "./chunk-MF6LSVWK.js";
+} from "./chunk-IXC4NQGC.js";
 
 // src/connectors/mariadb/index.ts
 import * as mariadb from "mariadb";
@@ -515,7 +516,7 @@ var MariaDBConnector = class _MariaDBConnector {
           let probes = [];
           if (options.maxRows) {
             const rewrites = statements.map(
-              (statement) => SQLRowLimiter.applyMaxRowsWithTruncationProbe(statement, options.maxRows)
+              (statement) => SQLRowLimiter.applyMaxRowsWithTruncationProbe(statement, options.maxRows, "mariadb")
             );
             probes = rewrites.map((rewrite) => rewrite.probeApplied);
             processedSQL = rewrites.map((rewrite) => rewrite.sql).join("; ");

package/dist/{mysql-GXDUNLJ6.js → mysql-XXIHWQGR.js}

@@ -6,21 +6,22 @@ import {
   parseQueryResultSets,
   requireDatabaseInDSN,
   withReadOnlyTransaction
-} from "./chunk-A4TSDCPY.js";
+} from "./chunk-6SZKW6GD.js";
 import "./chunk-FU2ZJE4E.js";
 import {
   quoteIdentifier
-} from "./chunk-JFWX35TB.js";
+} from "./chunk-X2EC47W4.js";
 import {
   SQLRowLimiter,
   closeQuietly
-} from "./chunk-6QETXS44.js";
+} from "./chunk-YIGJHEU6.js";
+import "./chunk-ABTVMHD6.js";
 import {
   ConnectorRegistry,
   SafeURL,
   obfuscateDSNPassword,
   splitSQLStatements
-} from "./chunk-MF6LSVWK.js";
+} from "./chunk-IXC4NQGC.js";
 
 // src/connectors/mysql/index.ts
 import mysql from "mysql2/promise";
@@ -520,7 +521,7 @@ var _MySQLConnector = class _MySQLConnector {
           let probes = [];
           if (options.maxRows) {
             const rewrites = statements.map(
-              (statement) => SQLRowLimiter.applyMaxRowsWithTruncationProbe(statement, options.maxRows)
+              (statement) => SQLRowLimiter.applyMaxRowsWithTruncationProbe(statement, options.maxRows, "mysql")
             );
             probes = rewrites.map((rewrite) => rewrite.probeApplied);
             processedSQL = rewrites.map((rewrite) => rewrite.sql).join("; ");