@depup/sentry__server-utils 10.67.0-depup.0

package/build/esm/integrations/tracing-channel/aws-sdk/index.js

@@ -0,0 +1,152 @@
+import * as diagnosticsChannel from 'node:diagnostics_channel';
+import { defineIntegration, debug, waitForTracingChannelBinding, startInactiveSpan, SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, SPAN_KIND } from '@sentry/core';
+import { CLOUD_REGION, _AWS_REQUEST_ID, HTTP_STATUS_CODE, AWS_REQUEST_EXTENDED_ID } from '@sentry/conventions/attributes';
+import { DEBUG_BUILD } from '../../../debug-build.js';
+import { CHANNELS } from '../../../orchestrion/channels.js';
+import { bindTracingChannelToSpan } from '../../../tracing-channel.js';
+import { AWS_SDK_ORIGIN } from './constants.js';
+import { ServicesExtensions } from './services/ServicesExtensions.js';
+import { removeSuffixFromStringIfExists, normalizeV3Request, extractAttributesFromNormalizedRequest } from './utils.js';
+
+const INTEGRATION_NAME = "Aws";
+function safe(fn) {
+  try {
+    return fn();
+  } catch (error) {
+    DEBUG_BUILD && debug.warn("[orchestrion:aws-sdk] error building span", error);
+    return void 0;
+  }
+}
+function setMetadataAttributes(span, metadata) {
+  if (!metadata) {
+    return;
+  }
+  if (metadata.requestId) {
+    span.setAttribute(_AWS_REQUEST_ID, metadata.requestId);
+  }
+  if (metadata.httpStatusCode) {
+    span.setAttribute(HTTP_STATUS_CODE, metadata.httpStatusCode);
+  }
+  if (metadata.extendedRequestId) {
+    span.setAttribute(AWS_REQUEST_EXTENDED_ID, metadata.extendedRequestId);
+  }
+}
+const _awsChannelIntegration = (() => {
+  const servicesExtensions = new ServicesExtensions();
+  return {
+    name: INTEGRATION_NAME,
+    setupOnce() {
+      if (!diagnosticsChannel.tracingChannel) {
+        return;
+      }
+      const getSpan = (data) => safe(() => {
+        const command = data.arguments[0];
+        const commandName = command?.constructor?.name;
+        if (!command || !commandName) {
+          return void 0;
+        }
+        const clientConfig = data.self?.config;
+        const serviceName = clientConfig?.serviceId ?? // `clientName` isn't available at the `send` boundary; fall back to the client's
+        // constructor name (e.g. `S3Client` -> `S3`). `serviceId` is set for all AWS clients.
+        removeSuffixFromStringIfExists(data.self?.constructor?.name || "AWS", "Client");
+        if (!command.input) {
+          command.input = {};
+        }
+        const normalizedRequest = normalizeV3Request(serviceName, commandName, command.input, void 0);
+        const requestMetadata = servicesExtensions.requestPreSpanHook(normalizedRequest);
+        const span = startInactiveSpan({
+          name: requestMetadata.spanName ?? `${normalizedRequest.serviceName}.${normalizedRequest.commandName}`,
+          kind: requestMetadata.spanKind ?? SPAN_KIND.CLIENT,
+          // `rpc` matches what the exporter infers from `rpc.service` for the OTel aws-sdk spans;
+          // service extensions override it where inference yields a different op (DynamoDB: `db`).
+          op: requestMetadata.spanOp || "rpc",
+          attributes: {
+            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: AWS_SDK_ORIGIN,
+            ...extractAttributesFromNormalizedRequest(normalizedRequest),
+            ...requestMetadata.spanAttributes
+          }
+        });
+        data._sentryNormalizedRequest = normalizedRequest;
+        data._sentryRequestMetadata = requestMetadata;
+        let regionResult;
+        try {
+          regionResult = clientConfig?.region?.();
+        } catch {
+        }
+        const regionHolder = {
+          settled: false,
+          promise: Promise.resolve(regionResult).then((region) => {
+            if (region) {
+              normalizedRequest.region = region;
+              span.setAttribute(CLOUD_REGION, region);
+            }
+          }).catch(() => {
+          }).finally(() => {
+            regionHolder.settled = true;
+          })
+        };
+        data._sentryRegion = regionHolder;
+        safe(() => servicesExtensions.requestPostSpanHook(normalizedRequest, span));
+        return span;
+      });
+      const opts = {
+        deferSpanEnd({ span, data, end }) {
+          const normalizedRequest = data._sentryNormalizedRequest;
+          const requestMetadata = data._sentryRequestMetadata;
+          if (!normalizedRequest) {
+            return false;
+          }
+          const failed = "error" in data;
+          safe(() => {
+            if (failed) {
+              const err = data.error;
+              const errMetadata = err?.$metadata;
+              setMetadataAttributes(span, {
+                requestId: err?.RequestId ?? errMetadata?.requestId,
+                httpStatusCode: errMetadata?.httpStatusCode,
+                extendedRequestId: err?.extendedRequestId ?? errMetadata?.extendedRequestId
+              });
+              return;
+            }
+            const output = data.result;
+            setMetadataAttributes(span, output?.$metadata);
+            const normalizedResponse = {
+              data: output,
+              request: normalizedRequest,
+              requestId: output?.$metadata?.requestId
+            };
+            servicesExtensions.responseHook(normalizedResponse, span);
+          });
+          if (requestMetadata?.isStream && !failed) {
+            return true;
+          }
+          const region = data._sentryRegion;
+          if (region && !region.settled) {
+            void region.promise.then(() => end());
+            return true;
+          }
+          return false;
+        }
+      };
+      const awsSendChannels = [
+        CHANNELS.AWS_SMITHY_CORE_SEND,
+        CHANNELS.AWS_SMITHY_CLIENT_SEND,
+        CHANNELS.AWS_SDK_SMITHY_CLIENT_SEND
+      ];
+      DEBUG_BUILD && debug.log(`[orchestrion:aws-sdk] subscribing to channels "${awsSendChannels.join('", "')}"`);
+      waitForTracingChannelBinding(() => {
+        for (const channelName of awsSendChannels) {
+          bindTracingChannelToSpan(
+            diagnosticsChannel.tracingChannel(channelName),
+            getSpan,
+            opts
+          );
+        }
+      });
+    }
+  };
+});
+const awsChannelIntegration = defineIntegration(_awsChannelIntegration);
+
+export { awsChannelIntegration };
+//# sourceMappingURL=index.js.map

