@woltz/rich-domain-drizzle 0.1.0

package/dist/esm/query-builder.js

@@ -0,0 +1,142 @@
+import { eq, ne, gt, gte, lt, lte, ilike, like, inArray, notInArray, isNull, isNotNull, between, and, or, asc, desc, } from "drizzle-orm";
+import { DrizzleAdapterError } from "./errors";
+const OPERATOR_MAP = {
+    equals: (col, val) => eq(col, val),
+    notEquals: (col, val) => ne(col, val),
+    greaterThan: (col, val) => gt(col, val),
+    greaterThanOrEqual: (col, val) => gte(col, val),
+    lessThan: (col, val) => lt(col, val),
+    lessThanOrEqual: (col, val) => lte(col, val),
+    contains: (col, val) => ilike(col, `%${val}%`),
+    startsWith: (col, val) => ilike(col, `${val}%`),
+    endsWith: (col, val) => ilike(col, `%${val}`),
+    in: (col, val) => inArray(col, val),
+    notIn: (col, val) => notInArray(col, val),
+    isNull: (col) => isNull(col),
+    isNotNull: (col) => isNotNull(col),
+    between: (col, val) => between(col, val[0], val[1]),
+};
+export class DrizzleQueryBuilder {
+    /**
+     * Apply Criteria to a Drizzle select query.
+     *
+     * @param criteria - The Criteria instance with filters, orders, pagination, search
+     * @param table - The Drizzle table object (e.g., usersTable)
+     * @param searchableFields - Fields to search when criteria.hasSearch()
+     * @returns Object with { where, orderBy, limit, offset } ready for Drizzle query
+     */
+    static apply(criteria, table, searchableFields) {
+        const result = {};
+        // Build filter conditions
+        const filterConditions = [];
+        for (const filter of criteria.getFilters()) {
+            if (filter.field.includes(".")) {
+                throw new DrizzleAdapterError(`Filtering on relation field "${filter.field}" is not supported. ` +
+                    `Drizzle requires explicit JOINs to filter across relations. ` +
+                    `Add a custom method to your repository using the Drizzle SQL API instead.`);
+            }
+            if (filter.options?.quantifier) {
+                throw new DrizzleAdapterError(`Quantifier "${filter.options.quantifier}" on field "${filter.field}" is not supported. ` +
+                    `Drizzle does not have a built-in some/every/none operator. ` +
+                    `Add a custom method to your repository using EXISTS subqueries instead.`);
+            }
+            const col = DrizzleQueryBuilder.resolveColumn(table, filter.field);
+            if (col === undefined) {
+                throw new DrizzleAdapterError(`Column "${filter.field}" not found on table. ` +
+                    `Available columns: ${Object.keys(table)
+                        .filter((k) => typeof table[k] === "object")
+                        .join(", ")}`);
+            }
+            const operatorFn = OPERATOR_MAP[filter.operator];
+            if (!operatorFn) {
+                throw new DrizzleAdapterError(`Operator "${filter.operator}" is not supported by the Drizzle adapter.`);
+            }
+            filterConditions.push(operatorFn(col, filter.value));
+        }
+        // Build search conditions
+        let searchCondition;
+        if (criteria.hasSearch() &&
+            searchableFields &&
+            searchableFields.length > 0) {
+            const search = criteria.getSearch();
+            const searchConditions = [];
+            for (const searchField of searchableFields) {
+                let fieldName;
+                let caseSensitive = false;
+                if (typeof searchField === "object" && "field" in searchField) {
+                    fieldName = searchField.field;
+                    caseSensitive = searchField.caseSensitive ?? false;
+                }
+                else {
+                    fieldName = String(searchField);
+                }
+                if (fieldName.includes(".")) {
+                    throw new DrizzleAdapterError(`Search field "${fieldName}" references a relation. ` +
+                        `Drizzle does not support cross-table search via Criteria. ` +
+                        `Use only top-level columns in getSearchableFields().`);
+                }
+                const col = DrizzleQueryBuilder.resolveColumn(table, fieldName);
+                if (col === undefined) {
+                    throw new DrizzleAdapterError(`Search field "${fieldName}" not found on table. ` +
+                        `Available columns: ${Object.keys(table)
+                            .filter((k) => typeof table[k] === "object")
+                            .join(", ")}`);
+                }
+                if (caseSensitive) {
+                    searchConditions.push(like(col, `%${search}%`));
+                }
+                else {
+                    searchConditions.push(ilike(col, `%${search}%`));
+                }
+            }
+            if (searchConditions.length > 0) {
+                searchCondition =
+                    searchConditions.length === 1
+                        ? searchConditions[0]
+                        : or(...searchConditions);
+            }
+        }
+        // Combine filter and search conditions
+        if (filterConditions.length > 0 || searchCondition) {
+            const allConditions = [...filterConditions];
+            if (searchCondition)
+                allConditions.push(searchCondition);
+            if (allConditions.length === 1) {
+                result.where = allConditions[0];
+            }
+            else {
+                result.where = and(...allConditions);
+            }
+        }
+        // Build order conditions
+        const orders = criteria.getOrders();
+        if (orders.length > 0) {
+            result.orderBy = orders.map((o) => {
+                if (o.field.includes(".")) {
+                    throw new DrizzleAdapterError(`Ordering by relation field "${o.field}" is not supported. ` +
+                        `Drizzle requires explicit JOINs to order across relations. ` +
+                        `Add a custom method to your repository using the Drizzle SQL API instead.`);
+                }
+                const col = DrizzleQueryBuilder.resolveColumn(table, o.field);
+                if (col === undefined) {
+                    throw new DrizzleAdapterError(`Order field "${o.field}" not found on table. ` +
+                        `Available columns: ${Object.keys(table)
+                            .filter((k) => typeof table[k] === "object")
+                            .join(", ")}`);
+                }
+                return o.direction === "desc" ? desc(col) : asc(col);
+            });
+        }
+        // Build pagination
+        const pagination = criteria.getPagination();
+        if (pagination) {
+            result.limit = pagination.limit;
+            result.offset = pagination.offset;
+        }
+        return result;
+    }
+    static resolveColumn(table, fieldPath) {
+        return table[fieldPath];
+    }
+}
+//# sourceMappingURL=query-builder.js.map

