@m1212e/rumble 0.16.16 → 0.16.17

package/out/index.mjs

@@ -3,6 +3,7 @@ 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";
+import { AttributeNames, SpanNames, createOpenTelemetryWrapper } from "@pothos/tracing-opentelemetry";
 import { capitalize, cloneDeep, debounce, merge, toMerged } from "es-toolkit";
 import { createPubSub, createYoga } from "graphql-yoga";
 import { useSofa } from "sofa-api";
@@ -15,6 +16,7 @@ import { SQLiteTable } from "drizzle-orm/sqlite-core";
 import SchemaBuilder, { BasePlugin } from "@pothos/core";
 import DrizzlePlugin from "@pothos/plugin-drizzle";
 import SmartSubscriptionsPlugin, { subscribeOptionsFromIterator } from "@pothos/plugin-smart-subscriptions";
+import TracingPlugin, { isRootField } from "@pothos/plugin-tracing";
 import { DateResolver, DateTimeISOResolver, JSONResolver } from "graphql-scalars";
 
 //#region lib/types/rumbleError.ts
@@ -342,7 +344,7 @@ Warning! No abilities have been registered for
 but has been accessed. This will block everything. If this is intended, you can ignore this warning. If not, please ensure that you register the ability in your ability builder.
 `);
 }, 1e3);
-const createAbilityBuilder = ({ db, actions, defaultLimit }) => {
+const createAbilityBuilder = ({ db, actions, defaultLimit, otel }) => {
 	let hasBeenBuilt = false;
 	const createBuilderForTable = () => {
 		const queryFilters = /* @__PURE__ */ new Map();
@@ -474,27 +476,49 @@ const createAbilityBuilder = ({ db, actions, defaultLimit }) => {
 					}
 					return { withContext: (userContext) => {
 						return { filter: (action) => {
-							const filters = queryFilters.get(action);
-							if (filters === "unrestricted") return transformToResponse();
-							if (!filters) {
-								nothingRegisteredWarningLogger(tableName.toString(), action);
-								return transformToResponse(blockEverythingFilter);
-							}
-							const dynamicResults = new Array(dynamicQueryFilters[action].length);
-							let filtersReturned = 0;
-							for (let i = 0; i < dynamicQueryFilters[action].length; i++) {
-								const func = dynamicQueryFilters[action][i];
-								const result = func(userContext);
-								if (result === "allow") return transformToResponse();
-								if (result === void 0) continue;
-								dynamicResults[filtersReturned++] = result;
-							}
-							dynamicResults.length = filtersReturned;
-							const allQueryFilters = [...simpleQueryFilters[action], ...dynamicResults];
-							if (allQueryFilters.length === 0) return transformToResponse(blockEverythingFilter);
-							return transformToResponse(allQueryFilters.length === 1 ? allQueryFilters[0] : allQueryFilters.reduce((a, b) => {
-								return mergeFilters(a, b);
-							}, {}));
+							const assembleAbilities = (span) => {
+								const filters = queryFilters.get(action);
+								if (filters === "unrestricted") {
+									span?.setAttribute("abilities.status", "unrestricted");
+									return transformToResponse();
+								}
+								if (!filters) {
+									span?.setAttribute("abilities.status", "blocked_everything");
+									nothingRegisteredWarningLogger(tableName.toString(), action);
+									return transformToResponse(blockEverythingFilter);
+								}
+								const dynamicResults = new Array(dynamicQueryFilters[action].length);
+								let filtersReturned = 0;
+								for (let i = 0; i < dynamicQueryFilters[action].length; i++) {
+									const func = dynamicQueryFilters[action][i];
+									const result = func(userContext);
+									if (result === "allow") return transformToResponse();
+									if (result === void 0) continue;
+									dynamicResults[filtersReturned++] = result;
+								}
+								dynamicResults.length = filtersReturned;
+								span?.setAttribute("abilities.dynamic", dynamicResults.length);
+								span?.setAttribute("abilities.static", simpleQueryFilters[action].length);
+								const allQueryFilters = [...simpleQueryFilters[action], ...dynamicResults];
+								span?.setAttribute("abilities.total", allQueryFilters.length);
+								if (allQueryFilters.length === 0) {
+									span?.setAttribute("abilities.status", "blocked_everything");
+									return transformToResponse(blockEverythingFilter);
+								}
+								const mergedFilters = allQueryFilters.length === 1 ? allQueryFilters[0] : allQueryFilters.reduce((a, b) => {
+									return mergeFilters(a, b);
+								}, {});
+								span?.setAttribute("abilities.status", "applied");
+								return transformToResponse(mergedFilters);
+							};
+							if (otel?.tracer) return otel.tracer.startActiveSpan(`prepare_query_abilities_${action}`, (span) => {
+								try {
+									return assembleAbilities(span);
+								} finally {
+									span.end();
+								}
+							});
+							else return assembleAbilities();
 						} };
 					} };
 				};
@@ -1399,22 +1423,43 @@ const applyFilters = async ({ filters, entities, context }) => {
 //#region lib/runtimeFiltersPlugin/runtimeFiltersPlugin.ts
 const applyFiltersKey = "applyFilters";
 var RuntimeFiltersPlugin = class extends BasePlugin {
+	tracer;
+	onTypeConfig(typeConfig) {
+		this.tracer = this.builder.options.otelTracer;
+		return typeConfig;
+	}
 	wrapResolve(resolver, fieldConfig) {
 		return async (parent, args, context, info) => {
-			let filters;
-			const fieldType = fieldConfig?.type;
-			if (fieldType.kind === "List") filters = fieldType.type?.ref.currentConfig.pothosOptions[applyFiltersKey];
-			else if (fieldType.kind === "Object") filters = fieldType.ref.currentConfig.pothosOptions[applyFiltersKey];
-			if (!filters || !Array.isArray(filters) || filters.length === 0) return resolver(parent, args, context, info);
-			const resolved = await resolver(parent, args, context, info);
-			const allResolvedValues = Array.isArray(resolved) ? resolved : [resolved];
-			const allowed = await applyFilters({
-				filters: Array.isArray(filters) ? filters : [filters],
-				entities: allResolvedValues,
-				context
+			const runFilters = async (span) => {
+				let filters;
+				const fieldType = fieldConfig?.type;
+				if (fieldType.kind === "List") filters = fieldType.type?.ref.currentConfig.pothosOptions[applyFiltersKey];
+				else if (fieldType.kind === "Object") filters = fieldType.ref.currentConfig.pothosOptions[applyFiltersKey];
+				if (!filters || !Array.isArray(filters) || filters.length === 0) {
+					span?.setAttribute("filters.status", "unrestricted");
+					return resolver(parent, args, context, info);
+				}
+				const resolved = await resolver(parent, args, context, info);
+				const allResolvedValues = Array.isArray(resolved) ? resolved : [resolved];
+				const allFilters = Array.isArray(filters) ? filters : [filters];
+				span?.setAttribute("filters.total", allFilters.length);
+				const allowed = await applyFilters({
+					filters: allFilters,
+					entities: allResolvedValues,
+					context
+				});
+				span?.setAttribute("filters.allowed", allowed.length);
+				if (Array.isArray(resolved)) return allowed;
+				return allowed[0] ?? null;
+			};
+			if (this.tracer) return this.tracer.startActiveSpan(`apply_filters_${fieldConfig.name}`, async (span) => {
+				try {
+					return await runFilters();
+				} finally {
+					span.end();
+				}
 			});
-			if (Array.isArray(resolved)) return allowed;
-			return allowed[0] ?? null;
+			else return runFilters();
 		};
 	}
 };
@@ -1428,7 +1473,8 @@ function registerRuntimeFiltersPlugin() {
 
 //#endregion
 //#region lib/schemaBuilder.ts
-const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig }) => {
+const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig, otel }) => {
+	const createSpan = otel?.tracer ? createOpenTelemetryWrapper(otel.tracer, otel.options) : void 0;
 	registerRuntimeFiltersPlugin();
 	const schemaBuilder = new SchemaBuilder({
 		...pothosConfig,
@@ -1436,6 +1482,7 @@ const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig }
 			pluginName,
 			DrizzlePlugin,
 			SmartSubscriptionsPlugin,
+			TracingPlugin,
 			...pothosConfig?.plugins ?? []
 		],
 		drizzle: {
@@ -1451,7 +1498,12 @@ const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig }
 		smartSubscriptions: { ...subscribeOptionsFromIterator((name, _context) => {
 			return pubsub.subscribe(name);
 		}) },
-		defaultFieldNullability: false
+		defaultFieldNullability: false,
+		tracing: {
+			default: otel?.tracer ? (config) => isRootField(config) : () => false,
+			wrap: createSpan ? (resolver, options) => createSpan(resolver, options) : (resolver) => resolver
+		},
+		otelTracer: otel?.tracer
 	});
 	schemaBuilder.addScalarType("JSON", JSONResolver);
 	schemaBuilder.addScalarType("Date", DateResolver);
@@ -1487,7 +1539,7 @@ export const db = drizzle(
   "postgres://postgres:postgres@localhost:5432/postgres",
   {
     relations, // <--- add this line
-    schema,   
+    schema,
   },
 );
 
@@ -1550,7 +1602,22 @@ export const db = drizzle(
 		return createYoga({
 			...args,
 			graphiql: enableApiDocs,
-			plugins: [...args?.plugins ?? [], ...enableApiDocs ? [] : [useDisableIntrospection(), EnvelopArmorPlugin()]].filter(Boolean),
+			plugins: [
+				...args?.plugins ?? [],
+				...enableApiDocs ? [] : [useDisableIntrospection(), EnvelopArmorPlugin()],
+				rumbleInput.otel?.tracer ? { onExecute: ({ setExecuteFn, executeFn }) => {
+					setExecuteFn((options) => rumbleInput.otel.tracer.startActiveSpan(SpanNames.EXECUTE, { attributes: { [AttributeNames.OPERATION_NAME]: options.operationName ?? "anonymous" } }, async (span) => {
+						try {
+							return await executeFn(options);
+						} catch (error) {
+							span.recordException(error);
+							throw error;
+						} finally {
+							span.end();
+						}
+					}));
+				} } : false
+			].filter(Boolean),
 			schema: builtSchema(),
 			context
 		});