package/build/esm/integrations/tracing-channel/aws-sdk/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../../../../../src/integrations/tracing-channel/aws-sdk/index.ts"],"sourcesContent":["import * as diagnosticsChannel from 'node:diagnostics_channel';\nimport type { IntegrationFn, Span } from '@sentry/core';\nimport {\n  debug,\n  defineIntegration,\n  SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN,\n  SPAN_KIND,\n  startInactiveSpan,\n  waitForTracingChannelBinding,\n} from '@sentry/core';\nimport {\n  _AWS_REQUEST_ID as AWS_REQUEST_ID,\n  AWS_REQUEST_EXTENDED_ID,\n  CLOUD_REGION,\n  HTTP_STATUS_CODE,\n} from '@sentry/conventions/attributes';\nimport { DEBUG_BUILD } from '../../../debug-build';\nimport { CHANNELS } from '../../../orchestrion/channels';\nimport type { TracingChannelLifeCycleOptions } from '../../../tracing-channel';\nimport { bindTracingChannelToSpan } from '../../../tracing-channel';\nimport { AWS_SDK_ORIGIN } from './constants';\nimport { ServicesExtensions } from './services';\nimport type { NormalizedRequest, NormalizedResponse, RequestMetadata } from './types';\nimport { extractAttributesFromNormalizedRequest, normalizeV3Request, removeSuffixFromStringIfExists } from './utils';\n\n// Same name as the OTel `Aws` integration by design, so enabling injection swaps this in for it.\nconst INTEGRATION_NAME = 'Aws' as const;\n\n// The context orchestrion's transform attaches to the channel: `arguments` is the live args of the\n// wrapped `Client.prototype.send` call (`[command, ...]`), `self` the client, `result`/`error` the\n// settled value. The `_sentry*` fields are stashed by us across the call's lifecycle.\ninterface AwsSendChannelContext {\n  arguments: unknown[];\n  self?: { config?: AwsClientConfig; constructor?: { name?: string } };\n  result?: unknown;\n  error?: unknown;\n  _sentryNormalizedRequest?: NormalizedRequest;\n  _sentryRequestMetadata?: RequestMetadata;\n  _sentryRegion?: { settled: boolean; promise: Promise<void> };\n}\n\ninterface AwsClientConfig {\n  serviceId?: string;\n  region?: () => string | Promise<string> | undefined;\n}\n\ninterface AwsV3Command {\n  input?: Record<string, unknown>;\n  constructor?: { name?: string };\n}\n\n/** Runs a span-building callback so a throw inside it can never break the user's aws-sdk call. */\nfunction safe<T>(fn: () => T): T | undefined {\n  try {\n    return fn();\n  } catch (error) {\n    DEBUG_BUILD && debug.warn('[orchestrion:aws-sdk] error building span', error);\n    return undefined;\n  }\n}\n\n// `metadata` is smithy's `ResponseMetadata`, read off the untyped channel result/error (`any` for the\n// same reason as `CommandInput`, see types.ts).\nfunction setMetadataAttributes(span: Span, metadata: Record<string, any> | undefined): void {\n  if (!metadata) {\n    return;\n  }\n  if (metadata.requestId) {\n    // oxlint-disable-next-line typescript/no-deprecated\n    span.setAttribute(AWS_REQUEST_ID, metadata.requestId);\n  }\n  if (metadata.httpStatusCode) {\n    // oxlint-disable-next-line typescript/no-deprecated\n    span.setAttribute(HTTP_STATUS_CODE, metadata.httpStatusCode);\n  }\n  if (metadata.extendedRequestId) {\n    // oxlint-disable-next-line typescript/no-deprecated\n    span.setAttribute(AWS_REQUEST_EXTENDED_ID, metadata.extendedRequestId);\n  }\n}\n\nconst _awsChannelIntegration = (() => {\n  const servicesExtensions = new ServicesExtensions();\n\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      // `tracingChannel` is unavailable before Node 18.19 so do nothing in that case.\n      if (!diagnosticsChannel.tracingChannel) {\n        return;\n      }\n\n      const getSpan = (data: AwsSendChannelContext): Span | undefined =>\n        safe(() => {\n          const command = data.arguments[0] as AwsV3Command | undefined;\n          const commandName = command?.constructor?.name;\n          if (!command || !commandName) {\n            // Not a recognizable v3 command call; leave the active context untouched.\n            return undefined;\n          }\n\n          const clientConfig = data.self?.config;\n          const serviceName =\n            clientConfig?.serviceId ??\n            // `clientName` isn't available at the `send` boundary; fall back to the client's\n            // constructor name (e.g. `S3Client` -> `S3`). `serviceId` is set for all AWS clients.\n            removeSuffixFromStringIfExists(data.self?.constructor?.name || 'AWS', 'Client');\n\n          // Commands with all-optional members can be constructed without an input (`new\n          // ListBucketsCommand()`); the OTel path traces those too, so default rather than bail.\n          // The default is assigned back onto the command (not kept detached) because service hooks\n          // mutate `commandInput` (trace-propagation headers, `MessageAttributeNames`) and those\n          // writes must reach the serialized request. Current smithy clients already default `input`\n          // to `{}` in the command constructor; this only affects older clients in our range.\n          if (!command.input) {\n            command.input = {};\n          }\n          const normalizedRequest = normalizeV3Request(serviceName, commandName, command.input, undefined);\n          const requestMetadata = servicesExtensions.requestPreSpanHook(normalizedRequest);\n\n          const span = startInactiveSpan({\n            name: requestMetadata.spanName ?? `${normalizedRequest.serviceName}.${normalizedRequest.commandName}`,\n            kind: requestMetadata.spanKind ?? SPAN_KIND.CLIENT,\n            // `rpc` matches what the exporter infers from `rpc.service` for the OTel aws-sdk spans;\n            // service extensions override it where inference yields a different op (DynamoDB: `db`).\n            op: requestMetadata.spanOp || 'rpc',\n            attributes: {\n              [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: AWS_SDK_ORIGIN,\n              ...extractAttributesFromNormalizedRequest(normalizedRequest),\n              ...requestMetadata.spanAttributes,\n            },\n          });\n\n          data._sentryNormalizedRequest = normalizedRequest;\n          data._sentryRequestMetadata = requestMetadata;\n\n          // `region` resolves asynchronously while `send` proceeds (a channel subscriber cannot delay\n          // the traced call the way the OTel middleware does). Backfill it onto the span and the\n          // normalized request once available; `deferSpanEnd` holds the span open until this settles\n          // so `cloud.region` cannot be lost when `send` settles first (e.g. an early failure).\n          //\n          // The provider call is guarded separately: the span is already started, so a synchronous\n          // throw bubbling into the enclosing `safe` would discard it without ending it (a leaked\n          // open span).\n          let regionResult: string | Promise<string> | undefined;\n          try {\n            regionResult = clientConfig?.region?.();\n          } catch {\n            // Nothing to do; continue without a region.\n          }\n          // The `.finally` self-reference is safe: the callback only runs after initialization.\n          const regionHolder: { settled: boolean; promise: Promise<void> } = {\n            settled: false,\n            promise: Promise.resolve(regionResult)\n              .then(region => {\n                if (region) {\n                  normalizedRequest.region = region;\n                  span.setAttribute(CLOUD_REGION, region);\n                }\n              })\n              .catch(() => {\n                // Nothing to do; continue without a region.\n              })\n              .finally(() => {\n                regionHolder.settled = true;\n              }),\n          };\n          data._sentryRegion = regionHolder;\n\n          // Inject trace-propagation headers into outgoing messages (SQS/SNS/Lambda). Runs before\n          // `send` proceeds, so the mutated `commandInput` is used to build the request.\n          safe(() => servicesExtensions.requestPostSpanHook(normalizedRequest, span));\n\n          return span;\n        });\n\n      const opts: TracingChannelLifeCycleOptions<AwsSendChannelContext> = {\n        deferSpanEnd({ span, data, end }) {\n          const normalizedRequest = data._sentryNormalizedRequest;\n          const requestMetadata = data._sentryRequestMetadata;\n          if (!normalizedRequest) {\n            return false;\n          }\n\n          const failed = 'error' in data;\n\n          // The channel `result`/`error` are untyped; the `$metadata` casts below name smithy's\n          // `ResponseMetadata` shape (`any`-valued, see `setMetadataAttributes`).\n          safe(() => {\n            if (failed) {\n              const err = data.error as\n                | { $metadata?: Record<string, any>; RequestId?: string; extendedRequestId?: string }\n                | undefined;\n              const errMetadata = err?.$metadata;\n              // Like the OTel path, read RequestId/extendedRequestId off the error itself, with\n              // `$metadata` (which smithy service errors also carry) as the fallback. A spread won't\n              // do: `$metadata` includes these keys with `undefined` values, clobbering the fallback.\n              setMetadataAttributes(span, {\n                requestId: err?.RequestId ?? errMetadata?.requestId,\n                httpStatusCode: errMetadata?.httpStatusCode,\n                extendedRequestId: err?.extendedRequestId ?? errMetadata?.extendedRequestId,\n              });\n              return;\n            }\n\n            const output = data.result as { $metadata?: Record<string, any> } | undefined;\n            setMetadataAttributes(span, output?.$metadata);\n\n            const normalizedResponse: NormalizedResponse = {\n              data: output,\n              request: normalizedRequest,\n              requestId: output?.$metadata?.requestId,\n            };\n            servicesExtensions.responseHook(normalizedResponse, span);\n          });\n\n          // Streaming responses end the span when their wrapped stream is consumed (see\n          // bedrock-runtime); the helper must not end it on `send` settling. Errors always end here.\n          if (requestMetadata?.isStream && !failed) {\n            return true;\n          }\n\n          // Normally the region settles long before `send` does (the SDK awaits it internally to\n          // build the endpoint), but when `send` settles first (e.g. an early failure) hold the span\n          // open until the region backfill lands. The error status was already applied by the\n          // helper's `error` subscriber, so a plain `end()` suffices.\n          const region = data._sentryRegion;\n          if (region && !region.settled) {\n            void region.promise.then(() => end());\n            return true;\n          }\n\n          return false;\n        },\n      };\n\n      // The AWS SDK's `Client.prototype.send` lives in different smithy packages across versions; the\n      // transform injects one channel per package. Only the package hosting the app's client fires, so\n      // subscribing to all of them is safe and never double-instruments a single call.\n      const awsSendChannels = [\n        CHANNELS.AWS_SMITHY_CORE_SEND,\n        CHANNELS.AWS_SMITHY_CLIENT_SEND,\n        CHANNELS.AWS_SDK_SMITHY_CLIENT_SEND,\n      ] as const;\n\n      DEBUG_BUILD && debug.log(`[orchestrion:aws-sdk] subscribing to channels \"${awsSendChannels.join('\", \"')}\"`);\n\n      waitForTracingChannelBinding(() => {\n        for (const channelName of awsSendChannels) {\n          bindTracingChannelToSpan(\n            diagnosticsChannel.tracingChannel<AwsSendChannelContext>(channelName),\n            getSpan,\n            opts,\n          );\n        }\n      });\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * EXPERIMENTAL — orchestrion-driven aws-sdk (v3) integration.\n *\n * Subscribes to the `orchestrion:@smithy/smithy-client:send` (and equivalent) diagnostics_channel\n * the orchestrion code transform injects into the AWS SDK's smithy `Client.prototype.send`, emitting\n * spans identical to the OTel `@opentelemetry/instrumentation-aws-sdk` integration (with a distinct\n * `auto.aws.orchestrion.aws_sdk` origin). Requires the orchestrion runtime hook or bundler plugin —\n * wire it up via `experimentalUseDiagnosticsChannelInjection()`.\n *\n * @experimental\n */\nexport const awsChannelIntegration = defineIntegration(_awsChannelIntegration);\n"],"names":["AWS_REQUEST_ID"],"mappings":";;;;;;;;;;AA0BA,MAAM,gBAAA,GAAmB,KAAA;AA0BzB,SAAS,KAAQ,EAAA,EAA4B;AAC3C,EAAA,IAAI;AACF,IAAA,OAAO,EAAA,EAAG;AAAA,EACZ,SAAS,KAAA,EAAO;AACd,IAAA,WAAA,IAAe,KAAA,CAAM,IAAA,CAAK,2CAAA,EAA6C,KAAK,CAAA;AAC5E,IAAA,OAAO,MAAA;AAAA,EACT;AACF;AAIA,SAAS,qBAAA,CAAsB,MAAY,QAAA,EAAiD;AAC1F,EAAA,IAAI,CAAC,QAAA,EAAU;AACb,IAAA;AAAA,EACF;AACA,EAAA,IAAI,SAAS,SAAA,EAAW;AAEtB,IAAA,IAAA,CAAK,YAAA,CAAaA,eAAA,EAAgB,QAAA,CAAS,SAAS,CAAA;AAAA,EACtD;AACA,EAAA,IAAI,SAAS,cAAA,EAAgB;AAE3B,IAAA,IAAA,CAAK,YAAA,CAAa,gBAAA,EAAkB,QAAA,CAAS,cAAc,CAAA;AAAA,EAC7D;AACA,EAAA,IAAI,SAAS,iBAAA,EAAmB;AAE9B,IAAA,IAAA,CAAK,YAAA,CAAa,uBAAA,EAAyB,QAAA,CAAS,iBAAiB,CAAA;AAAA,EACvE;AACF;AAEA,MAAM,0BAA0B,MAAM;AACpC,EAAA,MAAM,kBAAA,GAAqB,IAAI,kBAAA,EAAmB;AAElD,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,gBAAA;AAAA,IACN,SAAA,GAAY;AAEV,MAAA,IAAI,CAAC,mBAAmB,cAAA,EAAgB;AACtC,QAAA;AAAA,MACF;AAEA,MAAA,MAAM,OAAA,GAAU,CAAC,IAAA,KACf,IAAA,CAAK,MAAM;AACT,QAAA,MAAM,OAAA,GAAU,IAAA,CAAK,SAAA,CAAU,CAAC,CAAA;AAChC,QAAA,MAAM,WAAA,GAAc,SAAS,WAAA,EAAa,IAAA;AAC1C,QAAA,IAAI,CAAC,OAAA,IAAW,CAAC,WAAA,EAAa;AAE5B,UAAA,OAAO,KAAA,CAAA;AAAA,QACT;AAEA,QAAA,MAAM,YAAA,GAAe,KAAK,IAAA,EAAM,MAAA;AAChC,QAAA,MAAM,cACJ,YAAA,EAAc,SAAA;AAAA;AAAA,QAGd,+BAA+B,IAAA,CAAK,IAAA,EAAM,WAAA,EAAa,IAAA,IAAQ,OAAO,QAAQ,CAAA;AAQhF,QAAA,IAAI,CAAC,QAAQ,KAAA,EAAO;AAClB,UAAA,OAAA,CAAQ,QAAQ,EAAC;AAAA,QACnB;AACA,QAAA,MAAM,oBAAoB,kBAAA,CAAmB,WAAA,EAAa,WAAA,EAAa,OAAA,CAAQ,OAAO,KAAA,CAAS,CAAA;AAC/F,QAAA,MAAM,eAAA,GAAkB,kBAAA,CAAmB,kBAAA,CAAmB,iBAAiB,CAAA;AAE/E,QAAA,MAAM,OAAO,iBAAA,CAAkB;AAAA,UAC7B,IAAA,EAAM,gBAAgB,QAAA,IAAY,CAAA,EAAG,kBAAkB,WAAW,CAAA,CAAA,EAAI,kBAAkB,WAAW,CAAA,CAAA;AAAA,UACnG,IAAA,EAAM,eAAA,CAAgB,QAAA,IAAY,SAAA,CAAU,MAAA;AAAA;AAAA;AAAA,UAG5C,EAAA,EAAI,gBAAgB,MAAA,IAAU,KAAA;AAAA,UAC9B,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,cAAA;AAAA,YACpC,GAAG,uCAAuC,iBAAiB,CAAA;AAAA,YAC3D,GAAG,eAAA,CAAgB;AAAA;AACrB,SACD,CAAA;AAED,QAAA,IAAA,CAAK,wBAAA,GAA2B,iBAAA;AAChC,QAAA,IAAA,CAAK,sBAAA,GAAyB,eAAA;AAU9B,QAAA,IAAI,YAAA;AACJ,QAAA,IAAI;AACF,UAAA,YAAA,GAAe,cAAc,MAAA,IAAS;AAAA,QACxC,CAAA,CAAA,MAAQ;AAAA,QAER;AAEA,QAAA,MAAM,YAAA,GAA6D;AAAA,UACjE,OAAA,EAAS,KAAA;AAAA,UACT,SAAS,OAAA,CAAQ,OAAA,CAAQ,YAAY,CAAA,CAClC,KAAK,CAAA,MAAA,KAAU;AACd,YAAA,IAAI,MAAA,EAAQ;AACV,cAAA,iBAAA,CAAkB,MAAA,GAAS,MAAA;AAC3B,cAAA,IAAA,CAAK,YAAA,CAAa,cAAc,MAAM,CAAA;AAAA,YACxC;AAAA,UACF,CAAC,CAAA,CACA,KAAA,CAAM,MAAM;AAAA,UAEb,CAAC,CAAA,CACA,OAAA,CAAQ,MAAM;AACb,YAAA,YAAA,CAAa,OAAA,GAAU,IAAA;AAAA,UACzB,CAAC;AAAA,SACL;AACA,QAAA,IAAA,CAAK,aAAA,GAAgB,YAAA;AAIrB,QAAA,IAAA,CAAK,MAAM,kBAAA,CAAmB,mBAAA,CAAoB,iBAAA,EAAmB,IAAI,CAAC,CAAA;AAE1E,QAAA,OAAO,IAAA;AAAA,MACT,CAAC,CAAA;AAEH,MAAA,MAAM,IAAA,GAA8D;AAAA,QAClE,YAAA,CAAa,EAAE,IAAA,EAAM,IAAA,EAAM,KAAI,EAAG;AAChC,UAAA,MAAM,oBAAoB,IAAA,CAAK,wBAAA;AAC/B,UAAA,MAAM,kBAAkB,IAAA,CAAK,sBAAA;AAC7B,UAAA,IAAI,CAAC,iBAAA,EAAmB;AACtB,YAAA,OAAO,KAAA;AAAA,UACT;AAEA,UAAA,MAAM,SAAS,OAAA,IAAW,IAAA;AAI1B,UAAA,IAAA,CAAK,MAAM;AACT,YAAA,IAAI,MAAA,EAAQ;AACV,cAAA,MAAM,MAAM,IAAA,CAAK,KAAA;AAGjB,cAAA,MAAM,cAAc,GAAA,EAAK,SAAA;AAIzB,cAAA,qBAAA,CAAsB,IAAA,EAAM;AAAA,gBAC1B,SAAA,EAAW,GAAA,EAAK,SAAA,IAAa,WAAA,EAAa,SAAA;AAAA,gBAC1C,gBAAgB,WAAA,EAAa,cAAA;AAAA,gBAC7B,iBAAA,EAAmB,GAAA,EAAK,iBAAA,IAAqB,WAAA,EAAa;AAAA,eAC3D,CAAA;AACD,cAAA;AAAA,YACF;AAEA,YAAA,MAAM,SAAS,IAAA,CAAK,MAAA;AACpB,YAAA,qBAAA,CAAsB,IAAA,EAAM,QAAQ,SAAS,CAAA;AAE7C,YAAA,MAAM,kBAAA,GAAyC;AAAA,cAC7C,IAAA,EAAM,MAAA;AAAA,cACN,OAAA,EAAS,iBAAA;AAAA,cACT,SAAA,EAAW,QAAQ,SAAA,EAAW;AAAA,aAChC;AACA,YAAA,kBAAA,CAAmB,YAAA,CAAa,oBAAoB,IAAI,CAAA;AAAA,UAC1D,CAAC,CAAA;AAID,UAAA,IAAI,eAAA,EAAiB,QAAA,IAAY,CAAC,MAAA,EAAQ;AACxC,YAAA,OAAO,IAAA;AAAA,UACT;AAMA,UAAA,MAAM,SAAS,IAAA,CAAK,aAAA;AACpB,UAAA,IAAI,MAAA,IAAU,CAAC,MAAA,CAAO,OAAA,EAAS;AAC7B,YAAA,KAAK,MAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,MAAM,KAAK,CAAA;AACpC,YAAA,OAAO,IAAA;AAAA,UACT;AAEA,UAAA,OAAO,KAAA;AAAA,QACT;AAAA,OACF;AAKA,MAAA,MAAM,eAAA,GAAkB;AAAA,QACtB,QAAA,CAAS,oBAAA;AAAA,QACT,QAAA,CAAS,sBAAA;AAAA,QACT,QAAA,CAAS;AAAA,OACX;AAEA,MAAA,WAAA,IAAe,MAAM,GAAA,CAAI,CAAA,+CAAA,EAAkD,gBAAgB,IAAA,CAAK,MAAM,CAAC,CAAA,CAAA,CAAG,CAAA;AAE1G,MAAA,4BAAA,CAA6B,MAAM;AACjC,QAAA,KAAA,MAAW,eAAe,eAAA,EAAiB;AACzC,UAAA,wBAAA;AAAA,YACE,kBAAA,CAAmB,eAAsC,WAAW,CAAA;AAAA,YACpE,OAAA;AAAA,YACA;AAAA,WACF;AAAA,QACF;AAAA,MACF,CAAC,CAAA;AAAA,IACH;AAAA,GACF;AACF,CAAA,CAAA;AAaO,MAAM,qBAAA,GAAwB,kBAAkB,sBAAsB;;;;"}