package/dist/esm/query-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-builder.js","sourceRoot":"","sources":["../../src/query-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,EAAE,EACF,EAAE,EACF,EAAE,EACF,GAAG,EACH,EAAE,EACF,GAAG,EACH,KAAK,EACL,IAAI,EACJ,OAAO,EACP,UAAU,EACV,MAAM,EACN,SAAS,EACT,OAAO,EACP,GAAG,EACH,EAAE,EACF,GAAG,EACH,IAAI,GAEL,MAAM,aAAa,CAAC;AAErB,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAQ/C,MAAM,YAAY,GAA+B;IAC/C,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IAClC,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IACrC,WAAW,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IACvC,kBAAkB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC;IAC/C,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC;IACpC,eAAe,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC;IAC5C,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,GAAG,GAAG,CAAC;IAC9C,UAAU,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,CAAC;IAC/C,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,GAAG,EAAE,CAAC;IAC7C,EAAE,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;IACnC,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC;IACzC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC;IAC5B,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC;IAClC,OAAO,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;CACpD,CAAC;AAEF,MAAM,OAAO,mBAAmB;IAC9B;;;;;;;OAOG;IACH,MAAM,CAAC,KAAK,CACV,QAAqB,EACrB,KAAU,EACV,gBAAyC;QAOzC,MAAM,MAAM,GAKR,EAAE,CAAC;QAEP,0BAA0B;QAC1B,MAAM,gBAAgB,GAAU,EAAE,CAAC;QACnC,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/B,MAAM,IAAI,mBAAmB,CAC3B,gCAAgC,MAAM,CAAC,KAAK,sBAAsB;oBAChE,8DAA8D;oBAC9D,2EAA2E,CAC9E,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,CAAC;gBAC/B,MAAM,IAAI,mBAAmB,CAC3B,eAAe,MAAM,CAAC,OAAO,CAAC,UAAU,eAAe,MAAM,CAAC,KAAK,sBAAsB;oBACvF,6DAA6D;oBAC7D,yEAAyE,CAC5E,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,mBAAmB,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACnE,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;gBACtB,MAAM,IAAI,mBAAmB,CAC3B,WAAW,MAAM,CAAC,KAAK,wBAAwB;oBAC7C,sBAAsB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;yBACrC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;yBAC3C,IAAI,CAAC,IAAI,CAAC,EAAE,CAClB,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,MAAM,IAAI,mBAAmB,CAC3B,aAAa,MAAM,CAAC,QAAQ,4CAA4C,CACzE,CAAC;YACJ,CAAC;YAED,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACvD,CAAC;QAED,0BAA0B;QAC1B,IAAI,eAAgC,CAAC;QACrC,IACE,QAAQ,CAAC,SAAS,EAAE;YACpB,gBAAgB;YAChB,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAC3B,CAAC;YACD,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAG,CAAC;YACrC,MAAM,gBAAgB,GAAU,EAAE,CAAC;YAEnC,KAAK,MAAM,WAAW,IAAI,gBAAgB,EAAE,CAAC;gBAC3C,IAAI,SAAiB,CAAC;gBACtB,IAAI,aAAa,GAAG,KAAK,CAAC;gBAE1B,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;oBAC9D,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC;oBAC9B,aAAa,GAAG,WAAW,CAAC,aAAa,IAAI,KAAK,CAAC;gBACrD,CAAC;qBAAM,CAAC;oBACN,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;gBAClC,CAAC;gBAED,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC5B,MAAM,IAAI,mBAAmB,CAC3B,iBAAiB,SAAS,2BAA2B;wBACnD,4DAA4D;wBAC5D,sDAAsD,CACzD,CAAC;gBACJ,CAAC;gBAED,MAAM,GAAG,GAAG,mBAAmB,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;gBAChE,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,MAAM,IAAI,mBAAmB,CAC3B,iBAAiB,SAAS,wBAAwB;wBAChD,sBAAsB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;6BACrC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;6BAC3C,IAAI,CAAC,IAAI,CAAC,EAAE,CAClB,CAAC;gBACJ,CAAC;gBAED,IAAI,aAAa,EAAE,CAAC;oBAClB,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC;gBAClD,CAAC;qBAAM,CAAC;oBACN,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YAED,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChC,eAAe;oBACb,gBAAgB,CAAC,MAAM,KAAK,CAAC;wBAC3B,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;wBACrB,CAAC,CAAC,EAAE,CAAC,GAAG,gBAAgB,CAAE,CAAC;YACjC,CAAC;QACH,CAAC;QAED,uCAAuC;QACvC,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,IAAI,eAAe,EAAE,CAAC;YACnD,MAAM,aAAa,GAAU,CAAC,GAAG,gBAAgB,CAAC,CAAC;YACnD,IAAI,eAAe;gBAAE,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAEzD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/B,MAAM,CAAC,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,aAAa,CAAE,CAAC;YACxC,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBAChC,IAAI,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,mBAAmB,CAC3B,+BAA+B,CAAC,CAAC,KAAK,sBAAsB;wBAC1D,6DAA6D;wBAC7D,2EAA2E,CAC9E,CAAC;gBACJ,CAAC;gBAED,MAAM,GAAG,GAAG,mBAAmB,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,MAAM,IAAI,mBAAmB,CAC3B,gBAAgB,CAAC,CAAC,KAAK,wBAAwB;wBAC7C,sBAAsB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;6BACrC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;6BAC3C,IAAI,CAAC,IAAI,CAAC,EAAE,CAClB,CAAC;gBACJ,CAAC;gBAED,OAAO,CAAC,CAAC,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;QACL,CAAC;QAED,mBAAmB;QACnB,MAAM,UAAU,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC5C,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;YAChC,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QACpC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,KAAU,EAAE,SAAiB;QAChD,OAAO,KAAK,CAAC,SAAS,CAAC,CAAC;IAC1B,CAAC;CACF"}

