bun-query-builder 0.2.5 → 0.2.7

package/dist/bin/cli.js

@@ -11643,7 +11643,7 @@ var init_dist = __esm(() => {
 // src/config.ts
 import process19 from "process";
 function isMysqlLike(dialect = config5.dialect) {
-  return dialect === "mysql" || dialect === "singlestore";
+  return dialect === "mysql" || dialect === "singlestore" || dialect === "vitess";
 }
 function resolveDialect(explicit) {
   if (explicit)
@@ -11701,6 +11701,9 @@ function applyConfig(userConfig) {
   if (userConfig.softDeletes) {
     config5.softDeletes = { ...config5.softDeletes, ...userConfig.softDeletes };
   }
+  if (userConfig.vitess) {
+    config5.vitess = { ...config5.vitess, ...userConfig.vitess };
+  }
 }
 function setConfig(userConfig) {
   if (userConfig.dialect && _lastConfiguredDialect && userConfig.dialect !== _lastConfiguredDialect) {
@@ -11730,6 +11733,9 @@ var init_config = __esm(() => {
       host: "localhost",
       port: 5432
     },
+    vitess: {
+      sharded: true
+    },
     timestamps: {
       createdAt: "created_at",
       updatedAt: "updated_at",
@@ -12094,7 +12100,7 @@ function resolveDatabaseName(dbConfig = config5.database) {
   return String(raw).replace(/^['"]|['"]$/g, "");
 }
 function createConnectionString(dialect, dbConfig) {
-  if ((dialect === "postgres" || dialect === "mysql" || dialect === "singlestore") && process21.env.DB_CONNECTION === dialect) {
+  if ((dialect === "postgres" || dialect === "mysql" || dialect === "singlestore" || dialect === "vitess") && process21.env.DB_CONNECTION === dialect) {
     const e = process21.env;
     const envDb = (e.DB_DATABASE || dbConfig.database || "").replace(/^['"]|['"]$/g, "");
     const envUser = e.DB_USERNAME || dbConfig.username;
@@ -12115,6 +12121,7 @@ function createConnectionString(dialect, dbConfig) {
       return `postgres://${username}:${password}@${host}${port ? `:${port}` : ""}/${database}${sslQ}`;
     case "mysql":
     case "singlestore":
+    case "vitess":
       return `mysql://${username}:${password}@${host}${port ? `:${port}` : ""}/${database}${sslQ}`;
     case "sqlite":
       if (database === ":memory:") {
@@ -17969,7 +17976,7 @@ function parseMysqlEnum(columnType) {
     return;
   return m[1].split(",").map((s) => s.trim().replace(/^'(.*)'$/s, "$1").replace(/''/g, "'")).filter((s) => s.length > 0);
 }
-function groupPhysicalIndexes(table, rows) {
+function groupPhysicalIndexes(_table, rows) {
   const byName = new Map;
   for (const r of rows) {
     if (r.isPrimary)
@@ -18525,6 +18532,42 @@ var init_singlestore = __esm(() => {
   };
 });
 
+// src/drivers/vitess.ts
+var VitessDriver;
+var init_vitess = __esm(() => {
+  init_mysql();
+  VitessDriver = class VitessDriver extends MySQLDriver {
+    sharded;
+    constructor(sharded = true) {
+      super();
+      this.sharded = sharded;
+    }
+    getAutoIncrementClause(column) {
+      return this.sharded ? "" : super.getAutoIncrementClause(column);
+    }
+    renderColumn(column) {
+      if (!this.sharded || !column.references)
+        return super.renderColumn(column);
+      const { references: _references, ...withoutReference } = column;
+      return super.renderColumn(withoutReference);
+    }
+    createTable(table) {
+      return super.createTable(table);
+    }
+    addForeignKey(...args) {
+      return this.sharded ? "" : super.addForeignKey(...args);
+    }
+    createMigrationsTable() {
+      if (!this.sharded)
+        return super.createMigrationsTable();
+      return `CREATE TABLE IF NOT EXISTS migrations (
+      migration VARCHAR(255) NOT NULL PRIMARY KEY,
+      executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+    )`;
+    }
+  };
+});
+
 // src/drivers/sqlite.ts
 function isNumericPlanType(type) {
   return type === "integer" || type === "bigint" || type === "float" || type === "double" || type === "decimal";
@@ -18736,6 +18779,8 @@ function getDialectDriver(dialect) {
       return new MySQLDriver;
     case "singlestore":
       return new SingleStoreDriver;
+    case "vitess":
+      return new VitessDriver(config5.vitess?.sharded ?? true);
     case "sqlite":
       return new SQLiteDriver;
     default:
@@ -18743,14 +18788,17 @@ function getDialectDriver(dialect) {
   }
 }
 var init_drivers = __esm(() => {
+  init_config();
   init_mysql();
   init_postgres();
   init_singlestore();
+  init_vitess();
   init_sqlite();
   init_mysql();
   init_postgres();
   init_singlestore();
   init_sqlite();
+  init_vitess();
 });
 
 // src/workspace.ts
@@ -29238,7 +29286,7 @@ function generateDiffOperations(previous, next, opts = {}) {
     if (preserveUnknownColumns && removedCols.length > 0) {
       const preserved = removedCols.map((name) => prevCols[name]);
       const preservedNames = new Set(removedCols);
-      const keptIndexes = prev.indexes.filter((idx) => idx.columns.some((c2) => preservedNames.has(c2)) && !currIdx[indexKey(idx)]);
+      const keptIndexes = prev.indexes.filter((idx) => idx.columns.some((c2) => preservedNames.has(c2)) && !currIdx[indexKey(idx, prev.table)]);
       info2(`-- Keeping ${preserved.length} column(s) on "${curr.table}" no model declares: ${removedCols.join(", ")} \u2014 reconciled from the database, where an undeclared column is out of scope rather than removed`);
       curr = { ...curr, columns: [...curr.columns, ...preserved], indexes: [...curr.indexes, ...keptIndexes] };
       currCols = mapColumnsByName(curr.columns);
@@ -31480,7 +31528,7 @@ function getPrefix() {
 }
 var prefix = getPrefix();
 // package.json
-var version2 = "0.2.5";
+var version2 = "0.2.7";
 
 // bin/cli.ts
 init_actions();

package/dist/config.d.ts

@@ -1,11 +1,12 @@
 import type { QueryBuilderConfig, SupportedDialect } from './types';
 /**
- * Whether a dialect belongs to the MySQL wire-protocol family (MySQL itself or
- * SingleStore). These share `?` placeholders, backtick identifier quoting,
- * `ON DUPLICATE KEY UPDATE` upserts, `LAST_INSERT_ID()` id recovery, and the
- * `YYYY-MM-DD HH:MM:SS` datetime literal shape. Runtime DML branches should key
- * off this helper rather than `dialect === 'mysql'` so SingleStore inherits the
- * same behavior; only DDL (see `SingleStoreDriver`) diverges.
+ * Whether a dialect belongs to the MySQL wire-protocol family (MySQL itself,
+ * SingleStore, or Vitess). These share `?` placeholders, backtick identifier
+ * quoting, `ON DUPLICATE KEY UPDATE` upserts, `LAST_INSERT_ID()` id recovery,
+ * and the `YYYY-MM-DD HH:MM:SS` datetime literal shape. Runtime DML branches
+ * should key off this helper rather than `dialect === 'mysql'` so the
+ * distributed dialects inherit the same behavior; only DDL (see
+ * `SingleStoreDriver` / `VitessDriver`) diverges.
  */
 export declare function isMysqlLike(dialect?: SupportedDialect): boolean;
 /**

package/dist/drivers/index.d.ts

@@ -2,6 +2,7 @@ import { MySQLDriver } from './mysql';
 import { PostgresDriver } from './postgres';
 import { SingleStoreDriver } from './singlestore';
 import { SQLiteDriver } from './sqlite';
+import { VitessDriver } from './vitess';
 import type { DialectDriver } from './postgres';
 import type { SupportedDialect } from '../types';
 export type { DialectDriver } from './postgres';
@@ -32,6 +33,7 @@ export { MySQLDriver } from './mysql';
 export { PostgresDriver } from './postgres';
 export { SingleStoreDriver } from './singlestore';
 export { SQLiteDriver } from './sqlite';
+export { VitessDriver } from './vitess';
 // DynamoDB driver (NoSQL)
 export {
   createDynamoDBDriver,

package/dist/drivers/vitess.d.ts

@@ -0,0 +1,41 @@
+import { MySQLDriver } from './mysql';
+import type { ColumnPlan, TablePlan } from '../migrations';
+/**
+ * Vitess DDL driver.
+ *
+ * Vitess shards MySQL behind vtgate, which speaks the MySQL wire protocol, so
+ * runtime DML (placeholders, backtick quoting, `ON DUPLICATE KEY UPDATE`,
+ * `LAST_INSERT_ID`) is identical — see `isMysqlLike` in `config.ts`. Only DDL
+ * diverges only when the keyspace is sharded. An unsharded keyspace retains
+ * normal MySQL AUTO_INCREMENT and foreign-key support, so this driver takes
+ * the topology explicitly instead of conflating "behind vtgate" with
+ * "distributed across shards".
+ *
+ *  - **No foreign keys.** Enforcing one would need a cross-shard read on every
+ *    write, so vtgate rejects them in a sharded keyspace. Both forms are
+ *    suppressed: the `ALTER TABLE … ADD CONSTRAINT` statement (`addForeignKey`)
+ *    and the inline `REFERENCES` clause the MySQL driver emits inside a column
+ *    definition. Missing the inline form would be the subtler bug — the table
+ *    plan carries the reference on the column, so a `CREATE TABLE` would fail
+ *    on its own without any FK statement being generated.
+ *  - **No AUTO_INCREMENT.** Each shard would independently hand out the same
+ *    values and collide. This is not a parse error, which makes it worse than
+ *    one: the DDL succeeds and the collisions appear later, under load, as
+ *    duplicate-key failures on inserts that were previously fine. Primary keys
+ *    must come from the application or from a sequence table in an unsharded
+ *    keyspace, declared in the VSchema.
+ *
+ * **Sharding is not expressed in DDL.** Unlike SingleStore, where the shard key
+ * is a `CREATE TABLE` clause, Vitess keeps the topology in a separate VSchema
+ * document that maps each table's sharding column to a vindex. So `createTable`
+ * here is plain MySQL minus the two constructs above, and nothing in this
+ * driver needs to know how a table is sharded.
+ */
+export declare class VitessDriver extends MySQLDriver {
+  constructor(sharded?: boolean);
+  protected getAutoIncrementClause(column: ColumnPlan): string;
+  protected renderColumn(column: ColumnPlan): string;
+  createTable(table: TablePlan): string;
+  addForeignKey(...args: Parameters<MySQLDriver['addForeignKey']>): string;
+  createMigrationsTable(): string;
+}

package/dist/src/index.js

@@ -11642,7 +11642,7 @@ var init_dist = __esm(() => {
 // src/config.ts
 import process19 from "process";
 function isMysqlLike(dialect = config5.dialect) {
-  return dialect === "mysql" || dialect === "singlestore";
+  return dialect === "mysql" || dialect === "singlestore" || dialect === "vitess";
 }
 function resolveDialect(explicit) {
   if (explicit)
@@ -11700,6 +11700,9 @@ function applyConfig(userConfig) {
   if (userConfig.softDeletes) {
     config5.softDeletes = { ...config5.softDeletes, ...userConfig.softDeletes };
   }
+  if (userConfig.vitess) {
+    config5.vitess = { ...config5.vitess, ...userConfig.vitess };
+  }
 }
 function setConfig(userConfig) {
   if (userConfig.dialect && _lastConfiguredDialect && userConfig.dialect !== _lastConfiguredDialect) {
@@ -11729,6 +11732,9 @@ var init_config = __esm(() => {
       host: "localhost",
       port: 5432
     },
+    vitess: {
+      sharded: true
+    },
     timestamps: {
       createdAt: "created_at",
       updatedAt: "updated_at",
@@ -12093,7 +12099,7 @@ function resolveDatabaseName(dbConfig = config5.database) {
   return String(raw).replace(/^['"]|['"]$/g, "");
 }
 function createConnectionString(dialect, dbConfig) {
-  if ((dialect === "postgres" || dialect === "mysql" || dialect === "singlestore") && process21.env.DB_CONNECTION === dialect) {
+  if ((dialect === "postgres" || dialect === "mysql" || dialect === "singlestore" || dialect === "vitess") && process21.env.DB_CONNECTION === dialect) {
     const e = process21.env;
     const envDb = (e.DB_DATABASE || dbConfig.database || "").replace(/^['"]|['"]$/g, "");
     const envUser = e.DB_USERNAME || dbConfig.username;
@@ -12114,6 +12120,7 @@ function createConnectionString(dialect, dbConfig) {
       return `postgres://${username}:${password}@${host}${port ? `:${port}` : ""}/${database}${sslQ}`;
     case "mysql":
     case "singlestore":
+    case "vitess":
       return `mysql://${username}:${password}@${host}${port ? `:${port}` : ""}/${database}${sslQ}`;
     case "sqlite":
       if (database === ":memory:") {
@@ -17968,7 +17975,7 @@ function parseMysqlEnum(columnType) {
     return;
   return m[1].split(",").map((s) => s.trim().replace(/^'(.*)'$/s, "$1").replace(/''/g, "'")).filter((s) => s.length > 0);
 }
-function groupPhysicalIndexes(table, rows) {
+function groupPhysicalIndexes(_table, rows) {
   const byName = new Map;
   for (const r of rows) {
     if (r.isPrimary)
@@ -18524,6 +18531,42 @@ var init_singlestore = __esm(() => {
   };
 });
 
+// src/drivers/vitess.ts
+var VitessDriver;
+var init_vitess = __esm(() => {
+  init_mysql();
+  VitessDriver = class VitessDriver extends MySQLDriver {
+    sharded;
+    constructor(sharded = true) {
+      super();
+      this.sharded = sharded;
+    }
+    getAutoIncrementClause(column) {
+      return this.sharded ? "" : super.getAutoIncrementClause(column);
+    }
+    renderColumn(column) {
+      if (!this.sharded || !column.references)
+        return super.renderColumn(column);
+      const { references: _references, ...withoutReference } = column;
+      return super.renderColumn(withoutReference);
+    }
+    createTable(table) {
+      return super.createTable(table);
+    }
+    addForeignKey(...args) {
+      return this.sharded ? "" : super.addForeignKey(...args);
+    }
+    createMigrationsTable() {
+      if (!this.sharded)
+        return super.createMigrationsTable();
+      return `CREATE TABLE IF NOT EXISTS migrations (
+      migration VARCHAR(255) NOT NULL PRIMARY KEY,
+      executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+    )`;
+    }
+  };
+});
+
 // src/drivers/sqlite.ts
 function isNumericPlanType(type) {
   return type === "integer" || type === "bigint" || type === "float" || type === "double" || type === "decimal";
@@ -18735,6 +18778,8 @@ function getDialectDriver(dialect) {
       return new MySQLDriver;
     case "singlestore":
       return new SingleStoreDriver;
+    case "vitess":
+      return new VitessDriver(config5.vitess?.sharded ?? true);
     case "sqlite":
       return new SQLiteDriver;
     default:
@@ -18742,14 +18787,17 @@ function getDialectDriver(dialect) {
   }
 }
 var init_drivers = __esm(() => {
+  init_config();
   init_mysql();
   init_postgres();
   init_singlestore();
+  init_vitess();
   init_sqlite();
   init_mysql();
   init_postgres();
   init_singlestore();
   init_sqlite();
+  init_vitess();
 });
 
 // src/workspace.ts
@@ -29237,7 +29285,7 @@ function generateDiffOperations(previous, next, opts = {}) {
     if (preserveUnknownColumns && removedCols.length > 0) {
       const preserved = removedCols.map((name) => prevCols[name]);
       const preservedNames = new Set(removedCols);
-      const keptIndexes = prev.indexes.filter((idx) => idx.columns.some((c2) => preservedNames.has(c2)) && !currIdx[indexKey(idx)]);
+      const keptIndexes = prev.indexes.filter((idx) => idx.columns.some((c2) => preservedNames.has(c2)) && !currIdx[indexKey(idx, prev.table)]);
       info2(`-- Keeping ${preserved.length} column(s) on "${curr.table}" no model declares: ${removedCols.join(", ")} \u2014 reconciled from the database, where an undeclared column is out of scope rather than removed`);
       curr = { ...curr, columns: [...curr.columns, ...preserved], indexes: [...curr.indexes, ...keptIndexes] };
       currCols = mapColumnsByName(curr.columns);
@@ -29580,6 +29628,7 @@ export {
   browserQuery,
   browserAuth,
   applySqliteBootstrapPragmas,
+  VitessDriver,
   SingleTableRepository,
   SingleTableQueryBuilder,
   SingleTablePatterns,

package/dist/types.d.ts

@@ -192,6 +192,10 @@ export declare interface DatabaseConfig {
   ssl?: boolean
   pool?: PoolConfig
 }
+/** Vitess topology controls that affect generated DDL. */
+export declare interface VitessConfig {
+  sharded: boolean
+}
 /**
  * # `BrowserConfig`
  *
@@ -236,6 +240,7 @@ export declare interface QueryBuilderConfig {
   transactionDefaults: TransactionDefaultsConfig
   sql: SqlConfig
   sqlite?: SqliteConfig
+  vitess?: VitessConfig
   features: FeatureToggles
   debug?: {
     captureText: boolean
@@ -293,7 +298,11 @@ export declare interface UnsafeOptions {
  *   but its DDL adds distributed-table concepts (`SHARD KEY`, `SORT KEY`,
  *   `ROWSTORE`/columnstore) and drops foreign keys — handled by the dedicated
  *   `SingleStoreDriver`.
+ * - 'vitess': MySQL behind vtgate. Shares MySQL's wire protocol and DML (see
+ *   `isMysqlLike`). Unsharded keyspaces retain normal MySQL DDL, while a
+ *   sharded keyspace rejects foreign keys and AUTO_INCREMENT; select that
+ *   stricter profile with `vitess.sharded`.
  * - 'sqlite': Lightweight engine; some features are limited or emulated
  * - 'browser': Browser-compatible mode that uses fetch() API calls instead of direct database connections
  */
-export type SupportedDialect = 'postgres' | 'mysql' | 'singlestore' | 'sqlite' | 'browser';
+export type SupportedDialect = 'postgres' | 'mysql' | 'singlestore' | 'vitess' | 'sqlite' | 'browser';

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "bun-query-builder",
   "type": "module",
-  "version": "0.2.5",
+  "version": "0.2.7",
   "description": "A simple yet performant query builder for TypeScript. Built with Bun.",
   "author": "Chris Breuer <chris@stacksjs.org>",
   "license": "MIT",