@fragno-dev/db 0.1.15 → 0.2.0

package/src/sql-driver/sql-driver.ts

@@ -0,0 +1,53 @@
+/**
+ * These interfaces are adapted from Kysely.
+ * Modified from: https://github.com/kysely-org/kysely
+ * License: MIT
+ * Date obtained: December 3 2025
+ * Copyright (c) 2022 Sami Koskimäki
+ */
+
+import type { DialectAdapter } from "./dialect-adapter/dialect-adapter";
+
+export interface CompiledQuery {
+  readonly sql: string;
+  readonly parameters: ReadonlyArray<unknown>;
+}
+
+export interface DatabaseConnection {
+  executeQuery<R>(compiledQuery: CompiledQuery): Promise<QueryResult<R>>;
+}
+
+export interface QueryResult<O> {
+  readonly numAffectedRows?: bigint;
+  readonly numChangedRows?: bigint;
+  readonly insertId?: bigint;
+
+  readonly rows: O[];
+}
+
+export interface TransactionSettings {
+  //
+}
+
+export interface Driver {
+  init(): Promise<void>;
+
+  acquireConnection(): Promise<DatabaseConnection>;
+  releaseConnection(connection: DatabaseConnection): Promise<void>;
+
+  beginTransaction(connection: DatabaseConnection, settings: TransactionSettings): Promise<void>;
+  commitTransaction(connection: DatabaseConnection): Promise<void>;
+  rollbackTransaction(connection: DatabaseConnection): Promise<void>;
+
+  destroy(): Promise<void>;
+}
+
+export interface QueryCompiler {
+  compileQuery(node: unknown, queryId: unknown): CompiledQuery;
+}
+
+export interface Dialect {
+  createDriver(): Driver;
+  createAdapter(): DialectAdapter;
+  createQueryCompiler(): QueryCompiler;
+}

package/src/sql-driver/sql.ts

@@ -0,0 +1,57 @@
+import type { CompiledQuery, Dialect } from "./sql-driver";
+import {
+  sql as kyselySql,
+  type QueryExecutor,
+  type CompiledQuery as KyselyCompiledQuery,
+} from "kysely";
+
+/**
+ * Wrapper around Kysely's RawBuilder that provides a compile() method with a dialect parameter.
+ */
+export class RawBuilder {
+  #kyselyBuilder: ReturnType<typeof kyselySql>;
+
+  constructor(kyselyBuilder: ReturnType<typeof kyselySql>) {
+    this.#kyselyBuilder = kyselyBuilder;
+  }
+
+  /**
+   * Compiles the SQL query using the provided Kysely dialect.
+   * Creates a minimal query executor with the dialect's adapter and query compiler.
+   *
+   * @param dialect - Kysely dialect (e.g., SqliteDialect, PostgresDialect, MysqlDialect)
+   * @returns Compiled query with SQL string and parameters
+   */
+  compile(dialect: Dialect): CompiledQuery {
+    const queryCompiler = dialect.createQueryCompiler();
+
+    return this.#kyselyBuilder.compile({
+      getExecutor(): QueryExecutor {
+        return {
+          transformQuery(node, _queryId) {
+            return node;
+          },
+          compileQuery(node, queryId) {
+            return queryCompiler.compileQuery(node, queryId) as KyselyCompiledQuery;
+          },
+        } as QueryExecutor;
+      },
+    });
+  }
+}
+
+/**
+ * Tagged template function for building SQL queries with parameters.
+ * Wraps Kysely's sql function to provide a compile() method without arguments.
+ *
+ * @example
+ * ```ts
+ * const userId = 123;
+ * const query = sql`SELECT * FROM users WHERE id = ${userId}`;
+ * const compiled = query.compile();
+ * ```
+ */
+export function sql(strings: TemplateStringsArray, ...values: unknown[]): RawBuilder {
+  const kyselyBuilder = kyselySql(strings, ...values);
+  return new RawBuilder(kyselyBuilder);
+}

package/src/sql-driver/sqlocal.test.ts