package/dist/esm/repository.d.ts

@@ -0,0 +1,46 @@
+import { Aggregate, Repository, Mapper, Criteria, PaginatedResult } from "@woltz/rich-domain";
+import { DrizzleClient, DrizzleUnitOfWork } from "./unit-of-work";
+import { DrizzleToPersistence } from "./mappers/to-persistence";
+import { SearchableField } from "./query-builder";
+export interface DrizzleRepositoryConfig<TDomain, TPersistence> {
+    db: DrizzleClient;
+    table: any;
+    toDomainMapper: Mapper<TPersistence, TDomain>;
+    toPersistenceMapper: DrizzleToPersistence<TDomain>;
+    uow: DrizzleUnitOfWork;
+}
+export declare abstract class DrizzleRepository<TDomain extends Aggregate<any>, TPersistence> extends Repository<TDomain> {
+    protected readonly db: DrizzleClient;
+    protected readonly table: any;
+    protected readonly toDomainMapper: Mapper<TPersistence, TDomain>;
+    protected readonly toPersistenceMapper: DrizzleToPersistence<TDomain>;
+    protected readonly uow: DrizzleUnitOfWork;
+    constructor(config: DrizzleRepositoryConfig<TDomain, TPersistence>);
+    /**
+     * Returns tx from UOWStorage if inside a transaction, otherwise the raw db.
+     */
+    protected get context(): DrizzleClient;
+    /**
+     * The table name string used by EntitySchemaRegistry and db.query accessor.
+     */
+    protected abstract get model(): string;
+    /**
+     * Search conditions for full-text search via Criteria.search().
+     */
+    protected abstract getSearchableFields(): SearchableField<TPersistence>[];
+    /**
+     * Relations to include when fetching (for Drizzle relational query API).
+     */
+    protected getDefaultRelations(): Record<string, any>;
+    find(criteria: Criteria<TDomain>): Promise<PaginatedResult<TDomain>>;
+    findById(id: string): Promise<TDomain | null>;
+    findManyByIds(ids: string[]): Promise<TDomain[]>;
+    count(criteria?: Criteria<TDomain>): Promise<number>;
+    exists(id: string): Promise<boolean>;
+    save(entity: TDomain): Promise<void>;
+    delete(entity: TDomain): Promise<void>;
+    deleteById(id: string): Promise<void>;
+    transaction<T>(work: () => Promise<T>): Promise<T>;
+    private markArrayOfAggregateWithClean;
+}
+//# sourceMappingURL=repository.d.ts.map