package/build/esm/integrations/tracing-channel/aws-sdk/services/MessageAttributes.js

@@ -0,0 +1,41 @@
+import { uniq, debug } from '@sentry/core';
+import { DEBUG_BUILD } from '../../../../debug-build.js';
+
+const MAX_MESSAGE_ATTRIBUTES = 10;
+const SENTRY_TRACE_HEADER = "sentry-trace";
+const BAGGAGE_HEADER = "baggage";
+const PROPAGATION_FIELDS = [SENTRY_TRACE_HEADER, BAGGAGE_HEADER];
+function injectPropagationContext(attributesMap, traceData) {
+  const attributes = attributesMap ?? {};
+  const headerKeys = Object.keys(traceData);
+  if (Object.keys(attributes).length + headerKeys.length <= MAX_MESSAGE_ATTRIBUTES) {
+    for (const key of headerKeys) {
+      const value = traceData[key];
+      if (value) {
+        attributes[key] = { DataType: "String", StringValue: value };
+      }
+    }
+  } else {
+    DEBUG_BUILD && debug.warn(
+      "[orchestrion:aws-sdk] cannot set trace propagation on SQS/SNS message due to maximum amount of MessageAttributes"
+    );
+  }
+  return attributes;
+}
+function extractPropagationHeaders(message) {
+  const carrier = message.MessageAttributes ?? {};
+  const sentryTrace = carrier[SENTRY_TRACE_HEADER]?.StringValue ?? carrier[SENTRY_TRACE_HEADER]?.Value;
+  if (!sentryTrace) {
+    return void 0;
+  }
+  return {
+    sentryTrace,
+    baggage: carrier[BAGGAGE_HEADER]?.StringValue ?? carrier[BAGGAGE_HEADER]?.Value
+  };
+}
+function addPropagationFieldsToAttributeNames(messageAttributeNames = []) {
+  return uniq([...messageAttributeNames, ...PROPAGATION_FIELDS]);
+}
+
+export { MAX_MESSAGE_ATTRIBUTES, addPropagationFieldsToAttributeNames, extractPropagationHeaders, injectPropagationContext };
+//# sourceMappingURL=MessageAttributes.js.map