@@ -0,0 +1,117 @@
+import { SQLocalKysely } from "sqlocal/kysely";
+import { describe, expect, it } from "vitest";
+import { SqlDriverAdapter } from "./sql-driver-adapter";
+import type { GenericSQLPlugin } from "./query-executor/plugin";
+import { sql } from "./sql";
+
+describe("SQLocal", () => {
+  it("should create a new SQLocal instance", async () => {
+    const { dialect } = new SQLocalKysely(":memory:");
+
+    const adapter = new SqlDriverAdapter(dialect);
+
+    const query = sql`SELECT 5`.compile(dialect);
+    const result = await adapter.executeQuery(query);
+    expect(result.rows).toEqual([{ 5: 5 }]);
+
+    console.log(result);
+
+    await adapter.destroy();
+  });
+
+  it("should execute queries in a transaction and commit", async () => {
+    const { dialect } = new SQLocalKysely(":memory:");
+    const adapter = new SqlDriverAdapter(dialect);
+
+    // Create table outside transaction
+    await adapter.executeQuery(sql`CREATE TABLE test (id INTEGER, name TEXT)`.compile(dialect));
+
+    // Execute queries in transaction
+    await adapter.transaction(async (trx) => {
+      await trx.executeQuery(sql`INSERT INTO test VALUES (1, 'Alice')`.compile(dialect));
+      await trx.executeQuery(sql`INSERT INTO test VALUES (2, 'Bob')`.compile(dialect));
+    });
+
+    // Verify data persists after transaction
+    const result = await adapter.executeQuery(sql`SELECT * FROM test ORDER BY id`.compile(dialect));
+    expect(result.rows).toEqual([
+      { id: 1, name: "Alice" },
+      { id: 2, name: "Bob" },
+    ]);
+
+    await adapter.destroy();
+  });
+
+  it("should rollback transaction on error", async () => {
+    const { dialect } = new SQLocalKysely(":memory:");
+    const adapter = new SqlDriverAdapter(dialect);
+
+    // Create table outside transaction
+    await adapter.executeQuery(
+      sql`CREATE TABLE test (id INTEGER PRIMARY KEY, name TEXT)`.compile(dialect),
+    );
+    await adapter.executeQuery(sql`INSERT INTO test VALUES (1, 'Alice')`.compile(dialect));
+
+    // Try to execute queries in transaction with duplicate key error
+    await expect(
+      adapter.transaction(async (trx) => {
+        await trx.executeQuery(sql`INSERT INTO test VALUES (2, 'Bob')`.compile(dialect));
+        // This should cause a duplicate key error
+        await trx.executeQuery(sql`INSERT INTO test VALUES (1, 'Charlie')`.compile(dialect));
+      }),
+    ).rejects.toThrow();
+
+    // Verify Bob was not inserted (transaction rolled back)
+    const result = await adapter.executeQuery(sql`SELECT * FROM test ORDER BY id`.compile(dialect));
+    expect(result.rows).toEqual([{ id: 1, name: "Alice" }]);
+
+    await adapter.destroy();
+  });
+
+  it("should support plugins that transform results", async () => {
+    const { dialect } = new SQLocalKysely(":memory:");
+    const adapter = new SqlDriverAdapter(dialect);
+
+    // Create a plugin that adds metadata to results
+    const metadataPlugin: GenericSQLPlugin = {
+      async transformResult({ result }) {
+        return {
+          ...result,
+          rows: result.rows.map((row) => ({
+            ...(row as object),
+            __metadata: { transformed: true },
+          })),
+        };
+      },
+    };
+
+    const adapterWithPlugin = adapter.withPlugin(metadataPlugin);
+
+    await adapterWithPlugin.executeQuery(
+      sql`CREATE TABLE test (id INTEGER, name TEXT)`.compile(dialect),
+    );
+    await adapterWithPlugin.executeQuery(
+      sql`INSERT INTO test VALUES (1, 'Alice')`.compile(dialect),
+    );
+
+    const result = await adapterWithPlugin.executeQuery(sql`SELECT * FROM test`.compile(dialect));
+
+    expect(result.rows).toEqual([{ id: 1, name: "Alice", __metadata: { transformed: true } }]);
+
+    await adapter.destroy();
+  });
+
+  it("should properly destroy and release resources", async () => {
+    const { dialect } = new SQLocalKysely(":memory:");
+    const adapter = new SqlDriverAdapter(dialect);
+
+    // Execute some queries
+    await adapter.executeQuery(sql`SELECT 1`.compile(dialect));
+
+    // Destroy should complete without error
+    await expect(adapter.destroy()).resolves.toBeUndefined();
+
+    // Multiple destroy calls should be safe
+    await expect(adapter.destroy()).resolves.toBeUndefined();
+  });
+});