package/dist/esm/repository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"repository.d.ts","sourceRoot":"","sources":["../../src/repository.ts"],"names":[],"mappings":"AACA,OAAO,EACL,SAAS,EACT,UAAU,EACV,MAAM,EACN,QAAQ,EACR,eAAe,EAChB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAc,MAAM,gBAAgB,CAAC;AAE9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAAuB,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAEvE,MAAM,WAAW,uBAAuB,CAAC,OAAO,EAAE,YAAY;IAC5D,EAAE,EAAE,aAAa,CAAC;IAClB,KAAK,EAAE,GAAG,CAAC;IACX,cAAc,EAAE,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAC9C,mBAAmB,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACnD,GAAG,EAAE,iBAAiB,CAAC;CACxB;AAED,8BAAsB,iBAAiB,CACrC,OAAO,SAAS,SAAS,CAAC,GAAG,CAAC,EAC9B,YAAY,CACZ,SAAQ,UAAU,CAAC,OAAO,CAAC;IAC3B,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,aAAa,CAAC;IACrC,SAAS,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC;IAC9B,SAAS,CAAC,QAAQ,CAAC,cAAc,EAAE,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACjE,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACtE,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,iBAAiB,CAAC;gBAE9B,MAAM,EAAE,uBAAuB,CAAC,OAAO,EAAE,YAAY,CAAC;IASlE;;OAEG;IACH,SAAS,KAAK,OAAO,IAAI,aAAa,CAGrC;IAED;;OAEG;IACH,SAAS,CAAC,QAAQ,KAAK,KAAK,IAAI,MAAM,CAAC;IAEvC;;OAEG;IACH,SAAS,CAAC,QAAQ,CAAC,mBAAmB,IAAI,eAAe,CAAC,YAAY,CAAC,EAAE;IAEzE;;OAEG;IACH,SAAS,CAAC,mBAAmB,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAI9C,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAqDpE,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC;IA8B7C,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IA4BhD,KAAK,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAkBpD,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQpC,IAAI,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAKpC,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBtC,UAAU,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBrC,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAIxD,OAAO,CAAC,6BAA6B;CAOtC"}

package/dist/esm/repository.js

