@fragno-dev/db 0.1.6 → 0.1.7

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @fragno-dev/db
 
+## 0.1.7
+
+### Patch Changes
+
+- e36dbcd: fix: allow async lazy initialization of `db` in adapters
+- ab6c4bf: fix: make Fragment loading in the CLI more robust
+- d1feecd: fix: allow lazy initialization of `db` in adapters
+
 ## 0.1.6
 
 ### Patch Changes

package/dist/adapters/adapters.d.ts

@@ -4,7 +4,11 @@ import { AbstractQuery } from "../query/query.js";
 import { SchemaGenerator } from "../schema-generator/schema-generator.js";
 
 //#region src/adapters/adapters.d.ts
+declare const fragnoDatabaseAdapterNameFakeSymbol: "$fragno-database-adapter-name";
+declare const fragnoDatabaseAdapterVersionFakeSymbol: "$fragno-database-adapter-version";
 interface DatabaseAdapter<TUOWConfig = void> {
+  [fragnoDatabaseAdapterNameFakeSymbol]: string;
+  [fragnoDatabaseAdapterVersionFakeSymbol]: number;
   /**
    * Get current schema version, undefined if not initialized.
    */
@@ -22,7 +26,8 @@ interface DatabaseAdapter<TUOWConfig = void> {
     path?: string;
   }) => SchemaGenerator;
   isConnectionHealthy: () => Promise<boolean>;
+  close: () => Promise<void>;
 }
 //#endregion
-export { DatabaseAdapter };
+export { DatabaseAdapter, fragnoDatabaseAdapterNameFakeSymbol, fragnoDatabaseAdapterVersionFakeSymbol };
 //# sourceMappingURL=adapters.d.ts.map

package/dist/adapters/adapters.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"adapters.d.ts","names":[],"sources":["../../src/adapters/adapters.ts"],"sourcesContent":[],"mappings":";;;;;;UAKiB;;AAAjB;;EAMsC,gBAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAFC,OAED,CAAA,MAAA,GAAA,SAAA,CAAA;EAC1B,iBAAA,EAAA,CAAA,gBAD0B,SAC1B,CAAA,CAAA,MAAA,EAAA,CAAA,EAAA,SAAA,EAAA,MAAA,EAAA,GAEL,aAFK,CAES,CAFT,EAEY,UAFZ,CAAA;EAES,qBAAA,CAAA,EAAA,CAAA,gBAEsB,SAFtB,CAAA,CAAA,MAAA,EAEyC,CAFzC,EAAA,SAAA,EAAA,MAAA,EAAA,GAEkE,QAFlE;EAAG;;;;EAE+D,qBAAA,CAAA,EAAA,CAAA,SAAA,EAAA;IAO9D,MAAA,EAAA,SAAA;IAElB,SAAA,EAAA,MAAA;EAEsB,CAAA,EAAA,EAAA,QAAA,EAAA;IAAO,IAAA,CAAA,EAAA,MAAA;QAF7B;6BAEsB"}
+{"version":3,"file":"adapters.d.ts","names":[],"sources":["../../src/adapters/adapters.ts"],"sourcesContent":[],"mappings":";;;;;;cAKa;cACA;AADA,UAGI,eAHJ,CAAA,aAA8E,IAAA,CAAA,CAAA;EAC9E,CAGV,mCAAA,CAHU,EAAA,MAAoF;EAEhF,CAEd,sCAAA,CAF6B,EAAA,MAAA;EAC7B;;;EAQmC,gBAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAFC,OAED,CAAA,MAAA,GAAA,SAAA,CAAA;EAC1B,iBAAA,EAAA,CAAA,gBAD0B,SAC1B,CAAA,CAAA,MAAA,EAAA,CAAA,EAAA,SAAA,EAAA,MAAA,EAAA,GAEL,aAFK,CAES,CAFT,EAEY,UAFZ,CAAA;EAES,qBAAA,CAAA,EAAA,CAAA,gBAEsB,SAFtB,CAAA,CAAA,MAAA,EAEyC,CAFzC,EAAA,SAAA,EAAA,MAAA,EAAA,GAEkE,QAFlE;EAAG;;;;EAE+D,qBAAA,CAAA,EAAA,CAAA,SAAA,EAAA;IAO9D,MAAA,EAAA,SAAA;IAElB,SAAA,EAAA,MAAA;EAEsB,CAAA,EAAA,EAAA,OAEP,CAFO,EAAA;IAEd,IAAA,CAAA,EAAA,MAAA;EAAO,CAAA,EAAA,GAJf,eAIe;6BAFO;eAEd"}

package/dist/adapters/adapters.js

@@ -1 +1,7 @@
-export {  };
+//#region src/adapters/adapters.ts
+const fragnoDatabaseAdapterNameFakeSymbol = "$fragno-database-adapter-name";
+const fragnoDatabaseAdapterVersionFakeSymbol = "$fragno-database-adapter-version";
+
+//#endregion
+export { fragnoDatabaseAdapterNameFakeSymbol, fragnoDatabaseAdapterVersionFakeSymbol };
+//# sourceMappingURL=adapters.js.map

package/dist/adapters/adapters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapters.js","names":[],"sources":["../../src/adapters/adapters.ts"],"sourcesContent":["import type { Migrator } from \"../migration-engine/create\";\nimport type { AbstractQuery } from \"../query/query\";\nimport type { SchemaGenerator } from \"../schema-generator/schema-generator\";\nimport type { AnySchema } from \"../schema/create\";\n\nexport const fragnoDatabaseAdapterNameFakeSymbol = \"$fragno-database-adapter-name\" as const;\nexport const fragnoDatabaseAdapterVersionFakeSymbol = \"$fragno-database-adapter-version\" as const;\n\nexport interface DatabaseAdapter<TUOWConfig = void> {\n  [fragnoDatabaseAdapterNameFakeSymbol]: string;\n  [fragnoDatabaseAdapterVersionFakeSymbol]: number;\n\n  /**\n   * Get current schema version, undefined if not initialized.\n   */\n  getSchemaVersion(namespace: string): Promise<string | undefined>;\n\n  createQueryEngine: <const T extends AnySchema>(\n    schema: T,\n    namespace: string,\n  ) => AbstractQuery<T, TUOWConfig>;\n\n  createMigrationEngine?: <const T extends AnySchema>(schema: T, namespace: string) => Migrator;\n\n  /**\n   * Generate a combined schema file from one or more fragments.\n   * If not implemented, schema generation is not supported for this adapter.\n   */\n  createSchemaGenerator?: (\n    fragments: { schema: AnySchema; namespace: string }[],\n    options?: { path?: string },\n  ) => SchemaGenerator;\n\n  isConnectionHealthy: () => Promise<boolean>;\n\n  close: () => Promise<void>;\n}\n"],"mappings":";AAKA,MAAa,sCAAsC;AACnD,MAAa,yCAAyC"}

package/dist/adapters/drizzle/drizzle-adapter.d.ts

@@ -1,17 +1,20 @@
 import { AnySchema } from "../../schema/create.js";
 import { AbstractQuery } from "../../query/query.js";
 import { SchemaGenerator } from "../../schema-generator/schema-generator.js";