package/src/with-database.ts

@@ -1,5 +1,9 @@
 import type { AnySchema } from "./schema/create";
-import type { RequestThisContext, FragnoPublicConfig } from "@fragno-dev/core";
+import type {
+  RequestThisContext,
+  FragnoPublicConfig,
+  AnyFragnoInstantiatedFragment,
+} from "@fragno-dev/core";
 import { FragmentDefinitionBuilder, instantiate } from "@fragno-dev/core";
 import {
   DatabaseFragmentDefinitionBuilder,
@@ -9,7 +13,8 @@ import {
   type FragnoPublicConfigWithDatabase,
   type DatabaseRequestStorage,
 } from "./db-fragment-definition-builder";
-import { internalFragmentDef } from "./fragments/internal-fragment";
+import { internalFragmentDef, type InternalFragmentInstance } from "./fragments/internal-fragment";
+import type { HooksMap } from "./hooks/hooks";
 
 /**
  * Helper to add database support to a fragment builder.
@@ -42,6 +47,7 @@ export function withDatabase<TSchema extends AnySchema>(
   TServiceThisContext extends RequestThisContext,
   THandlerThisContext extends RequestThisContext,
   TRequestStorage,
+  TLinkedFragments extends Record<string, AnyFragnoInstantiatedFragment>,
 >(
   builder: FragmentDefinitionBuilder<
     TConfig,
@@ -53,7 +59,8 @@ export function withDatabase<TSchema extends AnySchema>(
     TPrivateServices,
     TServiceThisContext,
     THandlerThisContext,
-    TRequestStorage
+    TRequestStorage,
+    TLinkedFragments
   >,
 ) => DatabaseFragmentDefinitionBuilder<
   TSchema,
@@ -63,8 +70,10 @@ export function withDatabase<TSchema extends AnySchema>(
   TServices,
   TServiceDeps,
   TPrivateServices,
-  DatabaseServiceContext,
-  DatabaseHandlerContext
+  HooksMap,
+  DatabaseServiceContext<HooksMap>,
+  DatabaseHandlerContext,
+  TLinkedFragments & { _fragno_internal: InternalFragmentInstance }
 > {
   return <
     TConfig,
@@ -76,6 +85,7 @@ export function withDatabase<TSchema extends AnySchema>(
     TServiceThisContext extends RequestThisContext,
     THandlerThisContext extends RequestThisContext,
     TRequestStorage,
+    TLinkedFragments extends Record<string, AnyFragnoInstantiatedFragment>,
   >(
     builder: FragmentDefinitionBuilder<
       TConfig,
@@ -87,22 +97,21 @@ export function withDatabase<TSchema extends AnySchema>(
       TPrivateServices,
       TServiceThisContext,
       THandlerThisContext,
-      TRequestStorage
+      TRequestStorage,
+      TLinkedFragments
     >,
   ) => {
-    // Link the internal fragment to all database fragments (except the internal fragment itself)
-    // No circular dependency: internal-fragment.ts uses DatabaseFragmentDefinitionBuilder directly
-    const isInternalFragment = builder.name === "$fragno-internal-fragment";
-    const builderWithInternal = isInternalFragment
-      ? builder
-      : builder.withLinkedFragment("_fragno_internal", ({ config, options }) => {
-          // Cast is safe: by the time this callback is invoked during fragment instantiation,
-          // the options will be FragnoPublicConfigWithDatabase (enforced by DatabaseFragmentDefinitionBuilder)
-          return instantiate(internalFragmentDef)
-            .withConfig(config as {})
-            .withOptions(options as FragnoPublicConfigWithDatabase)
-            .build();
-        });
+    const builderWithInternal = builder.withLinkedFragment(
+      "_fragno_internal",
+      ({ config, options }) => {
+        // Cast is safe: by the time this callback is invoked during fragment instantiation,
+        // the options will be FragnoPublicConfigWithDatabase (enforced by DatabaseFragmentDefinitionBuilder)
+        return instantiate(internalFragmentDef)
+          .withConfig(config as {})
+          .withOptions(options as FragnoPublicConfigWithDatabase)
+          .build();
+      },
+    );
 
     // Cast is safe: we're creating a DatabaseFragmentDefinitionBuilder which internally uses
     // FragnoPublicConfigWithDatabase, but the input builder uses FragnoPublicConfig.
@@ -118,8 +127,10 @@ export function withDatabase<TSchema extends AnySchema>(
       TServices,
       TServiceDeps,
       TPrivateServices,
-      DatabaseServiceContext,
-      DatabaseHandlerContext
+      {}, // Start with empty hooks, provideHooks() will update this
+      DatabaseServiceContext<{}>,
+      DatabaseHandlerContext,
+      TLinkedFragments & { _fragno_internal: InternalFragmentInstance }
     >(
       builderWithInternal as unknown as FragmentDefinitionBuilder<
         TConfig,
@@ -129,9 +140,10 @@ export function withDatabase<TSchema extends AnySchema>(
         TServices,
         TServiceDeps,
         TPrivateServices,
-        DatabaseServiceContext,
+        DatabaseServiceContext<{}>,
         DatabaseHandlerContext,
-        DatabaseRequestStorage
+        DatabaseRequestStorage,
+        TLinkedFragments & { _fragno_internal: InternalFragmentInstance }
       >,
       schema,
       namespace,

package/tsdown.config.ts

@@ -5,12 +5,17 @@ export default defineConfig({
     "./src/mod.ts",
     "./src/id.ts",
     "./src/schema/create.ts",
-    "./src/query/query.ts",
+    "./src/query/simple-query-interface.ts",
     "./src/adapters/adapters.ts",
     "./src/adapters/kysely/kysely-adapter.ts",
     "./src/adapters/drizzle/drizzle-adapter.ts",
     "./src/adapters/drizzle/generate.ts",
-    "./src/query/unit-of-work.ts",
+    "./src/adapters/generic-sql/generic-sql-adapter.ts",
+    "./src/adapters/generic-sql/driver-config.ts",
+    "./src/sql-driver/sql-driver.ts",
+    "./src/sql-driver/dialects/dialects.ts",
+    "./src/sql-driver/dialects/durable-object-dialect.ts",
+    "./src/query/unit-of-work/unit-of-work.ts",
     "./src/query/cursor.ts",
     "./src/fragment.ts",
     "./src/db-fragment-definition-builder.ts",

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

@@ -1,40 +0,0 @@
-//#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

@@ -1 +0,0 @@
-{"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.d.ts

@@ -1,23 +0,0 @@
-import "./shared.js";
-import { DrizzleCompiledQuery } from "./drizzle-uow-compiler.js";
-
-//#region src/adapters/drizzle/drizzle-query.d.ts
-
-/**
- * Configuration options for creating a Drizzle Unit of Work
- */
-interface DrizzleUOWConfig {
-  /**
-   * Optional callback to receive compiled SQL queries for logging/debugging
-   * This callback is invoked for each query as it's compiled
-   */
-  onQuery?: (query: DrizzleCompiledQuery) => void;
-  /**
-   * If true, the query will not be executed and the query will be returned. Not respected for UOWs
-   * since those have to be manually executed.
-   */
-  dryRun?: boolean;
-}
-//#endregion
-export { DrizzleUOWConfig };
-//# sourceMappingURL=drizzle-query.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"drizzle-query.d.ts","names":[],"sources":["../../../src/adapters/drizzle/drizzle-query.ts"],"sourcesContent":[],"mappings":";;;;;;;;UAeiB,gBAAA;;;;;oBAKG"}

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

@@ -1 +0,0 @@
-{"version":3,"file":"drizzle-query.js","names":["#indexName","#condition","#setValues","executor: UOWExecutor<DrizzleCompiledQuery, DrizzleResult>"],"sources":["../../../src/adapters/drizzle/drizzle-query.ts"],"sourcesContent":["import type { AbstractQuery } from \"../../query/query\";\nimport type { AnySchema, AnyTable } from \"../../schema/create\";\nimport type { CompiledMutation, UOWExecutor, ValidIndexName } 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\";\nimport type { TableToUpdateValues } from \"../../query/query\";\nimport type { CursorResult } from \"../../query/cursor\";\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   * If true, the query will not be executed and the query will be returned. Not respected for UOWs\n   * since those have to be manually executed.\n   */\n  dryRun?: boolean;\n}\n\n/**\n * Special builder for updateMany operations that captures configuration\n */\nclass UpdateManySpecialBuilder<TTable extends AnyTable> {\n  #indexName?: string;\n  #condition?: unknown;\n  #setValues?: TableToUpdateValues<TTable>;\n\n  whereIndex<TIndexName extends ValidIndexName<TTable>>(\n    indexName: TIndexName,\n    condition?: unknown,\n  ): this {\n    this.#indexName = indexName as string;\n    this.#condition = condition;\n    return this;\n  }\n\n  set(values: TableToUpdateValues<TTable>): this {\n    this.#setValues = values;\n    return this;\n  }\n\n  getConfig() {\n    return {\n      indexName: this.#indexName,\n      condition: this.#condition,\n      setValues: this.#setValues,\n    };\n  }\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 * @param uowConfig - Optional UOW configuration\n * @param schemaNamespaceMap - Optional WeakMap for schema-to-namespace lookups\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  uowConfig?: DrizzleUOWConfig,\n  schemaNamespaceMap?: WeakMap<AnySchema, string>,\n): AbstractQuery<T, DrizzleUOWConfig> {\n  function createUOW(opts: { name?: string; config?: DrizzleUOWConfig }) {\n    const uowCompiler = createDrizzleUOWCompiler(pool, provider, mapper);\n\n    const executor: UOWExecutor<DrizzleCompiledQuery, DrizzleResult> = {\n      async executeRetrievalPhase(retrievalBatch: DrizzleCompiledQuery[]) {\n        // In dryRun mode, skip execution and return empty results\n        if (opts.config?.dryRun) {\n          return retrievalBatch.map(() => ({\n            rows: [],\n            affectedRows: 0,\n          }));\n        }\n\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        // In dryRun mode, skip execution and return success with mock internal IDs\n        if (opts.config?.dryRun) {\n          return {\n            success: true,\n            createdInternalIds: mutationBatch.map(() => null),\n          };\n        }\n\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(provider);\n\n    const { onQuery, ...restUowConfig } = opts.config ?? {};\n\n    return new UnitOfWork(\n      uowCompiler,\n      executor,\n      decoder,\n      opts.name,\n      {\n        ...restUowConfig,\n        onQuery: (query) => {\n          // Handle both CompiledQuery and CompiledMutation structures\n          // Retrieval operations return DrizzleCompiledQuery directly: { sql, params }\n          // Mutation operations return CompiledMutation: { query: DrizzleCompiledQuery, expectedAffectedRows }\n          const actualQuery =\n            query && typeof query === \"object\" && \"query\" in query\n              ? (query as CompiledMutation<DrizzleCompiledQuery>).query\n              : (query as DrizzleCompiledQuery);\n\n          opts.config?.onQuery?.(actualQuery);\n        },\n      },\n      schemaNamespaceMap,\n    ).forSchema(schema);\n  }\n\n  return {\n    async find(tableName, builderFn) {\n      // Safe: builderFn returns a FindBuilder (or void), which matches UnitOfWork signature\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      const uow = createUOW({ config: uowConfig }).find(tableName, builderFn as any);\n      const [result] = await uow.executeRetrieve();\n      return result;\n    },\n\n    async findWithCursor(tableName, builderFn) {\n      // Safe: builderFn returns a FindBuilder, which matches UnitOfWork signature\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      const uow = createUOW({ config: uowConfig }).findWithCursor(tableName, builderFn as any);\n      const [result] = await uow.executeRetrieve();\n      return result as CursorResult<unknown>;\n    },\n\n    async findFirst(tableName, builderFn) {\n      const uow = createUOW({ config: uowConfig });\n      if (builderFn) {\n        uow.find(tableName, (b) => {\n          builderFn(b);\n          return b.pageSize(1);\n        });\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({ config: uowConfig });\n      uow.create(tableName, values);\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({ config: uowConfig });\n      for (const values of valuesArray) {\n        uow.create(tableName, values);\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({ config: uowConfig });\n      uow.update(tableName, id, builderFn);\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      const table = schema.tables[tableName];\n      if (!table) {\n        throw new Error(`Table ${tableName} not found in schema`);\n      }\n\n      const specialBuilder = new UpdateManySpecialBuilder<typeof table>();\n      builderFn(specialBuilder);\n\n      const { indexName, condition, setValues } = specialBuilder.getConfig();\n\n      if (!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({ config: uowConfig });\n      findUow.find(tableName, (b) => {\n        if (condition) {\n          // Safe: condition is captured from whereIndex call with proper typing\n          // eslint-disable-next-line @typescript-eslint/no-explicit-any\n          return b.whereIndex(indexName as ValidIndexName<typeof table>, condition as any);\n        }\n        return b.whereIndex(indexName as ValidIndexName<typeof table>);\n      });\n      const [records]: unknown[][] = await findUow.executeRetrieve();\n\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const updateUow = createUOW({ config: uowConfig });\n      for (const record of records as Array<{ id: unknown }>) {\n        updateUow.update(tableName, record.id as string, (b) => b.set(setValues));\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({ config: uowConfig });\n      uow.delete(tableName, id, builderFn);\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      const findUow = createUOW({ config: uowConfig });\n      findUow.find(tableName, builderFn);\n      const [records]: unknown[][] = await findUow.executeRetrieve();\n\n      if (!records || records.length === 0) {\n        return;\n      }\n\n      const deleteUow = createUOW({ config: uowConfig });\n      for (const record of records as Array<{ id: unknown }>) {\n        deleteUow.delete(tableName, 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, nestedUowConfig) {\n      return createUOW({\n        name,\n        config: {\n          ...uowConfig,\n          ...nestedUowConfig,\n        },\n      });\n    },\n  } as AbstractQuery<T, DrizzleUOWConfig>;\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAM,2BAAN,MAAwD;CACtD;CACA;CACA;CAEA,WACE,WACA,WACM;AACN,QAAKA,YAAa;AAClB,QAAKC,YAAa;AAClB,SAAO;;CAGT,IAAI,QAA2C;AAC7C,QAAKC,YAAa;AAClB,SAAO;;CAGT,YAAY;AACV,SAAO;GACL,WAAW,MAAKF;GAChB,WAAW,MAAKC;GAChB,WAAW,MAAKC;GACjB;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BL,SAAgB,YACd,QACA,MACA,UACA,QACA,WACA,oBACoC;CACpC,SAAS,UAAU,MAAoD;EACrE,MAAM,cAAc,yBAAyB,MAAM,UAAU,OAAO;EAEpE,MAAMC,WAA6D;GACjE,MAAM,sBAAsB,gBAAwC;AAElE,QAAI,KAAK,QAAQ,OACf,QAAO,eAAe,WAAW;KAC/B,MAAM,EAAE;KACR,cAAc;KACf,EAAE;IAGL,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;AAElF,QAAI,KAAK,QAAQ,OACf,QAAO;KACL,SAAS;KACT,oBAAoB,cAAc,UAAU,KAAK;KAClD;IAGH,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;EAED,MAAM,UAAU,wBAAwB,SAAS;EAEjD,MAAM,EAAE,QAAS,GAAG,kBAAkB,KAAK,UAAU,EAAE;AAEvD,SAAO,IAAI,WACT,aACA,UACA,SACA,KAAK,MACL;GACE,GAAG;GACH,UAAU,UAAU;IAIlB,MAAM,cACJ,SAAS,OAAO,UAAU,YAAY,WAAW,QAC5C,MAAiD,QACjD;AAEP,SAAK,QAAQ,UAAU,YAAY;;GAEtC,EACD,mBACD,CAAC,UAAU,OAAO;;AAGrB,QAAO;EACL,MAAM,KAAK,WAAW,WAAW;GAI/B,MAAM,CAAC,UAAU,MADL,UAAU,EAAE,QAAQ,WAAW,CAAC,CAAC,KAAK,WAAW,UAAiB,CACnD,iBAAiB;AAC5C,UAAO;;EAGT,MAAM,eAAe,WAAW,WAAW;GAIzC,MAAM,CAAC,UAAU,MADL,UAAU,EAAE,QAAQ,WAAW,CAAC,CAAC,eAAe,WAAW,UAAiB,CAC7D,iBAAiB;AAC5C,UAAO;;EAGT,MAAM,UAAU,WAAW,WAAW;GACpC,MAAM,MAAM,UAAU,EAAE,QAAQ,WAAW,CAAC;AAC5C,OAAI,UACF,KAAI,KAAK,YAAY,MAAM;AACzB,cAAU,EAAE;AACZ,WAAO,EAAE,SAAS,EAAE;KACpB;OAEF,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,UAAU,EAAE,QAAQ,WAAW,CAAC;AAC5C,OAAI,OAAO,WAAW,OAAO;GAC7B,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,UAAU,EAAE,QAAQ,WAAW,CAAC;AAC5C,QAAK,MAAM,UAAU,YACnB,KAAI,OAAO,WAAW,OAAO;GAE/B,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,UAAU,EAAE,QAAQ,WAAW,CAAC;AAC5C,OAAI,OAAO,WAAW,IAAI,UAAU;GACpC,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;AAErC,OAAI,CADU,OAAO,OAAO,WAE1B,OAAM,IAAI,MAAM,SAAS,UAAU,sBAAsB;GAG3D,MAAM,iBAAiB,IAAI,0BAAwC;AACnE,aAAU,eAAe;GAEzB,MAAM,EAAE,WAAW,WAAW,cAAc,eAAe,WAAW;AAEtE,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,4CAA4C;AAE9D,OAAI,CAAC,UACH,OAAM,IAAI,MAAM,qCAAqC;GAGvD,MAAM,UAAU,UAAU,EAAE,QAAQ,WAAW,CAAC;AAChD,WAAQ,KAAK,YAAY,MAAM;AAC7B,QAAI,UAGF,QAAO,EAAE,WAAW,WAA2C,UAAiB;AAElF,WAAO,EAAE,WAAW,UAA0C;KAC9D;GACF,MAAM,CAAC,WAAwB,MAAM,QAAQ,iBAAiB;AAE9D,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,UAAU,EAAE,QAAQ,WAAW,CAAC;AAClD,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAW,OAAO,KAAe,MAAM,EAAE,IAAI,UAAU,CAAC;GAE3E,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,MAAM,OAAO,WAAW,IAAI,WAAW;GACrC,MAAM,MAAM,UAAU,EAAE,QAAQ,WAAW,CAAC;AAC5C,OAAI,OAAO,WAAW,IAAI,UAAU;GACpC,MAAM,EAAE,YAAY,MAAM,IAAI,kBAAkB;AAChD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,iEAAiE;;EAIrF,MAAM,WAAW,WAAW,WAAW;GACrC,MAAM,UAAU,UAAU,EAAE,QAAQ,WAAW,CAAC;AAChD,WAAQ,KAAK,WAAW,UAAU;GAClC,MAAM,CAAC,WAAwB,MAAM,QAAQ,iBAAiB;AAE9D,OAAI,CAAC,WAAW,QAAQ,WAAW,EACjC;GAGF,MAAM,YAAY,UAAU,EAAE,QAAQ,WAAW,CAAC;AAClD,QAAK,MAAM,UAAU,QACnB,WAAU,OAAO,WAAW,OAAO,GAAa;GAElD,MAAM,EAAE,YAAY,MAAM,UAAU,kBAAkB;AACtD,OAAI,CAAC,QACH,OAAM,IAAI,MAAM,8CAA8C;;EAIlE,iBAAiB,MAAM,iBAAiB;AACtC,UAAO,UAAU;IACf;IACA,QAAQ;KACN,GAAG;KACH,GAAG;KACJ;IACF,CAAC;;EAEL"}

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

@@ -1,10 +0,0 @@
-import "./shared.js";
-
-//#region src/adapters/drizzle/drizzle-uow-compiler.d.ts
-type DrizzleCompiledQuery = {
-  sql: string;
-  params: unknown[];
-};
-//#endregion
-export { DrizzleCompiledQuery };
-//# sourceMappingURL=drizzle-uow-compiler.d.ts.map

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

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