@@ -0,0 +1,188 @@
+import { eq, inArray, count } from "drizzle-orm";
+import { Aggregate, Repository, PaginatedResult, } from "@woltz/rich-domain";
+import { UOWStorage } from "./unit-of-work";
+import { NoRecordsAffectedError } from "./errors";
+import { DrizzleQueryBuilder } from "./query-builder";
+export class DrizzleRepository extends Repository {
+    db;
+    table;
+    toDomainMapper;
+    toPersistenceMapper;
+    uow;
+    constructor(config) {
+        super();
+        this.db = config.db;
+        this.table = config.table;
+        this.toDomainMapper = config.toDomainMapper;
+        this.toPersistenceMapper = config.toPersistenceMapper;
+        this.uow = config.uow;
+    }
+    /**
+     * Returns tx from UOWStorage if inside a transaction, otherwise the raw db.
+     */
+    get context() {
+        const ctx = UOWStorage.getStore()?.ctx;
+        return ctx?.client ?? this.db;
+    }
+    /**
+     * Relations to include when fetching (for Drizzle relational query API).
+     */
+    getDefaultRelations() {
+        return {};
+    }
+    async find(criteria) {
+        const { where, orderBy, limit, offset } = DrizzleQueryBuilder.apply(criteria, this.table, this.getSearchableFields());
+        const queryModel = this.context.query?.[this.model];
+        let data;
+        let total;
+        if (queryModel) {
+            [data, total] = await Promise.all([
+                queryModel.findMany({
+                    where,
+                    orderBy,
+                    limit,
+                    offset,
+                    with: this.getDefaultRelations(),
+                }),
+                this.context
+                    .select({ value: count() })
+                    .from(this.table)
+                    .where(where)
+                    .then((r) => Number(r[0]?.value ?? 0)),
+            ]);
+        }
+        else {
+            const selectQuery = this.context.select().from(this.table).$dynamic();
+            if (where)
+                selectQuery.where(where);
+            if (orderBy && orderBy.length > 0)
+                selectQuery.orderBy(...orderBy);
+            if (limit !== undefined)
+                selectQuery.limit(limit);
+            if (offset !== undefined)
+                selectQuery.offset(offset);
+            const countQuery = this.context
+                .select({ value: count() })
+                .from(this.table);
+            if (where)
+                countQuery.where(where);
+            [data, [{ value: total }]] = await Promise.all([selectQuery, countQuery]);
+            total = Number(total ?? 0);
+        }
+        const toDomain = data.map((item) => this.toDomainMapper.build(item));
+        this.markArrayOfAggregateWithClean(toDomain);
+        return PaginatedResult.create(toDomain, criteria.getPagination(), total);
+    }
+    async findById(id) {
+        const queryModel = this.context.query?.[this.model];
+        let data;
+        if (queryModel) {
+            data = await queryModel.findFirst({
+                where: eq(this.table.id, id),
+                with: this.getDefaultRelations(),
+            });
+        }
+        else {
+            const rows = await this.context
+                .select()
+                .from(this.table)
+                .where(eq(this.table.id, id))
+                .limit(1);
+            data = rows[0] ?? null;
+        }
+        if (!data)
+            return null;
+        const result = this.toDomainMapper.build(data);
+        if (result instanceof Aggregate) {
+            result.markAsClean();
+        }
+        return result;
+    }
+    async findManyByIds(ids) {
+        if (ids.length === 0)
+            return [];
+        const queryModel = this.context.query?.[this.model];
+        let data;
+        if (queryModel) {
+            data = await queryModel.findMany({
+                where: inArray(this.table.id, ids),
+                with: this.getDefaultRelations(),
+            });
+        }
+        else {
+            data = await this.context
+                .select()
+                .from(this.table)
+                .where(inArray(this.table.id, ids));
+        }
+        const toDomain = data.map((item) => this.toDomainMapper.build(item));
+        this.markArrayOfAggregateWithClean(toDomain);
+        return toDomain;
+    }
+    async count(criteria) {
+        let where;
+        if (criteria) {
+            ({ where } = DrizzleQueryBuilder.apply(criteria, this.table, this.getSearchableFields()));
+        }
+        const query = this.context.select({ value: count() }).from(this.table);
+        if (where)
+            query.where(where);
+        const result = await query;
+        return Number(result[0]?.value ?? 0);
+    }
+    async exists(id) {
+        const result = await this.context
+            .select({ value: count() })
+            .from(this.table)
+            .where(eq(this.table.id, id));
+        return Number(result[0]?.value ?? 0) > 0;
+    }
+    async save(entity) {
+        await this.toPersistenceMapper.build(entity);
+        entity.markAsPersisted();
+    }
+    async delete(entity) {
+        const id = entity.id.value;
+        try {
+            const result = await this.context
+                .delete(this.table)
+                .where(eq(this.table.id, id))
+                .returning({ id: this.table.id });
+            if (!result || result.length === 0) {
+                throw new NoRecordsAffectedError("Delete", this.model, String(id));
+            }
+        }
+        catch (error) {
+            if (error instanceof NoRecordsAffectedError)
+                throw error;
+            throw error;
+        }
+    }
+    async deleteById(id) {
+        try {
+            const result = await this.context
+                .delete(this.table)
+                .where(eq(this.table.id, id))
+                .returning({ id: this.table.id });
+            if (!result || result.length === 0) {
+                throw new NoRecordsAffectedError("Delete", this.model, id);
+            }
+        }
+        catch (error) {
+            if (error instanceof NoRecordsAffectedError)
+                throw error;
+            throw error;
+        }
+    }
+    async transaction(work) {
+        return this.uow.transaction(work);
+    }
+    markArrayOfAggregateWithClean(entities) {
+        for (const entity of entities) {
+            if (entity instanceof Aggregate) {
+                entity.markAsClean();
+            }
+        }
+    }
+}
+//# sourceMappingURL=repository.js.map