-import { DatabaseAdapter } from "../adapters.js";
+import { DatabaseAdapter, fragnoDatabaseAdapterNameFakeSymbol, fragnoDatabaseAdapterVersionFakeSymbol } from "../adapters.js";
 import { DrizzleUOWConfig } from "./drizzle-query.js";
 
 //#region src/adapters/drizzle/drizzle-adapter.d.ts
 interface DrizzleConfig {
-  db: unknown | (() => unknown);
+  db: unknown | (() => unknown | Promise<unknown>);
   provider: "sqlite" | "mysql" | "postgresql";
 }
 declare class DrizzleAdapter implements DatabaseAdapter<DrizzleUOWConfig> {
   #private;
   constructor(config: DrizzleConfig);
+  get [fragnoDatabaseAdapterNameFakeSymbol](): string;
+  get [fragnoDatabaseAdapterVersionFakeSymbol](): number;
+  close(): Promise<void>;
   get provider(): "sqlite" | "mysql" | "postgresql";
   isConnectionHealthy(): Promise<boolean>;
   getSchemaVersion(namespace: string): Promise<string | undefined>;

package/dist/adapters/drizzle/drizzle-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-adapter.d.ts","names":[],"sources":["../../../src/adapters/drizzle/drizzle-adapter.ts"],"sourcesContent":[],"mappings":";;;;;;;UAUiB,aAAA;;EAAA,QAAA,EAAA,QAAa,GAAA,OAAA,GAAA,YAAA;AAK9B;AAAuD,cAA1C,cAAA,YAA0B,eAAgB,CAAA,gBAAA,CAAA,CAAA;EAGjC,CAAA,OAAA;EAaS,WAAA,CAAA,MAAA,EAbT,aAaS;EAkBc,IAAA,QAAA,CAAA,CAAA,EAAA,QAAA,GAAA,OAAA,GAAA,YAAA;EAST,mBAAA,CAAA,CAAA,EA3BL,OA2BK,CAAA,OAAA,CAAA;EACxB,gBAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAViC,OAUjC,CAAA,MAAA,GAAA,SAAA,CAAA;EAEO,iBAAA,CAAA,gBAHiB,SAGjB,CAAA,CAAA,MAAA,EAFP,OAEO,EAAA,SAAA,EAAA,MAAA,CAAA,EAAd,aAAc,CAAA,OAAA,EAAS,gBAAT,CAAA;EAAS,qBAAA,CAAA,SAAA,EAAA;IAAvB,MAAA,EAYoB,SAZpB;IAYoB,SAAA,EAAA,MAAA;EAEpB,CAAA,EAAA,EAAA,OA5DiD,CA4DjD,EAAA;IA5DkC,IAAA,CAAA,EAAA,MAAA;EAAe,CAAA,CAAA,EA4DjD,eA5DiD"}
+{"version":3,"file":"drizzle-adapter.d.ts","names":[],"sources":["../../../src/adapters/drizzle/drizzle-adapter.ts"],"sourcesContent":[],"mappings":";;;;;;;UAgBiB,aAAA;EAAA,EAAA,EAAA,OAAA,GAAA,CAAA,GAAa,GAAA,OAAA,GACG,OAAO,CAAA,OAAA,CAAA,CAAA;EAI3B,QAAA,EAAA,QAAe,GAAA,OAAA,GAAA,YAAA;;AAIN,cAJT,cAAA,YAA0B,eAIjB,CAJiC,gBAIjC,CAAA,CAAA;EAOf,CAAA,OAAA;EAIA,WAAA,CAAA,MAAA,EAXe,aAWf;EAIU,KARV,mCAAA,GAQU,EAAA,MAAA;EAQc,KAZxB,sCAAA,GAYwB,EAAA,MAAA;EAqBc,KAAA,CAAA,CAAA,EA7B5B,OA6B4B,CAAA,IAAA,CAAA;EAST,IAAA,QAAA,CAAA,CAAA,EAAA,QAAA,GAAA,OAAA,GAAA,YAAA;EACxB,mBAAA,CAAA,CAAA,EA/BmB,OA+BnB,CAAA,OAAA,CAAA;EAEO,gBAAA,CAAA,SAAA,EAAA,MAAA,CAAA,EAZ0B,OAY1B,CAAA,MAAA,GAAA,SAAA,CAAA;EAAS,iBAAA,CAAA,gBAHQ,SAGR,CAAA,CAAA,MAAA,EAFhB,OAEgB,EAAA,SAAA,EAAA,MAAA,CAAA,EAAvB,aAAuB,CAAT,OAAS,EAAA,gBAAA,CAAA;EAAvB,qBAAA,CAAA,SAAA,EAAA;IAOoB,MAAA,EAAA,SAAA;IAEpB,SAAA,EAAA,MAAA;EArEkC,CAAA,EAAA,EAAA,QAAA,EAAA;IAAe,IAAA,CAAA,EAAA,MAAA;MAqEjD"}

package/dist/adapters/drizzle/drizzle-adapter.js

@@ -1,30 +1,42 @@
+import { fragnoDatabaseAdapterNameFakeSymbol, fragnoDatabaseAdapterVersionFakeSymbol } from "../adapters.js";
 import "../../schema/create.js";
 import { createSettingsManager, settingsSchema } from "../../shared/settings-schema.js";
-import { sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/sql.js";
+import { sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/sql.js";
 import { createTableNameMapper } from "./shared.js";
 import { generateSchema } from "./generate.js";
 import { fromDrizzle } from "./drizzle-query.js";
+import { createDrizzleConnectionPool } from "./drizzle-connection-pool.js";
 
 //#region src/adapters/drizzle/drizzle-adapter.ts
 var DrizzleAdapter = class {
-	#drizzleConfig;
+	#connectionPool;
+	#provider;
 	constructor(config) {
-		this.#drizzleConfig = config;
+		this.#connectionPool = createDrizzleConnectionPool(config.db);
+		this.#provider = config.provider;
 	}
-	get provider() {
-		return this.#drizzleConfig.provider;
+	get [fragnoDatabaseAdapterNameFakeSymbol]() {
+		return "drizzle";
+	}
+	get [fragnoDatabaseAdapterVersionFakeSymbol]() {
+		return 0;
 	}
-	#getDb() {
-		const db = this.#drizzleConfig.db;
-		return typeof db === "function" ? db() : db;
+	async close() {
+		await this.#connectionPool.close();
+	}
+	get provider() {
+		return this.#provider;
 	}
 	async isConnectionHealthy() {
+		const conn = await this.#connectionPool.connect();
 		try {
-			const result = await this.#getDb().execute(sql`SELECT 1 as healthy`);
+			const result = await conn.db.execute(sql`SELECT 1 as healthy`);
 			if (Array.isArray(result)) return result.length > 0 && result[0]["healthy"] === 1;
 			else return result.rows[0]["healthy"] === 1;
 		} catch {
 			return false;
+		} finally {
+			await conn.release();
 		}
 	}
 	async getSchemaVersion(namespace) {
@@ -32,16 +44,13 @@ var DrizzleAdapter = class {
 	}
 	createQueryEngine(schema, namespace) {
 		const mapper = namespace ? createTableNameMapper(namespace) : void 0;
-		return fromDrizzle(schema, {
-			db: this.#getDb(),
-			provider: this.#drizzleConfig.provider
-		}, mapper);
+		return fromDrizzle(schema, this.#connectionPool, this.#provider, mapper);
 	}
 	createSchemaGenerator(fragments, options) {
 		return { generateSchema: (genOptions) => {
 			const path = genOptions?.path ?? options?.path ?? "fragno-schema.ts";
 			return {
-				schema: generateSchema(fragments, this.#drizzleConfig.provider),
+				schema: generateSchema(fragments, this.#provider),
 				path
 			};
 		} };

package/dist/adapters/drizzle/drizzle-adapter.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-adapter.js","names":["#drizzleConfig","#getDb"],"sources":["../../../src/adapters/drizzle/drizzle-adapter.ts"],"sourcesContent":["import type { DatabaseAdapter } from \"../adapters\";\nimport { type AnySchema } from \"../../schema/create\";\nimport type { AbstractQuery } from \"../../query/query\";\nimport type { SchemaGenerator } from \"../../schema-generator/schema-generator\";\nimport { generateSchema } from \"./generate\";\nimport { fromDrizzle, type DrizzleUOWConfig } from \"./drizzle-query\";\nimport { createTableNameMapper, type DBType, type DrizzleResult } from \"./shared\";\nimport { createSettingsManager, settingsSchema } from \"../../shared/settings-schema\";\nimport { sql } from \"drizzle-orm\";\n\nexport interface DrizzleConfig {\n  db: unknown | (() => unknown);\n  provider: \"sqlite\" | \"mysql\" | \"postgresql\";\n}\n\nexport class DrizzleAdapter implements DatabaseAdapter<DrizzleUOWConfig> {\n  #drizzleConfig: DrizzleConfig;\n\n  constructor(config: DrizzleConfig) {\n    this.#drizzleConfig = config;\n  }\n\n  get provider(): \"sqlite\" | \"mysql\" | \"postgresql\" {\n    return this.#drizzleConfig.provider;\n  }\n\n  #getDb(): DBType {\n    const db = this.#drizzleConfig.db;\n    return (typeof db === \"function\" ? db() : db) as DBType;\n  }\n\n  async isConnectionHealthy(): Promise<boolean> {\n    try {\n      const result = await this.#getDb().execute(sql`SELECT 1 as healthy`);\n\n      // Handle different result formats across providers\n      // PostgreSQL/MySQL: { rows: [...] }\n      // SQLite: array directly or { rows: [...] }\n      if (Array.isArray(result)) {\n        return result.length > 0 && result[0][\"healthy\"] === 1;\n      } else {\n        const drizzleResult = result as DrizzleResult;\n        return drizzleResult.rows[0][\"healthy\"] === 1;\n      }\n    } catch {\n      return false;\n    }\n  }\n\n  async getSchemaVersion(namespace: string): Promise<string | undefined> {\n    const queryEngine = this.createQueryEngine(settingsSchema, namespace);\n    const manager = createSettingsManager(queryEngine, namespace);\n\n    // Try to read the version key directly\n    const result = await manager.get(\"version\");\n    return result?.value;\n  }\n\n  createQueryEngine<TSchema extends AnySchema>(\n    schema: TSchema,\n    namespace: string,\n  ): AbstractQuery<TSchema, DrizzleUOWConfig> {\n    // Only create mapper if namespace is non-empty\n    const mapper = namespace ? createTableNameMapper(namespace) : undefined;\n    // Resolve the db instance if it's a function\n    const resolvedConfig: DrizzleConfig = {\n      db: this.#getDb(),\n      provider: this.#drizzleConfig.provider,\n    };\n    return fromDrizzle(schema, resolvedConfig, mapper);\n  }\n\n  createSchemaGenerator(\n    fragments: { schema: AnySchema; namespace: string }[],\n    options?: { path?: string },\n  ): SchemaGenerator {\n    return {\n      generateSchema: (genOptions) => {\n        const path = genOptions?.path ?? options?.path ?? \"fragno-schema.ts\";\n\n        return {\n          schema: generateSchema(fragments, this.#drizzleConfig.provider),\n          path,\n        };\n      },\n    };\n  }\n}\n"],"mappings":";;;;;;;;AAeA,IAAa,iBAAb,MAAyE;CACvE;CAEA,YAAY,QAAuB;AACjC,QAAKA,gBAAiB;;CAGxB,IAAI,WAA8C;AAChD,SAAO,MAAKA,cAAe;;CAG7B,SAAiB;EACf,MAAM,KAAK,MAAKA,cAAe;AAC/B,SAAQ,OAAO,OAAO,aAAa,IAAI,GAAG;;CAG5C,MAAM,sBAAwC;AAC5C,MAAI;GACF,MAAM,SAAS,MAAM,MAAKC,OAAQ,CAAC,QAAQ,GAAG,sBAAsB;AAKpE,OAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OAAO,SAAS,KAAK,OAAO,GAAG,eAAe;OAGrD,QADsB,OACD,KAAK,GAAG,eAAe;UAExC;AACN,UAAO;;;CAIX,MAAM,iBAAiB,WAAgD;AAMrE,UADe,MAHC,sBADI,KAAK,kBAAkB,gBAAgB,UAAU,EAClB,UAAU,CAGhC,IAAI,UAAU,GAC5B;;CAGjB,kBACE,QACA,WAC0C;EAE1C,MAAM,SAAS,YAAY,sBAAsB,UAAU,GAAG;AAM9D,SAAO,YAAY,QAJmB;GACpC,IAAI,MAAKA,OAAQ;GACjB,UAAU,MAAKD,cAAe;GAC/B,EAC0C,OAAO;;CAGpD,sBACE,WACA,SACiB;AACjB,SAAO,EACL,iBAAiB,eAAe;GAC9B,MAAM,OAAO,YAAY,QAAQ,SAAS,QAAQ;AAElD,UAAO;IACL,QAAQ,eAAe,WAAW,MAAKA,cAAe,SAAS;IAC/D;IACD;KAEJ"}
+{"version":3,"file":"drizzle-adapter.js","names":["#connectionPool","#provider"],"sources":["../../../src/adapters/drizzle/drizzle-adapter.ts"],"sourcesContent":["import type { DatabaseAdapter } from \"../adapters\";\nimport { type AnySchema } from \"../../schema/create\";\nimport type { AbstractQuery } from \"../../query/query\";\nimport type { SchemaGenerator } from \"../../schema-generator/schema-generator\";\nimport { generateSchema } from \"./generate\";\nimport { fromDrizzle, type DrizzleUOWConfig } from \"./drizzle-query\";\nimport { createTableNameMapper, type DBType, type DrizzleResult } from \"./shared\";\nimport { createSettingsManager, settingsSchema } from \"../../shared/settings-schema\";\nimport { sql } from \"drizzle-orm\";\nimport {\n  fragnoDatabaseAdapterNameFakeSymbol,\n  fragnoDatabaseAdapterVersionFakeSymbol,\n} from \"../adapters\";\nimport type { ConnectionPool } from \"../../shared/connection-pool\";\nimport { createDrizzleConnectionPool } from \"./drizzle-connection-pool\";\n\nexport interface DrizzleConfig {\n  db: unknown | (() => unknown | Promise<unknown>);\n  provider: \"sqlite\" | \"mysql\" | \"postgresql\";\n}\n\nexport class DrizzleAdapter implements DatabaseAdapter<DrizzleUOWConfig> {\n  #connectionPool: ConnectionPool<DBType>;\n  #provider: \"sqlite\" | \"mysql\" | \"postgresql\";\n\n  constructor(config: DrizzleConfig) {\n    this.#connectionPool = createDrizzleConnectionPool(\n      config.db as DBType | (() => DBType | Promise<DBType>),\n    );\n    this.#provider = config.provider;\n  }\n\n  get [fragnoDatabaseAdapterNameFakeSymbol](): string {\n    return \"drizzle\";\n  }\n\n  get [fragnoDatabaseAdapterVersionFakeSymbol](): number {\n    return 0;\n  }\n\n  async close(): Promise<void> {\n    await this.#connectionPool.close();\n  }\n\n  get provider(): \"sqlite\" | \"mysql\" | \"postgresql\" {\n    return this.#provider;\n  }\n\n  async isConnectionHealthy(): Promise<boolean> {\n    const conn = await this.#connectionPool.connect();\n    try {\n      const result = await conn.db.execute(sql`SELECT 1 as healthy`);\n\n      // Handle different result formats across providers\n      // PostgreSQL/MySQL: { rows: [...] }\n      // SQLite: array directly or { rows: [...] }\n      if (Array.isArray(result)) {\n        return result.length > 0 && result[0][\"healthy\"] === 1;\n      } else {\n        const drizzleResult = result as DrizzleResult;\n        return drizzleResult.rows[0][\"healthy\"] === 1;\n      }\n    } catch {\n      return false;\n    } finally {\n      await conn.release();\n    }\n  }\n\n  async getSchemaVersion(namespace: string): Promise<string | undefined> {\n    const queryEngine = this.createQueryEngine(settingsSchema, namespace);\n    const manager = createSettingsManager(queryEngine, namespace);\n\n    // Try to read the version key directly\n    const result = await manager.get(\"version\");\n    return result?.value;\n  }\n\n  createQueryEngine<TSchema extends AnySchema>(\n    schema: TSchema,\n    namespace: string,\n  ): AbstractQuery<TSchema, DrizzleUOWConfig> {\n    // Only create mapper if namespace is non-empty\n    const mapper = namespace ? createTableNameMapper(namespace) : undefined;\n    return fromDrizzle(schema, this.#connectionPool, this.#provider, mapper);\n  }\n\n  createSchemaGenerator(\n    fragments: { schema: AnySchema; namespace: string }[],\n    options?: { path?: string },\n  ): SchemaGenerator {\n    return {\n      generateSchema: (genOptions) => {\n        const path = genOptions?.path ?? options?.path ?? \"fragno-schema.ts\";\n\n        return {\n          schema: generateSchema(fragments, this.#provider),\n          path,\n        };\n      },\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;AAqBA,IAAa,iBAAb,MAAyE;CACvE;CACA;CAEA,YAAY,QAAuB;AACjC,QAAKA,iBAAkB,4BACrB,OAAO,GACR;AACD,QAAKC,WAAY,OAAO;;CAG1B,KAAK,uCAA+C;AAClD,SAAO;;CAGT,KAAK,0CAAkD;AACrD,SAAO;;CAGT,MAAM,QAAuB;AAC3B,QAAM,MAAKD,eAAgB,OAAO;;CAGpC,IAAI,WAA8C;AAChD,SAAO,MAAKC;;CAGd,MAAM,sBAAwC;EAC5C,MAAM,OAAO,MAAM,MAAKD,eAAgB,SAAS;AACjD,MAAI;GACF,MAAM,SAAS,MAAM,KAAK,GAAG,QAAQ,GAAG,sBAAsB;AAK9D,OAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OAAO,SAAS,KAAK,OAAO,GAAG,eAAe;OAGrD,QADsB,OACD,KAAK,GAAG,eAAe;UAExC;AACN,UAAO;YACC;AACR,SAAM,KAAK,SAAS;;;CAIxB,MAAM,iBAAiB,WAAgD;AAMrE,UADe,MAHC,sBADI,KAAK,kBAAkB,gBAAgB,UAAU,EAClB,UAAU,CAGhC,IAAI,UAAU,GAC5B;;CAGjB,kBACE,QACA,WAC0C;EAE1C,MAAM,SAAS,YAAY,sBAAsB,UAAU,GAAG;AAC9D,SAAO,YAAY,QAAQ,MAAKA,gBAAiB,MAAKC,UAAW,OAAO;;CAG1E,sBACE,WACA,SACiB;AACjB,SAAO,EACL,iBAAiB,eAAe;GAC9B,MAAM,OAAO,YAAY,QAAQ,SAAS,QAAQ;AAElD,UAAO;IACL,QAAQ,eAAe,WAAW,MAAKA,SAAU;IACjD;IACD;KAEJ"}

package/dist/adapters/drizzle/drizzle-connection-pool.js

@@ -0,0 +1,40 @@
+//#region src/adapters/drizzle/drizzle-connection-pool.ts
+/**
+* Creates a Drizzle-specific connection pool.
+* Drizzle doesn't have a standard destroy() method, as cleanup depends on the underlying driver.
+*/
+function createDrizzleConnectionPool(dbOrFactory) {
+	let cachedDb;
+	let initPromise;
+	const ensureInitialized = async () => {
+		if (cachedDb) return cachedDb;
+		if (!initPromise) initPromise = (async () => {
+			const db = typeof dbOrFactory === "function" ? await dbOrFactory() : dbOrFactory;
+			cachedDb = db;
+			return db;
+		})();
+		return initPromise;
+	};
+	if (typeof dbOrFactory === "function") ensureInitialized();
+	else cachedDb = dbOrFactory;
+	return {
+		async connect() {
+			return {
+				db: await ensureInitialized(),
+				release: () => Promise.resolve()
+			};
+		},
+		getDatabaseSync() {
+			if (!cachedDb) throw new Error("Cannot get database synchronously: database not initialized.");
+			return cachedDb;
+		},
+		async close() {
+			cachedDb = void 0;
+			initPromise = void 0;
+		}
+	};
+}
+
+//#endregion
+export { createDrizzleConnectionPool };
+//# sourceMappingURL=drizzle-connection-pool.js.map

package/dist/adapters/drizzle/drizzle-connection-pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"drizzle-connection-pool.js","names":["cachedDb: DBType | undefined","initPromise: Promise<DBType> | undefined"],"sources":["../../../src/adapters/drizzle/drizzle-connection-pool.ts"],"sourcesContent":["import type { ConnectionPool } from \"../../shared/connection-pool\";\nimport type { DBType } from \"./shared\";\n\n/**\n * Creates a Drizzle-specific connection pool.\n * Drizzle doesn't have a standard destroy() method, as cleanup depends on the underlying driver.\n */\nexport function createDrizzleConnectionPool(\n  dbOrFactory: DBType | (() => DBType | Promise<DBType>),\n): ConnectionPool<DBType> {\n  let cachedDb: DBType | undefined;\n  let initPromise: Promise<DBType> | undefined;\n\n  const ensureInitialized = async (): Promise<DBType> => {\n    if (cachedDb) {\n      return cachedDb;\n    }\n\n    if (!initPromise) {\n      initPromise = (async () => {\n        const db =\n          typeof dbOrFactory === \"function\"\n            ? await (dbOrFactory as () => DBType | Promise<DBType>)()\n            : dbOrFactory;\n        cachedDb = db;\n        return db;\n      })();\n    }\n\n    return initPromise;\n  };\n\n  // Eagerly start initialization if it's a factory function\n  if (typeof dbOrFactory === \"function\") {\n    void ensureInitialized();\n  } else {\n    // Direct instance - cache it immediately\n    cachedDb = dbOrFactory;\n  }\n\n  return {\n    async connect() {\n      const db = await ensureInitialized();\n\n      return {\n        db,\n        release: () => Promise.resolve(),\n      };\n    },\n\n    getDatabaseSync() {\n      if (!cachedDb) {\n        throw new Error(\"Cannot get database synchronously: database not initialized.\");\n      }\n      return cachedDb;\n    },\n\n    async close() {\n      // Drizzle doesn't have a standard destroy() method\n      // Cleanup depends on the underlying driver (e.g., PGlite, node-postgres, mysql2)\n      // For now, we just clear the cache\n      cachedDb = undefined;\n      initPromise = undefined;\n    },\n  };\n}\n"],"mappings":";;;;;AAOA,SAAgB,4BACd,aACwB;CACxB,IAAIA;CACJ,IAAIC;CAEJ,MAAM,oBAAoB,YAA6B;AACrD,MAAI,SACF,QAAO;AAGT,MAAI,CAAC,YACH,gBAAe,YAAY;GACzB,MAAM,KACJ,OAAO,gBAAgB,aACnB,MAAO,aAAgD,GACvD;AACN,cAAW;AACX,UAAO;MACL;AAGN,SAAO;;AAIT,KAAI,OAAO,gBAAgB,WACzB,CAAK,mBAAmB;KAGxB,YAAW;AAGb,QAAO;EACL,MAAM,UAAU;AAGd,UAAO;IACL,IAHS,MAAM,mBAAmB;IAIlC,eAAe,QAAQ,SAAS;IACjC;;EAGH,kBAAkB;AAChB,OAAI,CAAC,SACH,OAAM,IAAI,MAAM,+DAA+D;AAEjF,UAAO;;EAGT,MAAM,QAAQ;AAIZ,cAAW;AACX,iBAAc;;EAEjB"}

package/dist/adapters/drizzle/drizzle-query.js

@@ -13,27 +13,40 @@ import { createDrizzleUOWDecoder } from "./drizzle-uow-decoder.js";
 * enabling features like SQL snapshot testing.
 *
 * @param schema - The database schema definition
-* @param config - Drizzle configuration containing the database instance and provider
+* @param pool - Connection pool for acquiring database connections
+* @param provider - SQL provider (sqlite, mysql, postgresql)
 * @param mapper - Optional table name mapper for namespace prefixing
 * @returns An AbstractQuery instance for performing database operations
 *
 * @example
 * ```ts
-* const queryEngine = fromDrizzle(mySchema, {
-*   db: drizzle,
-*   provider: 'postgresql'
-* });
+* const pool = createSimpleConnectionPool(drizzle);
+* const queryEngine = fromDrizzle(mySchema, pool, 'postgresql');
 *
 * const uow = queryEngine.createUnitOfWork('myOperation');
 * ```
 */
-function fromDrizzle(schema, config, mapper) {
-	const [db] = parseDrizzle(config.db);
-	const { provider } = config;
+function fromDrizzle(schema, pool, provider, mapper) {
 	function createUOW(name, uowConfig) {
-		return new UnitOfWork(schema, createDrizzleUOWCompiler(schema, config, mapper, uowConfig?.onQuery), {
-			executeRetrievalPhase: (retrievalBatch) => executeDrizzleRetrievalPhase(db, retrievalBatch, provider),
-			executeMutationPhase: (mutationBatch) => executeDrizzleMutationPhase(db, mutationBatch, provider)
+		return new UnitOfWork(schema, createDrizzleUOWCompiler(schema, pool, provider, mapper, uowConfig?.onQuery), {
+			async executeRetrievalPhase(retrievalBatch) {
+				const conn = await pool.connect();
+				try {
+					const db = parseDrizzle(conn.db)[0];
+					return await executeDrizzleRetrievalPhase(db, retrievalBatch, provider);
+				} finally {
+					await conn.release();
+				}
+			},
+			async executeMutationPhase(mutationBatch) {
+				const conn = await pool.connect();
+				try {
+					const db = parseDrizzle(conn.db)[0];
+					return await executeDrizzleMutationPhase(db, mutationBatch, provider);
+				} finally {
+					await conn.release();
+				}
+			}
 		}, createDrizzleUOWDecoder(schema, provider), name);
 	}
 	return {

package/dist/adapters/drizzle/drizzle-query.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-query.js","names":["whereConfig: { indexName?: string; condition?: unknown }","setValues: unknown"],"sources":["../../../src/adapters/drizzle/drizzle-query.ts"],"sourcesContent":["import type { AbstractQuery } from \"../../query/query\";\nimport type { AnySchema } from \"../../schema/create\";\nimport type { DrizzleConfig } from \"./drizzle-adapter\";\nimport type { CompiledMutation, UOWExecutor } from \"../../query/unit-of-work\";\nimport { createDrizzleUOWCompiler, type DrizzleCompiledQuery } from \"./drizzle-uow-compiler\";\nimport { executeDrizzleRetrievalPhase, executeDrizzleMutationPhase } from \"./drizzle-uow-executor\";\nimport { UnitOfWork } from \"../../query/unit-of-work\";\nimport { parseDrizzle, type DrizzleResult, type TableNameMapper } from \"./shared\";\nimport { createDrizzleUOWDecoder } from \"./drizzle-uow-decoder\";\n\n/**\n * Configuration options for creating a Drizzle Unit of Work\n */\nexport interface DrizzleUOWConfig {\n  /**\n   * Optional callback to receive compiled SQL queries for logging/debugging\n   * This callback is invoked for each query as it's compiled\n   */\n  onQuery?: (query: DrizzleCompiledQuery) => void;\n}\n\n/**\n * Creates a Drizzle-based query engine for the given schema.\n *\n * This is the main entry point for creating a database query interface using Drizzle.\n * It uses a compiler-based architecture where queries are compiled to SQL and then executed,\n * enabling features like SQL snapshot testing.\n *\n * @param schema - The database schema definition\n * @param config - Drizzle configuration containing the database instance and provider\n * @param mapper - Optional table name mapper for namespace prefixing\n * @returns An AbstractQuery instance for performing database operations\n *\n * @example\n * ```ts\n * const queryEngine = fromDrizzle(mySchema, {\n *   db: drizzle,\n *   provider: 'postgresql'\n * });\n *\n * const uow = queryEngine.createUnitOfWork('myOperation');\n * ```\n */\nexport function fromDrizzle<T extends AnySchema>(\n  schema: T,\n  config: DrizzleConfig,\n  mapper?: TableNameMapper,\n): AbstractQuery<T, DrizzleUOWConfig> {\n  const [db] = parseDrizzle(config.db);\n  const { provider } = config;\n\n  function createUOW(name?: string, uowConfig?: DrizzleUOWConfig) {\n    const uowCompiler = createDrizzleUOWCompiler(schema, config, mapper, uowConfig?.onQuery);\n\n    const executor: UOWExecutor<DrizzleCompiledQuery, DrizzleResult> = {\n      executeRetrievalPhase: (retrievalBatch: DrizzleCompiledQuery[]) =>\n        executeDrizzleRetrievalPhase(db, retrievalBatch, provider),\n      executeMutationPhase: (mutationBatch: CompiledMutation<DrizzleCompiledQuery>[]) =>\n        executeDrizzleMutationPhase(db, mutationBatch, provider),\n    };\n\n    const decoder = createDrizzleUOWDecoder(schema, provider);\n\n    return new UnitOfWork(schema, uowCompiler, executor, decoder, name);\n  }\n\n  return {\n    find(tableName, builderFn) {\n      const uow = createUOW();\n      uow.find(tableName, builderFn);\n      return uow.executeRetrieve();\n    },\n\n    async findFirst(tableName, builderFn) {\n      const uow = createUOW();\n      if (builderFn) {\n        uow.find(tableName, (b) => builderFn(b as never).pageSize(1));\n      } else {\n        uow.find(tableName, (b) => b.whereIndex(\"primary\").pageSize(1));\n      }\n      // executeRetrieve runs an array of `find` operation results, which each return an array of rows\n      const [result]: unknown[][] = await uow.executeRetrieve();\n      return result?.[0] ?? null;\n    },\n\n    async create(tableName, values) {\n      const uow = createUOW();\n      uow.create(tableName as string, values as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to create record\");\n      }\n\n      const createdIds = uow.getCreatedIds();\n      const createdId = createdIds[0];\n      if (!createdId) {\n        throw new Error(\"Failed to get created ID\");\n      }\n      return createdId;\n    },\n\n    async createMany(tableName, valuesArray) {\n      const uow = createUOW();\n      for (const values of valuesArray) {\n        uow.create(tableName as string, values as never);\n      }\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to create records\");\n      }\n\n      return uow.getCreatedIds();\n    },\n\n    async update(tableName, id, builderFn) {\n      const uow = createUOW();\n      uow.update(tableName as string, id, builderFn as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to update record (version conflict or record not found)\");\n      }\n    },\n\n    async updateMany(tableName, builderFn) {\n      // FIXME: This is not correct\n\n      let whereConfig: { indexName?: string; condition?: unknown } = {};\n      let setValues: unknown;\n\n      const specialBuilder = {\n        whereIndex(indexName: string, condition?: unknown) {\n          whereConfig = { indexName, condition };\n          return this;\n        },\n        set(values: unknown) {\n          setValues = values;\n          return this;\n        },\n      };\n\n      builderFn(specialBuilder);\n\n      if (!whereConfig.indexName) {\n        throw new Error(\"whereIndex() must be called in updateMany\");\n      }\n      if (!setValues) {\n        throw new Error(\"set() must be called in updateMany\");\n      }\n\n      const findUow = createUOW();\n      findUow.find(tableName, (b) => {\n        if (whereConfig.condition) {\n          return b.whereIndex(whereConfig.indexName as never, whereConfig.condition as never);\n        }\n        return b.whereIndex(whereConfig.indexName as never);\n      });\n      const findResults = await findUow.executeRetrieve();\n      const records = (findResults as unknown as [unknown])[0];\n\n      // @ts-expect-error - Type narrowing doesn't work through unknown cast\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const updateUow = createUOW();\n      for (const record of records as never as Array<{ id: unknown }>) {\n        updateUow.update(tableName as string, record.id as string, (b) =>\n          b.set(setValues as never),\n        );\n      }\n      const { success } = await updateUow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to update records (version conflict)\");\n      }\n    },\n\n    async delete(tableName, id, builderFn) {\n      const uow = createUOW();\n      uow.delete(tableName as string, id, builderFn as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to delete record (version conflict or record not found)\");\n      }\n    },\n\n    async deleteMany(tableName, builderFn) {\n      let whereConfig: { indexName?: string; condition?: unknown } = {};\n\n      const specialBuilder = {\n        whereIndex(indexName: string, condition?: unknown) {\n          whereConfig = { indexName, condition };\n          return this;\n        },\n      };\n\n      builderFn(specialBuilder as never);\n\n      if (!whereConfig.indexName) {\n        throw new Error(\"whereIndex() must be called in deleteMany\");\n      }\n\n      const findUow = createUOW();\n      findUow.find(tableName as string, (b) => {\n        if (whereConfig.condition) {\n          return b.whereIndex(whereConfig.indexName as never, whereConfig.condition as never);\n        }\n        return b.whereIndex(whereConfig.indexName as never);\n      });\n      const findResults2 = await findUow.executeRetrieve();\n      const records = (findResults2 as unknown as [unknown])[0];\n\n      // @ts-expect-error - Type narrowing doesn't work through unknown cast\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const deleteUow = createUOW();\n      for (const record of records as never as Array<{ id: unknown }>) {\n        deleteUow.delete(tableName as string, record.id as string);\n      }\n      const { success } = await deleteUow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to delete records (version conflict)\");\n      }\n    },\n\n    createUnitOfWork(name, uowConfig) {\n      return createUOW(name, uowConfig);\n    },\n  } as AbstractQuery<T, DrizzleUOWConfig>;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CA,SAAgB,YACd,QACA,QACA,QACoC;CACpC,MAAM,CAAC,MAAM,aAAa,OAAO,GAAG;CACpC,MAAM,EAAE,aAAa;CAErB,SAAS,UAAU,MAAe,WAA8B;AAY9D,SAAO,IAAI,WAAW,QAXF,yBAAyB,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,EAErB;GACjE,wBAAwB,mBACtB,6BAA6B,IAAI,gBAAgB,SAAS;GAC5D,uBAAuB,kBACrB,4BAA4B,IAAI,eAAe,SAAS;GAC3D,EAEe,wBAAwB,QAAQ,SAAS,EAEK,KAAK;;AAGrE,QAAO;EACL,KAAK,WAAW,WAAW;GACzB,MAAM,MAAM,WAAW;AACvB,OAAI,KAAK,WAAW,UAAU;AAC9B,UAAO,IAAI,iBAAiB;;EAG9B,MAAM,UAAU,WAAW,WAAW;GACpC,MAAM,MAAM,WAAW;AACvB,OAAI,UACF,KAAI,KAAK,YAAY,MAAM,UAAU,EAAW,CAAC,SAAS,EAAE,CAAC;OAE7D,KAAI,KAAK,YAAY,MAAM,EAAE,WAAW,UAAU,CAAC,SAAS,EAAE,CAAC;GAGjE,MAAM,CAAC,UAAuB,MAAM,IAAI,iBAAiB;AACzD,UAAO,SAAS,MAAM;;EAGxB,MAAM,OAAO,WAAW,QAAQ;GAC9B,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,OAAgB;GAChD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,0BAA0B;GAI5C,MAAM,YADa,IAAI,eAAe,CACT;AAC7B,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,2BAA2B;AAE7C,UAAO;;EAGT,MAAM,WAAW,WAAW,aAAa;GACvC,MAAM,MAAM,WAAW;AACvB,QAAK,MAAM,UAAU,YACnB,KAAI,OAAO,WAAqB,OAAgB;GAElD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,UAAO,IAAI,eAAe;;EAG5B,MAAM,OAAO,WAAW,IAAI,WAAW;GACrC,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,IAAI,UAAmB;GACvD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;GAGrC,IAAIA,cAA2D,EAAE;GACjE,IAAIC;AAaJ,aAXuB;IACrB,WAAW,WAAmB,WAAqB;AACjD,mBAAc;MAAE;MAAW;MAAW;AACtC,YAAO;;IAET,IAAI,QAAiB;AACnB,iBAAY;AACZ,YAAO;;IAEV,CAEwB;AAEzB,OAAI,CAAC,YAAY,UACf,OAAM,IAAI,MAAM,4CAA4C;AAE9D,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,qCAAqC;GAGvD,MAAM,UAAU,WAAW;AAC3B,WAAQ,KAAK,YAAY,MAAM;AAC7B,QAAI,YAAY,UACd,QAAO,EAAE,WAAW,YAAY,WAAoB,YAAY,UAAmB;AAErF,WAAO,EAAE,WAAW,YAAY,UAAmB;KACnD;GAEF,MAAM,WADc,MAAM,QAAQ,iBAAiB,EACG;AAGtD,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,WAAW;AAC7B,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAqB,OAAO,KAAe,MAC1D,EAAE,IAAI,UAAmB,CAC1B;GAEH,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,MAAM,OAAO,WAAW,IAAI,WAAW;GACrC,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,IAAI,UAAmB;GACvD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;GACrC,IAAID,cAA2D,EAAE;AASjE,aAPuB,EACrB,WAAW,WAAmB,WAAqB;AACjD,kBAAc;KAAE;KAAW;KAAW;AACtC,WAAO;MAEV,CAEiC;AAElC,OAAI,CAAC,YAAY,UACf,OAAM,IAAI,MAAM,4CAA4C;GAG9D,MAAM,UAAU,WAAW;AAC3B,WAAQ,KAAK,YAAsB,MAAM;AACvC,QAAI,YAAY,UACd,QAAO,EAAE,WAAW,YAAY,WAAoB,YAAY,UAAmB;AAErF,WAAO,EAAE,WAAW,YAAY,UAAmB;KACnD;GAEF,MAAM,WADe,MAAM,QAAQ,iBAAiB,EACG;AAGvD,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,WAAW;AAC7B,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAqB,OAAO,GAAa;GAE5D,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,iBAAiB,MAAM,WAAW;AAChC,UAAO,UAAU,MAAM,UAAU;;EAEpC"}
+{"version":3,"file":"drizzle-query.js","names":["whereConfig: { indexName?: string; condition?: unknown }","setValues: unknown"],"sources":["../../../src/adapters/drizzle/drizzle-query.ts"],"sourcesContent":["import type { AbstractQuery } from \"../../query/query\";\nimport type { AnySchema } from \"../../schema/create\";\nimport type { CompiledMutation, UOWExecutor } from \"../../query/unit-of-work\";\nimport { createDrizzleUOWCompiler, type DrizzleCompiledQuery } from \"./drizzle-uow-compiler\";\nimport { executeDrizzleRetrievalPhase, executeDrizzleMutationPhase } from \"./drizzle-uow-executor\";\nimport { UnitOfWork } from \"../../query/unit-of-work\";\nimport { parseDrizzle, type DrizzleResult, type TableNameMapper, type DBType } from \"./shared\";\nimport { createDrizzleUOWDecoder } from \"./drizzle-uow-decoder\";\nimport type { ConnectionPool } from \"../../shared/connection-pool\";\n\n/**\n * Configuration options for creating a Drizzle Unit of Work\n */\nexport interface DrizzleUOWConfig {\n  /**\n   * Optional callback to receive compiled SQL queries for logging/debugging\n   * This callback is invoked for each query as it's compiled\n   */\n  onQuery?: (query: DrizzleCompiledQuery) => void;\n}\n\n/**\n * Creates a Drizzle-based query engine for the given schema.\n *\n * This is the main entry point for creating a database query interface using Drizzle.\n * It uses a compiler-based architecture where queries are compiled to SQL and then executed,\n * enabling features like SQL snapshot testing.\n *\n * @param schema - The database schema definition\n * @param pool - Connection pool for acquiring database connections\n * @param provider - SQL provider (sqlite, mysql, postgresql)\n * @param mapper - Optional table name mapper for namespace prefixing\n * @returns An AbstractQuery instance for performing database operations\n *\n * @example\n * ```ts\n * const pool = createSimpleConnectionPool(drizzle);\n * const queryEngine = fromDrizzle(mySchema, pool, 'postgresql');\n *\n * const uow = queryEngine.createUnitOfWork('myOperation');\n * ```\n */\nexport function fromDrizzle<T extends AnySchema>(\n  schema: T,\n  pool: ConnectionPool<DBType>,\n  provider: \"sqlite\" | \"mysql\" | \"postgresql\",\n  mapper?: TableNameMapper,\n): AbstractQuery<T, DrizzleUOWConfig> {\n  function createUOW(name?: string, uowConfig?: DrizzleUOWConfig) {\n    const uowCompiler = createDrizzleUOWCompiler(\n      schema,\n      pool,\n      provider,\n      mapper,\n      uowConfig?.onQuery,\n    );\n\n    const executor: UOWExecutor<DrizzleCompiledQuery, DrizzleResult> = {\n      async executeRetrievalPhase(retrievalBatch: DrizzleCompiledQuery[]) {\n        const conn = await pool.connect();\n        try {\n          const db = parseDrizzle(conn.db)[0];\n          return await executeDrizzleRetrievalPhase(db, retrievalBatch, provider);\n        } finally {\n          await conn.release();\n        }\n      },\n      async executeMutationPhase(mutationBatch: CompiledMutation<DrizzleCompiledQuery>[]) {\n        const conn = await pool.connect();\n        try {\n          const db = parseDrizzle(conn.db)[0];\n          return await executeDrizzleMutationPhase(db, mutationBatch, provider);\n        } finally {\n          await conn.release();\n        }\n      },\n    };\n\n    const decoder = createDrizzleUOWDecoder(schema, provider);\n\n    return new UnitOfWork(schema, uowCompiler, executor, decoder, name);\n  }\n\n  return {\n    find(tableName, builderFn) {\n      const uow = createUOW();\n      uow.find(tableName, builderFn);\n      return uow.executeRetrieve();\n    },\n\n    async findFirst(tableName, builderFn) {\n      const uow = createUOW();\n      if (builderFn) {\n        uow.find(tableName, (b) => builderFn(b as never).pageSize(1));\n      } else {\n        uow.find(tableName, (b) => b.whereIndex(\"primary\").pageSize(1));\n      }\n      // executeRetrieve runs an array of `find` operation results, which each return an array of rows\n      const [result]: unknown[][] = await uow.executeRetrieve();\n      return result?.[0] ?? null;\n    },\n\n    async create(tableName, values) {\n      const uow = createUOW();\n      uow.create(tableName as string, values as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to create record\");\n      }\n\n      const createdIds = uow.getCreatedIds();\n      const createdId = createdIds[0];\n      if (!createdId) {\n        throw new Error(\"Failed to get created ID\");\n      }\n      return createdId;\n    },\n\n    async createMany(tableName, valuesArray) {\n      const uow = createUOW();\n      for (const values of valuesArray) {\n        uow.create(tableName as string, values as never);\n      }\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to create records\");\n      }\n\n      return uow.getCreatedIds();\n    },\n\n    async update(tableName, id, builderFn) {\n      const uow = createUOW();\n      uow.update(tableName as string, id, builderFn as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to update record (version conflict or record not found)\");\n      }\n    },\n\n    async updateMany(tableName, builderFn) {\n      // FIXME: This is not correct\n\n      let whereConfig: { indexName?: string; condition?: unknown } = {};\n      let setValues: unknown;\n\n      const specialBuilder = {\n        whereIndex(indexName: string, condition?: unknown) {\n          whereConfig = { indexName, condition };\n          return this;\n        },\n        set(values: unknown) {\n          setValues = values;\n          return this;\n        },\n      };\n\n      builderFn(specialBuilder);\n\n      if (!whereConfig.indexName) {\n        throw new Error(\"whereIndex() must be called in updateMany\");\n      }\n      if (!setValues) {\n        throw new Error(\"set() must be called in updateMany\");\n      }\n\n      const findUow = createUOW();\n      findUow.find(tableName, (b) => {\n        if (whereConfig.condition) {\n          return b.whereIndex(whereConfig.indexName as never, whereConfig.condition as never);\n        }\n        return b.whereIndex(whereConfig.indexName as never);\n      });\n      const findResults = await findUow.executeRetrieve();\n      const records = (findResults as unknown as [unknown])[0];\n\n      // @ts-expect-error - Type narrowing doesn't work through unknown cast\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const updateUow = createUOW();\n      for (const record of records as never as Array<{ id: unknown }>) {\n        updateUow.update(tableName as string, record.id as string, (b) =>\n          b.set(setValues as never),\n        );\n      }\n      const { success } = await updateUow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to update records (version conflict)\");\n      }\n    },\n\n    async delete(tableName, id, builderFn) {\n      const uow = createUOW();\n      uow.delete(tableName as string, id, builderFn as never);\n      const { success } = await uow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to delete record (version conflict or record not found)\");\n      }\n    },\n\n    async deleteMany(tableName, builderFn) {\n      let whereConfig: { indexName?: string; condition?: unknown } = {};\n\n      const specialBuilder = {\n        whereIndex(indexName: string, condition?: unknown) {\n          whereConfig = { indexName, condition };\n          return this;\n        },\n      };\n\n      builderFn(specialBuilder as never);\n\n      if (!whereConfig.indexName) {\n        throw new Error(\"whereIndex() must be called in deleteMany\");\n      }\n\n      const findUow = createUOW();\n      findUow.find(tableName as string, (b) => {\n        if (whereConfig.condition) {\n          return b.whereIndex(whereConfig.indexName as never, whereConfig.condition as never);\n        }\n        return b.whereIndex(whereConfig.indexName as never);\n      });\n      const findResults2 = await findUow.executeRetrieve();\n      const records = (findResults2 as unknown as [unknown])[0];\n\n      // @ts-expect-error - Type narrowing doesn't work through unknown cast\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const deleteUow = createUOW();\n      for (const record of records as never as Array<{ id: unknown }>) {\n        deleteUow.delete(tableName as string, record.id as string);\n      }\n      const { success } = await deleteUow.executeMutations();\n      if (!success) {\n        throw new Error(\"Failed to delete records (version conflict)\");\n      }\n    },\n\n    createUnitOfWork(name, uowConfig) {\n      return createUOW(name, uowConfig);\n    },\n  } as AbstractQuery<T, DrizzleUOWConfig>;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AA0CA,SAAgB,YACd,QACA,MACA,UACA,QACoC;CACpC,SAAS,UAAU,MAAe,WAA8B;AAgC9D,SAAO,IAAI,WAAW,QA/BF,yBAClB,QACA,MACA,UACA,QACA,WAAW,QACZ,EAEkE;GACjE,MAAM,sBAAsB,gBAAwC;IAClE,MAAM,OAAO,MAAM,KAAK,SAAS;AACjC,QAAI;KACF,MAAM,KAAK,aAAa,KAAK,GAAG,CAAC;AACjC,YAAO,MAAM,6BAA6B,IAAI,gBAAgB,SAAS;cAC/D;AACR,WAAM,KAAK,SAAS;;;GAGxB,MAAM,qBAAqB,eAAyD;IAClF,MAAM,OAAO,MAAM,KAAK,SAAS;AACjC,QAAI;KACF,MAAM,KAAK,aAAa,KAAK,GAAG,CAAC;AACjC,YAAO,MAAM,4BAA4B,IAAI,eAAe,SAAS;cAC7D;AACR,WAAM,KAAK,SAAS;;;GAGzB,EAEe,wBAAwB,QAAQ,SAAS,EAEK,KAAK;;AAGrE,QAAO;EACL,KAAK,WAAW,WAAW;GACzB,MAAM,MAAM,WAAW;AACvB,OAAI,KAAK,WAAW,UAAU;AAC9B,UAAO,IAAI,iBAAiB;;EAG9B,MAAM,UAAU,WAAW,WAAW;GACpC,MAAM,MAAM,WAAW;AACvB,OAAI,UACF,KAAI,KAAK,YAAY,MAAM,UAAU,EAAW,CAAC,SAAS,EAAE,CAAC;OAE7D,KAAI,KAAK,YAAY,MAAM,EAAE,WAAW,UAAU,CAAC,SAAS,EAAE,CAAC;GAGjE,MAAM,CAAC,UAAuB,MAAM,IAAI,iBAAiB;AACzD,UAAO,SAAS,MAAM;;EAGxB,MAAM,OAAO,WAAW,QAAQ;GAC9B,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,OAAgB;GAChD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,0BAA0B;GAI5C,MAAM,YADa,IAAI,eAAe,CACT;AAC7B,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,2BAA2B;AAE7C,UAAO;;EAGT,MAAM,WAAW,WAAW,aAAa;GACvC,MAAM,MAAM,WAAW;AACvB,QAAK,MAAM,UAAU,YACnB,KAAI,OAAO,WAAqB,OAAgB;GAElD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,2BAA2B;AAG7C,UAAO,IAAI,eAAe;;EAG5B,MAAM,OAAO,WAAW,IAAI,WAAW;GACrC,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,IAAI,UAAmB;GACvD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;GAGrC,IAAIA,cAA2D,EAAE;GACjE,IAAIC;AAaJ,aAXuB;IACrB,WAAW,WAAmB,WAAqB;AACjD,mBAAc;MAAE;MAAW;MAAW;AACtC,YAAO;;IAET,IAAI,QAAiB;AACnB,iBAAY;AACZ,YAAO;;IAEV,CAEwB;AAEzB,OAAI,CAAC,YAAY,UACf,OAAM,IAAI,MAAM,4CAA4C;AAE9D,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,qCAAqC;GAGvD,MAAM,UAAU,WAAW;AAC3B,WAAQ,KAAK,YAAY,MAAM;AAC7B,QAAI,YAAY,UACd,QAAO,EAAE,WAAW,YAAY,WAAoB,YAAY,UAAmB;AAErF,WAAO,EAAE,WAAW,YAAY,UAAmB;KACnD;GAEF,MAAM,WADc,MAAM,QAAQ,iBAAiB,EACG;AAGtD,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,WAAW;AAC7B,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAqB,OAAO,KAAe,MAC1D,EAAE,IAAI,UAAmB,CAC1B;GAEH,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,MAAM,OAAO,WAAW,IAAI,WAAW;GACrC,MAAM,MAAM,WAAW;AACvB,OAAI,OAAO,WAAqB,IAAI,UAAmB;GACvD,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;GACrC,IAAID,cAA2D,EAAE;AASjE,aAPuB,EACrB,WAAW,WAAmB,WAAqB;AACjD,kBAAc;KAAE;KAAW;KAAW;AACtC,WAAO;MAEV,CAEiC;AAElC,OAAI,CAAC,YAAY,UACf,OAAM,IAAI,MAAM,4CAA4C;GAG9D,MAAM,UAAU,WAAW;AAC3B,WAAQ,KAAK,YAAsB,MAAM;AACvC,QAAI,YAAY,UACd,QAAO,EAAE,WAAW,YAAY,WAAoB,YAAY,UAAmB;AAErF,WAAO,EAAE,WAAW,YAAY,UAAmB;KACnD;GAEF,MAAM,WADe,MAAM,QAAQ,iBAAiB,EACG;AAGvD,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,WAAW;AAC7B,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAqB,OAAO,GAAa;GAE5D,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,iBAAiB,MAAM,WAAW;AAChC,UAAO,UAAU,MAAM,UAAU;;EAEpC"}

package/dist/adapters/drizzle/drizzle-uow-compiler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-uow-compiler.d.ts","names":[],"sources":["../../../src/adapters/drizzle/drizzle-uow-compiler.ts"],"sourcesContent":[],"mappings":";;KAkBY,oBAAA"}
+{"version":3,"file":"drizzle-uow-compiler.d.ts","names":[],"sources":["../../../src/adapters/drizzle/drizzle-uow-compiler.ts"],"sourcesContent":[],"mappings":";;KAwBY,oBAAA"}

package/dist/adapters/drizzle/drizzle-uow-compiler.js

@@ -2,10 +2,10 @@ import { Column } from "../../schema/create.js";
 import { serialize } from "../../schema/serialize.js";
 import { decodeCursor, serializeCursorValues } from "../../query/cursor.js";
 import { buildCondition } from "../../query/condition-builder.js";
-import { sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/sql.js";
-import { and, eq, gt, gte, inArray, isNotNull, isNull, like, lt, lte, ne, not, notInArray, notLike, or } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/expressions/conditions.js";
-import { asc, desc } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/expressions/select.js";
-import { count } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/functions/aggregate.js";
+import { sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/sql.js";
+import { and, eq, gt, gte, inArray, isNotNull, isNull, like, lt, lte, ne, not, notInArray, notLike, or } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/expressions/conditions.js";
+import { asc, desc } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/expressions/select.js";
+import { count } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/functions/aggregate.js";
 import { parseDrizzle } from "./shared.js";
 import { ReferenceSubquery, encodeValues } from "../../query/result-transform.js";
 import { getOrderedJoinColumns } from "./join-column-utils.js";
@@ -18,14 +18,14 @@ import { getOrderedJoinColumns } from "./join-column-utils.js";
 * that can be executed as a batch/transaction.
 *
 * @param schema - The database schema
-* @param config - Drizzle configuration
+* @param pool - Connection pool for acquiring database connections
+* @param provider - SQL provider (sqlite, mysql, postgresql)
 * @param mapper - Optional table name mapper for namespace prefixing
 * @param onQuery - Optional callback to receive compiled queries for logging/debugging
 * @returns A UOWCompiler instance for Drizzle
 */
-function createDrizzleUOWCompiler(schema, config, mapper, onQuery) {
-	const [db, drizzleTables] = parseDrizzle(config.db);
-	const { provider } = config;
+function createDrizzleUOWCompiler(schema, pool, provider, mapper, onQuery) {
+	const [db, drizzleTables] = parseDrizzle(pool.getDatabaseSync());
 	/**
 	* Convert a Fragno table to a Drizzle table
 	* @throws Error if table is not found in Drizzle schema

package/dist/adapters/drizzle/drizzle-uow-compiler.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle-uow-compiler.js","names":["Drizzle.eq","Drizzle.ne","Drizzle.gt","Drizzle.gte","Drizzle.lt","Drizzle.lte","Drizzle.inArray","Drizzle.notInArray","Drizzle.isNull","Drizzle.isNotNull","Drizzle.like","Drizzle.notLike","Drizzle.and","Drizzle.not","Drizzle.or","processed: Record<string, unknown>","result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>>","joinColumns: Record<string, boolean>","joinOrderBy: Drizzle.SQL[] | undefined","Drizzle.asc","Drizzle.desc","joinWhere: Drizzle.SQL | undefined","joinConfig: Drizzle.DBQueryConfig<\"many\", boolean>","whereClause: Drizzle.SQL | undefined","Drizzle.count","indexColumns: AnyColumn[]","orderDirection: \"asc\" | \"desc\"","orderBy: Drizzle.SQL[] | undefined","columns: Record<string, boolean>","whereClauses: Drizzle.SQL[]","queryConfig: Drizzle.DBQueryConfig<\"many\", boolean>"],"sources":["../../../src/adapters/drizzle/drizzle-uow-compiler.ts"],"sourcesContent":["import * as Drizzle from \"drizzle-orm\";\nimport type { AnyColumn, AnySchema, AnyTable, FragnoId } from \"../../schema/create\";\nimport { Column } from \"../../schema/create\";\nimport type {\n  CompiledMutation,\n  MutationOperation,\n  RetrievalOperation,\n  UOWCompiler,\n} from \"../../query/unit-of-work\";\nimport { buildCondition, type Condition } from \"../../query/condition-builder\";\nimport type { DrizzleConfig } from \"./drizzle-adapter\";\nimport { type ColumnType, type TableType, type TableNameMapper, parseDrizzle } from \"./shared\";\nimport { encodeValues, ReferenceSubquery } from \"../../query/result-transform\";\nimport { serialize } from \"../../schema/serialize\";\nimport { decodeCursor, serializeCursorValues } from \"../../query/cursor\";\nimport type { CompiledJoin } from \"../../query/orm/orm\";\nimport { getOrderedJoinColumns } from \"./join-column-utils\";\n\nexport type DrizzleCompiledQuery = {\n  sql: string;\n  params: unknown[];\n};\n\n/**\n * Create a Drizzle-specific Unit of Work compiler\n *\n * This compiler translates UOW operations into Drizzle query functions\n * that can be executed as a batch/transaction.\n *\n * @param schema - The database schema\n * @param config - Drizzle configuration\n * @param mapper - Optional table name mapper for namespace prefixing\n * @param onQuery - Optional callback to receive compiled queries for logging/debugging\n * @returns A UOWCompiler instance for Drizzle\n */\nexport function createDrizzleUOWCompiler<TSchema extends AnySchema>(\n  schema: TSchema,\n  config: DrizzleConfig,\n  mapper?: TableNameMapper,\n  onQuery?: (query: DrizzleCompiledQuery) => void,\n): UOWCompiler<TSchema, DrizzleCompiledQuery> {\n  const [db, drizzleTables] = parseDrizzle(config.db);\n  const { provider } = config;\n\n  /**\n   * Convert a Fragno table to a Drizzle table\n   * @throws Error if table is not found in Drizzle schema\n   */\n  function toDrizzleTable(table: AnyTable): TableType {\n    // Map logical table name to physical table name using the mapper\n    const physicalTableName = mapper ? mapper.toPhysical(table.ormName) : table.ormName;\n    const out = drizzleTables[physicalTableName];\n    if (out) {\n      return out;\n    }\n\n    throw new Error(\n      `[Drizzle] Unknown table name ${physicalTableName} (logical: ${table.ormName}), is it included in your Drizzle schema?`,\n    );\n  }\n\n  /**\n   * Convert a Fragno column to a Drizzle column\n   * @throws Error if column is not found in Drizzle table\n   */\n  function toDrizzleColumn(col: AnyColumn): ColumnType {\n    const fragnoTable = schema.tables[col.tableName];\n    if (!fragnoTable) {\n      throw new Error(`[Drizzle] Unknown table ${col.tableName} for column ${col.ormName}.`);\n    }\n\n    const table = toDrizzleTable(fragnoTable);\n    const out = table[col.ormName];\n    if (out) {\n      return out;\n    }\n\n    throw new Error(`[Drizzle] Unknown column name ${col.ormName} in ${fragnoTable.ormName}.`);\n  }\n\n  /**\n   * Build a WHERE clause from a condition using Drizzle's query builder\n   */\n  function buildWhere(condition: Condition): Drizzle.SQL | undefined {\n    if (condition.type === \"compare\") {\n      const left = toDrizzleColumn(condition.a);\n      const op = condition.operator;\n      let right = condition.b;\n      if (right instanceof Column) {\n        right = toDrizzleColumn(right);\n      } else {\n        // Serialize non-Column values (e.g., FragnoId -> string, Date -> number for SQLite)\n        right = serialize(right, condition.a, provider);\n      }\n\n      switch (op) {\n        case \"=\":\n          return Drizzle.eq(left, right);\n        case \"!=\":\n          return Drizzle.ne(left, right);\n        case \">\":\n          return Drizzle.gt(left, right);\n        case \">=\":\n          return Drizzle.gte(left, right);\n        case \"<\":\n          return Drizzle.lt(left, right);\n        case \"<=\":\n          return Drizzle.lte(left, right);\n        case \"in\": {\n          return Drizzle.inArray(left, right as never[]);\n        }\n        case \"not in\":\n          return Drizzle.notInArray(left, right as never[]);\n        case \"is\":\n          return right === null ? Drizzle.isNull(left) : Drizzle.eq(left, right);\n        case \"is not\":\n          return right === null ? Drizzle.isNotNull(left) : Drizzle.ne(left, right);\n        case \"contains\": {\n          right =\n            typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not contains\": {\n          right =\n            typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n          return Drizzle.notLike(left, right as string);\n        }\n        case \"ends with\": {\n          right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not ends with\": {\n          right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n          return Drizzle.notLike(left, right as string);\n        }\n        case \"starts with\": {\n          right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not starts with\": {\n          right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n          return Drizzle.notLike(left, right as string);\n        }\n\n        default:\n          throw new Error(`Unsupported operator: ${op}`);\n      }\n    }\n\n    if (condition.type === \"and\") {\n      return Drizzle.and(...condition.items.map((item) => buildWhere(item)));\n    }\n\n    if (condition.type === \"not\") {\n      const result = buildWhere(condition.item);\n      if (!result) {\n        return;\n      }\n\n      return Drizzle.not(result);\n    }\n\n    return Drizzle.or(...condition.items.map((item) => buildWhere(item)));\n  }\n\n  /**\n   * Process reference subqueries in encoded values, converting them to Drizzle SQL subqueries\n   */\n  function processReferenceSubqueries(values: Record<string, unknown>): Record<string, unknown> {\n    const processed: Record<string, unknown> = {};\n\n    for (const [key, value] of Object.entries(values)) {\n      if (value instanceof ReferenceSubquery) {\n        const refTable = value.referencedTable;\n        const externalId = value.externalIdValue;\n        const internalIdCol = refTable.getInternalIdColumn();\n        const idCol = refTable.getIdColumn();\n        const drizzleRefTable = toDrizzleTable(refTable);\n        const drizzleIdCol = toDrizzleColumn(idCol);\n\n        // Create a parameterized SQL subquery using Drizzle's query builder\n        // Safe cast: we're building a SQL subquery that returns a single bigint value\n        processed[key] = db\n          .select({ value: drizzleRefTable[internalIdCol.ormName] })\n          .from(drizzleRefTable)\n          .where(Drizzle.eq(drizzleIdCol, externalId))\n          .limit(1) as unknown;\n      } else {\n        processed[key] = value;\n      }\n    }\n\n    return processed;\n  }\n\n  /**\n   * Get table from schema by name\n   * @throws Error if table is not found in schema\n   */\n  function getTable(name: unknown): AnyTable {\n    const table = schema.tables[name as string];\n    if (!table) {\n      throw new Error(`Invalid table name ${name}.`);\n    }\n    return table;\n  }\n\n  /**\n   * Get the version to check for a given ID and checkVersion flag.\n   * @returns The version to check or undefined if no check is required.\n   * @throws Error if the ID is a string and checkVersion is true.\n   */\n  function getVersionToCheck(id: FragnoId | string, checkVersion: boolean): number | undefined {\n    if (!checkVersion) {\n      return undefined;\n    }\n\n    if (typeof id === \"string\") {\n      throw new Error(\n        `Cannot use checkVersion with a string ID. Version checking requires a FragnoId with version information.`,\n      );\n    }\n\n    return id.version;\n  }\n\n  /**\n   * Process joins recursively to support nested joins with orderBy and limit\n   */\n  function processJoins(\n    joins: CompiledJoin[],\n  ): Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> {\n    const result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> = {};\n\n    for (const join of joins) {\n      const { options, relation } = join;\n\n      if (!options) {\n        continue;\n      }\n\n      const targetTable = relation.table;\n      const joinName = relation.name;\n\n      // Build columns for this join using shared utility\n      const selectOption = options.select === undefined ? true : options.select;\n      const orderedColumns = getOrderedJoinColumns(targetTable, selectOption);\n      const joinColumns: Record<string, boolean> = {};\n      for (const colName of orderedColumns) {\n        joinColumns[colName] = true;\n      }\n\n      // Build orderBy for this join\n      let joinOrderBy: Drizzle.SQL[] | undefined;\n      if (options.orderBy && options.orderBy.length > 0) {\n        joinOrderBy = options.orderBy.map(([col, direction]) => {\n          const drizzleCol = toDrizzleColumn(col);\n          return direction === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n        });\n      }\n\n      // Build WHERE clause for this join if provided\n      let joinWhere: Drizzle.SQL | undefined;\n      if (options.where) {\n        joinWhere = buildWhere(options.where);\n      }\n\n      // Build the join config\n      const joinConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n        columns: joinColumns,\n        orderBy: joinOrderBy,\n        limit: options.limit,\n        where: joinWhere,\n      };\n\n      // Recursively process nested joins\n      if (options.join && options.join.length > 0) {\n        joinConfig.with = processJoins(options.join);\n      }\n\n      result[joinName] = joinConfig;\n    }\n\n    return result;\n  }\n\n  return {\n    compileRetrievalOperation(op: RetrievalOperation<TSchema>): DrizzleCompiledQuery | null {\n      switch (op.type) {\n        case \"count\": {\n          // Build WHERE clause\n          let whereClause: Drizzle.SQL | undefined;\n          if (op.options.where) {\n            const condition = buildCondition(op.table.columns, op.options.where);\n            if (condition === false) {\n              // Never matches - return null\n              return null;\n            }\n            if (condition !== true) {\n              whereClause = buildWhere(condition);\n            }\n          }\n\n          const drizzleTable = toDrizzleTable(op.table);\n          const query = db.select({ count: Drizzle.count() }).from(drizzleTable);\n\n          const compiledQuery = whereClause ? query.where(whereClause).toSQL() : query.toSQL();\n          onQuery?.(compiledQuery);\n          return compiledQuery;\n        }\n\n        case \"find\": {\n          const {\n            useIndex: _useIndex,\n            orderByIndex,\n            joins,\n            after,\n            before,\n            pageSize,\n            ...findOptions\n          } = op.options;\n\n          // Get index columns for ordering and cursor pagination\n          let indexColumns: AnyColumn[] = [];\n          let orderDirection: \"asc\" | \"desc\" = \"asc\";\n\n          if (orderByIndex) {\n            const index = op.table.indexes[orderByIndex.indexName];\n            orderDirection = orderByIndex.direction;\n\n            if (!index) {\n              // If _primary index doesn't exist, fall back to ID column\n              if (orderByIndex.indexName === \"_primary\") {\n                indexColumns = [op.table.getIdColumn()];\n              } else {\n                throw new Error(\n                  `Index \"${orderByIndex.indexName}\" not found on table \"${op.table.name}\"`,\n                );\n              }\n            } else {\n              indexColumns = index.columns;\n            }\n          }\n\n          // Convert orderByIndex to orderBy format\n          let orderBy: Drizzle.SQL[] | undefined;\n          if (indexColumns.length > 0) {\n            orderBy = indexColumns.map((col) => {\n              const drizzleCol = toDrizzleColumn(col);\n              return orderDirection === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n            });\n          }\n\n          // Build query configuration\n          const columns: Record<string, boolean> = {};\n          const select = findOptions.select;\n\n          if (select === true || select === undefined) {\n            for (const col of Object.values(op.table.columns)) {\n              columns[col.ormName] = true;\n            }\n          } else {\n            for (const k of select) {\n              columns[op.table.columns[k].ormName] = true;\n            }\n            // Always include hidden columns (for FragnoId construction with internal ID and version)\n            for (const col of Object.values(op.table.columns)) {\n              if (col.isHidden && !columns[col.ormName]) {\n                columns[col.ormName] = true;\n              }\n            }\n          }\n\n          // Build WHERE clause with cursor conditions\n          const whereClauses: Drizzle.SQL[] = [];\n\n          // Add user-defined where clause\n          if (findOptions.where) {\n            const condition = buildCondition(op.table.columns, findOptions.where);\n            if (condition === false) {\n              // Never matches - return null to indicate this query should be skipped\n              return null;\n            }\n            if (condition !== true) {\n              const clause = buildWhere(condition);\n              if (clause) {\n                whereClauses.push(clause);\n              }\n            }\n          }\n\n          // Add cursor-based pagination conditions\n          if ((after || before) && indexColumns.length > 0) {\n            const cursor = after || before;\n            const cursorData = decodeCursor(cursor!);\n            const serializedValues = serializeCursorValues(cursorData, indexColumns, provider);\n\n            // Build tuple comparison for cursor pagination\n            // For \"after\" with \"asc\": (col1, col2, ...) > (val1, val2, ...)\n            // For \"before\" with \"desc\": reverse the comparison\n            const isAfter = !!after;\n            const useGreaterThan =\n              (isAfter && orderDirection === \"asc\") || (!isAfter && orderDirection === \"desc\");\n\n            if (indexColumns.length === 1) {\n              // Simple single-column case\n              const col = toDrizzleColumn(indexColumns[0]!);\n              const val = serializedValues[indexColumns[0]!.ormName];\n              whereClauses.push(useGreaterThan ? Drizzle.gt(col, val) : Drizzle.lt(col, val));\n            } else {\n              // Multi-column tuple comparison using SQL\n              const drizzleCols = indexColumns.map((c) => toDrizzleColumn(c));\n              const vals = indexColumns.map((c) => serializedValues[c.ormName]);\n              const operator = useGreaterThan ? \">\" : \"<\";\n              // Safe cast: building a SQL comparison expression for cursor pagination\n              // Build the tuple comparison: (col1, col2) > (val1, val2)\n              const colsSQL = Drizzle.sql.join(drizzleCols, Drizzle.sql.raw(\", \"));\n              const valsSQL = Drizzle.sql.join(\n                vals.map((v) => Drizzle.sql`${v}`),\n                Drizzle.sql.raw(\", \"),\n              );\n              whereClauses.push(\n                Drizzle.sql`(${colsSQL}) ${Drizzle.sql.raw(operator)} (${valsSQL})`,\n              );\n            }\n          }\n\n          const whereClause = whereClauses.length > 0 ? Drizzle.and(...whereClauses) : undefined;\n\n          const queryConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n            columns,\n            limit: pageSize,\n            where: whereClause,\n            orderBy,\n            with: {},\n          };\n\n          // Process joins recursively to support nested joins\n          if (joins) {\n            queryConfig.with = processJoins(joins);\n          }\n\n          const physicalTableName = mapper ? mapper.toPhysical(op.table.ormName) : op.table.ormName;\n          const compiledQuery = db.query[physicalTableName].findMany(queryConfig).toSQL();\n          onQuery?.(compiledQuery);\n          return compiledQuery;\n        }\n      }\n    },\n\n    compileMutationOperation(\n      op: MutationOperation<TSchema>,\n    ): CompiledMutation<DrizzleCompiledQuery> | null {\n      switch (op.type) {\n        case \"create\": {\n          const table = getTable(op.table);\n          const drizzleTable = toDrizzleTable(table);\n          // encodeValues now handles runtime defaults automatically\n          const encodedValues = encodeValues(op.values, table, true, provider);\n          const values = processReferenceSubqueries(encodedValues);\n\n          const compiledQuery = db.insert(drizzleTable).values(values).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: null, // creates don't need affected row checks\n          };\n        }\n\n        case \"update\": {\n          const table = getTable(op.table);\n          const idColumn = table.getIdColumn();\n          const versionColumn = table.getVersionColumn();\n          const drizzleTable = toDrizzleTable(table);\n\n          const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n          const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n          // Build WHERE clause that filters by ID and optionally by version\n          const condition =\n            versionToCheck !== undefined\n              ? buildCondition(table.columns, (eb) =>\n                  eb.and(\n                    eb(idColumn.ormName, \"=\", externalId),\n                    eb(versionColumn.ormName, \"=\", versionToCheck),\n                  ),\n                )\n              : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n          // Handle boolean cases\n          if (condition === false) {\n            // Never matches - skip this operation\n            return null;\n          }\n\n          const whereClause = condition === true ? undefined : buildWhere(condition);\n          const encodedSetValues = encodeValues(op.set, table, false, provider);\n          const setValues = processReferenceSubqueries(encodedSetValues);\n\n          // Automatically increment _version for optimistic concurrency control\n          // Safe cast: we're building a SQL expression for incrementing the version\n          setValues[versionColumn.ormName] = Drizzle.sql.raw(\n            `COALESCE(${versionColumn.ormName}, 0) + 1`,\n          ) as unknown;\n\n          const compiledQuery = db.update(drizzleTable).set(setValues).where(whereClause).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: op.checkVersion ? 1 : null,\n          };\n        }\n\n        case \"delete\": {\n          const table = getTable(op.table);\n          const idColumn = table.getIdColumn();\n          const versionColumn = table.getVersionColumn();\n          const drizzleTable = toDrizzleTable(table);\n\n          const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n          const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n          // Build WHERE clause that filters by ID and optionally by version\n          const condition =\n            versionToCheck !== undefined\n              ? buildCondition(table.columns, (eb) =>\n                  eb.and(\n                    eb(idColumn.ormName, \"=\", externalId),\n                    eb(versionColumn.ormName, \"=\", versionToCheck),\n                  ),\n                )\n              : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n          // Handle boolean cases\n          if (condition === false) {\n            // Never matches - skip this operation\n            return null;\n          }\n\n          const whereClause = condition === true ? undefined : buildWhere(condition);\n\n          const compiledQuery = db.delete(drizzleTable).where(whereClause).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: op.checkVersion ? 1 : null,\n          };\n        }\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,SAAgB,yBACd,QACA,QACA,QACA,SAC4C;CAC5C,MAAM,CAAC,IAAI,iBAAiB,aAAa,OAAO,GAAG;CACnD,MAAM,EAAE,aAAa;;;;;CAMrB,SAAS,eAAe,OAA4B;EAElD,MAAM,oBAAoB,SAAS,OAAO,WAAW,MAAM,QAAQ,GAAG,MAAM;EAC5E,MAAM,MAAM,cAAc;AAC1B,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MACR,gCAAgC,kBAAkB,aAAa,MAAM,QAAQ,2CAC9E;;;;;;CAOH,SAAS,gBAAgB,KAA4B;EACnD,MAAM,cAAc,OAAO,OAAO,IAAI;AACtC,MAAI,CAAC,YACH,OAAM,IAAI,MAAM,2BAA2B,IAAI,UAAU,cAAc,IAAI,QAAQ,GAAG;EAIxF,MAAM,MADQ,eAAe,YAAY,CACvB,IAAI;AACtB,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MAAM,iCAAiC,IAAI,QAAQ,MAAM,YAAY,QAAQ,GAAG;;;;;CAM5F,SAAS,WAAW,WAA+C;AACjE,MAAI,UAAU,SAAS,WAAW;GAChC,MAAM,OAAO,gBAAgB,UAAU,EAAE;GACzC,MAAM,KAAK,UAAU;GACrB,IAAI,QAAQ,UAAU;AACtB,OAAI,iBAAiB,OACnB,SAAQ,gBAAgB,MAAM;OAG9B,SAAQ,UAAU,OAAO,UAAU,GAAG,SAAS;AAGjD,WAAQ,IAAR;IACE,KAAK,IACH,QAAOA,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,KACH,QAAOC,QAAgB,MAAM,MAAiB;IAEhD,KAAK,SACH,QAAOC,WAAmB,MAAM,MAAiB;IACnD,KAAK,KACH,QAAO,UAAU,OAAOC,OAAe,KAAK,GAAGR,GAAW,MAAM,MAAM;IACxE,KAAK,SACH,QAAO,UAAU,OAAOS,UAAkB,KAAK,GAAGR,GAAW,MAAM,MAAM;IAC3E,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOS,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAG/C,QACE,OAAM,IAAI,MAAM,yBAAyB,KAAK;;;AAIpD,MAAI,UAAU,SAAS,MACrB,QAAOC,IAAY,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;AAGxE,MAAI,UAAU,SAAS,OAAO;GAC5B,MAAM,SAAS,WAAW,UAAU,KAAK;AACzC,OAAI,CAAC,OACH;AAGF,UAAOC,IAAY,OAAO;;AAG5B,SAAOC,GAAW,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;;;;;CAMvE,SAAS,2BAA2B,QAA0D;EAC5F,MAAMC,YAAqC,EAAE;AAE7C,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,OAAO,CAC/C,KAAI,iBAAiB,mBAAmB;GACtC,MAAM,WAAW,MAAM;GACvB,MAAM,aAAa,MAAM;GACzB,MAAM,gBAAgB,SAAS,qBAAqB;GACpD,MAAM,QAAQ,SAAS,aAAa;GACpC,MAAM,kBAAkB,eAAe,SAAS;GAChD,MAAM,eAAe,gBAAgB,MAAM;AAI3C,aAAU,OAAO,GACd,OAAO,EAAE,OAAO,gBAAgB,cAAc,UAAU,CAAC,CACzD,KAAK,gBAAgB,CACrB,MAAMf,GAAW,cAAc,WAAW,CAAC,CAC3C,MAAM,EAAE;QAEX,WAAU,OAAO;AAIrB,SAAO;;;;;;CAOT,SAAS,SAAS,MAAyB;EACzC,MAAM,QAAQ,OAAO,OAAO;AAC5B,MAAI,CAAC,MACH,OAAM,IAAI,MAAM,sBAAsB,KAAK,GAAG;AAEhD,SAAO;;;;;;;CAQT,SAAS,kBAAkB,IAAuB,cAA2C;AAC3F,MAAI,CAAC,aACH;AAGF,MAAI,OAAO,OAAO,SAChB,OAAM,IAAI,MACR,2GACD;AAGH,SAAO,GAAG;;;;;CAMZ,SAAS,aACP,OACwD;EACxD,MAAMgB,SAAiE,EAAE;AAEzE,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,EAAE,SAAS,aAAa;AAE9B,OAAI,CAAC,QACH;GAGF,MAAM,cAAc,SAAS;GAC7B,MAAM,WAAW,SAAS;GAI1B,MAAM,iBAAiB,sBAAsB,aADxB,QAAQ,WAAW,SAAY,OAAO,QAAQ,OACI;GACvE,MAAMC,cAAuC,EAAE;AAC/C,QAAK,MAAM,WAAW,eACpB,aAAY,WAAW;GAIzB,IAAIC;AACJ,OAAI,QAAQ,WAAW,QAAQ,QAAQ,SAAS,EAC9C,eAAc,QAAQ,QAAQ,KAAK,CAAC,KAAK,eAAe;IACtD,MAAM,aAAa,gBAAgB,IAAI;AACvC,WAAO,cAAc,QAAQC,IAAY,WAAW,GAAGC,KAAa,WAAW;KAC/E;GAIJ,IAAIC;AACJ,OAAI,QAAQ,MACV,aAAY,WAAW,QAAQ,MAAM;GAIvC,MAAMC,aAAqD;IACzD,SAAS;IACT,SAAS;IACT,OAAO,QAAQ;IACf,OAAO;IACR;AAGD,OAAI,QAAQ,QAAQ,QAAQ,KAAK,SAAS,EACxC,YAAW,OAAO,aAAa,QAAQ,KAAK;AAG9C,UAAO,YAAY;;AAGrB,SAAO;;AAGT,QAAO;EACL,0BAA0B,IAA8D;AACtF,WAAQ,GAAG,MAAX;IACE,KAAK,SAAS;KAEZ,IAAIC;AACJ,SAAI,GAAG,QAAQ,OAAO;MACpB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,GAAG,QAAQ,MAAM;AACpE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,KAChB,eAAc,WAAW,UAAU;;KAIvC,MAAM,eAAe,eAAe,GAAG,MAAM;KAC7C,MAAM,QAAQ,GAAG,OAAO,EAAE,OAAOC,OAAe,EAAE,CAAC,CAAC,KAAK,aAAa;KAEtE,MAAM,gBAAgB,cAAc,MAAM,MAAM,YAAY,CAAC,OAAO,GAAG,MAAM,OAAO;AACpF,eAAU,cAAc;AACxB,YAAO;;IAGT,KAAK,QAAQ;KACX,MAAM,EACJ,UAAU,WACV,cACA,OACA,OACA,QACA,SACA,GAAG,gBACD,GAAG;KAGP,IAAIC,eAA4B,EAAE;KAClC,IAAIC,iBAAiC;AAErC,SAAI,cAAc;MAChB,MAAM,QAAQ,GAAG,MAAM,QAAQ,aAAa;AAC5C,uBAAiB,aAAa;AAE9B,UAAI,CAAC,MAEH,KAAI,aAAa,cAAc,WAC7B,gBAAe,CAAC,GAAG,MAAM,aAAa,CAAC;UAEvC,OAAM,IAAI,MACR,UAAU,aAAa,UAAU,wBAAwB,GAAG,MAAM,KAAK,GACxE;UAGH,gBAAe,MAAM;;KAKzB,IAAIC;AACJ,SAAI,aAAa,SAAS,EACxB,WAAU,aAAa,KAAK,QAAQ;MAClC,MAAM,aAAa,gBAAgB,IAAI;AACvC,aAAO,mBAAmB,QAAQR,IAAY,WAAW,GAAGC,KAAa,WAAW;OACpF;KAIJ,MAAMQ,UAAmC,EAAE;KAC3C,MAAM,SAAS,YAAY;AAE3B,SAAI,WAAW,QAAQ,WAAW,OAChC,MAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,SAAQ,IAAI,WAAW;UAEpB;AACL,WAAK,MAAM,KAAK,OACd,SAAQ,GAAG,MAAM,QAAQ,GAAG,WAAW;AAGzC,WAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,KAAI,IAAI,YAAY,CAAC,QAAQ,IAAI,SAC/B,SAAQ,IAAI,WAAW;;KAM7B,MAAMC,eAA8B,EAAE;AAGtC,SAAI,YAAY,OAAO;MACrB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,YAAY,MAAM;AACrE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,MAAM;OACtB,MAAM,SAAS,WAAW,UAAU;AACpC,WAAI,OACF,cAAa,KAAK,OAAO;;;AAM/B,UAAK,SAAS,WAAW,aAAa,SAAS,GAAG;MAGhD,MAAM,mBAAmB,sBADN,aADJ,SAAS,OACgB,EACmB,cAAc,SAAS;MAKlF,MAAM,UAAU,CAAC,CAAC;MAClB,MAAM,iBACH,WAAW,mBAAmB,SAAW,CAAC,WAAW,mBAAmB;AAE3E,UAAI,aAAa,WAAW,GAAG;OAE7B,MAAM,MAAM,gBAAgB,aAAa,GAAI;OAC7C,MAAM,MAAM,iBAAiB,aAAa,GAAI;AAC9C,oBAAa,KAAK,iBAAiB3B,GAAW,KAAK,IAAI,GAAGE,GAAW,KAAK,IAAI,CAAC;aAC1E;OAEL,MAAM,cAAc,aAAa,KAAK,MAAM,gBAAgB,EAAE,CAAC;OAC/D,MAAM,OAAO,aAAa,KAAK,MAAM,iBAAiB,EAAE,SAAS;OACjE,MAAM,WAAW,iBAAiB,MAAM;OAGxC,MAAM,cAAsB,KAAK,iBAAyB,IAAI,KAAK,CAAC;OACpE,MAAM,cAAsB,KAC1B,KAAK,KAAK,MAAM,GAAW,GAAG,IAAI,MACtB,IAAI,KAAK,CACtB;AACD,oBAAa,KACX,GAAW,IAAI,QAAQ,QAAgB,IAAI,SAAS,CAAC,IAAI,QAAQ,GAClE;;;KAML,MAAM0B,cAAsD;MAC1D;MACA,OAAO;MACP,OALkB,aAAa,SAAS,IAAIlB,IAAY,GAAG,aAAa,GAAG;MAM3E;MACA,MAAM,EAAE;MACT;AAGD,SAAI,MACF,aAAY,OAAO,aAAa,MAAM;KAGxC,MAAM,oBAAoB,SAAS,OAAO,WAAW,GAAG,MAAM,QAAQ,GAAG,GAAG,MAAM;KAClF,MAAM,gBAAgB,GAAG,MAAM,mBAAmB,SAAS,YAAY,CAAC,OAAO;AAC/E,eAAU,cAAc;AACxB,YAAO;;;;EAKb,yBACE,IAC+C;AAC/C,WAAQ,GAAG,MAAX;IACE,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,eAAe,eAAe,MAAM;KAG1C,MAAM,SAAS,2BADO,aAAa,GAAG,QAAQ,OAAO,MAAM,SAAS,CACZ;KAExD,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,OAAO,OAAO,CAAC,OAAO;AACpE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB;MACvB;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,YAAY,2BADO,aAAa,GAAG,KAAK,OAAO,OAAO,SAAS,CACP;AAI9D,eAAU,cAAc,eAAuB,IAC7C,YAAY,cAAc,QAAQ,UACnC;KAED,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,IAAI,UAAU,CAAC,MAAM,YAAY,CAAC,OAAO;AACvF,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,MAAM,YAAY,CAAC,OAAO;AACxE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;;;EAIR"}
+{"version":3,"file":"drizzle-uow-compiler.js","names":["Drizzle.eq","Drizzle.ne","Drizzle.gt","Drizzle.gte","Drizzle.lt","Drizzle.lte","Drizzle.inArray","Drizzle.notInArray","Drizzle.isNull","Drizzle.isNotNull","Drizzle.like","Drizzle.notLike","Drizzle.and","Drizzle.not","Drizzle.or","processed: Record<string, unknown>","result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>>","joinColumns: Record<string, boolean>","joinOrderBy: Drizzle.SQL[] | undefined","Drizzle.asc","Drizzle.desc","joinWhere: Drizzle.SQL | undefined","joinConfig: Drizzle.DBQueryConfig<\"many\", boolean>","whereClause: Drizzle.SQL | undefined","Drizzle.count","indexColumns: AnyColumn[]","orderDirection: \"asc\" | \"desc\"","orderBy: Drizzle.SQL[] | undefined","columns: Record<string, boolean>","whereClauses: Drizzle.SQL[]","queryConfig: Drizzle.DBQueryConfig<\"many\", boolean>"],"sources":["../../../src/adapters/drizzle/drizzle-uow-compiler.ts"],"sourcesContent":["import * as Drizzle from \"drizzle-orm\";\nimport type { AnyColumn, AnySchema, AnyTable, FragnoId } from \"../../schema/create\";\nimport { Column } from \"../../schema/create\";\nimport type {\n  CompiledMutation,\n  MutationOperation,\n  RetrievalOperation,\n  UOWCompiler,\n} from \"../../query/unit-of-work\";\nimport { buildCondition, type Condition } from \"../../query/condition-builder\";\nimport {\n  type ColumnType,\n  type TableType,\n  type TableNameMapper,\n  parseDrizzle,\n  type DBType,\n} from \"./shared\";\nimport { encodeValues, ReferenceSubquery } from \"../../query/result-transform\";\nimport { serialize } from \"../../schema/serialize\";\nimport { decodeCursor, serializeCursorValues } from \"../../query/cursor\";\nimport type { CompiledJoin } from \"../../query/orm/orm\";\nimport { getOrderedJoinColumns } from \"./join-column-utils\";\nimport type { ConnectionPool } from \"../../shared/connection-pool\";\n\nexport type DrizzleCompiledQuery = {\n  sql: string;\n  params: unknown[];\n};\n\n/**\n * Create a Drizzle-specific Unit of Work compiler\n *\n * This compiler translates UOW operations into Drizzle query functions\n * that can be executed as a batch/transaction.\n *\n * @param schema - The database schema\n * @param pool - Connection pool for acquiring database connections\n * @param provider - SQL provider (sqlite, mysql, postgresql)\n * @param mapper - Optional table name mapper for namespace prefixing\n * @param onQuery - Optional callback to receive compiled queries for logging/debugging\n * @returns A UOWCompiler instance for Drizzle\n */\nexport function createDrizzleUOWCompiler<TSchema extends AnySchema>(\n  schema: TSchema,\n  pool: ConnectionPool<DBType>,\n  provider: \"sqlite\" | \"mysql\" | \"postgresql\",\n  mapper?: TableNameMapper,\n  onQuery?: (query: DrizzleCompiledQuery) => void,\n): UOWCompiler<TSchema, DrizzleCompiledQuery> {\n  // Get db synchronously for compilation (doesn't execute, just builds SQL)\n  // TODO: We don't even need a Drizzle instance with a db client attached here. `drizzle({ schema })` is enough.\n  const dbRaw = pool.getDatabaseSync();\n  const [db, drizzleTables] = parseDrizzle(dbRaw);\n\n  /**\n   * Convert a Fragno table to a Drizzle table\n   * @throws Error if table is not found in Drizzle schema\n   */\n  function toDrizzleTable(table: AnyTable): TableType {\n    // Map logical table name to physical table name using the mapper\n    const physicalTableName = mapper ? mapper.toPhysical(table.ormName) : table.ormName;\n    const out = drizzleTables[physicalTableName];\n    if (out) {\n      return out;\n    }\n\n    throw new Error(\n      `[Drizzle] Unknown table name ${physicalTableName} (logical: ${table.ormName}), is it included in your Drizzle schema?`,\n    );\n  }\n\n  /**\n   * Convert a Fragno column to a Drizzle column\n   * @throws Error if column is not found in Drizzle table\n   */\n  function toDrizzleColumn(col: AnyColumn): ColumnType {\n    const fragnoTable = schema.tables[col.tableName];\n    if (!fragnoTable) {\n      throw new Error(`[Drizzle] Unknown table ${col.tableName} for column ${col.ormName}.`);\n    }\n\n    const table = toDrizzleTable(fragnoTable);\n    const out = table[col.ormName];\n    if (out) {\n      return out;\n    }\n\n    throw new Error(`[Drizzle] Unknown column name ${col.ormName} in ${fragnoTable.ormName}.`);\n  }\n\n  /**\n   * Build a WHERE clause from a condition using Drizzle's query builder\n   */\n  function buildWhere(condition: Condition): Drizzle.SQL | undefined {\n    if (condition.type === \"compare\") {\n      const left = toDrizzleColumn(condition.a);\n      const op = condition.operator;\n      let right = condition.b;\n      if (right instanceof Column) {\n        right = toDrizzleColumn(right);\n      } else {\n        // Serialize non-Column values (e.g., FragnoId -> string, Date -> number for SQLite)\n        right = serialize(right, condition.a, provider);\n      }\n\n      switch (op) {\n        case \"=\":\n          return Drizzle.eq(left, right);\n        case \"!=\":\n          return Drizzle.ne(left, right);\n        case \">\":\n          return Drizzle.gt(left, right);\n        case \">=\":\n          return Drizzle.gte(left, right);\n        case \"<\":\n          return Drizzle.lt(left, right);\n        case \"<=\":\n          return Drizzle.lte(left, right);\n        case \"in\": {\n          return Drizzle.inArray(left, right as never[]);\n        }\n        case \"not in\":\n          return Drizzle.notInArray(left, right as never[]);\n        case \"is\":\n          return right === null ? Drizzle.isNull(left) : Drizzle.eq(left, right);\n        case \"is not\":\n          return right === null ? Drizzle.isNotNull(left) : Drizzle.ne(left, right);\n        case \"contains\": {\n          right =\n            typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not contains\": {\n          right =\n            typeof right === \"string\" ? `%${right}%` : Drizzle.sql`concat('%', ${right}, '%')`;\n          return Drizzle.notLike(left, right as string);\n        }\n        case \"ends with\": {\n          right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not ends with\": {\n          right = typeof right === \"string\" ? `%${right}` : Drizzle.sql`concat('%', ${right})`;\n          return Drizzle.notLike(left, right as string);\n        }\n        case \"starts with\": {\n          right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n          return Drizzle.like(left, right as string);\n        }\n        case \"not starts with\": {\n          right = typeof right === \"string\" ? `${right}%` : Drizzle.sql`concat(${right}, '%')`;\n          return Drizzle.notLike(left, right as string);\n        }\n\n        default:\n          throw new Error(`Unsupported operator: ${op}`);\n      }\n    }\n\n    if (condition.type === \"and\") {\n      return Drizzle.and(...condition.items.map((item) => buildWhere(item)));\n    }\n\n    if (condition.type === \"not\") {\n      const result = buildWhere(condition.item);\n      if (!result) {\n        return;\n      }\n\n      return Drizzle.not(result);\n    }\n\n    return Drizzle.or(...condition.items.map((item) => buildWhere(item)));\n  }\n\n  /**\n   * Process reference subqueries in encoded values, converting them to Drizzle SQL subqueries\n   */\n  function processReferenceSubqueries(values: Record<string, unknown>): Record<string, unknown> {\n    const processed: Record<string, unknown> = {};\n\n    for (const [key, value] of Object.entries(values)) {\n      if (value instanceof ReferenceSubquery) {\n        const refTable = value.referencedTable;\n        const externalId = value.externalIdValue;\n        const internalIdCol = refTable.getInternalIdColumn();\n        const idCol = refTable.getIdColumn();\n        const drizzleRefTable = toDrizzleTable(refTable);\n        const drizzleIdCol = toDrizzleColumn(idCol);\n\n        // Create a parameterized SQL subquery using Drizzle's query builder\n        // Safe cast: we're building a SQL subquery that returns a single bigint value\n        processed[key] = db\n          .select({ value: drizzleRefTable[internalIdCol.ormName] })\n          .from(drizzleRefTable)\n          .where(Drizzle.eq(drizzleIdCol, externalId))\n          .limit(1) as unknown;\n      } else {\n        processed[key] = value;\n      }\n    }\n\n    return processed;\n  }\n\n  /**\n   * Get table from schema by name\n   * @throws Error if table is not found in schema\n   */\n  function getTable(name: unknown): AnyTable {\n    const table = schema.tables[name as string];\n    if (!table) {\n      throw new Error(`Invalid table name ${name}.`);\n    }\n    return table;\n  }\n\n  /**\n   * Get the version to check for a given ID and checkVersion flag.\n   * @returns The version to check or undefined if no check is required.\n   * @throws Error if the ID is a string and checkVersion is true.\n   */\n  function getVersionToCheck(id: FragnoId | string, checkVersion: boolean): number | undefined {\n    if (!checkVersion) {\n      return undefined;\n    }\n\n    if (typeof id === \"string\") {\n      throw new Error(\n        `Cannot use checkVersion with a string ID. Version checking requires a FragnoId with version information.`,\n      );\n    }\n\n    return id.version;\n  }\n\n  /**\n   * Process joins recursively to support nested joins with orderBy and limit\n   */\n  function processJoins(\n    joins: CompiledJoin[],\n  ): Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> {\n    const result: Record<string, Drizzle.DBQueryConfig<\"many\", boolean>> = {};\n\n    for (const join of joins) {\n      const { options, relation } = join;\n\n      if (!options) {\n        continue;\n      }\n\n      const targetTable = relation.table;\n      const joinName = relation.name;\n\n      // Build columns for this join using shared utility\n      const selectOption = options.select === undefined ? true : options.select;\n      const orderedColumns = getOrderedJoinColumns(targetTable, selectOption);\n      const joinColumns: Record<string, boolean> = {};\n      for (const colName of orderedColumns) {\n        joinColumns[colName] = true;\n      }\n\n      // Build orderBy for this join\n      let joinOrderBy: Drizzle.SQL[] | undefined;\n      if (options.orderBy && options.orderBy.length > 0) {\n        joinOrderBy = options.orderBy.map(([col, direction]) => {\n          const drizzleCol = toDrizzleColumn(col);\n          return direction === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n        });\n      }\n\n      // Build WHERE clause for this join if provided\n      let joinWhere: Drizzle.SQL | undefined;\n      if (options.where) {\n        joinWhere = buildWhere(options.where);\n      }\n\n      // Build the join config\n      const joinConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n        columns: joinColumns,\n        orderBy: joinOrderBy,\n        limit: options.limit,\n        where: joinWhere,\n      };\n\n      // Recursively process nested joins\n      if (options.join && options.join.length > 0) {\n        joinConfig.with = processJoins(options.join);\n      }\n\n      result[joinName] = joinConfig;\n    }\n\n    return result;\n  }\n\n  return {\n    compileRetrievalOperation(op: RetrievalOperation<TSchema>): DrizzleCompiledQuery | null {\n      switch (op.type) {\n        case \"count\": {\n          // Build WHERE clause\n          let whereClause: Drizzle.SQL | undefined;\n          if (op.options.where) {\n            const condition = buildCondition(op.table.columns, op.options.where);\n            if (condition === false) {\n              // Never matches - return null\n              return null;\n            }\n            if (condition !== true) {\n              whereClause = buildWhere(condition);\n            }\n          }\n\n          const drizzleTable = toDrizzleTable(op.table);\n          const query = db.select({ count: Drizzle.count() }).from(drizzleTable);\n\n          const compiledQuery = whereClause ? query.where(whereClause).toSQL() : query.toSQL();\n          onQuery?.(compiledQuery);\n          return compiledQuery;\n        }\n\n        case \"find\": {\n          const {\n            useIndex: _useIndex,\n            orderByIndex,\n            joins,\n            after,\n            before,\n            pageSize,\n            ...findOptions\n          } = op.options;\n\n          // Get index columns for ordering and cursor pagination\n          let indexColumns: AnyColumn[] = [];\n          let orderDirection: \"asc\" | \"desc\" = \"asc\";\n\n          if (orderByIndex) {\n            const index = op.table.indexes[orderByIndex.indexName];\n            orderDirection = orderByIndex.direction;\n\n            if (!index) {\n              // If _primary index doesn't exist, fall back to ID column\n              if (orderByIndex.indexName === \"_primary\") {\n                indexColumns = [op.table.getIdColumn()];\n              } else {\n                throw new Error(\n                  `Index \"${orderByIndex.indexName}\" not found on table \"${op.table.name}\"`,\n                );\n              }\n            } else {\n              indexColumns = index.columns;\n            }\n          }\n\n          // Convert orderByIndex to orderBy format\n          let orderBy: Drizzle.SQL[] | undefined;\n          if (indexColumns.length > 0) {\n            orderBy = indexColumns.map((col) => {\n              const drizzleCol = toDrizzleColumn(col);\n              return orderDirection === \"asc\" ? Drizzle.asc(drizzleCol) : Drizzle.desc(drizzleCol);\n            });\n          }\n\n          // Build query configuration\n          const columns: Record<string, boolean> = {};\n          const select = findOptions.select;\n\n          if (select === true || select === undefined) {\n            for (const col of Object.values(op.table.columns)) {\n              columns[col.ormName] = true;\n            }\n          } else {\n            for (const k of select) {\n              columns[op.table.columns[k].ormName] = true;\n            }\n            // Always include hidden columns (for FragnoId construction with internal ID and version)\n            for (const col of Object.values(op.table.columns)) {\n              if (col.isHidden && !columns[col.ormName]) {\n                columns[col.ormName] = true;\n              }\n            }\n          }\n\n          // Build WHERE clause with cursor conditions\n          const whereClauses: Drizzle.SQL[] = [];\n\n          // Add user-defined where clause\n          if (findOptions.where) {\n            const condition = buildCondition(op.table.columns, findOptions.where);\n            if (condition === false) {\n              // Never matches - return null to indicate this query should be skipped\n              return null;\n            }\n            if (condition !== true) {\n              const clause = buildWhere(condition);\n              if (clause) {\n                whereClauses.push(clause);\n              }\n            }\n          }\n\n          // Add cursor-based pagination conditions\n          if ((after || before) && indexColumns.length > 0) {\n            const cursor = after || before;\n            const cursorData = decodeCursor(cursor!);\n            const serializedValues = serializeCursorValues(cursorData, indexColumns, provider);\n\n            // Build tuple comparison for cursor pagination\n            // For \"after\" with \"asc\": (col1, col2, ...) > (val1, val2, ...)\n            // For \"before\" with \"desc\": reverse the comparison\n            const isAfter = !!after;\n            const useGreaterThan =\n              (isAfter && orderDirection === \"asc\") || (!isAfter && orderDirection === \"desc\");\n\n            if (indexColumns.length === 1) {\n              // Simple single-column case\n              const col = toDrizzleColumn(indexColumns[0]!);\n              const val = serializedValues[indexColumns[0]!.ormName];\n              whereClauses.push(useGreaterThan ? Drizzle.gt(col, val) : Drizzle.lt(col, val));\n            } else {\n              // Multi-column tuple comparison using SQL\n              const drizzleCols = indexColumns.map((c) => toDrizzleColumn(c));\n              const vals = indexColumns.map((c) => serializedValues[c.ormName]);\n              const operator = useGreaterThan ? \">\" : \"<\";\n              // Safe cast: building a SQL comparison expression for cursor pagination\n              // Build the tuple comparison: (col1, col2) > (val1, val2)\n              const colsSQL = Drizzle.sql.join(drizzleCols, Drizzle.sql.raw(\", \"));\n              const valsSQL = Drizzle.sql.join(\n                vals.map((v) => Drizzle.sql`${v}`),\n                Drizzle.sql.raw(\", \"),\n              );\n              whereClauses.push(\n                Drizzle.sql`(${colsSQL}) ${Drizzle.sql.raw(operator)} (${valsSQL})`,\n              );\n            }\n          }\n\n          const whereClause = whereClauses.length > 0 ? Drizzle.and(...whereClauses) : undefined;\n\n          const queryConfig: Drizzle.DBQueryConfig<\"many\", boolean> = {\n            columns,\n            limit: pageSize,\n            where: whereClause,\n            orderBy,\n            with: {},\n          };\n\n          // Process joins recursively to support nested joins\n          if (joins) {\n            queryConfig.with = processJoins(joins);\n          }\n\n          const physicalTableName = mapper ? mapper.toPhysical(op.table.ormName) : op.table.ormName;\n          const compiledQuery = db.query[physicalTableName].findMany(queryConfig).toSQL();\n          onQuery?.(compiledQuery);\n          return compiledQuery;\n        }\n      }\n    },\n\n    compileMutationOperation(\n      op: MutationOperation<TSchema>,\n    ): CompiledMutation<DrizzleCompiledQuery> | null {\n      switch (op.type) {\n        case \"create\": {\n          const table = getTable(op.table);\n          const drizzleTable = toDrizzleTable(table);\n          // encodeValues now handles runtime defaults automatically\n          const encodedValues = encodeValues(op.values, table, true, provider);\n          const values = processReferenceSubqueries(encodedValues);\n\n          const compiledQuery = db.insert(drizzleTable).values(values).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: null, // creates don't need affected row checks\n          };\n        }\n\n        case \"update\": {\n          const table = getTable(op.table);\n          const idColumn = table.getIdColumn();\n          const versionColumn = table.getVersionColumn();\n          const drizzleTable = toDrizzleTable(table);\n\n          const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n          const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n          // Build WHERE clause that filters by ID and optionally by version\n          const condition =\n            versionToCheck !== undefined\n              ? buildCondition(table.columns, (eb) =>\n                  eb.and(\n                    eb(idColumn.ormName, \"=\", externalId),\n                    eb(versionColumn.ormName, \"=\", versionToCheck),\n                  ),\n                )\n              : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n          // Handle boolean cases\n          if (condition === false) {\n            // Never matches - skip this operation\n            return null;\n          }\n\n          const whereClause = condition === true ? undefined : buildWhere(condition);\n          const encodedSetValues = encodeValues(op.set, table, false, provider);\n          const setValues = processReferenceSubqueries(encodedSetValues);\n\n          // Automatically increment _version for optimistic concurrency control\n          // Safe cast: we're building a SQL expression for incrementing the version\n          setValues[versionColumn.ormName] = Drizzle.sql.raw(\n            `COALESCE(${versionColumn.ormName}, 0) + 1`,\n          ) as unknown;\n\n          const compiledQuery = db.update(drizzleTable).set(setValues).where(whereClause).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: op.checkVersion ? 1 : null,\n          };\n        }\n\n        case \"delete\": {\n          const table = getTable(op.table);\n          const idColumn = table.getIdColumn();\n          const versionColumn = table.getVersionColumn();\n          const drizzleTable = toDrizzleTable(table);\n\n          const externalId = typeof op.id === \"string\" ? op.id : op.id.externalId;\n          const versionToCheck = getVersionToCheck(op.id, op.checkVersion);\n\n          // Build WHERE clause that filters by ID and optionally by version\n          const condition =\n            versionToCheck !== undefined\n              ? buildCondition(table.columns, (eb) =>\n                  eb.and(\n                    eb(idColumn.ormName, \"=\", externalId),\n                    eb(versionColumn.ormName, \"=\", versionToCheck),\n                  ),\n                )\n              : buildCondition(table.columns, (eb) => eb(idColumn.ormName, \"=\", externalId));\n\n          // Handle boolean cases\n          if (condition === false) {\n            // Never matches - skip this operation\n            return null;\n          }\n\n          const whereClause = condition === true ? undefined : buildWhere(condition);\n\n          const compiledQuery = db.delete(drizzleTable).where(whereClause).toSQL();\n          onQuery?.(compiledQuery);\n          return {\n            query: compiledQuery,\n            expectedAffectedRows: op.checkVersion ? 1 : null,\n          };\n        }\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AA0CA,SAAgB,yBACd,QACA,MACA,UACA,QACA,SAC4C;CAI5C,MAAM,CAAC,IAAI,iBAAiB,aADd,KAAK,iBAAiB,CACW;;;;;CAM/C,SAAS,eAAe,OAA4B;EAElD,MAAM,oBAAoB,SAAS,OAAO,WAAW,MAAM,QAAQ,GAAG,MAAM;EAC5E,MAAM,MAAM,cAAc;AAC1B,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MACR,gCAAgC,kBAAkB,aAAa,MAAM,QAAQ,2CAC9E;;;;;;CAOH,SAAS,gBAAgB,KAA4B;EACnD,MAAM,cAAc,OAAO,OAAO,IAAI;AACtC,MAAI,CAAC,YACH,OAAM,IAAI,MAAM,2BAA2B,IAAI,UAAU,cAAc,IAAI,QAAQ,GAAG;EAIxF,MAAM,MADQ,eAAe,YAAY,CACvB,IAAI;AACtB,MAAI,IACF,QAAO;AAGT,QAAM,IAAI,MAAM,iCAAiC,IAAI,QAAQ,MAAM,YAAY,QAAQ,GAAG;;;;;CAM5F,SAAS,WAAW,WAA+C;AACjE,MAAI,UAAU,SAAS,WAAW;GAChC,MAAM,OAAO,gBAAgB,UAAU,EAAE;GACzC,MAAM,KAAK,UAAU;GACrB,IAAI,QAAQ,UAAU;AACtB,OAAI,iBAAiB,OACnB,SAAQ,gBAAgB,MAAM;OAG9B,SAAQ,UAAU,OAAO,UAAU,GAAG,SAAS;AAGjD,WAAQ,IAAR;IACE,KAAK,IACH,QAAOA,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,IACH,QAAOC,GAAW,MAAM,MAAM;IAChC,KAAK,KACH,QAAOC,IAAY,MAAM,MAAM;IACjC,KAAK,KACH,QAAOC,QAAgB,MAAM,MAAiB;IAEhD,KAAK,SACH,QAAOC,WAAmB,MAAM,MAAiB;IACnD,KAAK,KACH,QAAO,UAAU,OAAOC,OAAe,KAAK,GAAGR,GAAW,MAAM,MAAM;IACxE,KAAK,SACH,QAAO,UAAU,OAAOS,UAAkB,KAAK,GAAGR,GAAW,MAAM,MAAM;IAC3E,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOS,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aACE,OAAO,UAAU,WAAW,IAAI,MAAM,KAAK,GAAW,eAAe,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,IAAI,UAAU,GAAW,eAAe,MAAM;AAClF,YAAOC,QAAgB,MAAM,MAAgB;IAE/C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOD,KAAa,MAAM,MAAgB;IAE5C,KAAK;AACH,aAAQ,OAAO,UAAU,WAAW,GAAG,MAAM,KAAK,GAAW,UAAU,MAAM;AAC7E,YAAOC,QAAgB,MAAM,MAAgB;IAG/C,QACE,OAAM,IAAI,MAAM,yBAAyB,KAAK;;;AAIpD,MAAI,UAAU,SAAS,MACrB,QAAOC,IAAY,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;AAGxE,MAAI,UAAU,SAAS,OAAO;GAC5B,MAAM,SAAS,WAAW,UAAU,KAAK;AACzC,OAAI,CAAC,OACH;AAGF,UAAOC,IAAY,OAAO;;AAG5B,SAAOC,GAAW,GAAG,UAAU,MAAM,KAAK,SAAS,WAAW,KAAK,CAAC,CAAC;;;;;CAMvE,SAAS,2BAA2B,QAA0D;EAC5F,MAAMC,YAAqC,EAAE;AAE7C,OAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,OAAO,CAC/C,KAAI,iBAAiB,mBAAmB;GACtC,MAAM,WAAW,MAAM;GACvB,MAAM,aAAa,MAAM;GACzB,MAAM,gBAAgB,SAAS,qBAAqB;GACpD,MAAM,QAAQ,SAAS,aAAa;GACpC,MAAM,kBAAkB,eAAe,SAAS;GAChD,MAAM,eAAe,gBAAgB,MAAM;AAI3C,aAAU,OAAO,GACd,OAAO,EAAE,OAAO,gBAAgB,cAAc,UAAU,CAAC,CACzD,KAAK,gBAAgB,CACrB,MAAMf,GAAW,cAAc,WAAW,CAAC,CAC3C,MAAM,EAAE;QAEX,WAAU,OAAO;AAIrB,SAAO;;;;;;CAOT,SAAS,SAAS,MAAyB;EACzC,MAAM,QAAQ,OAAO,OAAO;AAC5B,MAAI,CAAC,MACH,OAAM,IAAI,MAAM,sBAAsB,KAAK,GAAG;AAEhD,SAAO;;;;;;;CAQT,SAAS,kBAAkB,IAAuB,cAA2C;AAC3F,MAAI,CAAC,aACH;AAGF,MAAI,OAAO,OAAO,SAChB,OAAM,IAAI,MACR,2GACD;AAGH,SAAO,GAAG;;;;;CAMZ,SAAS,aACP,OACwD;EACxD,MAAMgB,SAAiE,EAAE;AAEzE,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,EAAE,SAAS,aAAa;AAE9B,OAAI,CAAC,QACH;GAGF,MAAM,cAAc,SAAS;GAC7B,MAAM,WAAW,SAAS;GAI1B,MAAM,iBAAiB,sBAAsB,aADxB,QAAQ,WAAW,SAAY,OAAO,QAAQ,OACI;GACvE,MAAMC,cAAuC,EAAE;AAC/C,QAAK,MAAM,WAAW,eACpB,aAAY,WAAW;GAIzB,IAAIC;AACJ,OAAI,QAAQ,WAAW,QAAQ,QAAQ,SAAS,EAC9C,eAAc,QAAQ,QAAQ,KAAK,CAAC,KAAK,eAAe;IACtD,MAAM,aAAa,gBAAgB,IAAI;AACvC,WAAO,cAAc,QAAQC,IAAY,WAAW,GAAGC,KAAa,WAAW;KAC/E;GAIJ,IAAIC;AACJ,OAAI,QAAQ,MACV,aAAY,WAAW,QAAQ,MAAM;GAIvC,MAAMC,aAAqD;IACzD,SAAS;IACT,SAAS;IACT,OAAO,QAAQ;IACf,OAAO;IACR;AAGD,OAAI,QAAQ,QAAQ,QAAQ,KAAK,SAAS,EACxC,YAAW,OAAO,aAAa,QAAQ,KAAK;AAG9C,UAAO,YAAY;;AAGrB,SAAO;;AAGT,QAAO;EACL,0BAA0B,IAA8D;AACtF,WAAQ,GAAG,MAAX;IACE,KAAK,SAAS;KAEZ,IAAIC;AACJ,SAAI,GAAG,QAAQ,OAAO;MACpB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,GAAG,QAAQ,MAAM;AACpE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,KAChB,eAAc,WAAW,UAAU;;KAIvC,MAAM,eAAe,eAAe,GAAG,MAAM;KAC7C,MAAM,QAAQ,GAAG,OAAO,EAAE,OAAOC,OAAe,EAAE,CAAC,CAAC,KAAK,aAAa;KAEtE,MAAM,gBAAgB,cAAc,MAAM,MAAM,YAAY,CAAC,OAAO,GAAG,MAAM,OAAO;AACpF,eAAU,cAAc;AACxB,YAAO;;IAGT,KAAK,QAAQ;KACX,MAAM,EACJ,UAAU,WACV,cACA,OACA,OACA,QACA,SACA,GAAG,gBACD,GAAG;KAGP,IAAIC,eAA4B,EAAE;KAClC,IAAIC,iBAAiC;AAErC,SAAI,cAAc;MAChB,MAAM,QAAQ,GAAG,MAAM,QAAQ,aAAa;AAC5C,uBAAiB,aAAa;AAE9B,UAAI,CAAC,MAEH,KAAI,aAAa,cAAc,WAC7B,gBAAe,CAAC,GAAG,MAAM,aAAa,CAAC;UAEvC,OAAM,IAAI,MACR,UAAU,aAAa,UAAU,wBAAwB,GAAG,MAAM,KAAK,GACxE;UAGH,gBAAe,MAAM;;KAKzB,IAAIC;AACJ,SAAI,aAAa,SAAS,EACxB,WAAU,aAAa,KAAK,QAAQ;MAClC,MAAM,aAAa,gBAAgB,IAAI;AACvC,aAAO,mBAAmB,QAAQR,IAAY,WAAW,GAAGC,KAAa,WAAW;OACpF;KAIJ,MAAMQ,UAAmC,EAAE;KAC3C,MAAM,SAAS,YAAY;AAE3B,SAAI,WAAW,QAAQ,WAAW,OAChC,MAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,SAAQ,IAAI,WAAW;UAEpB;AACL,WAAK,MAAM,KAAK,OACd,SAAQ,GAAG,MAAM,QAAQ,GAAG,WAAW;AAGzC,WAAK,MAAM,OAAO,OAAO,OAAO,GAAG,MAAM,QAAQ,CAC/C,KAAI,IAAI,YAAY,CAAC,QAAQ,IAAI,SAC/B,SAAQ,IAAI,WAAW;;KAM7B,MAAMC,eAA8B,EAAE;AAGtC,SAAI,YAAY,OAAO;MACrB,MAAM,YAAY,eAAe,GAAG,MAAM,SAAS,YAAY,MAAM;AACrE,UAAI,cAAc,MAEhB,QAAO;AAET,UAAI,cAAc,MAAM;OACtB,MAAM,SAAS,WAAW,UAAU;AACpC,WAAI,OACF,cAAa,KAAK,OAAO;;;AAM/B,UAAK,SAAS,WAAW,aAAa,SAAS,GAAG;MAGhD,MAAM,mBAAmB,sBADN,aADJ,SAAS,OACgB,EACmB,cAAc,SAAS;MAKlF,MAAM,UAAU,CAAC,CAAC;MAClB,MAAM,iBACH,WAAW,mBAAmB,SAAW,CAAC,WAAW,mBAAmB;AAE3E,UAAI,aAAa,WAAW,GAAG;OAE7B,MAAM,MAAM,gBAAgB,aAAa,GAAI;OAC7C,MAAM,MAAM,iBAAiB,aAAa,GAAI;AAC9C,oBAAa,KAAK,iBAAiB3B,GAAW,KAAK,IAAI,GAAGE,GAAW,KAAK,IAAI,CAAC;aAC1E;OAEL,MAAM,cAAc,aAAa,KAAK,MAAM,gBAAgB,EAAE,CAAC;OAC/D,MAAM,OAAO,aAAa,KAAK,MAAM,iBAAiB,EAAE,SAAS;OACjE,MAAM,WAAW,iBAAiB,MAAM;OAGxC,MAAM,cAAsB,KAAK,iBAAyB,IAAI,KAAK,CAAC;OACpE,MAAM,cAAsB,KAC1B,KAAK,KAAK,MAAM,GAAW,GAAG,IAAI,MACtB,IAAI,KAAK,CACtB;AACD,oBAAa,KACX,GAAW,IAAI,QAAQ,QAAgB,IAAI,SAAS,CAAC,IAAI,QAAQ,GAClE;;;KAML,MAAM0B,cAAsD;MAC1D;MACA,OAAO;MACP,OALkB,aAAa,SAAS,IAAIlB,IAAY,GAAG,aAAa,GAAG;MAM3E;MACA,MAAM,EAAE;MACT;AAGD,SAAI,MACF,aAAY,OAAO,aAAa,MAAM;KAGxC,MAAM,oBAAoB,SAAS,OAAO,WAAW,GAAG,MAAM,QAAQ,GAAG,GAAG,MAAM;KAClF,MAAM,gBAAgB,GAAG,MAAM,mBAAmB,SAAS,YAAY,CAAC,OAAO;AAC/E,eAAU,cAAc;AACxB,YAAO;;;;EAKb,yBACE,IAC+C;AAC/C,WAAQ,GAAG,MAAX;IACE,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,eAAe,eAAe,MAAM;KAG1C,MAAM,SAAS,2BADO,aAAa,GAAG,QAAQ,OAAO,MAAM,SAAS,CACZ;KAExD,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,OAAO,OAAO,CAAC,OAAO;AACpE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB;MACvB;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,YAAY,2BADO,aAAa,GAAG,KAAK,OAAO,OAAO,SAAS,CACP;AAI9D,eAAU,cAAc,eAAuB,IAC7C,YAAY,cAAc,QAAQ,UACnC;KAED,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,IAAI,UAAU,CAAC,MAAM,YAAY,CAAC,OAAO;AACvF,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;IAGH,KAAK,UAAU;KACb,MAAM,QAAQ,SAAS,GAAG,MAAM;KAChC,MAAM,WAAW,MAAM,aAAa;KACpC,MAAM,gBAAgB,MAAM,kBAAkB;KAC9C,MAAM,eAAe,eAAe,MAAM;KAE1C,MAAM,aAAa,OAAO,GAAG,OAAO,WAAW,GAAG,KAAK,GAAG,GAAG;KAC7D,MAAM,iBAAiB,kBAAkB,GAAG,IAAI,GAAG,aAAa;KAGhE,MAAM,YACJ,mBAAmB,SACf,eAAe,MAAM,UAAU,OAC7B,GAAG,IACD,GAAG,SAAS,SAAS,KAAK,WAAW,EACrC,GAAG,cAAc,SAAS,KAAK,eAAe,CAC/C,CACF,GACD,eAAe,MAAM,UAAU,OAAO,GAAG,SAAS,SAAS,KAAK,WAAW,CAAC;AAGlF,SAAI,cAAc,MAEhB,QAAO;KAGT,MAAM,cAAc,cAAc,OAAO,SAAY,WAAW,UAAU;KAE1E,MAAM,gBAAgB,GAAG,OAAO,aAAa,CAAC,MAAM,YAAY,CAAC,OAAO;AACxE,eAAU,cAAc;AACxB,YAAO;MACL,OAAO;MACP,sBAAsB,GAAG,eAAe,IAAI;MAC7C;;;;EAIR"}

package/dist/adapters/drizzle/drizzle-uow-executor.js

@@ -1,6 +1,6 @@
-import { SQL, StringChunk, sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sql/sql.js";
-import { SQLiteSyncDialect } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sqlite-core/dialect.js";
-import { BaseSQLiteDatabase } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@types_better-sqlite3@7.6.13_better-sqlite3@12.4.1_kysely@0.28.8/node_modules/drizzle-orm/sqlite-core/db.js";
+import { SQL, StringChunk, sql } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sql/sql.js";
+import { SQLiteSyncDialect } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sqlite-core/dialect.js";
+import { BaseSQLiteDatabase } from "../../node_modules/.pnpm/drizzle-orm@0.44.7_@electric-sql_pglite@0.3.11_@libsql_client@0.15.15_@types_better-sql_aee8b50e4fe3dba0cf9f533329f36260/node_modules/drizzle-orm/sqlite-core/db.js";
 
 //#region src/adapters/drizzle/drizzle-uow-executor.ts
 function isSyncSQLite(db) {
@@ -42,6 +42,7 @@ function toSQL(query, provider) {
 function getAffectedRows(result) {
 	if (Array.isArray(result)) return result.length;
 	if (result && typeof result === "object") {
+		if ("rowsAffected" in result && typeof result["rowsAffected"] === "number") return result["rowsAffected"];
 		if ("affectedRows" in result && typeof result["affectedRows"] === "number") return result["affectedRows"];
 		if ("rowCount" in result && (typeof result["rowCount"] === "number" || typeof result["rowCount"] === "bigint")) {
 			const rowCount = result["rowCount"];
@@ -94,7 +95,10 @@ async function executeDrizzleRetrievalPhase(db, retrievalBatch, provider) {
 		}
 	}, async (tx) => {
 		for (const query of retrievalBatch) {
-			const result = await tx.execute(toSQL(query, provider));
+			const sqlObj = toSQL(query, provider);
+			const executeMethod = tx.execute ?? tx.run;
+			if (!executeMethod) throw new Error("Transaction object has neither execute nor run method");
+			const result = await executeMethod.call(tx, sqlObj);
 			retrievalResults.push(result);
 		}
 	});
@@ -123,7 +127,10 @@ async function executeDrizzleMutationPhase(db, mutationBatch, provider) {
 			}
 		}, async (tx) => {
 			for (const { query, expectedAffectedRows } of mutationBatch) {
-				const result = await tx.execute(toSQL(query, provider));
+				const sqlObj = toSQL(query, provider);
+				const executeMethod = tx.execute ?? tx.run;
+				if (!executeMethod) throw new Error("Transaction object has neither execute nor run method");
+				const result = await executeMethod.call(tx, sqlObj);
 				if (expectedAffectedRows === null) createdInternalIds.push(extractCreatedInternalId(result));
 				else validateAffectedRows(result, expectedAffectedRows);
 			}