@m1212e/rumble 0.16.15 → 0.16.17

package/out/index.cjs

@@ -31,6 +31,7 @@ const require_generate = require('./generate-9zSO5f7n.cjs');
 let graphql = require("graphql");
 let _escape_tech_graphql_armor = require("@escape.tech/graphql-armor");
 let _graphql_yoga_plugin_disable_introspection = require("@graphql-yoga/plugin-disable-introspection");
+let _pothos_tracing_opentelemetry = require("@pothos/tracing-opentelemetry");
 let es_toolkit = require("es-toolkit");
 let graphql_yoga = require("graphql-yoga");
 let sofa_api = require("sofa-api");
@@ -47,6 +48,8 @@ let _pothos_plugin_drizzle = require("@pothos/plugin-drizzle");
 _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 _pothos_plugin_tracing = require("@pothos/plugin-tracing");
+_pothos_plugin_tracing = __toESM(_pothos_plugin_tracing);
 let graphql_scalars = require("graphql-scalars");
 
 //#region lib/types/rumbleError.ts
@@ -374,7 +377,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();
@@ -506,27 +509,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();
 						} };
 					} };
 				};
@@ -949,6 +974,10 @@ async function initSearchIfApplicable(input) {
 		const dbName = (await input.db.execute(drizzle_orm.sql`SELECT current_database()`)).rows[0].current_database;
 		await input.db.execute(drizzle_orm.sql.raw(`ALTER DATABASE ${dbName} SET pg_trgm.similarity_threshold = ${threshold};`));
 	}
+	if (input.search?.cpu_operator_cost) {
+		if (typeof input.search.cpu_operator_cost !== "number") throw new Error(`CPU operator cost must be a number`);
+		if (input.search.cpu_operator_cost <= 0) throw new Error(`CPU operator cost must be a positive number`);
+	}
 }
 /**
 * Performs adjustments to the query args to issue a full text search in case the
@@ -1274,8 +1303,8 @@ 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(drizzle_orm.sql`SET LOCAL cpu_operator_cost = ${search.cpu_operator_cost};`);
+						if (search?.enabled && args.search && args.search.length > 0 && search?.cpu_operator_cost) return db.transaction(async (tx) => {
+							if (isPostgresDB(tx)) await tx.execute(drizzle_orm.sql.raw(`SET 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);
 						});
@@ -1427,22 +1456,43 @@ const applyFilters = async ({ filters, entities, context }) => {
 //#region lib/runtimeFiltersPlugin/runtimeFiltersPlugin.ts
 const applyFiltersKey = "applyFilters";
 var RuntimeFiltersPlugin = class extends _pothos_core.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();
 		};
 	}
 };
@@ -1456,7 +1506,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 ? (0, _pothos_tracing_opentelemetry.createOpenTelemetryWrapper)(otel.tracer, otel.options) : void 0;
 	registerRuntimeFiltersPlugin();
 	const schemaBuilder = new _pothos_core.default({
 		...pothosConfig,
@@ -1464,6 +1515,7 @@ const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig }
 			pluginName,
 			_pothos_plugin_drizzle.default,
 			_pothos_plugin_smart_subscriptions.default,
+			_pothos_plugin_tracing.default,
 			...pothosConfig?.plugins ?? []
 		],
 		drizzle: {
@@ -1479,7 +1531,12 @@ const createSchemaBuilder = ({ db, disableDefaultObjects, pubsub, pothosConfig }
 		smartSubscriptions: { ...(0, _pothos_plugin_smart_subscriptions.subscribeOptionsFromIterator)((name, _context) => {
 			return pubsub.subscribe(name);
 		}) },
-		defaultFieldNullability: false
+		defaultFieldNullability: false,
+		tracing: {
+			default: otel?.tracer ? (config) => (0, _pothos_plugin_tracing.isRootField)(config) : () => false,
+			wrap: createSpan ? (resolver, options) => createSpan(resolver, options) : (resolver) => resolver
+		},
+		otelTracer: otel?.tracer
 	});
 	schemaBuilder.addScalarType("JSON", graphql_scalars.JSONResolver);
 	schemaBuilder.addScalarType("Date", graphql_scalars.DateResolver);
@@ -1515,7 +1572,7 @@ export const db = drizzle(
   "postgres://postgres:postgres@localhost:5432/postgres",
   {
     relations, // <--- add this line
-    schema,   
+    schema,
   },
 );
 
@@ -1578,7 +1635,22 @@ export const db = drizzle(
 		return (0, graphql_yoga.createYoga)({
 			...args,
 			graphiql: enableApiDocs,
-			plugins: [...args?.plugins ?? [], ...enableApiDocs ? [] : [(0, _graphql_yoga_plugin_disable_introspection.useDisableIntrospection)(), (0, _escape_tech_graphql_armor.EnvelopArmorPlugin)()]].filter(Boolean),
+			plugins: [
+				...args?.plugins ?? [],
+				...enableApiDocs ? [] : [(0, _graphql_yoga_plugin_disable_introspection.useDisableIntrospection)(), (0, _escape_tech_graphql_armor.EnvelopArmorPlugin)()],
+				rumbleInput.otel?.tracer ? { onExecute: ({ setExecuteFn, executeFn }) => {
+					setExecuteFn((options) => rumbleInput.otel.tracer.startActiveSpan(_pothos_tracing_opentelemetry.SpanNames.EXECUTE, { attributes: { [_pothos_tracing_opentelemetry.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
 		});