@m1212e/rumble 0.12.10 → 0.12.13

package/out/index.cjs

@@ -22,35 +22,21 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 
 //#endregion
 let node_fs_promises = require("node:fs/promises");
-node_fs_promises = __toESM(node_fs_promises);
 let node_path = require("node:path");
-node_path = __toESM(node_path);
 let graphql = require("graphql");
-graphql = __toESM(graphql);
 let es_toolkit = require("es-toolkit");
-es_toolkit = __toESM(es_toolkit);
 let wonka = require("wonka");
-wonka = __toESM(wonka);
 let __escape_tech_graphql_armor = require("@escape.tech/graphql-armor");
-__escape_tech_graphql_armor = __toESM(__escape_tech_graphql_armor);
 let __graphql_yoga_plugin_disable_introspection = require("@graphql-yoga/plugin-disable-introspection");
-__graphql_yoga_plugin_disable_introspection = __toESM(__graphql_yoga_plugin_disable_introspection);
 let graphql_yoga = require("graphql-yoga");
-graphql_yoga = __toESM(graphql_yoga);
 let sofa_api = require("sofa-api");
-sofa_api = __toESM(sofa_api);
 let drizzle_orm = require("drizzle-orm");
-drizzle_orm = __toESM(drizzle_orm);
 let drizzle_orm_casing = require("drizzle-orm/casing");
-drizzle_orm_casing = __toESM(drizzle_orm_casing);
 let drizzle_orm_pg_core = require("drizzle-orm/pg-core");
-drizzle_orm_pg_core = __toESM(drizzle_orm_pg_core);
 let pluralize = require("pluralize");
 pluralize = __toESM(pluralize);
 let drizzle_orm_mysql_core = require("drizzle-orm/mysql-core");
-drizzle_orm_mysql_core = __toESM(drizzle_orm_mysql_core);
 let drizzle_orm_sqlite_core = require("drizzle-orm/sqlite-core");
-drizzle_orm_sqlite_core = __toESM(drizzle_orm_sqlite_core);
 let __pothos_core = require("@pothos/core");
 __pothos_core = __toESM(__pothos_core);
 let __pothos_plugin_drizzle = require("@pothos/plugin-drizzle");
@@ -58,7 +44,6 @@ __pothos_plugin_drizzle = __toESM(__pothos_plugin_drizzle);
 let __pothos_plugin_smart_subscriptions = require("@pothos/plugin-smart-subscriptions");
 __pothos_plugin_smart_subscriptions = __toESM(__pothos_plugin_smart_subscriptions);
 let graphql_scalars = require("graphql-scalars");