package/build/esm/integrations/tracing-channel/aws-sdk/services/MessageAttributes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MessageAttributes.js","sources":["../../../../../../src/integrations/tracing-channel/aws-sdk/services/MessageAttributes.ts"],"sourcesContent":["import type { SerializedTraceData } from '@sentry/core';\nimport { debug, uniq } from '@sentry/core';\nimport { DEBUG_BUILD } from '../../../../debug-build';\nimport type { SNS, SQS } from '../aws-sdk.types';\n\n// https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-quotas.html\nexport const MAX_MESSAGE_ATTRIBUTES = 10;\n\n// Sentry trace-propagation headers written into / read from AWS message attributes.\nconst SENTRY_TRACE_HEADER = 'sentry-trace';\nconst BAGGAGE_HEADER = 'baggage';\nconst PROPAGATION_FIELDS = [SENTRY_TRACE_HEADER, BAGGAGE_HEADER];\n\nexport interface AwsSdkContextObject {\n  [key: string]: {\n    StringValue?: string;\n    Value?: string;\n  };\n}\n\n/**\n * Inject trace-propagation headers (from `getTraceData({ span })`) into an SQS/SNS message-attribute\n * map, so the consumer can continue the trace. Respects the SQS 10-attribute quota. Mirrors the OTel\n * integration's `injectPropagationContext`, but writes Sentry's `sentry-trace`/`baggage` instead of\n * W3C headers. Callers pass the precomputed headers so batch sends serialize them only once.\n */\nexport function injectPropagationContext(\n  attributesMap: SQS.MessageBodyAttributeMap | SNS.MessageAttributeMap | undefined,\n  traceData: SerializedTraceData,\n): SQS.MessageBodyAttributeMap | SNS.MessageAttributeMap {\n  const attributes = attributesMap ?? {};\n  const headerKeys = Object.keys(traceData) as (keyof SerializedTraceData)[];\n\n  if (Object.keys(attributes).length + headerKeys.length <= MAX_MESSAGE_ATTRIBUTES) {\n    for (const key of headerKeys) {\n      const value = traceData[key];\n      if (value) {\n        // Index-assigning into the SQS/SNS map union needs one concrete map type; the written value\n        // shape is valid for both.\n        (attributes as SQS.MessageBodyAttributeMap)[key] = { DataType: 'String', StringValue: value };\n      }\n    }\n  } else {\n    DEBUG_BUILD &&\n      debug.warn(\n        '[orchestrion:aws-sdk] cannot set trace propagation on SQS/SNS message due to maximum amount of MessageAttributes',\n      );\n  }\n  return attributes;\n}\n\n/** Read the propagation headers back off a received SQS message, if present. */\nexport function extractPropagationHeaders(\n  message: SQS.Message,\n): { sentryTrace?: string; baggage?: string } | undefined {\n  const carrier = (message.MessageAttributes ?? {}) as AwsSdkContextObject;\n  const sentryTrace = carrier[SENTRY_TRACE_HEADER]?.StringValue ?? carrier[SENTRY_TRACE_HEADER]?.Value;\n  if (!sentryTrace) {\n    return undefined;\n  }\n  return {\n    sentryTrace,\n    baggage: carrier[BAGGAGE_HEADER]?.StringValue ?? carrier[BAGGAGE_HEADER]?.Value,\n  };\n}\n\nexport function addPropagationFieldsToAttributeNames(messageAttributeNames: string[] = []): string[] {\n  return uniq([...messageAttributeNames, ...PROPAGATION_FIELDS]);\n}\n"],"names":[],"mappings":";;;AAMO,MAAM,sBAAA,GAAyB;AAGtC,MAAM,mBAAA,GAAsB,cAAA;AAC5B,MAAM,cAAA,GAAiB,SAAA;AACvB,MAAM,kBAAA,GAAqB,CAAC,mBAAA,EAAqB,cAAc,CAAA;AAexD,SAAS,wBAAA,CACd,eACA,SAAA,EACuD;AACvD,EAAA,MAAM,UAAA,GAAa,iBAAiB,EAAC;AACrC,EAAA,MAAM,UAAA,GAAa,MAAA,CAAO,IAAA,CAAK,SAAS,CAAA;AAExC,EAAA,IAAI,OAAO,IAAA,CAAK,UAAU,EAAE,MAAA,GAAS,UAAA,CAAW,UAAU,sBAAA,EAAwB;AAChF,IAAA,KAAA,MAAW,OAAO,UAAA,EAAY;AAC5B,MAAA,MAAM,KAAA,GAAQ,UAAU,GAAG,CAAA;AAC3B,MAAA,IAAI,KAAA,EAAO;AAGT,QAAC,WAA2C,GAAG,CAAA,GAAI,EAAE,QAAA,EAAU,QAAA,EAAU,aAAa,KAAA,EAAM;AAAA,MAC9F;AAAA,IACF;AAAA,EACF,CAAA,MAAO;AACL,IAAA,WAAA,IACE,KAAA,CAAM,IAAA;AAAA,MACJ;AAAA,KACF;AAAA,EACJ;AACA,EAAA,OAAO,UAAA;AACT;AAGO,SAAS,0BACd,OAAA,EACwD;AACxD,EAAA,MAAM,OAAA,GAAW,OAAA,CAAQ,iBAAA,IAAqB,EAAC;AAC/C,EAAA,MAAM,cAAc,OAAA,CAAQ,mBAAmB,GAAG,WAAA,IAAe,OAAA,CAAQ,mBAAmB,CAAA,EAAG,KAAA;AAC/F,EAAA,IAAI,CAAC,WAAA,EAAa;AAChB,IAAA,OAAO,MAAA;AAAA,EACT;AACA,EAAA,OAAO;AAAA,IACL,WAAA;AAAA,IACA,SAAS,OAAA,CAAQ,cAAc,GAAG,WAAA,IAAe,OAAA,CAAQ,cAAc,CAAA,EAAG;AAAA,GAC5E;AACF;AAEO,SAAS,oCAAA,CAAqC,qBAAA,GAAkC,EAAC,EAAa;AACnG,EAAA,OAAO,KAAK,CAAC,GAAG,qBAAA,EAAuB,GAAG,kBAAkB,CAAC,CAAA;AAC/D;;;;"}

