@m1212e/rumble 0.23.5 → 0.23.7

package/out/index.mjs

@@ -180,7 +180,7 @@ function mapNullFieldsToUndefined(obj) {
 }
 //#endregion
 //#region package.json
-var version = "0.23.5";
+var version = "0.23.7";
 //#endregion
 //#region lib/helpers/mergeFilters.ts
 const EmptyFilter$1 = Symbol.for("drizzle:EmptyFilter");
@@ -1862,7 +1862,7 @@ const createPubSubInstance = ({ subscriptions }) => {
 //#region lib/query.ts
 const createQueryImplementer = ({ db, schemaBuilder, search, logger: loggerConfig, whereArgImplementer, orderArgImplementer, makePubSubInstance }) => {
 	const registeredQueryFieldNames = /* @__PURE__ */ new Set();
-	return ({ table, readAction = "read", listAction = "read" }) => {
+	return ({ table, readAction = "read", listAction = "read", findFirst, findMany }) => {
 		const WhereArg = whereArgImplementer({ table });
 		const OrderArg = orderArgImplementer({ table });
 		const tableSchema = tableHelper({
@@ -1921,8 +1921,13 @@ const createQueryImplementer = ({ db, schemaBuilder, search, logger: loggerConfi
 						const filter = ctx.abilities[table].filter(listAction).merge(mappedArgs).query.many;
 						if (mappedArgs.offset) filter.offset = mappedArgs.offset;
 						if (mappedArgs.orderBy) filter.orderBy = mappedArgs.orderBy;
-						const queryInstance = query(filter);
-						if (filter.columns) queryInstance.columns = filter.columns;
+						const queryArgs = query(filter);
+						if (filter.columns) queryArgs.columns = filter.columns;
+						const run = findMany ? (tx) => findMany({
+							tx,
+							filter,
+							queryArgs
+						}) : (tx) => tx.query[table].findMany(queryArgs);
 						if (search?.enabled && args.search && args.search.length > 0 && search?.cpu_operator_cost) return db.transaction(async (tx) => {
 							if (isPostgresDB(tx)) await tx.execute(sql.raw(`SET cpu_operator_cost = ${search.cpu_operator_cost};`));
 							else {
@@ -1930,9 +1935,9 @@ const createQueryImplementer = ({ db, schemaBuilder, search, logger: loggerConfi
 								const log = loggerConfig?.enabled ? loggerConfig.logger : void 0;
 								log ? log.info({}, msg) : console.info(msg);
 							}
-							return tx.query[table].findMany(queryInstance);
+							return run(tx);
 						});
-						return db.query[table].findMany(queryInstance);
+						return run(db);
 					}
 				}),
 				[singularName]: t.drizzleField({
@@ -1944,9 +1949,13 @@ const createQueryImplementer = ({ db, schemaBuilder, search, logger: loggerConfi
 					resolve: (query, _root, args, ctx, _info) => {
 						deepSetProto(args);
 						const filter = ctx.abilities[table].filter(readAction).merge({ where: { [primaryKeyField.name]: args.id } }).query.single;
-						const q = query(filter);
-						if (filter.columns) q.columns = filter.columns;
-						return db.query[table].findFirst(q).then(assertFindFirstExists);
+						const queryArgs = query(filter);
+						if (filter.columns) queryArgs.columns = filter.columns;
+						return findFirst ? findFirst({
+							tx: db,
+							filter,
+							queryArgs
+						}) : db.query[table].findFirst(queryArgs).then(assertFindFirstExists);
 					}
 				})
 			};