@m1212e/rumble 0.23.2 → 0.23.5

package/out/index.mjs

@@ -1,10 +1,9 @@
 import { t as lazy } from "./lazy-C7IyM4M4.mjs";
 import { t as generateFromSchema } from "./generate-0QFVc3Sg.mjs";
-import { GraphQLError, GraphQLSchema } from "graphql";
+import { GraphQLError, GraphQLSchema, execute as defaultExecute, subscribe as defaultSubscribe } from "graphql";
 import { EnvelopArmorPlugin } from "@escape.tech/graphql-armor";
 import { useDisableIntrospection } from "@graphql-yoga/plugin-disable-introspection";
 import { SpanStatusCode, trace } from "@opentelemetry/api";
-import { AttributeNames, SpanNames, createOpenTelemetryWrapper } from "@pothos/tracing-opentelemetry";
 import { capitalize, cloneDeep, debounce, merge, toMerged } from "es-toolkit";
 import { createPubSub, createYoga } from "graphql-yoga";
 import { One, count, getTableColumns, getTableName, is, isTable, relationsFilterToSQL, sql } from "drizzle-orm";
@@ -12,6 +11,7 @@ import { toCamelCase } from "drizzle-orm/casing";
 import { PgColumn, PgEnumColumn, PgEnumObjectColumn, PgTable, isPgEnum } from "drizzle-orm/pg-core";
 import { MySqlEnumColumn, MySqlEnumObjectColumn, MySqlTable } from "drizzle-orm/mysql-core";
 import pluralize from "pluralize";
+import { AttributeNames, SpanNames, createOpenTelemetryWrapper } from "@pothos/tracing-opentelemetry";
 import { SQLiteTable } from "drizzle-orm/sqlite-core";
 import SchemaBuilder, { BasePlugin } from "@pothos/core";
 import DrizzlePlugin from "@pothos/plugin-drizzle";
@@ -180,12 +180,25 @@ function mapNullFieldsToUndefined(obj) {
 }
 //#endregion
 //#region package.json
-var version = "0.23.2";
+var version = "0.23.5";
 //#endregion
 //#region lib/helpers/mergeFilters.ts
