@m1212e/rumble 0.16.7 → 0.16.15

package/out/index.mjs

@@ -1,4 +1,5 @@
-import { t as generateFromSchema } from "./generate-DcdqYHG4.mjs";
+import { t as lazy } from "./lazy-CoMVcY4X.mjs";
+import { t as generateFromSchema } from "./generate-PDRNfY6V.mjs";
 import { GraphQLError } from "graphql";
 import { EnvelopArmorPlugin } from "@escape.tech/graphql-armor";
 import { useDisableIntrospection } from "@graphql-yoga/plugin-disable-introspection";
@@ -37,14 +38,14 @@ var RumbleErrorSafe = class extends GraphQLError {};
 //#endregion
 //#region lib/helpers/asserts.ts
 /**
-* 
+*
 * Helper function to map a drizzle findFirst query result,
 * which may be optional, to a correct drizzle type.
-* 
+*
 * @throws RumbleError
-* 
+*
 * @example
-* 
+*
 * ```ts
 * schemaBuilder.queryFields((t) => {
 return {
@@ -71,14 +72,14 @@ const assertFindFirstExists = (value) => {
 	return value;
 };
 /**
-* 
+*
 * Helper function to map a drizzle findFirst query result,
 * which may be optional, to a correct drizzle type.
-* 
+*
 * @throws RumbleError
-* 
+*
 * @example
-* 
+*
 * ```ts
 schemaBuilder.mutationFields((t) => {
 return {
@@ -179,30 +180,6 @@ function mapNullFieldsToUndefined(obj) {
 	return Object.fromEntries(Object.entries(obj).map(([key, value]) => [key, value === null ? void 0 : value]));
 }
 
-//#endregion
-//#region lib/helpers/lazy.ts
-/**
-* Creates a lazily initialized function.
-*
-* The returned function will call the initializer function on its first call and
-* store the result. On subsequent calls, it will return the stored result.
-*
-* @param initializer The function to be called for initialization.
-* @returns A function that calls the initializer function on its first call and
-*          returns the stored result on subsequent calls.
-*/
-function lazy(initializer) {
-	let value;
-	let initialized = false;
-	return () => {
-		if (!initialized) {
-			value = initializer();
-			initialized = true;
-		}
-		return value;
-	};
-}
-
 //#endregion
 //#region lib/helpers/mergeFilters.ts
 function mergeFilters(filterA, filterB) {
@@ -374,25 +351,25 @@ const createAbilityBuilder = ({ db, actions, defaultLimit }) => {
 		return {
 			allow: (action) => {
 				if (hasBeenBuilt) throw new RumbleError("You can't call allow() after the ability builder has been built. Please ensure that you register all abilities before accessing them.");
-				const actions$1 = Array.isArray(action) ? action : [action];
-				for (const action$1 of actions$1) {
-					let filters = queryFilters.get(action$1);
+				const actions = Array.isArray(action) ? action : [action];
+				for (const action of actions) {
+					let filters = queryFilters.get(action);
 					if (!filters) {
 						filters = "unrestricted";
-						queryFilters.set(action$1, filters);
+						queryFilters.set(action, filters);
 					}
 				}
 				return { when: (queryFilter) => {
-					for (const action$1 of actions$1) {
-						if (queryFilters.get(action$1) === "unrestricted") queryFilters.set(action$1, []);
-						queryFilters.get(action$1).push(queryFilter);
+					for (const action of actions) {
+						if (queryFilters.get(action) === "unrestricted") queryFilters.set(action, []);
+						queryFilters.get(action).push(queryFilter);
 					}
 				} };
 			},
 			filter: (action) => {
-				const actions$1 = Array.isArray(action) ? action : [action];
+				const actions = Array.isArray(action) ? action : [action];
 				return { by: (explicitFilter) => {
-					for (const action$1 of actions$1) runtimeFilters.get(action$1).push(explicitFilter);
+					for (const action of actions) runtimeFilters.get(action).push(explicitFilter);
 				} };
 			},
 			_: {
@@ -432,16 +409,16 @@ const createAbilityBuilder = ({ db, actions, defaultLimit }) => {
 					/**
 					* Packs the filters into a response object that can be applied for queries by the user
 					*/
-					function transformToResponse(queryFilters$1) {
+					function transformToResponse(queryFilters) {
 						const internalTransformer = (filters, mergedLimit) => {
 							const limit = lazy(() => {
 								if (mergedLimit !== void 0) {
 									if (!filters?.limit) return mergedLimit;
 									if (filters.limit > mergedLimit) return mergedLimit;
 								}
-								let limit$1 = filters?.limit;
-								if (defaultLimit && (limit$1 === void 0 || limit$1 > defaultLimit)) limit$1 = defaultLimit;
-								return limit$1 ?? void 0;
+								let limit = filters?.limit;
+								if (defaultLimit && (limit === void 0 || limit > defaultLimit)) limit = defaultLimit;
+								return limit ?? void 0;
 							});
 							const sqlTransformedWhere = lazy(() => {
 								return filters?.where ? relationsFilterToSQL(tableSchema.foundRelation.table, filters.where) : void 0;
@@ -485,14 +462,14 @@ const createAbilityBuilder = ({ db, actions, defaultLimit }) => {
 								} }
 							};
 						};
-						const ret = internalTransformer(queryFilters$1);
+						const ret = internalTransformer(queryFilters);
 						/**
 						* Merges the current query filters with the provided filters for this call only
 						*/
-						function merge$1(p) {
+						function merge(p) {
 							return internalTransformer(mergeFilters(ret.query.many, p), p.limit);
 						}
-						ret.merge = merge$1;
+						ret.merge = merge;
 						return ret;
 					}
 					return { withContext: (userContext) => {
@@ -706,7 +683,13 @@ const createWhereArgImplementer = ({ db, schemaBuilder, enumImplementer }) => {
 //#region lib/client/client.ts
 const clientCreatorImplementer = ({ builtSchema }) => {
 	const clientCreator = async ({ apiUrl, outputPath, rumbleImportPath, useExternalUrqlClient, removeExisting, forceReactivity }) => {
-		if (process.env.NODE_ENV !== "development") console.warn("Running rumble client generation in non development mode. Are you sure this is correct?");
+		if (process.env.NODE_ENV !== "development") console.warn(`Running rumble client generation in non development mode. Are you sure this is correct? Called from ${__filename} with arguments: ${JSON.stringify({
+			outputPath,
+			apiUrl,
+			rumbleImportPath,
+			useExternalUrqlClient,
+			removeExisting
+		})}`);
 		await generateFromSchema({
 			schema: builtSchema(),
 			outputPath,
@@ -1008,7 +991,7 @@ const createObjectImplementer = ({ db, search, schemaBuilder, makePubSubInstance
 			fields: (t) => {
 				const columns = tableSchema.columns;
 				const configMap = /* @__PURE__ */ new Map();
-				const userAdjustments = adjust?.(new Proxy(t, { get: (target, prop) => {
+				const userAdjustments = adjust?.(new Proxy(t, { get: (target, prop, receiver) => {
 					if (typeof target[prop] !== "function" || prop === "arg" || prop === "builder" || prop === "graphqlKind" || prop === "kind" || prop === "listRef" || prop === "table" || prop === "typename" || prop === "variant" || prop.toString().startsWith("boolean") || prop.toString().startsWith("float") || prop.toString().startsWith("id") || prop.toString().startsWith("int") || prop.toString().startsWith("string") || prop.toString().startsWith("expose")) return target[prop];
 					return (...params) => {
 						const ref = target[prop](...params);
@@ -1098,21 +1081,21 @@ const createObjectImplementer = ({ db, search, schemaBuilder, makePubSubInstance
 						subscribe,
 						nullable,
 						description: `Get the ${pluralize.plural(relationSchema.tsName)} related to this ${pluralize.singular(tableSchema.tsName)}`,
-						query: (args$1, ctx) => {
-							args$1 = JSON.parse(JSON.stringify(args$1));
+						query: (args, ctx) => {
+							args = JSON.parse(JSON.stringify(args));
 							if (isMany) adjustQueryArgsForSearch({
 								search,
-								args: args$1,
+								args,
 								tableSchema: relationSchema,
 								abilities: ctx.abilities[relationSchema.tsName].filter(readAction)
 							});
 							const filter = ctx.abilities[relationSchema.tsName].filter(readAction).merge({
-								where: args$1.where,
-								limit: args$1.limit,
-								extras: args$1.extras
+								where: args.where,
+								limit: args.limit,
+								extras: args.extras
 							}).query[filterSpecifier];
-							if (args$1.offset) filter.offset = args$1.offset;
-							if (args$1.orderBy) filter.orderBy = args$1.orderBy;
+							if (args.offset) filter.offset = args.offset;
+							if (args.orderBy) filter.orderBy = args.orderBy;
 							return filter;
 						}
 					});
@@ -1185,10 +1168,10 @@ const createPubSubInstance = ({ subscriptions }) => {
 				return pubsub.publish(key);
 			},
 			updated(primaryKeyValue) {
-				const keys = (Array.isArray(primaryKeyValue) ? primaryKeyValue : [primaryKeyValue]).map((primaryKeyValue$1) => makePubSubKey({
+				const keys = (Array.isArray(primaryKeyValue) ? primaryKeyValue : [primaryKeyValue]).map((primaryKeyValue) => makePubSubKey({
 					tableName: table.toString(),
 					action: "updated",
-					primaryKeyValue: primaryKeyValue$1
+					primaryKeyValue
 				}));
 				const uniqueKeys = Array.from(new Set(keys));
 				for (const key of uniqueKeys) pubsub.publish(key);
@@ -1259,6 +1242,11 @@ const createQueryImplementer = ({ db, schemaBuilder, search, whereArgImplementer
 						if (mappedArgs.orderBy) filter.orderBy = mappedArgs.orderBy;
 						const queryInstance = query(filter);
 						if (filter.columns) queryInstance.columns = filter.columns;
+						if (search?.cpu_operator_cost) return db.transaction(async (tx) => {
+							if (isPostgresDB(tx)) await tx.execute(sql`SET LOCAL cpu_operator_cost = ${search.cpu_operator_cost};`);
+							else console.info("Database dialect is not postgresql, cannot set cpu_operator_cost.");
+							return tx.query[table].findMany(queryInstance);
+						});
 						return db.query[table].findMany(queryInstance);
 					}
 				}),