package/dist/esm/repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"repository.js","sourceRoot":"","sources":["../../src/repository.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAO,MAAM,aAAa,CAAC;AACtD,OAAO,EACL,SAAS,EACT,UAAU,EAGV,eAAe,GAChB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAoC,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC9E,OAAO,EAAE,sBAAsB,EAAE,MAAM,UAAU,CAAC;AAElD,OAAO,EAAE,mBAAmB,EAAmB,MAAM,iBAAiB,CAAC;AAUvE,MAAM,OAAgB,iBAGpB,SAAQ,UAAmB;IACR,EAAE,CAAgB;IAClB,KAAK,CAAM;IACX,cAAc,CAAgC;IAC9C,mBAAmB,CAAgC;IACnD,GAAG,CAAoB;IAE1C,YAAY,MAAsD;QAChE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,CAAC;QACtD,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,IAAc,OAAO;QACnB,MAAM,GAAG,GAAG,UAAU,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC;QACvC,OAAO,GAAG,EAAE,MAAM,IAAI,IAAI,CAAC,EAAE,CAAC;IAChC,CAAC;IAYD;;OAEG;IACO,mBAAmB;QAC3B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAA2B;QACpC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC,KAAK,CACjE,QAAQ,EACR,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,mBAAmB,EAAE,CAC3B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpD,IAAI,IAAoB,CAAC;QACzB,IAAI,KAAa,CAAC;QAElB,IAAI,UAAU,EAAE,CAAC;YACf,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChC,UAAU,CAAC,QAAQ,CAAC;oBAClB,KAAK;oBACL,OAAO;oBACP,KAAK;oBACL,MAAM;oBACN,IAAI,EAAE,IAAI,CAAC,mBAAmB,EAAE;iBACjC,CAAC;gBACF,IAAI,CAAC,OAAO;qBACT,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;qBAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;qBAChB,KAAK,CAAC,KAAK,CAAC;qBACZ,IAAI,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;aAChD,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;YAEtE,IAAI,KAAK;gBAAE,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpC,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;gBAAE,WAAW,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC;YACnE,IAAI,KAAK,KAAK,SAAS;gBAAE,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAClD,IAAI,MAAM,KAAK,SAAS;gBAAE,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAErD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO;iBAC5B,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;iBAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,KAAK;gBAAE,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAEnC,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;YAC1E,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,MAAM,QAAQ,GAAe,IAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACvD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAChC,CAAC;QAEF,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;QAE7C,OAAO,eAAe,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpD,IAAI,IAAS,CAAC;QAEd,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC;gBAChC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC;gBAC5B,IAAI,EAAE,IAAI,CAAC,mBAAmB,EAAE;aACjC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO;iBAC5B,MAAM,EAAE;iBACR,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;iBAChB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC5B,KAAK,CAAC,CAAC,CAAC,CAAC;YACZ,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;QACzB,CAAC;QAED,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAE/C,IAAI,MAAM,YAAY,SAAS,EAAE,CAAC;YAChC,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAa;QAC/B,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEhC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpD,IAAI,IAAW,CAAC;QAEhB,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC;gBAC/B,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC;gBAClC,IAAI,EAAE,IAAI,CAAC,mBAAmB,EAAE;aACjC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO;iBACtB,MAAM,EAAE;iBACR,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;iBAChB,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;QACxC,CAAC;QAED,MAAM,QAAQ,GAAc,IAAI,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CACjD,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAChC,CAAC;QAEF,IAAI,CAAC,6BAA6B,CAAC,QAAQ,CAAC,CAAC;QAE7C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAA4B;QACtC,IAAI,KAAsB,CAAC;QAE3B,IAAI,QAAQ,EAAE,CAAC;YACb,CAAC,EAAE,KAAK,EAAE,GAAG,mBAAmB,CAAC,KAAK,CACpC,QAAQ,EACR,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,mBAAmB,EAAE,CAC3B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,IAAI,KAAK;YAAE,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC;QAC3B,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO;aAC9B,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;aAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;aAChB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAChC,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAe;QACxB,MAAM,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,CAAC,eAAe,EAAE,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAe;QAC1B,MAAM,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO;iBAC9B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC5B,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YAEpC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,sBAAsB,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,sBAAsB;gBAAE,MAAM,KAAK,CAAC;YACzD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,EAAU;QACzB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO;iBAC9B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClB,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC5B,SAAS,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YAEpC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,sBAAsB,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,YAAY,sBAAsB;gBAAE,MAAM,KAAK,CAAC;YACzD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAI,IAAsB;QACzC,OAAO,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAEO,6BAA6B,CAAC,QAAmB;QACvD,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,IAAI,MAAM,YAAY,SAAS,EAAE,CAAC;gBAChC,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/dist/esm/unit-of-work.d.ts

@@ -0,0 +1,49 @@
+import { AsyncLocalStorage } from "async_hooks";
+/**
+ * Drizzle database instance type.
+ * Generic to support pg, mysql, sqlite, and other drivers.
+ */
+export type DrizzleClient = {
+    transaction: <T>(fn: (tx: any) => Promise<T>) => Promise<T>;
+    query: Record<string, any>;
+    select: (...args: any[]) => any;
+    insert: (table: any) => any;
+    update: (table: any) => any;
+    delete: (table: any) => any;
+    [key: string]: any;
+};
+export type DrizzleTransactionClient = DrizzleClient;
+export declare class DrizzleTransactionContext {
+    readonly client: DrizzleTransactionClient;
+    constructor(client: DrizzleTransactionClient);
+}
+export declare const UOWStorage: AsyncLocalStorage<{
+    ctx: DrizzleTransactionContext | null;
+}>;
+export declare class DrizzleUnitOfWork {
+    private readonly db;
+    constructor(db: DrizzleClient);
+    /**
+     * Get current transaction context (if any).
+     */
+    getCurrentContext(): DrizzleTransactionContext | null;
+    /**
+     * Check if currently inside a transaction.
+     */
+    isInTransaction(): boolean;
+    /**
+     * Execute work inside a transaction.
+     * If already in a transaction, reuses the existing context (idempotent nesting).
+     */
+    transaction<T>(work: () => Promise<T>): Promise<T>;
+}
+/**
+ * Decorator that wraps a method in a transaction.
+ * If already inside a transaction, reuses the existing one.
+ */
+export declare function Transactional(inputUow?: DrizzleUnitOfWork): MethodDecorator;
+/**
+ * Helper to get current transaction client from anywhere.
+ */
+export declare function getCurrentDrizzleContext(): DrizzleTransactionClient | null;
+//# sourceMappingURL=unit-of-work.d.ts.map

package/dist/esm/unit-of-work.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"unit-of-work.d.ts","sourceRoot":"","sources":["../../src/unit-of-work.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEhD;;;GAGG;AACH,MAAM,MAAM,aAAa,GAAG;IAC1B,WAAW,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;IAC5D,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3B,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC;IAChC,MAAM,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAAC;IAC5B,MAAM,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAAC;IAC5B,MAAM,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,GAAG,CAAC;IAC5B,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG,aAAa,CAAC;AAErD,qBAAa,yBAAyB;aACR,MAAM,EAAE,wBAAwB;gBAAhC,MAAM,EAAE,wBAAwB;CAC7D;AAED,eAAO,MAAM,UAAU;SAChB,yBAAyB,GAAG,IAAI;EACnC,CAAC;AAEL,qBAAa,iBAAiB;IAChB,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAAF,EAAE,EAAE,aAAa;IAE9C;;OAEG;IACH,iBAAiB,IAAI,yBAAyB,GAAG,IAAI;IAIrD;;OAEG;IACH,eAAe,IAAI,OAAO;IAI1B;;;OAGG;IACG,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;CAazD;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAAC,QAAQ,CAAC,EAAE,iBAAiB,GAAG,eAAe,CA2B3E;AAwBD;;GAEG;AACH,wBAAgB,wBAAwB,IAAI,wBAAwB,GAAG,IAAI,CAE1E"}

package/dist/esm/unit-of-work.js

@@ -0,0 +1,87 @@
+import { AsyncLocalStorage } from "async_hooks";
+export class DrizzleTransactionContext {
+    client;
+    constructor(client) {
+        this.client = client;
+    }
+}
+export const UOWStorage = new AsyncLocalStorage();
+export class DrizzleUnitOfWork {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    /**
+     * Get current transaction context (if any).
+     */
+    getCurrentContext() {
+        return UOWStorage.getStore()?.ctx ?? null;
+    }
+    /**
+     * Check if currently inside a transaction.
+     */
+    isInTransaction() {
+        return this.getCurrentContext() !== null;
+    }
+    /**
+     * Execute work inside a transaction.
+     * If already in a transaction, reuses the existing context (idempotent nesting).
+     */
+    async transaction(work) {
+        if (this.isInTransaction()) {
+            return work();
+        }
+        return this.db.transaction(async (tx) => {
+            const ctx = new DrizzleTransactionContext(tx);
+            return UOWStorage.run({ ctx }, async () => {
+                return await work();
+            });
+        });
+    }
+}
+/**
+ * Decorator that wraps a method in a transaction.
+ * If already inside a transaction, reuses the existing one.
+ */
+export function Transactional(inputUow) {
+    return function (_target, _propertyKey, descriptor) {
+        const original = descriptor.value;
+        descriptor.value = async function (...args) {
+            if (UOWStorage.getStore()?.ctx) {
+                return original.apply(this, args);
+            }
+            const uow = inputUow ?? findUnitOfWork(this);
+            if (!uow) {
+                throw new Error(`@Transactional: DrizzleUnitOfWork not found in ${this.constructor.name} instance. ` +
+                    "Ensure your class has a 'uow' property. eg: constructor(private readonly uow: DrizzleUnitOfWork) {}");
+            }
+            return uow.transaction(() => original.apply(this, args));
+        };
+        return descriptor;
+    };
+}
+/**
+ * Find DrizzleUnitOfWork in instance.
+ */
+function findUnitOfWork(instance) {
+    if (instance.uow instanceof DrizzleUnitOfWork) {
+        return instance.uow;
+    }
+    if (instance._uow instanceof DrizzleUnitOfWork) {
+        return instance._uow;
+    }
+    for (const key of Object.keys(instance)) {
+        const value = instance[key];
+        if (value instanceof DrizzleUnitOfWork) {
+            return value;
+        }
+    }
+    return null;
+}
+/**
+ * Helper to get current transaction client from anywhere.
+ */
+export function getCurrentDrizzleContext() {
+    return UOWStorage.getStore()?.ctx?.client ?? null;
+}
+//# sourceMappingURL=unit-of-work.js.map

package/dist/esm/unit-of-work.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"unit-of-work.js","sourceRoot":"","sources":["../../src/unit-of-work.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAkBhD,MAAM,OAAO,yBAAyB;IACR;IAA5B,YAA4B,MAAgC;QAAhC,WAAM,GAAN,MAAM,CAA0B;IAAG,CAAC;CACjE;AAED,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,iBAAiB,EAE3C,CAAC;AAEL,MAAM,OAAO,iBAAiB;IACC;IAA7B,YAA6B,EAAiB;QAAjB,OAAE,GAAF,EAAE,CAAe;IAAG,CAAC;IAElD;;OAEG;IACH,iBAAiB;QACf,OAAO,UAAU,CAAC,QAAQ,EAAE,EAAE,GAAG,IAAI,IAAI,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,iBAAiB,EAAE,KAAK,IAAI,CAAC;IAC3C,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAI,IAAsB;QACzC,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;YAC3B,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;QAED,OAAO,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,EAAO,EAAE,EAAE;YAC3C,MAAM,GAAG,GAAG,IAAI,yBAAyB,CAAC,EAAE,CAAC,CAAC;YAE9C,OAAO,UAAU,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE,KAAK,IAAI,EAAE;gBACxC,OAAO,MAAM,IAAI,EAAE,CAAC;YACtB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAAC,QAA4B;IACxD,OAAO,UACL,OAAY,EACZ,YAA6B,EAC7B,UAA8B;QAE9B,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC;QAElC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAsB,GAAG,IAAW;YAC1D,IAAI,UAAU,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,CAAC;gBAC/B,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpC,CAAC;YAED,MAAM,GAAG,GAAG,QAAQ,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;YAE7C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,MAAM,IAAI,KAAK,CACb,kDAAkD,IAAI,CAAC,WAAW,CAAC,IAAI,aAAa;oBAClF,qGAAqG,CACxG,CAAC;YACJ,CAAC;YAED,OAAO,GAAG,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,QAAa;IACnC,IAAI,QAAQ,CAAC,GAAG,YAAY,iBAAiB,EAAE,CAAC;QAC9C,OAAO,QAAQ,CAAC,GAAG,CAAC;IACtB,CAAC;IAED,IAAI,QAAQ,CAAC,IAAI,YAAY,iBAAiB,EAAE,CAAC;QAC/C,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAED,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;YACvC,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB;IACtC,OAAO,UAAU,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,MAAM,IAAI,IAAI,CAAC;AACpD,CAAC"}