package/build/esm/integrations/tracing-channel/aws-sdk/services/ServicesExtensions.js

@@ -0,0 +1,45 @@
+import { BedrockRuntimeServiceExtension } from './bedrock-runtime.js';
+import { DynamodbServiceExtension } from './dynamodb.js';
+import { KinesisServiceExtension } from './kinesis.js';
+import { LambdaServiceExtension } from './lambda.js';
+import { S3ServiceExtension } from './s3.js';
+import { SecretsManagerServiceExtension } from './secretsmanager.js';
+import { SnsServiceExtension } from './sns.js';
+import { SqsServiceExtension } from './sqs.js';
+import { StepFunctionsServiceExtension } from './stepfunctions.js';
+
+class ServicesExtensions {
+  constructor() {
+    // Per-service extensions, keyed by the client's `serviceId` (e.g. `'S3'`). Services without a
+    // registered extension still get the base rpc span from the subscriber.
+    this._services = /* @__PURE__ */ new Map([
+      ["SecretsManager", new SecretsManagerServiceExtension()],
+      ["SFN", new StepFunctionsServiceExtension()],
+      ["SQS", new SqsServiceExtension()],
+      ["SNS", new SnsServiceExtension()],
+      ["DynamoDB", new DynamodbServiceExtension()],
+      ["Lambda", new LambdaServiceExtension()],
+      ["S3", new S3ServiceExtension()],
+      ["Kinesis", new KinesisServiceExtension()],
+      ["BedrockRuntime", new BedrockRuntimeServiceExtension()]
+    ]);
+  }
+  requestPreSpanHook(request) {
+    const serviceExtension = this._services.get(request.serviceName);
+    if (!serviceExtension) {
+      return {};
+    }
+    return serviceExtension.requestPreSpanHook(request);
+  }
+  requestPostSpanHook(request, span) {
+    const serviceExtension = this._services.get(request.serviceName);
+    serviceExtension?.requestPostSpanHook?.(request, span);
+  }
+  responseHook(response, span) {
+    const serviceExtension = this._services.get(response.request.serviceName);
+    serviceExtension?.responseHook?.(response, span);
+  }
+}
+
+export { ServicesExtensions };
+//# sourceMappingURL=ServicesExtensions.js.map