-graphql_scalars = __toESM(graphql_scalars);
 
 //#region lib/client/generate/client.ts
 function generateClient({ apiUrl, rumbleImportPath, useExternalUrqlClient, availableSubscriptions }) {
@@ -499,8 +484,65 @@ const assertFirstEntryExists = (value) => {
 
 //#endregion
 //#region lib/helpers/mapNullFieldsToUndefined.ts
+/**
+* Helper to map null fields to undefined
+* @param obj The object to map
+* @returns The mapped object with all fields of 'null' transformed to 'undefined'
+* 
+* This becomes useful for update mutations where you do not want to pass null (unset value in db)
+* but undefined (do not touch value in db) in case of a value not beeing set in the args of said mutation
+* @example
+* ```ts
+*  updateUser: t.drizzleField({
+type: User,
+args: {
+id: t.arg.string({ required: true }),
+email: t.arg.string(),
+lastName: t.arg.string(),
+firstName: t.arg.string(),
+},
+resolve: async (query, root, args, ctx, info) => {
+const mappedArgs = mapNullFieldsToUndefined(args);
+const user = await db.transaction(async (tx) => {
+const user = await tx
+.update(schema.user)
+.set({
+
+// email: args.email ?? undefined,
+// lastName: args.lastName ?? undefined,
+// firstName: args.firstName ?? undefined,
+
+// becomes this
+
+email: mappedArgs.email,
+lastName: mappedArgs.lastName,
+firstName: mappedArgs.firstName,
+})
+.returning()
+.then(assertFirstEntryExists);
+return user;
+});
+
+pubsub.updated(user.id);
+
+return db.query.user
+.findFirst(
+query(
+ctx.abilities.user.filter('read').merge(
+{
+where: { id: user.id },
+}
+1).query.single,
+),
+)
+.then(assertFindFirstExists);
+},
+}),
+* 
+* 
+* ```
+*/
 function mapNullFieldsToUndefined(obj) {
-	if (!obj) return;
 	return Object.fromEntries(Object.entries(obj).map(([key, value]) => [key, value === null ? void 0 : value]));
 }
 
@@ -659,6 +701,8 @@ function tableHelper({ db, table }) {
 	if (!foundRelation) throw new RumbleError(`Could not find schema for ${JSON.stringify(table)}`);
 	const foundSchema = Object.values(db._.schema).find((schema) => schema.dbName === foundRelation.table[drizzleOriginalNameSymbol]);
 	if (!foundSchema) throw new RumbleError(`Could not find schema for ${JSON.stringify(table)}`);
+	const fullSchema = db._.fullSchema?.[foundSchema.tsName];
+	if (!fullSchema) throw new RumbleError(`Could not find full schema for ${JSON.stringify(table)}`);
 	return {
 		columns: foundSchema.columns,
 		primaryKey: foundSchema.primaryKey,
@@ -666,7 +710,8 @@ function tableHelper({ db, table }) {
 		dbName: foundSchema.dbName,
 		tsName: foundSchema.tsName,
 		foundSchema,
-		foundRelation
+		foundRelation,
+		fullSchema
 	};
 }
 
@@ -1021,6 +1066,7 @@ const createWhereArgImplementer = ({ db, schemaBuilder, enumImplementer }) => {
 const clientCreatorImplementer = ({ builtSchema }) => {
 	if (process.env.NODE_ENV !== "development") console.warn("Running rumble client generation in non development mode. Are you sure this is correct?");
 	const clientCreator = async ({ apiUrl, outputPath, rumbleImportPath, useExternalUrqlClient }) => {
+		if (process.env.NODE_ENV !== "development") console.warn("Running rumble client generation in non development mode. Are you sure this is correct?");
 		await generateFromSchema({
 			schema: builtSchema(),
 			outputPath,
@@ -1045,6 +1091,46 @@ const createContextFunction = ({ context: makeUserContext, abilityBuilder }) =>
 	};
 };
 
+//#endregion
+//#region lib/countQuery.ts
+const createCountQueryImplementer = ({ db, schemaBuilder, whereArgImplementer, makePubSubInstance }) => {
+	return ({ table, listAction = "read", isAllowed }) => {
+		const WhereArg = whereArgImplementer({ table });
+		const { registerOnInstance } = makePubSubInstance({ table });
+		const tableSchema = tableHelper({
+			db,
+			table
+		});
+		return schemaBuilder.queryFields((t) => {
+			return { [`${pluralize.default.plural(table.toString())}Count`]: t.field({
+				type: "Int",
+				nullable: false,
+				smartSubscription: true,
+				description: `Count all ${pluralize.default.plural(table.toString())}`,
+				subscribe: (subscriptions, _root, _args, _ctx, _info) => {
+					registerOnInstance({
+						instance: subscriptions,
+						action: "created"
+					});
+					registerOnInstance({
+						instance: subscriptions,
+						action: "removed"
+					});
+				},
+				args: { where: t.arg({
+					type: WhereArg,
+					required: false
+				}) },
+				resolve: async (root, args, ctx, info) => {
+					if (isAllowed && !await isAllowed(ctx)) throw new RumbleErrorSafe("Not allowed to perform this action");
+					Object.setPrototypeOf(args, Object.prototype);
+					return db.select({ count: (0, drizzle_orm.count)() }).from(tableSchema.fullSchema).where(ctx.abilities[table].filter(listAction).merge(mapNullFieldsToUndefined(args)).sql.where).then(assertFirstEntryExists).then((r) => r.count);
+				}
+			}) };
+		});
+	};
+};
+
 //#endregion
 //#region lib/helpers/sofaOpenAPIWebhookDocs.ts
 const sofaOpenAPIWebhookDocs = {
@@ -1792,6 +1878,12 @@ export const db = drizzle(
 		orderArgImplementer: orderArg,
 		makePubSubInstance
 	});
+	const countQuery = createCountQueryImplementer({
+		...rumbleInput,
+		schemaBuilder,
+		whereArgImplementer: whereArg,
+		makePubSubInstance
+	});
 	const builtSchema = lazy(() => schemaBuilder.toSchema());
 	const createYoga = (args) => {
 		const enableApiDocs = args?.enableApiDocs ?? process?.env?.NODE_ENV === "development";
@@ -1825,7 +1917,8 @@ export const db = drizzle(
 		clientCreator: clientCreatorImplementer({
 			...rumbleInput,
 			builtSchema
-		})
+		}),
+		countQuery
 	};
 };