+const EmptyFilter$1 = Symbol.for("drizzle:EmptyFilter");
+/**
+* Normalizes drizzle-orm's `EmptyFilter` sentinel to `undefined` so it's
+* treated as "no filter" rather than a real filter value to combine with
+* AND/OR — otherwise merging an unrestricted (EmptyFilter) base filter with
+* a real `where` would needlessly wrap it as `{ AND: [EmptyFilter, where] }`
+* instead of just `where`.
+*/
+function realWhere(where) {
+	return where === EmptyFilter$1 ? void 0 : where;
+}
 function mergeFilters(filterA, filterB, mode = "AND") {
+	const filterAWhere = realWhere(filterA?.where);
+	const filterBWhere = realWhere(filterB?.where);
 	return {
-		where: filterA?.where && filterB?.where ? mode === "OR" ? { OR: [filterA.where, filterB.where] } : { AND: [filterA.where, filterB.where] } : filterA?.where ?? filterB?.where,
+		where: filterAWhere && filterBWhere ? mode === "OR" ? { OR: [filterAWhere, filterBWhere] } : { AND: [filterAWhere, filterBWhere] } : filterAWhere ?? filterBWhere,
 		columns: filterA?.columns || filterB?.columns ? new Set([Object.entries(filterA?.columns ?? {}), Object.entries(filterB?.columns ?? {})].flat().filter(([, v]) => v === true).map(([k]) => k)).entries().reduce((acc, [key]) => {
 			acc[key] = true;
 			return acc;
@@ -412,6 +425,21 @@ function tableHelper({ db, table }) {
 }
 //#endregion
 //#region lib/abilityBuilder.ts
+/**
+* drizzle-orm's sentinel value for "this filter field is deliberately
+* skipped", as opposed to `undefined`, which some drizzle-orm versions
+* reject when it ends up as an explicit (present but `undefined`) `where`
+* property nested inside a relational `with` query.
+*
+* drizzle-orm defines this via `Symbol.for("drizzle:EmptyFilter")` — a
+* globally registered symbol — so recreating it the same way here yields
+* the exact same value drizzle-orm's relational query engine checks for,
+* without importing it directly: not every drizzle-orm version rumble
+* supports exports it as a named export, and older versions that don't know
+* about it at all still handle it safely (`Object.entries` on a symbol
+* yields no entries, i.e. "no filter").
+*/
+const EmptyFilter = Symbol.for("drizzle:EmptyFilter");
 function isDynamicQueryFilter(filter) {
 	return typeof filter === "function" && filter.constructor.name !== "AsyncFunction";
 }
@@ -580,7 +608,7 @@ by: (explicitFilter) => {
 									*/
 									single: {
 										extras: filters?.extras,
-										where: filters?.where,
+										where: filters?.where ?? EmptyFilter,
 										columns: filters?.columns
 									},
 									/**
@@ -588,7 +616,7 @@ by: (explicitFilter) => {
 									*/
 									many: {
 										extras: filters?.extras,
-										where: filters?.where,
+										where: filters?.where ?? EmptyFilter,
 										columns: filters?.columns,
 										get limit() {
 											return limit();
@@ -634,14 +662,14 @@ by: (explicitFilter) => {
 									*/
 									single: {
 										extras: filters?.extras,
-										where: filters?.where
+										where: filters?.where ?? EmptyFilter
 									},
 									/**
 									* For find many calls
 									*/
 									many: {
 										extras: filters?.extras,
-										where: filters?.where,
+										where: filters?.where ?? EmptyFilter,
 										get limit() {
 											return limit();
 										}
@@ -1272,6 +1300,165 @@ const sofaOpenAPIWebhookDocs = {
 	} }
 };
 //#endregion
+//#region lib/helpers/tracing.ts
+function isAsyncIterable(value) {
+	return typeof value === "object" && value !== null && Symbol.asyncIterator in value;
+}
+async function* wrapSubscriptionIterator(iterator, log, operationName) {
+	let eventCount = 0;
+	try {
+		for await (const event of iterator) {
+			eventCount++;
+			if (event.errors?.length) log.error({
+				"graphql.operation.name": operationName,
+				eventCount,
+				errors: event.errors.map((e) => e.message)
+			}, "graphql subscription event error");
+			yield event;
+		}
+		log.info({
+			"graphql.operation.name": operationName,
+			eventCount
+		}, "graphql subscription completed");
+	} catch (error) {
+		log.error({
+			"graphql.operation.name": operationName,
+			eventCount,
+			err: error
+		}, "graphql subscription threw");
+		throw error;
+	}
+}
+function buildTracedExecute(executeFn, rumbleInput) {
+	return async (options) => {
+		let log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
+		const operationName = options.operationName ?? "anonymous";
+		const start = Date.now();
+		log?.info({ "graphql.operation.name": operationName }, "graphql execute start");
+		const run = async () => {
+			const result = await executeFn(options);
+			if (result && "errors" in result && result.errors?.length) log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				errors: result.errors.map((e) => e.message)
+			}, "graphql execute completed with errors");
+			else log?.info({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start
+			}, "graphql execute completed");
+			return result;
+		};
+		if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan(SpanNames.EXECUTE, { attributes: {
+			[AttributeNames.OPERATION_NAME]: operationName,
+			[AttributeNames.SOURCE]: options.document
+		} }, async (span) => {
+			if (log && rumbleInput.logger?.injectTraceId !== false) {
+				const { traceId, spanId } = span.spanContext();
+				log = log.child({
+					traceId,
+					spanId
+				});
+			}
+			try {
+				const result = await run();
+				if (result && "errors" in result && result.errors?.length) {
+					for (const error of result.errors) span.recordException(error);
+					span.setStatus({ code: SpanStatusCode.ERROR });
+				}
+				return result;
+			} catch (error) {
+				if (error instanceof Error) span.recordException(error);
+				log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					err: error
+				}, "graphql execute threw");
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				span.end();
+			}
+		});
+		try {
+			return await run();
+		} catch (error) {
+			log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				err: error
+			}, "graphql execute threw");
+			throw error;
+		}
+	};
+}
+function buildTracedSubscribe(subscribeFn, rumbleInput) {
+	return async (options) => {
+		let log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
+		const operationName = options.operationName ?? "anonymous";
+		const start = Date.now();
+		const doSubscribe = async () => {
+			log?.info({ "graphql.operation.name": operationName }, "graphql subscribe start");
+			const result = await subscribeFn(options);
+			if (!isAsyncIterable(result)) {
+				const execResult = result;
+				if (execResult.errors?.length) log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					errors: execResult.errors.map((e) => e.message)
+				}, "graphql subscribe completed with errors");
+				return execResult;
+			}
+			log?.info({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start
+			}, "graphql subscription established");
+			if (log) return wrapSubscriptionIterator(result, log, operationName);
+			return result;
+		};
+		if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan("graphql.subscribe", async (span) => {
+			if (log && rumbleInput.logger?.injectTraceId !== false) {
+				const { traceId, spanId } = span.spanContext();
+				log = log.child({
+					traceId,
+					spanId
+				});
+			}
+			try {
+				const result = await doSubscribe();
+				if (!isAsyncIterable(result)) {
+					const execResult = result;
+					if (execResult.errors?.length) {
+						for (const error of execResult.errors) span.recordException(error);
+						span.setStatus({ code: SpanStatusCode.ERROR });
+					}
+				}
+				return result;
+			} catch (error) {
+				if (error instanceof Error) span.recordException(error);
+				log?.error({
+					"graphql.operation.name": operationName,
+					durationMs: Date.now() - start,
+					err: error
+				}, "graphql subscribe threw");
+				span.setStatus({ code: SpanStatusCode.ERROR });
+				throw error;
+			} finally {
+				span.end();
+			}
+		});
+		try {
+			return await doSubscribe();
+		} catch (error) {
+			log?.error({
+				"graphql.operation.name": operationName,
+				durationMs: Date.now() - start,
+				err: error
+			}, "graphql subscribe threw");
+			throw error;
+		}
+	};
+}
+//#endregion
 //#region lib/helpers/sqlTypes/mapDrizzleTypeToGraphQlType.ts
 function buildPothosResponseTypeFromGraphQLType({ builder, sqlType, fieldName, nullable, isArray }) {
 	const gqlType = mapSQLTypeToGraphQLType({
@@ -2309,59 +2496,7 @@ export const r = rumble({
 				...args?.plugins ?? [],
 				...enableApiDocs ? [] : [useDisableIntrospection(), EnvelopArmorPlugin(args?.armorConfig)],
 				rumbleInput.otel?.enabled || rumbleInput.logger?.enabled ? { onExecute: ({ setExecuteFn, executeFn }) => {
-					setExecuteFn(async (options) => {
-						const log = rumbleInput.logger?.enabled ? rumbleInput.logger.logger : void 0;
-						const operationName = options.operationName ?? "anonymous";
-						const start = Date.now();
-						log?.info({ "graphql.operation.name": operationName }, "graphql execute start");
-						const run = async () => {
-							const result = await executeFn(options);
-							if (result && "errors" in result && result.errors?.length) log?.error({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start,
-								errors: result.errors.map((e) => e.message)
-							}, "graphql execute completed with errors");
-							else log?.info({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start
-							}, "graphql execute completed");
-							return result;
-						};
-						if (rumbleInput.otel?.enabled) return rumbleInput.otel.tracer.startActiveSpan(SpanNames.EXECUTE, { attributes: {
-							[AttributeNames.OPERATION_NAME]: operationName,
-							[AttributeNames.SOURCE]: options.document
-						} }, async (span) => {
-							try {
-								const result = await run();
-								if (result && "errors" in result && result.errors?.length) {
-									for (const error of result.errors) span.recordException(error);
-									span.setStatus({ code: SpanStatusCode.ERROR });
-								}
-								return result;
-							} catch (error) {
-								if (error instanceof Error) span.recordException(error);
-								log?.error({
-									"graphql.operation.name": operationName,
-									durationMs: Date.now() - start,
-									err: error
-								}, "graphql execute threw");
-								span.setStatus({ code: SpanStatusCode.ERROR });
-								throw error;
-							} finally {
-								span.end();
-							}
-						});
-						try {
-							return await run();
-						} catch (error) {
-							log?.error({
-								"graphql.operation.name": operationName,
-								durationMs: Date.now() - start,
-								err: error
-							}, "graphql execute threw");
-							throw error;
-						}
-					});
+					setExecuteFn(buildTracedExecute(executeFn, rumbleInput));
 				} } : false
 			].filter(Boolean)
 		});
@@ -2382,6 +2517,8 @@ export const r = rumble({
 			...args,
 			schema: builtSchema(),
 			context,
+			execute: buildTracedExecute(args.execute ?? defaultExecute, rumbleInput),
+			subscribe: buildTracedSubscribe(args.subscribe ?? defaultSubscribe, rumbleInput),
 			errorHandler(errors) {
 				const span = trace.getActiveSpan();
 				for (const error of errors) span?.recordException(error);
@@ -2394,7 +2531,9 @@ export const r = rumble({
 		return implementation({
 			...args,
 			schema: builtSchema(),
-			context
+			context,
+			execute: buildTracedExecute(args.execute ?? defaultExecute, rumbleInput),
+			subscribe: buildTracedSubscribe(args.subscribe ?? defaultSubscribe, rumbleInput)
 		}, ...rest);
 	};
 	return {