Export as part of handlers

Author: Luca Forstner

packages/node/src/handlers.ts

@@ -10,6 +10,7 @@ import {
   extractTraceparentData,
   isString,
   logger,
+  normalize,
 } from '@sentry/utils';
 import * as domain from 'domain';
 import type * as http from 'http';
@@ -315,6 +316,48 @@ export function errorHandler(options?: {
   };
 }
 
+interface SentryTrpcMiddlewareOptions {
+  attachRpcInput?: boolean;
+}
+
+interface TrpcMiddlewareArguments<T> {
+  path: string;
+  type: 'query' | 'mutation' | 'subscription';
+  next: () => T;
+  rawInput: unknown;
+}
+
+/**
+ * Sentry tRPC middleware that names the handling transaction after the called procedure.
+ *
+ * Use the Sentry tRPC middleware in combination with the Sentry server integration,
+ * e.g. Express Request Handlers or Next.js SDK.
+ */
+export async function trpcMiddleware(options: SentryTrpcMiddlewareOptions = {}) {
+  return function <T>({ path, type, next, rawInput }: TrpcMiddlewareArguments<T>): T {
+    const hub = getCurrentHub();
+    const clientOptions = hub.getClient()?.getOptions();
+    const sentryTransaction = hub.getScope()?.getTransaction();
+
+    if (sentryTransaction) {
+      sentryTransaction.setName(`trcp/${path}`, 'route');
+      sentryTransaction.op = 'rpc.server';
+
+      const trpcData: Record<string, unknown> = {
+        procedureType: type,
+      };
+
+      if (options.attachRpcInput !== undefined ? options.attachRpcInput : clientOptions?.sendDefaultPii) {
+        trpcData.procedureInput = normalize(rawInput);
+      }
+
+      sentryTransaction.setData('trpc', trpcData);
+    }
+
+    return next();
+  };
+}
+
 // TODO (v8 / #5257): Remove this
 // eslint-disable-next-line deprecation/deprecation
 export type { ParseRequestOptions, ExpressRequest } from './requestDataDeprecated';

packages/node/src/index.ts

@@ -65,8 +65,6 @@ const INTEGRATIONS = {
 
 export { INTEGRATIONS as Integrations, Handlers };
 
-export { trpcMiddleware } from './trpc';
-
 // We need to patch domain on the global __SENTRY__ object to make it work for node in cross-platform packages like
 // @sentry/core. If we don't do this, browser bundlers will have troubles resolving `require('domain')`.
 const carrier = getMainCarrier();