@m1212e/rumble 0.12.12 → 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 }) {
@@ -716,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,
@@ -723,7 +710,8 @@ function tableHelper({ db, table }) {
 		dbName: foundSchema.dbName,
 		tsName: foundSchema.tsName,
 		foundSchema,
-		foundRelation
+		foundRelation,
+		fullSchema
 	};
 }
 
@@ -1078,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,
@@ -1102,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 = {
@@ -1849,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";
@@ -1882,7 +1917,8 @@ export const db = drizzle(
 		clientCreator: clientCreatorImplementer({
 			...rumbleInput,
 			builtSchema
-		})
+		}),
+		countQuery
 	};
 };