package/build/esm/integrations/tracing-channel/aws-sdk/services/ServicesExtensions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ServicesExtensions.js","sources":["../../../../../../src/integrations/tracing-channel/aws-sdk/services/ServicesExtensions.ts"],"sourcesContent":["import type { Span } from '@sentry/core';\nimport type { NormalizedRequest, NormalizedResponse, RequestMetadata } from '../types';\nimport { BedrockRuntimeServiceExtension } from './bedrock-runtime';\nimport { DynamodbServiceExtension } from './dynamodb';\nimport { KinesisServiceExtension } from './kinesis';\nimport { LambdaServiceExtension } from './lambda';\nimport { S3ServiceExtension } from './s3';\nimport { SecretsManagerServiceExtension } from './secretsmanager';\nimport type { ServiceExtension } from './ServiceExtension';\nimport { SnsServiceExtension } from './sns';\nimport { SqsServiceExtension } from './sqs';\nimport { StepFunctionsServiceExtension } from './stepfunctions';\n\nexport class ServicesExtensions implements ServiceExtension {\n  // Per-service extensions, keyed by the client's `serviceId` (e.g. `'S3'`). Services without a\n  // registered extension still get the base rpc span from the subscriber.\n  private _services: Map<string, ServiceExtension> = new Map<string, ServiceExtension>([\n    ['SecretsManager', new SecretsManagerServiceExtension()],\n    ['SFN', new StepFunctionsServiceExtension()],\n    ['SQS', new SqsServiceExtension()],\n    ['SNS', new SnsServiceExtension()],\n    ['DynamoDB', new DynamodbServiceExtension()],\n    ['Lambda', new LambdaServiceExtension()],\n    ['S3', new S3ServiceExtension()],\n    ['Kinesis', new KinesisServiceExtension()],\n    ['BedrockRuntime', new BedrockRuntimeServiceExtension()],\n  ]);\n\n  public requestPreSpanHook(request: NormalizedRequest): RequestMetadata {\n    const serviceExtension = this._services.get(request.serviceName);\n    if (!serviceExtension) {\n      return {};\n    }\n    return serviceExtension.requestPreSpanHook(request);\n  }\n\n  public requestPostSpanHook(request: NormalizedRequest, span: Span): void {\n    const serviceExtension = this._services.get(request.serviceName);\n    serviceExtension?.requestPostSpanHook?.(request, span);\n  }\n\n  public responseHook(response: NormalizedResponse, span: Span): void {\n    const serviceExtension = this._services.get(response.request.serviceName);\n    serviceExtension?.responseHook?.(response, span);\n  }\n}\n"],"names":[],"mappings":";;;;;;;;;;AAaO,MAAM,kBAAA,CAA+C;AAAA,EAArD,WAAA,GAAA;AAGL;AAAA;AAAA,IAAA,IAAA,CAAQ,SAAA,uBAA+C,GAAA,CAA8B;AAAA,MACnF,CAAC,gBAAA,EAAkB,IAAI,8BAAA,EAAgC,CAAA;AAAA,MACvD,CAAC,KAAA,EAAO,IAAI,6BAAA,EAA+B,CAAA;AAAA,MAC3C,CAAC,KAAA,EAAO,IAAI,mBAAA,EAAqB,CAAA;AAAA,MACjC,CAAC,KAAA,EAAO,IAAI,mBAAA,EAAqB,CAAA;AAAA,MACjC,CAAC,UAAA,EAAY,IAAI,wBAAA,EAA0B,CAAA;AAAA,MAC3C,CAAC,QAAA,EAAU,IAAI,sBAAA,EAAwB,CAAA;AAAA,MACvC,CAAC,IAAA,EAAM,IAAI,kBAAA,EAAoB,CAAA;AAAA,MAC/B,CAAC,SAAA,EAAW,IAAI,uBAAA,EAAyB,CAAA;AAAA,MACzC,CAAC,gBAAA,EAAkB,IAAI,8BAAA,EAAgC;AAAA,KACxD,CAAA;AAAA,EAAA;AAAA,EAEM,mBAAmB,OAAA,EAA6C;AACrE,IAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,QAAQ,WAAW,CAAA;AAC/D,IAAA,IAAI,CAAC,gBAAA,EAAkB;AACrB,MAAA,OAAO,EAAC;AAAA,IACV;AACA,IAAA,OAAO,gBAAA,CAAiB,mBAAmB,OAAO,CAAA;AAAA,EACpD;AAAA,EAEO,mBAAA,CAAoB,SAA4B,IAAA,EAAkB;AACvE,IAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,QAAQ,WAAW,CAAA;AAC/D,IAAA,gBAAA,EAAkB,mBAAA,GAAsB,SAAS,IAAI,CAAA;AAAA,EACvD;AAAA,EAEO,YAAA,CAAa,UAA8B,IAAA,EAAkB;AAClE,IAAA,MAAM,mBAAmB,IAAA,CAAK,SAAA,CAAU,GAAA,CAAI,QAAA,CAAS,QAAQ,WAAW,CAAA;AACxE,IAAA,gBAAA,EAAkB,YAAA,GAAe,UAAU,IAAI,CAAA;AAAA,EACjD;AACF;;;;"}