@sentry/core 10.53.1 → 10.54.0

package/build/esm/integrations/mcp-server/correlation.js.map

@@ -1 +1 @@
-{"version":3,"file":"correlation.js","sources":["../../../../src/integrations/mcp-server/correlation.ts"],"sourcesContent":["/**\n * Request-span correlation system for MCP server instrumentation\n *\n * Handles mapping requestId to span data for correlation with handler execution.\n *\n * Uses sessionId as the primary key for stateful transports. This handles the wrapper\n * transport pattern (e.g., NodeStreamableHTTPServerTransport wrapping WebStandardStreamableHTTPServerTransport)\n * where onmessage and send may receive different `this` values but share the same sessionId.\n *\n * Falls back to WeakMap by transport instance for stateless transports (no sessionId).\n */\n\nimport { SPAN_STATUS_ERROR } from '../../tracing';\nimport type { Span } from '../../types-hoist/span';\nimport { MCP_PROTOCOL_VERSION_ATTRIBUTE } from './attributes';\nimport { extractPromptResultAttributes, extractToolResultAttributes } from './resultExtraction';\nimport { buildServerAttributesFromInfo, extractSessionDataFromInitializeResponse } from './sessionExtraction';\nimport type { MCPTransport, RequestId, RequestSpanMapValue, ResolvedMcpOptions } from './types';\n\n/**\n * Session-scoped correlation for stateful transports (with sessionId)\n * @internal Using sessionId as key handles wrapper transport patterns where\n * different transport objects share the same logical session\n */\nconst sessionToSpanMap = new Map<string, Map<RequestId, RequestSpanMapValue>>();\n\n/**\n * Transport-scoped correlation fallback for stateless transports (no sessionId)\n * @internal WeakMap allows automatic cleanup when transport is garbage collected\n */\nconst statelessSpanMap = new WeakMap<MCPTransport, Map<RequestId, RequestSpanMapValue>>();\n\n/**\n * Gets or creates the span map for a transport, using sessionId when available\n * @internal\n * @param transport - MCP transport instance\n * @returns Span map for the transport/session\n */\nfunction getOrCreateSpanMap(transport: MCPTransport): Map<RequestId, RequestSpanMapValue> {\n  const sessionId = transport.sessionId;\n\n  if (sessionId) {\n    // Stateful transport - use sessionId as key (handles wrapper pattern)\n    let spanMap = sessionToSpanMap.get(sessionId);\n    if (!spanMap) {\n      spanMap = new Map();\n      sessionToSpanMap.set(sessionId, spanMap);\n    }\n    return spanMap;\n  }\n\n  // Stateless fallback - use transport instance as key\n  let spanMap = statelessSpanMap.get(transport);\n  if (!spanMap) {\n    spanMap = new Map();\n    statelessSpanMap.set(transport, spanMap);\n  }\n  return spanMap;\n}\n\n/**\n * Stores span context for later correlation with handler execution\n * @param transport - MCP transport instance\n * @param requestId - Request identifier\n * @param span - Active span to correlate\n * @param method - MCP method name\n */\nexport function storeSpanForRequest(transport: MCPTransport, requestId: RequestId, span: Span, method: string): void {\n  const spanMap = getOrCreateSpanMap(transport);\n  spanMap.set(requestId, {\n    span,\n    method,\n    // oxlint-disable-next-line sdk/no-unsafe-random-apis\n    startTime: Date.now(),\n  });\n}\n\n/**\n * Completes span with results and cleans up correlation\n * @param transport - MCP transport instance\n * @param requestId - Request identifier\n * @param result - Execution result for attribute extraction\n * @param options - Resolved MCP options\n * @param hasError - Whether the JSON-RPC response contained an error\n */\nexport function completeSpanWithResults(\n  transport: MCPTransport,\n  requestId: RequestId,\n  result: unknown,\n  options: ResolvedMcpOptions,\n  hasError = false,\n): void {\n  const spanMap = getOrCreateSpanMap(transport);\n  const spanData = spanMap.get(requestId);\n  if (spanData) {\n    const { span, method } = spanData;\n\n    if (hasError) {\n      span.setStatus({ code: SPAN_STATUS_ERROR, message: 'internal_error' });\n    } else if (method === 'initialize') {\n      const sessionData = extractSessionDataFromInitializeResponse(result);\n      const serverAttributes = buildServerAttributesFromInfo(sessionData.serverInfo);\n\n      const initAttributes: Record<string, string | number> = {\n        ...serverAttributes,\n      };\n      if (sessionData.protocolVersion) {\n        initAttributes[MCP_PROTOCOL_VERSION_ATTRIBUTE] = sessionData.protocolVersion;\n      }\n\n      span.setAttributes(initAttributes);\n    } else if (method === 'tools/call') {\n      const toolAttributes = extractToolResultAttributes(result, options.recordOutputs);\n      span.setAttributes(toolAttributes);\n    } else if (method === 'prompts/get') {\n      const promptAttributes = extractPromptResultAttributes(result, options.recordOutputs);\n      span.setAttributes(promptAttributes);\n    }\n\n    span.end();\n    spanMap.delete(requestId);\n  }\n}\n\n/**\n * Cleans up pending spans for a specific transport (when that transport closes)\n * @param transport - MCP transport instance\n */\nexport function cleanupPendingSpansForTransport(transport: MCPTransport): void {\n  const sessionId = transport.sessionId;\n\n  // Try sessionId-based cleanup first (for stateful transports)\n  if (sessionId) {\n    const spanMap = sessionToSpanMap.get(sessionId);\n    if (spanMap) {\n      for (const [, spanData] of spanMap) {\n        spanData.span.setStatus({\n          code: SPAN_STATUS_ERROR,\n          message: 'cancelled',\n        });\n        spanData.span.end();\n      }\n      sessionToSpanMap.delete(sessionId);\n    }\n    return;\n  }\n\n  // Fallback to transport-based cleanup (for stateless transports)\n  const spanMap = statelessSpanMap.get(transport);\n  if (spanMap) {\n    for (const [, spanData] of spanMap) {\n      spanData.span.setStatus({\n        code: SPAN_STATUS_ERROR,\n        message: 'cancelled',\n      });\n      spanData.span.end();\n    }\n    spanMap.clear();\n    // Note: WeakMap entries are automatically cleaned up when transport is GC'd\n  }\n}\n"],"names":[],"mappings":";;;;;AAAA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;;;AASA;AACA;AACA;AACA;AACA;AACA,MAAM,gBAAA,GAAmB,IAAI,GAAG,EAA+C;;AAE/E;AACA;AACA;AACA;AACA,MAAM,gBAAA,GAAmB,IAAI,OAAO,EAAqD;;AAEzF;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,kBAAkB,CAAC,SAAS,EAAqD;AAC1F,EAAE,MAAM,SAAA,GAAY,SAAS,CAAC,SAAS;;AAEvC,EAAE,IAAI,SAAS,EAAE;AACjB;AACA,IAAI,IAAI,UAAU,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC;AACjD,IAAI,IAAI,CAAC,OAAO,EAAE;AAClB,MAAM,OAAA,GAAU,IAAI,GAAG,EAAE;AACzB,MAAM,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC;AAC9C,IAAI;AACJ,IAAI,OAAO,OAAO;AAClB,EAAE;;AAEF;AACA,EAAE,IAAI,UAAU,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC;AAC/C,EAAE,IAAI,CAAC,OAAO,EAAE;AAChB,IAAI,OAAA,GAAU,IAAI,GAAG,EAAE;AACvB,IAAI,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC;AAC5C,EAAE;AACF,EAAE,OAAO,OAAO;AAChB;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,SAAS,mBAAmB,CAAC,SAAS,EAAgB,SAAS,EAAa,IAAI,EAAQ,MAAM,EAAgB;AACrH,EAAE,MAAM,OAAA,GAAU,kBAAkB,CAAC,SAAS,CAAC;AAC/C,EAAE,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE;AACzB,IAAI,IAAI;AACR,IAAI,MAAM;AACV;AACA,IAAI,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;AACzB,GAAG,CAAC;AACJ;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,SAAS,uBAAuB;AACvC,EAAE,SAAS;AACX,EAAE,SAAS;AACX,EAAE,MAAM;AACR,EAAE,OAAO;AACT,EAAE,QAAA,GAAW,KAAK;AAClB,EAAQ;AACR,EAAE,MAAM,OAAA,GAAU,kBAAkB,CAAC,SAAS,CAAC;AAC/C,EAAE,MAAM,WAAW,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;AACzC,EAAE,IAAI,QAAQ,EAAE;AAChB,IAAI,MAAM,EAAE,IAAI,EAAE,MAAA,EAAO,GAAI,QAAQ;;AAErC,IAAI,IAAI,QAAQ,EAAE;AAClB,MAAM,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,gBAAA,EAAkB,CAAC;AAC5E,IAAI,OAAO,IAAI,MAAA,KAAW,YAAY,EAAE;AACxC,MAAM,MAAM,WAAA,GAAc,wCAAwC,CAAC,MAAM,CAAC;AAC1E,MAAM,MAAM,mBAAmB,6BAA6B,CAAC,WAAW,CAAC,UAAU,CAAC;;AAEpF,MAAM,MAAM,cAAc,GAAoC;AAC9D,QAAQ,GAAG,gBAAgB;AAC3B,OAAO;AACP,MAAM,IAAI,WAAW,CAAC,eAAe,EAAE;AACvC,QAAQ,cAAc,CAAC,8BAA8B,IAAI,WAAW,CAAC,eAAe;AACpF,MAAM;;AAEN,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;AACxC,IAAI,OAAO,IAAI,MAAA,KAAW,YAAY,EAAE;AACxC,MAAM,MAAM,cAAA,GAAiB,2BAA2B,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC;AACvF,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;AACxC,IAAI,OAAO,IAAI,MAAA,KAAW,aAAa,EAAE;AACzC,MAAM,MAAM,gBAAA,GAAmB,6BAA6B,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC;AAC3F,MAAM,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC;AAC1C,IAAI;;AAEJ,IAAI,IAAI,CAAC,GAAG,EAAE;AACd,IAAI,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;AAC7B,EAAE;AACF;;AAEA;AACA;AACA;AACA;AACO,SAAS,+BAA+B,CAAC,SAAS,EAAsB;AAC/E,EAAE,MAAM,SAAA,GAAY,SAAS,CAAC,SAAS;;AAEvC;AACA,EAAE,IAAI,SAAS,EAAE;AACjB,IAAI,MAAM,UAAU,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC;AACnD,IAAI,IAAI,OAAO,EAAE;AACjB,MAAM,KAAK,MAAM,GAAG,QAAQ,CAAA,IAAK,OAAO,EAAE;AAC1C,QAAQ,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC;AAChC,UAAU,IAAI,EAAE,iBAAiB;AACjC,UAAU,OAAO,EAAE,WAAW;AAC9B,SAAS,CAAC;AACV,QAAQ,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE;AAC3B,MAAM;AACN,MAAM,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC;AACxC,IAAI;AACJ,IAAI;AACJ,EAAE;;AAEF;AACA,EAAE,MAAM,UAAU,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC;AACjD,EAAE,IAAI,OAAO,EAAE;AACf,IAAI,KAAK,MAAM,GAAG,QAAQ,CAAA,IAAK,OAAO,EAAE;AACxC,MAAM,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC;AAC9B,QAAQ,IAAI,EAAE,iBAAiB;AAC/B,QAAQ,OAAO,EAAE,WAAW;AAC5B,OAAO,CAAC;AACR,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE;AACzB,IAAI;AACJ,IAAI,OAAO,CAAC,KAAK,EAAE;AACnB;AACA,EAAE;AACF;;;;"}
+{"version":3,"file":"correlation.js","sources":["../../../../src/integrations/mcp-server/correlation.ts"],"sourcesContent":["/**\n * Request-span correlation system for MCP server instrumentation\n *\n * Handles mapping requestId to span data for correlation with handler execution.\n *\n * Uses sessionId as the primary key for stateful transports. This handles the wrapper\n * transport pattern (e.g., NodeStreamableHTTPServerTransport wrapping WebStandardStreamableHTTPServerTransport)\n * where onmessage and send may receive different `this` values but share the same sessionId.\n *\n * Falls back to WeakMap by transport instance for stateless transports (no sessionId).\n */\n\nimport { SPAN_STATUS_ERROR } from '../../tracing';\nimport type { Span } from '../../types/span';\nimport { MCP_PROTOCOL_VERSION_ATTRIBUTE } from './attributes';\nimport { extractPromptResultAttributes, extractToolResultAttributes } from './resultExtraction';\nimport { buildServerAttributesFromInfo, extractSessionDataFromInitializeResponse } from './sessionExtraction';\nimport type { MCPTransport, RequestId, RequestSpanMapValue, ResolvedMcpOptions } from './types';\n\n/**\n * Session-scoped correlation for stateful transports (with sessionId)\n * @internal Using sessionId as key handles wrapper transport patterns where\n * different transport objects share the same logical session\n */\nconst sessionToSpanMap = new Map<string, Map<RequestId, RequestSpanMapValue>>();\n\n/**\n * Transport-scoped correlation fallback for stateless transports (no sessionId)\n * @internal WeakMap allows automatic cleanup when transport is garbage collected\n */\nconst statelessSpanMap = new WeakMap<MCPTransport, Map<RequestId, RequestSpanMapValue>>();\n\n/**\n * Gets or creates the span map for a transport, using sessionId when available\n * @internal\n * @param transport - MCP transport instance\n * @returns Span map for the transport/session\n */\nfunction getOrCreateSpanMap(transport: MCPTransport): Map<RequestId, RequestSpanMapValue> {\n  const sessionId = transport.sessionId;\n\n  if (sessionId) {\n    // Stateful transport - use sessionId as key (handles wrapper pattern)\n    let spanMap = sessionToSpanMap.get(sessionId);\n    if (!spanMap) {\n      spanMap = new Map();\n      sessionToSpanMap.set(sessionId, spanMap);\n    }\n    return spanMap;\n  }\n\n  // Stateless fallback - use transport instance as key\n  let spanMap = statelessSpanMap.get(transport);\n  if (!spanMap) {\n    spanMap = new Map();\n    statelessSpanMap.set(transport, spanMap);\n  }\n  return spanMap;\n}\n\n/**\n * Stores span context for later correlation with handler execution\n * @param transport - MCP transport instance\n * @param requestId - Request identifier\n * @param span - Active span to correlate\n * @param method - MCP method name\n */\nexport function storeSpanForRequest(transport: MCPTransport, requestId: RequestId, span: Span, method: string): void {\n  const spanMap = getOrCreateSpanMap(transport);\n  spanMap.set(requestId, {\n    span,\n    method,\n    // oxlint-disable-next-line sdk/no-unsafe-random-apis\n    startTime: Date.now(),\n  });\n}\n\n/**\n * Completes span with results and cleans up correlation\n * @param transport - MCP transport instance\n * @param requestId - Request identifier\n * @param result - Execution result for attribute extraction\n * @param options - Resolved MCP options\n * @param hasError - Whether the JSON-RPC response contained an error\n */\nexport function completeSpanWithResults(\n  transport: MCPTransport,\n  requestId: RequestId,\n  result: unknown,\n  options: ResolvedMcpOptions,\n  hasError = false,\n): void {\n  const spanMap = getOrCreateSpanMap(transport);\n  const spanData = spanMap.get(requestId);\n  if (spanData) {\n    const { span, method } = spanData;\n\n    if (hasError) {\n      span.setStatus({ code: SPAN_STATUS_ERROR, message: 'internal_error' });\n    } else if (method === 'initialize') {\n      const sessionData = extractSessionDataFromInitializeResponse(result);\n      const serverAttributes = buildServerAttributesFromInfo(sessionData.serverInfo);\n\n      const initAttributes: Record<string, string | number> = {\n        ...serverAttributes,\n      };\n      if (sessionData.protocolVersion) {\n        initAttributes[MCP_PROTOCOL_VERSION_ATTRIBUTE] = sessionData.protocolVersion;\n      }\n\n      span.setAttributes(initAttributes);\n    } else if (method === 'tools/call') {\n      const toolAttributes = extractToolResultAttributes(result, options.recordOutputs);\n      span.setAttributes(toolAttributes);\n    } else if (method === 'prompts/get') {\n      const promptAttributes = extractPromptResultAttributes(result, options.recordOutputs);\n      span.setAttributes(promptAttributes);\n    }\n\n    span.end();\n    spanMap.delete(requestId);\n  }\n}\n\n/**\n * Cleans up pending spans for a specific transport (when that transport closes)\n * @param transport - MCP transport instance\n */\nexport function cleanupPendingSpansForTransport(transport: MCPTransport): void {\n  const sessionId = transport.sessionId;\n\n  // Try sessionId-based cleanup first (for stateful transports)\n  if (sessionId) {\n    const spanMap = sessionToSpanMap.get(sessionId);\n    if (spanMap) {\n      for (const [, spanData] of spanMap) {\n        spanData.span.setStatus({\n          code: SPAN_STATUS_ERROR,\n          message: 'cancelled',\n        });\n        spanData.span.end();\n      }\n      sessionToSpanMap.delete(sessionId);\n    }\n    return;\n  }\n\n  // Fallback to transport-based cleanup (for stateless transports)\n  const spanMap = statelessSpanMap.get(transport);\n  if (spanMap) {\n    for (const [, spanData] of spanMap) {\n      spanData.span.setStatus({\n        code: SPAN_STATUS_ERROR,\n        message: 'cancelled',\n      });\n      spanData.span.end();\n    }\n    spanMap.clear();\n    // Note: WeakMap entries are automatically cleaned up when transport is GC'd\n  }\n}\n"],"names":["spanMap"],"mappings":";;;;;AAwBA,MAAM,gBAAA,uBAAuB,GAAA,EAAiD;AAM9E,MAAM,gBAAA,uBAAuB,OAAA,EAA2D;AAQxF,SAAS,mBAAmB,SAAA,EAA8D;AACxF,EAAA,MAAM,YAAY,SAAA,CAAU,SAAA;AAE5B,EAAA,IAAI,SAAA,EAAW;AAEb,IAAA,IAAIA,QAAAA,GAAU,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AAC5C,IAAA,IAAI,CAACA,QAAAA,EAAS;AACZ,MAAAA,QAAAA,uBAAc,GAAA,EAAI;AAClB,MAAA,gBAAA,CAAiB,GAAA,CAAI,WAAWA,QAAO,CAAA;AAAA,IACzC;AACA,IAAA,OAAOA,QAAAA;AAAA,EACT;AAGA,EAAA,IAAI,OAAA,GAAU,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AAC5C,EAAA,IAAI,CAAC,OAAA,EAAS;AACZ,IAAA,OAAA,uBAAc,GAAA,EAAI;AAClB,IAAA,gBAAA,CAAiB,GAAA,CAAI,WAAW,OAAO,CAAA;AAAA,EACzC;AACA,EAAA,OAAO,OAAA;AACT;AASO,SAAS,mBAAA,CAAoB,SAAA,EAAyB,SAAA,EAAsB,IAAA,EAAY,MAAA,EAAsB;AACnH,EAAA,MAAM,OAAA,GAAU,mBAAmB,SAAS,CAAA;AAC5C,EAAA,OAAA,CAAQ,IAAI,SAAA,EAAW;AAAA,IACrB,IAAA;AAAA,IACA,MAAA;AAAA;AAAA,IAEA,SAAA,EAAW,KAAK,GAAA;AAAI,GACrB,CAAA;AACH;AAUO,SAAS,wBACd,SAAA,EACA,SAAA,EACA,MAAA,EACA,OAAA,EACA,WAAW,KAAA,EACL;AACN,EAAA,MAAM,OAAA,GAAU,mBAAmB,SAAS,CAAA;AAC5C,EAAA,MAAM,QAAA,GAAW,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA;AACtC,EAAA,IAAI,QAAA,EAAU;AACZ,IAAA,MAAM,EAAE,IAAA,EAAM,MAAA,EAAO,GAAI,QAAA;AAEzB,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,IAAA,CAAK,UAAU,EAAE,IAAA,EAAM,iBAAA,EAAmB,OAAA,EAAS,kBAAkB,CAAA;AAAA,IACvE,CAAA,MAAA,IAAW,WAAW,YAAA,EAAc;AAClC,MAAA,MAAM,WAAA,GAAc,yCAAyC,MAAM,CAAA;AACnE,MAAA,MAAM,gBAAA,GAAmB,6BAAA,CAA8B,WAAA,CAAY,UAAU,CAAA;AAE7E,MAAA,MAAM,cAAA,GAAkD;AAAA,QACtD,GAAG;AAAA,OACL;AACA,MAAA,IAAI,YAAY,eAAA,EAAiB;AAC/B,QAAA,cAAA,CAAe,8BAA8B,IAAI,WAAA,CAAY,eAAA;AAAA,MAC/D;AAEA,MAAA,IAAA,CAAK,cAAc,cAAc,CAAA;AAAA,IACnC,CAAA,MAAA,IAAW,WAAW,YAAA,EAAc;AAClC,MAAA,MAAM,cAAA,GAAiB,2BAAA,CAA4B,MAAA,EAAQ,OAAA,CAAQ,aAAa,CAAA;AAChF,MAAA,IAAA,CAAK,cAAc,cAAc,CAAA;AAAA,IACnC,CAAA,MAAA,IAAW,WAAW,aAAA,EAAe;AACnC,MAAA,MAAM,gBAAA,GAAmB,6BAAA,CAA8B,MAAA,EAAQ,OAAA,CAAQ,aAAa,CAAA;AACpF,MAAA,IAAA,CAAK,cAAc,gBAAgB,CAAA;AAAA,IACrC;AAEA,IAAA,IAAA,CAAK,GAAA,EAAI;AACT,IAAA,OAAA,CAAQ,OAAO,SAAS,CAAA;AAAA,EAC1B;AACF;AAMO,SAAS,gCAAgC,SAAA,EAA+B;AAC7E,EAAA,MAAM,YAAY,SAAA,CAAU,SAAA;AAG5B,EAAA,IAAI,SAAA,EAAW;AACb,IAAA,MAAMA,QAAAA,GAAU,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AAC9C,IAAA,IAAIA,QAAAA,EAAS;AACX,MAAA,KAAA,MAAW,GAAG,QAAQ,CAAA,IAAKA,QAAAA,EAAS;AAClC,QAAA,QAAA,CAAS,KAAK,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,iBAAA;AAAA,UACN,OAAA,EAAS;AAAA,SACV,CAAA;AACD,QAAA,QAAA,CAAS,KAAK,GAAA,EAAI;AAAA,MACpB;AACA,MAAA,gBAAA,CAAiB,OAAO,SAAS,CAAA;AAAA,IACnC;AACA,IAAA;AAAA,EACF;AAGA,EAAA,MAAM,OAAA,GAAU,gBAAA,CAAiB,GAAA,CAAI,SAAS,CAAA;AAC9C,EAAA,IAAI,OAAA,EAAS;AACX,IAAA,KAAA,MAAW,GAAG,QAAQ,CAAA,IAAK,OAAA,EAAS;AAClC,MAAA,QAAA,CAAS,KAAK,SAAA,CAAU;AAAA,QACtB,IAAA,EAAM,iBAAA;AAAA,QACN,OAAA,EAAS;AAAA,OACV,CAAA;AACD,MAAA,QAAA,CAAS,KAAK,GAAA,EAAI;AAAA,IACpB;AACA,IAAA,OAAA,CAAQ,KAAA,EAAM;AAAA,EAEhB;AACF;;;;"}

package/build/esm/integrations/mcp-server/errorCapture.js

@@ -3,49 +3,30 @@ import { captureException } from '../../exports.js';
 import { getActiveSpan } from '../../utils/spanUtils.js';
 import { SPAN_STATUS_ERROR } from '../../tracing/spanstatus.js';
 
-/**
- * Safe error capture utilities for MCP server instrumentation
- *
- * Ensures error reporting never interferes with MCP server operation.
- * All capture operations are wrapped in try-catch to prevent side effects.
- */
-
-
-/**
- * Captures an error without affecting MCP server operation.
- *
- * The active span already contains all MCP context (method, tool, arguments, etc.)
- * @param error - Error to capture
- * @param errorType - Classification of error type for filtering
- * @param extraData - Additional context data to include
- */
 function captureError(error, errorType, extraData) {
   try {
     const client = getClient();
     if (!client) {
       return;
     }
-
     const activeSpan = getActiveSpan();
     if (activeSpan?.isRecording()) {
       activeSpan.setStatus({
         code: SPAN_STATUS_ERROR,
-        message: 'internal_error',
+        message: "internal_error"
       });
     }
-
     captureException(error, {
       mechanism: {
-        type: 'auto.ai.mcp_server',
+        type: "auto.ai.mcp_server",
         handled: false,
         data: {
-          error_type: errorType || 'handler_execution',
-          ...extraData,
-        },
-      },
+          error_type: errorType || "handler_execution",
+          ...extraData
+        }
+      }
     });
   } catch {
-    // noop
   }
 }
 

package/build/esm/integrations/mcp-server/errorCapture.js.map

@@ -1 +1 @@
-{"version":3,"file":"errorCapture.js","sources":["../../../../src/integrations/mcp-server/errorCapture.ts"],"sourcesContent":["/**\n * Safe error capture utilities for MCP server instrumentation\n *\n * Ensures error reporting never interferes with MCP server operation.\n * All capture operations are wrapped in try-catch to prevent side effects.\n */\n\nimport { getClient } from '../../currentScopes';\nimport { captureException } from '../../exports';\nimport { SPAN_STATUS_ERROR } from '../../tracing';\nimport { getActiveSpan } from '../../utils/spanUtils';\nimport type { McpErrorType } from './types';\n\n/**\n * Captures an error without affecting MCP server operation.\n *\n * The active span already contains all MCP context (method, tool, arguments, etc.)\n * @param error - Error to capture\n * @param errorType - Classification of error type for filtering\n * @param extraData - Additional context data to include\n */\nexport function captureError(error: Error, errorType?: McpErrorType, extraData?: Record<string, unknown>): void {\n  try {\n    const client = getClient();\n    if (!client) {\n      return;\n    }\n\n    const activeSpan = getActiveSpan();\n    if (activeSpan?.isRecording()) {\n      activeSpan.setStatus({\n        code: SPAN_STATUS_ERROR,\n        message: 'internal_error',\n      });\n    }\n\n    captureException(error, {\n      mechanism: {\n        type: 'auto.ai.mcp_server',\n        handled: false,\n        data: {\n          error_type: errorType || 'handler_execution',\n          ...extraData,\n        },\n      },\n    });\n  } catch {\n    // noop\n  }\n}\n"],"names":[],"mappings":";;;;;AAAA;AACA;AACA;AACA;AACA;AACA;;;AAQA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,SAAS,YAAY,CAAC,KAAK,EAAS,SAAS,EAAiB,SAAS,EAAkC;AAChH,EAAE,IAAI;AACN,IAAI,MAAM,MAAA,GAAS,SAAS,EAAE;AAC9B,IAAI,IAAI,CAAC,MAAM,EAAE;AACjB,MAAM;AACN,IAAI;;AAEJ,IAAI,MAAM,UAAA,GAAa,aAAa,EAAE;AACtC,IAAI,IAAI,UAAU,EAAE,WAAW,EAAE,EAAE;AACnC,MAAM,UAAU,CAAC,SAAS,CAAC;AAC3B,QAAQ,IAAI,EAAE,iBAAiB;AAC/B,QAAQ,OAAO,EAAE,gBAAgB;AACjC,OAAO,CAAC;AACR,IAAI;;AAEJ,IAAI,gBAAgB,CAAC,KAAK,EAAE;AAC5B,MAAM,SAAS,EAAE;AACjB,QAAQ,IAAI,EAAE,oBAAoB;AAClC,QAAQ,OAAO,EAAE,KAAK;AACtB,QAAQ,IAAI,EAAE;AACd,UAAU,UAAU,EAAE,SAAA,IAAa,mBAAmB;AACtD,UAAU,GAAG,SAAS;AACtB,SAAS;AACT,OAAO;AACP,KAAK,CAAC;AACN,EAAE,EAAE,MAAM;AACV;AACA,EAAE;AACF;;;;"}
+{"version":3,"file":"errorCapture.js","sources":["../../../../src/integrations/mcp-server/errorCapture.ts"],"sourcesContent":["/**\n * Safe error capture utilities for MCP server instrumentation\n *\n * Ensures error reporting never interferes with MCP server operation.\n * All capture operations are wrapped in try-catch to prevent side effects.\n */\n\nimport { getClient } from '../../currentScopes';\nimport { captureException } from '../../exports';\nimport { SPAN_STATUS_ERROR } from '../../tracing';\nimport { getActiveSpan } from '../../utils/spanUtils';\nimport type { McpErrorType } from './types';\n\n/**\n * Captures an error without affecting MCP server operation.\n *\n * The active span already contains all MCP context (method, tool, arguments, etc.)\n * @param error - Error to capture\n * @param errorType - Classification of error type for filtering\n * @param extraData - Additional context data to include\n */\nexport function captureError(error: Error, errorType?: McpErrorType, extraData?: Record<string, unknown>): void {\n  try {\n    const client = getClient();\n    if (!client) {\n      return;\n    }\n\n    const activeSpan = getActiveSpan();\n    if (activeSpan?.isRecording()) {\n      activeSpan.setStatus({\n        code: SPAN_STATUS_ERROR,\n        message: 'internal_error',\n      });\n    }\n\n    captureException(error, {\n      mechanism: {\n        type: 'auto.ai.mcp_server',\n        handled: false,\n        data: {\n          error_type: errorType || 'handler_execution',\n          ...extraData,\n        },\n      },\n    });\n  } catch {\n    // noop\n  }\n}\n"],"names":[],"mappings":";;;;;AAqBO,SAAS,YAAA,CAAa,KAAA,EAAc,SAAA,EAA0B,SAAA,EAA2C;AAC9G,EAAA,IAAI;AACF,IAAA,MAAM,SAAS,SAAA,EAAU;AACzB,IAAA,IAAI,CAAC,MAAA,EAAQ;AACX,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,aAAa,aAAA,EAAc;AACjC,IAAA,IAAI,UAAA,EAAY,aAAY,EAAG;AAC7B,MAAA,UAAA,CAAW,SAAA,CAAU;AAAA,QACnB,IAAA,EAAM,iBAAA;AAAA,QACN,OAAA,EAAS;AAAA,OACV,CAAA;AAAA,IACH;AAEA,IAAA,gBAAA,CAAiB,KAAA,EAAO;AAAA,MACtB,SAAA,EAAW;AAAA,QACT,IAAA,EAAM,oBAAA;AAAA,QACN,OAAA,EAAS,KAAA;AAAA,QACT,IAAA,EAAM;AAAA,UACJ,YAAY,SAAA,IAAa,mBAAA;AAAA,UACzB,GAAG;AAAA;AACL;AACF,KACD,CAAA;AAAA,EACH,CAAA,CAAA,MAAQ;AAAA,EAER;AACF;;;;"}

package/build/esm/integrations/mcp-server/handlers.js

@@ -3,237 +3,123 @@ import { debug } from '../../utils/debug-logger.js';
 import { fill } from '../../utils/object.js';
 import { captureError } from './errorCapture.js';
 
-/**
- * Handler method wrapping for MCP server instrumentation
- *
- * Provides automatic error capture and span correlation for tool, resource,
- * and prompt handlers.
- */
-
-
-/**
- * Generic function to wrap MCP server method handlers
- * @internal
- * @param serverInstance - MCP server instance
- * @param methodName - Method name to wrap (tool, resource, prompt)
- */
 function wrapMethodHandler(serverInstance, methodName) {
-  fill(serverInstance, methodName, originalMethod => {
-    return function ( name, ...args) {
+  fill(serverInstance, methodName, (originalMethod) => {
+    return function(name, ...args) {
       const handler = args[args.length - 1];
-
-      if (typeof handler !== 'function') {
-        return (originalMethod ).call(this, name, ...args);
+      if (typeof handler !== "function") {
+        return originalMethod.call(this, name, ...args);
       }
-
-      const wrappedHandler = createWrappedHandler(handler , methodName, name);
-      return (originalMethod ).call(this, name, ...args.slice(0, -1), wrappedHandler);
+      const wrappedHandler = createWrappedHandler(handler, methodName, name);
+      return originalMethod.call(this, name, ...args.slice(0, -1), wrappedHandler);
     };
   });
 }
-
-/**
- * Creates a wrapped handler with span correlation and error capture
- * @internal
- * @param originalHandler - Original handler function
- * @param methodName - MCP method name
- * @param handlerName - Handler identifier
- * @returns Wrapped handler function
- */
 function createWrappedHandler(originalHandler, methodName, handlerName) {
-  return function ( ...handlerArgs) {
+  return function(...handlerArgs) {
     try {
       return createErrorCapturingHandler.call(this, originalHandler, methodName, handlerName, handlerArgs);
     } catch (error) {
-      DEBUG_BUILD && debug.warn('MCP handler wrapping failed:', error);
+      DEBUG_BUILD && debug.warn("MCP handler wrapping failed:", error);
       return originalHandler.apply(this, handlerArgs);
     }
   };
 }
-
-/**
- * Creates an error-capturing wrapper for handler execution
- * @internal
- * @param originalHandler - Original handler function
- * @param methodName - MCP method name
- * @param handlerName - Handler identifier
- * @param handlerArgs - Handler arguments
- * @param extraHandlerData - Additional handler context
- * @returns Handler execution result
- */
-function createErrorCapturingHandler(
-
-  originalHandler,
-  methodName,
-  handlerName,
-  handlerArgs,
-) {
+function createErrorCapturingHandler(originalHandler, methodName, handlerName, handlerArgs) {
   try {
     const result = originalHandler.apply(this, handlerArgs);
-
-    if (result && typeof result === 'object' && typeof (result ).then === 'function') {
-      return Promise.resolve(result).catch(error => {
+    if (result && typeof result === "object" && typeof result.then === "function") {
+      return Promise.resolve(result).catch((error) => {
         captureHandlerError(error, methodName, handlerName);
         throw error;
       });
     }
-
     return result;
   } catch (error) {
-    captureHandlerError(error , methodName, handlerName);
+    captureHandlerError(error, methodName, handlerName);
     throw error;
   }
 }
-
-/**
- * Captures handler execution errors based on handler type
- * @internal
- * @param error - Error to capture
- * @param methodName - MCP method name
- * @param handlerName - Handler identifier
- */
 function captureHandlerError(error, methodName, handlerName) {
   try {
     const extraData = {};
-
-    if (methodName === 'tool' || methodName === 'registerTool') {
+    if (methodName === "tool" || methodName === "registerTool") {
       extraData.tool_name = handlerName;
-
-      if (
-        error.name === 'ProtocolValidationError' ||
-        error.message.includes('validation') ||
-        error.message.includes('protocol')
-      ) {
-        captureError(error, 'validation', extraData);
-      } else if (
-        error.name === 'ServerTimeoutError' ||
-        error.message.includes('timed out') ||
-        error.message.includes('timeout')
-      ) {
-        captureError(error, 'timeout', extraData);
+      if (error.name === "ProtocolValidationError" || error.message.includes("validation") || error.message.includes("protocol")) {
+        captureError(error, "validation", extraData);
+      } else if (error.name === "ServerTimeoutError" || error.message.includes("timed out") || error.message.includes("timeout")) {
+        captureError(error, "timeout", extraData);
       } else {
-        captureError(error, 'tool_execution', extraData);
+        captureError(error, "tool_execution", extraData);
       }
-    } else if (methodName === 'resource' || methodName === 'registerResource') {
+    } else if (methodName === "resource" || methodName === "registerResource") {
       extraData.resource_uri = handlerName;
-      captureError(error, 'resource_execution', extraData);
-    } else if (methodName === 'prompt' || methodName === 'registerPrompt') {
+      captureError(error, "resource_execution", extraData);
+    } else if (methodName === "prompt" || methodName === "registerPrompt") {
       extraData.prompt_name = handlerName;
-      captureError(error, 'prompt_execution', extraData);
+      captureError(error, "prompt_execution", extraData);
     }
   } catch (_captureErr) {
-    // noop
   }
 }
-
-/**
- * Wraps tool handlers to associate them with request spans.
- * Instruments both `tool` (legacy API) and `registerTool` (new API) if present.
- * @param serverInstance - MCP server instance
- */
 function wrapToolHandlers(serverInstance) {
-  if (typeof serverInstance.tool === 'function') wrapMethodHandler(serverInstance, 'tool');
-  if (typeof serverInstance.registerTool === 'function') wrapMethodHandler(serverInstance, 'registerTool');
+  if (typeof serverInstance.tool === "function") wrapMethodHandler(serverInstance, "tool");
+  if (typeof serverInstance.registerTool === "function") wrapMethodHandler(serverInstance, "registerTool");
 }
-
-/**
- * Wraps resource handlers to associate them with request spans.
- * Instruments both `resource` (legacy API) and `registerResource` (new API) if present.
- * @param serverInstance - MCP server instance
- */
 function wrapResourceHandlers(serverInstance) {
-  if (typeof serverInstance.resource === 'function') wrapMethodHandler(serverInstance, 'resource');
-  if (typeof serverInstance.registerResource === 'function') wrapMethodHandler(serverInstance, 'registerResource');
+  if (typeof serverInstance.resource === "function") wrapMethodHandler(serverInstance, "resource");
+  if (typeof serverInstance.registerResource === "function") wrapMethodHandler(serverInstance, "registerResource");
 }
-
-/**
- * Wraps prompt handlers to associate them with request spans.
- * Instruments both `prompt` (legacy API) and `registerPrompt` (new API) if present.
- * @param serverInstance - MCP server instance
- */
 function wrapPromptHandlers(serverInstance) {
-  if (typeof serverInstance.prompt === 'function') wrapMethodHandler(serverInstance, 'prompt');
-  if (typeof serverInstance.registerPrompt === 'function') wrapMethodHandler(serverInstance, 'registerPrompt');
+  if (typeof serverInstance.prompt === "function") wrapMethodHandler(serverInstance, "prompt");
+  if (typeof serverInstance.registerPrompt === "function") wrapMethodHandler(serverInstance, "registerPrompt");
 }
-
-/**
- * Wraps all MCP handler types for span correlation.
- * Supports both the legacy API (`tool`, `resource`, `prompt`) and the newer API
- * (`registerTool`, `registerResource`, `registerPrompt`), instrumenting whichever methods are present.
- * @param serverInstance - MCP server instance
- */
 function wrapAllMCPHandlers(serverInstance) {
   wrapToolHandlers(serverInstance);
   wrapResourceHandlers(serverInstance);
   wrapPromptHandlers(serverInstance);
 }
-
-/**
- * Retroactively wraps handlers on tools, resources, and prompts that were registered
- * before `wrapMcpServerWithSentry` was called.
- *
- * The MCP SDK stores registered entries in private maps and invokes them via the entry's
- * own property at call time — `executor` for tools, `readCallback` for resources, and
- * `handler` for prompts. Replacing those properties
- * in-place is therefore equivalent to having wrapped the original registration call.
- *
- * NOTE: This intentionally accesses private MCP SDK internals (`_registeredTools` etc.).
- * The properties and their shapes are verified against @modelcontextprotocol/sdk source:
- * https://github.com/modelcontextprotocol/typescript-sdk/blob/2c0c481cb9dbfd15c8613f765c940a5f5bace94d/packages/server/src/server/mcp.ts#L304
- * When upgrading the MCP SDK, re-verify that these internal maps and their callable
- * properties still exist and are invoked directly (not captured by closure at registration).
- * All access is defensive — if a property is absent or not a function we skip silently.
- * @internal
- */
 function wrapExistingHandlers(serverInstance) {
-  const server = serverInstance ;
-
-  // Tools: MCP SDK calls registeredTool.executor (generated from handler at registration time)
-  const registeredTools = server['_registeredTools'];
-  if (registeredTools && typeof registeredTools === 'object') {
-    for (const [name, tool] of Object.entries(registeredTools )) {
-      if (typeof tool['executor'] === 'function') {
-        tool['executor'] = createWrappedHandler(tool['executor'] , 'registerTool', name);
+  const server = serverInstance;
+  const registeredTools = server["_registeredTools"];
+  if (registeredTools && typeof registeredTools === "object") {
+    for (const [name, tool] of Object.entries(registeredTools)) {
+      if (typeof tool["executor"] === "function") {
+        tool["executor"] = createWrappedHandler(tool["executor"], "registerTool", name);
       }
     }
   }
-
-  // Resources: MCP SDK calls registeredResource.readCallback
-  const registeredResources = server['_registeredResources'];
-  if (registeredResources && typeof registeredResources === 'object') {
-    for (const [name, resource] of Object.entries(registeredResources )) {
-      if (typeof resource['readCallback'] === 'function') {
-        resource['readCallback'] = createWrappedHandler(
-          resource['readCallback'] ,
-          'registerResource',
-          name,
+  const registeredResources = server["_registeredResources"];
+  if (registeredResources && typeof registeredResources === "object") {
+    for (const [name, resource] of Object.entries(registeredResources)) {
+      if (typeof resource["readCallback"] === "function") {
+        resource["readCallback"] = createWrappedHandler(
+          resource["readCallback"],
+          "registerResource",
+          name
         );
       }
     }
   }
-
-  // Resource templates: MCP SDK calls registeredResourceTemplate.readCallback
-  const registeredResourceTemplates = server['_registeredResourceTemplates'];
-  if (registeredResourceTemplates && typeof registeredResourceTemplates === 'object') {
+  const registeredResourceTemplates = server["_registeredResourceTemplates"];
+  if (registeredResourceTemplates && typeof registeredResourceTemplates === "object") {
     for (const [name, template] of Object.entries(
-      registeredResourceTemplates ,
+      registeredResourceTemplates
     )) {
-      if (typeof template['readCallback'] === 'function') {
-        template['readCallback'] = createWrappedHandler(
-          template['readCallback'] ,
-          'registerResource',
-          name,
+      if (typeof template["readCallback"] === "function") {
+        template["readCallback"] = createWrappedHandler(
+          template["readCallback"],
+          "registerResource",
+          name
         );
       }
     }
   }
-
-  // Prompts: MCP SDK calls registeredPrompt.handler
-  const registeredPrompts = server['_registeredPrompts'];
-  if (registeredPrompts && typeof registeredPrompts === 'object') {
-    for (const [name, prompt] of Object.entries(registeredPrompts )) {
-      if (typeof prompt['handler'] === 'function') {
-        prompt['handler'] = createWrappedHandler(prompt['handler'] , 'registerPrompt', name);
+  const registeredPrompts = server["_registeredPrompts"];
+  if (registeredPrompts && typeof registeredPrompts === "object") {
+    for (const [name, prompt] of Object.entries(registeredPrompts)) {
+      if (typeof prompt["handler"] === "function") {
+        prompt["handler"] = createWrappedHandler(prompt["handler"], "registerPrompt", name);
       }
     }
   }

package/build/esm/integrations/mcp-server/handlers.js.map

@@ -1 +1 @@
-{"version":3,"file":"handlers.js","sources":["../../../../src/integrations/mcp-server/handlers.ts"],"sourcesContent":["/**\n * Handler method wrapping for MCP server instrumentation\n *\n * Provides automatic error capture and span correlation for tool, resource,\n * and prompt handlers.\n */\n\nimport { DEBUG_BUILD } from '../../debug-build';\nimport { debug } from '../../utils/debug-logger';\nimport { fill } from '../../utils/object';\nimport { captureError } from './errorCapture';\nimport type { MCPHandler, MCPServerInstance } from './types';\n\n/**\n * Generic function to wrap MCP server method handlers\n * @internal\n * @param serverInstance - MCP server instance\n * @param methodName - Method name to wrap (tool, resource, prompt)\n */\nfunction wrapMethodHandler(serverInstance: MCPServerInstance, methodName: keyof MCPServerInstance): void {\n  fill(serverInstance, methodName, originalMethod => {\n    return function (this: MCPServerInstance, name: string, ...args: unknown[]) {\n      const handler = args[args.length - 1];\n\n      if (typeof handler !== 'function') {\n        return (originalMethod as (...args: unknown[]) => unknown).call(this, name, ...args);\n      }\n\n      const wrappedHandler = createWrappedHandler(handler as MCPHandler, methodName, name);\n      return (originalMethod as (...args: unknown[]) => unknown).call(this, name, ...args.slice(0, -1), wrappedHandler);\n    };\n  });\n}\n\n/**\n * Creates a wrapped handler with span correlation and error capture\n * @internal\n * @param originalHandler - Original handler function\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n * @returns Wrapped handler function\n */\nfunction createWrappedHandler(originalHandler: MCPHandler, methodName: keyof MCPServerInstance, handlerName: string) {\n  return function (this: unknown, ...handlerArgs: unknown[]): unknown {\n    try {\n      return createErrorCapturingHandler.call(this, originalHandler, methodName, handlerName, handlerArgs);\n    } catch (error) {\n      DEBUG_BUILD && debug.warn('MCP handler wrapping failed:', error);\n      return originalHandler.apply(this, handlerArgs);\n    }\n  };\n}\n\n/**\n * Creates an error-capturing wrapper for handler execution\n * @internal\n * @param originalHandler - Original handler function\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n * @param handlerArgs - Handler arguments\n * @param extraHandlerData - Additional handler context\n * @returns Handler execution result\n */\nfunction createErrorCapturingHandler(\n  this: MCPServerInstance,\n  originalHandler: MCPHandler,\n  methodName: keyof MCPServerInstance,\n  handlerName: string,\n  handlerArgs: unknown[],\n): unknown {\n  try {\n    const result = originalHandler.apply(this, handlerArgs);\n\n    if (result && typeof result === 'object' && typeof (result as { then?: unknown }).then === 'function') {\n      return Promise.resolve(result).catch(error => {\n        captureHandlerError(error, methodName, handlerName);\n        throw error;\n      });\n    }\n\n    return result;\n  } catch (error) {\n    captureHandlerError(error as Error, methodName, handlerName);\n    throw error;\n  }\n}\n\n/**\n * Captures handler execution errors based on handler type\n * @internal\n * @param error - Error to capture\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n */\nfunction captureHandlerError(error: Error, methodName: keyof MCPServerInstance, handlerName: string): void {\n  try {\n    const extraData: Record<string, unknown> = {};\n\n    if (methodName === 'tool' || methodName === 'registerTool') {\n      extraData.tool_name = handlerName;\n\n      if (\n        error.name === 'ProtocolValidationError' ||\n        error.message.includes('validation') ||\n        error.message.includes('protocol')\n      ) {\n        captureError(error, 'validation', extraData);\n      } else if (\n        error.name === 'ServerTimeoutError' ||\n        error.message.includes('timed out') ||\n        error.message.includes('timeout')\n      ) {\n        captureError(error, 'timeout', extraData);\n      } else {\n        captureError(error, 'tool_execution', extraData);\n      }\n    } else if (methodName === 'resource' || methodName === 'registerResource') {\n      extraData.resource_uri = handlerName;\n      captureError(error, 'resource_execution', extraData);\n    } else if (methodName === 'prompt' || methodName === 'registerPrompt') {\n      extraData.prompt_name = handlerName;\n      captureError(error, 'prompt_execution', extraData);\n    }\n  } catch (_captureErr) {\n    // noop\n  }\n}\n\n/**\n * Wraps tool handlers to associate them with request spans.\n * Instruments both `tool` (legacy API) and `registerTool` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapToolHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.tool === 'function') wrapMethodHandler(serverInstance, 'tool');\n  if (typeof serverInstance.registerTool === 'function') wrapMethodHandler(serverInstance, 'registerTool');\n}\n\n/**\n * Wraps resource handlers to associate them with request spans.\n * Instruments both `resource` (legacy API) and `registerResource` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapResourceHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.resource === 'function') wrapMethodHandler(serverInstance, 'resource');\n  if (typeof serverInstance.registerResource === 'function') wrapMethodHandler(serverInstance, 'registerResource');\n}\n\n/**\n * Wraps prompt handlers to associate them with request spans.\n * Instruments both `prompt` (legacy API) and `registerPrompt` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapPromptHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.prompt === 'function') wrapMethodHandler(serverInstance, 'prompt');\n  if (typeof serverInstance.registerPrompt === 'function') wrapMethodHandler(serverInstance, 'registerPrompt');\n}\n\n/**\n * Wraps all MCP handler types for span correlation.\n * Supports both the legacy API (`tool`, `resource`, `prompt`) and the newer API\n * (`registerTool`, `registerResource`, `registerPrompt`), instrumenting whichever methods are present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapAllMCPHandlers(serverInstance: MCPServerInstance): void {\n  wrapToolHandlers(serverInstance);\n  wrapResourceHandlers(serverInstance);\n  wrapPromptHandlers(serverInstance);\n}\n\n/**\n * Retroactively wraps handlers on tools, resources, and prompts that were registered\n * before `wrapMcpServerWithSentry` was called.\n *\n * The MCP SDK stores registered entries in private maps and invokes them via the entry's\n * own property at call time — `executor` for tools, `readCallback` for resources, and\n * `handler` for prompts. Replacing those properties\n * in-place is therefore equivalent to having wrapped the original registration call.\n *\n * NOTE: This intentionally accesses private MCP SDK internals (`_registeredTools` etc.).\n * The properties and their shapes are verified against @modelcontextprotocol/sdk source:\n * https://github.com/modelcontextprotocol/typescript-sdk/blob/2c0c481cb9dbfd15c8613f765c940a5f5bace94d/packages/server/src/server/mcp.ts#L304\n * When upgrading the MCP SDK, re-verify that these internal maps and their callable\n * properties still exist and are invoked directly (not captured by closure at registration).\n * All access is defensive — if a property is absent or not a function we skip silently.\n * @internal\n */\nexport function wrapExistingHandlers(serverInstance: MCPServerInstance): void {\n  const server = serverInstance as unknown as Record<string, unknown>;\n\n  // Tools: MCP SDK calls registeredTool.executor (generated from handler at registration time)\n  const registeredTools = server['_registeredTools'];\n  if (registeredTools && typeof registeredTools === 'object') {\n    for (const [name, tool] of Object.entries(registeredTools as Record<string, Record<string, unknown>>)) {\n      if (typeof tool['executor'] === 'function') {\n        tool['executor'] = createWrappedHandler(tool['executor'] as MCPHandler, 'registerTool', name);\n      }\n    }\n  }\n\n  // Resources: MCP SDK calls registeredResource.readCallback\n  const registeredResources = server['_registeredResources'];\n  if (registeredResources && typeof registeredResources === 'object') {\n    for (const [name, resource] of Object.entries(registeredResources as Record<string, Record<string, unknown>>)) {\n      if (typeof resource['readCallback'] === 'function') {\n        resource['readCallback'] = createWrappedHandler(\n          resource['readCallback'] as MCPHandler,\n          'registerResource',\n          name,\n        );\n      }\n    }\n  }\n\n  // Resource templates: MCP SDK calls registeredResourceTemplate.readCallback\n  const registeredResourceTemplates = server['_registeredResourceTemplates'];\n  if (registeredResourceTemplates && typeof registeredResourceTemplates === 'object') {\n    for (const [name, template] of Object.entries(\n      registeredResourceTemplates as Record<string, Record<string, unknown>>,\n    )) {\n      if (typeof template['readCallback'] === 'function') {\n        template['readCallback'] = createWrappedHandler(\n          template['readCallback'] as MCPHandler,\n          'registerResource',\n          name,\n        );\n      }\n    }\n  }\n\n  // Prompts: MCP SDK calls registeredPrompt.handler\n  const registeredPrompts = server['_registeredPrompts'];\n  if (registeredPrompts && typeof registeredPrompts === 'object') {\n    for (const [name, prompt] of Object.entries(registeredPrompts as Record<string, Record<string, unknown>>)) {\n      if (typeof prompt['handler'] === 'function') {\n        prompt['handler'] = createWrappedHandler(prompt['handler'] as MCPHandler, 'registerPrompt', name);\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";;;;;AAAA;AACA;AACA;AACA;AACA;AACA;;;AAQA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,iBAAiB,CAAC,cAAc,EAAqB,UAAU,EAAiC;AACzG,EAAE,IAAI,CAAC,cAAc,EAAE,UAAU,EAAE,kBAAkB;AACrD,IAAI,OAAO,WAAmC,IAAI,EAAU,GAAG,IAAI,EAAa;AAChF,MAAM,MAAM,OAAA,GAAU,IAAI,CAAC,IAAI,CAAC,MAAA,GAAS,CAAC,CAAC;;AAE3C,MAAM,IAAI,OAAO,OAAA,KAAY,UAAU,EAAE;AACzC,QAAQ,OAAO,CAAC,cAAA,GAAmD,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;AAC5F,MAAM;;AAEN,MAAM,MAAM,cAAA,GAAiB,oBAAoB,CAAC,UAAuB,UAAU,EAAE,IAAI,CAAC;AAC1F,MAAM,OAAO,CAAC,cAAA,GAAmD,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,cAAc,CAAC;AACvH,IAAI,CAAC;AACL,EAAE,CAAC,CAAC;AACJ;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,oBAAoB,CAAC,eAAe,EAAc,UAAU,EAA2B,WAAW,EAAU;AACrH,EAAE,OAAO,WAAyB,GAAG,WAAW,EAAsB;AACtE,IAAI,IAAI;AACR,MAAM,OAAO,2BAA2B,CAAC,IAAI,CAAC,IAAI,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,CAAC;AAC1G,IAAI,CAAA,CAAE,OAAO,KAAK,EAAE;AACpB,MAAM,WAAA,IAAe,KAAK,CAAC,IAAI,CAAC,8BAA8B,EAAE,KAAK,CAAC;AACtE,MAAM,OAAO,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,WAAW,CAAC;AACrD,IAAI;AACJ,EAAE,CAAC;AACH;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,2BAA2B;;AAEpC,EAAE,eAAe;AACjB,EAAE,UAAU;AACZ,EAAE,WAAW;AACb,EAAE,WAAW;AACb,EAAW;AACX,EAAE,IAAI;AACN,IAAI,MAAM,MAAA,GAAS,eAAe,CAAC,KAAK,CAAC,IAAI,EAAE,WAAW,CAAC;;AAE3D,IAAI,IAAI,MAAA,IAAU,OAAO,WAAW,QAAA,IAAY,OAAO,CAAC,MAAA,GAA8B,IAAA,KAAS,UAAU,EAAE;AAC3G,MAAM,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,KAAA,IAAS;AACpD,QAAQ,mBAAmB,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,CAAC;AAC3D,QAAQ,MAAM,KAAK;AACnB,MAAM,CAAC,CAAC;AACR,IAAI;;AAEJ,IAAI,OAAO,MAAM;AACjB,EAAE,CAAA,CAAE,OAAO,KAAK,EAAE;AAClB,IAAI,mBAAmB,CAAC,KAAA,GAAgB,UAAU,EAAE,WAAW,CAAC;AAChE,IAAI,MAAM,KAAK;AACf,EAAE;AACF;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA,SAAS,mBAAmB,CAAC,KAAK,EAAS,UAAU,EAA2B,WAAW,EAAgB;AAC3G,EAAE,IAAI;AACN,IAAI,MAAM,SAAS,GAA4B,EAAE;;AAEjD,IAAI,IAAI,UAAA,KAAe,UAAU,UAAA,KAAe,cAAc,EAAE;AAChE,MAAM,SAAS,CAAC,SAAA,GAAY,WAAW;;AAEvC,MAAM;AACN,QAAQ,KAAK,CAAC,IAAA,KAAS,yBAAA;AACvB,QAAQ,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAA;AAC3C,QAAQ,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU;AACzC,QAAQ;AACR,QAAQ,YAAY,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC;AACpD,MAAM,CAAA,MAAO;AACb,QAAQ,KAAK,CAAC,IAAA,KAAS,oBAAA;AACvB,QAAQ,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAA;AAC1C,QAAQ,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS;AACxC,QAAQ;AACR,QAAQ,YAAY,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,CAAC;AACjD,MAAM,OAAO;AACb,QAAQ,YAAY,CAAC,KAAK,EAAE,gBAAgB,EAAE,SAAS,CAAC;AACxD,MAAM;AACN,IAAI,CAAA,MAAO,IAAI,UAAA,KAAe,UAAA,IAAc,UAAA,KAAe,kBAAkB,EAAE;AAC/E,MAAM,SAAS,CAAC,YAAA,GAAe,WAAW;AAC1C,MAAM,YAAY,CAAC,KAAK,EAAE,oBAAoB,EAAE,SAAS,CAAC;AAC1D,IAAI,CAAA,MAAO,IAAI,UAAA,KAAe,QAAA,IAAY,UAAA,KAAe,gBAAgB,EAAE;AAC3E,MAAM,SAAS,CAAC,WAAA,GAAc,WAAW;AACzC,MAAM,YAAY,CAAC,KAAK,EAAE,kBAAkB,EAAE,SAAS,CAAC;AACxD,IAAI;AACJ,EAAE,CAAA,CAAE,OAAO,WAAW,EAAE;AACxB;AACA,EAAE;AACF;;AAEA;AACA;AACA;AACA;AACA;AACO,SAAS,gBAAgB,CAAC,cAAc,EAA2B;AAC1E,EAAE,IAAI,OAAO,cAAc,CAAC,IAAA,KAAS,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,MAAM,CAAC;AAC1F,EAAE,IAAI,OAAO,cAAc,CAAC,YAAA,KAAiB,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,cAAc,CAAC;AAC1G;;AAEA;AACA;AACA;AACA;AACA;AACO,SAAS,oBAAoB,CAAC,cAAc,EAA2B;AAC9E,EAAE,IAAI,OAAO,cAAc,CAAC,QAAA,KAAa,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,UAAU,CAAC;AAClG,EAAE,IAAI,OAAO,cAAc,CAAC,gBAAA,KAAqB,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,kBAAkB,CAAC;AAClH;;AAEA;AACA;AACA;AACA;AACA;AACO,SAAS,kBAAkB,CAAC,cAAc,EAA2B;AAC5E,EAAE,IAAI,OAAO,cAAc,CAAC,MAAA,KAAW,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,QAAQ,CAAC;AAC9F,EAAE,IAAI,OAAO,cAAc,CAAC,cAAA,KAAmB,UAAU,EAAE,iBAAiB,CAAC,cAAc,EAAE,gBAAgB,CAAC;AAC9G;;AAEA;AACA;AACA;AACA;AACA;AACA;AACO,SAAS,kBAAkB,CAAC,cAAc,EAA2B;AAC5E,EAAE,gBAAgB,CAAC,cAAc,CAAC;AAClC,EAAE,oBAAoB,CAAC,cAAc,CAAC;AACtC,EAAE,kBAAkB,CAAC,cAAc,CAAC;AACpC;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACO,SAAS,oBAAoB,CAAC,cAAc,EAA2B;AAC9E,EAAE,MAAM,MAAA,GAAS,cAAA;;AAEjB;AACA,EAAE,MAAM,eAAA,GAAkB,MAAM,CAAC,kBAAkB,CAAC;AACpD,EAAE,IAAI,eAAA,IAAmB,OAAO,eAAA,KAAoB,QAAQ,EAAE;AAC9D,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAA,IAAK,MAAM,CAAC,OAAO,CAAC,eAAA,EAA2D,EAAE;AAC3G,MAAM,IAAI,OAAO,IAAI,CAAC,UAAU,CAAA,KAAM,UAAU,EAAE;AAClD,QAAQ,IAAI,CAAC,UAAU,CAAA,GAAI,oBAAoB,CAAC,IAAI,CAAC,UAAU,CAAA,GAAiB,cAAc,EAAE,IAAI,CAAC;AACrG,MAAM;AACN,IAAI;AACJ,EAAE;;AAEF;AACA,EAAE,MAAM,mBAAA,GAAsB,MAAM,CAAC,sBAAsB,CAAC;AAC5D,EAAE,IAAI,mBAAA,IAAuB,OAAO,mBAAA,KAAwB,QAAQ,EAAE;AACtE,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAA,IAAK,MAAM,CAAC,OAAO,CAAC,mBAAA,EAA+D,EAAE;AACnH,MAAM,IAAI,OAAO,QAAQ,CAAC,cAAc,CAAA,KAAM,UAAU,EAAE;AAC1D,QAAQ,QAAQ,CAAC,cAAc,CAAA,GAAI,oBAAoB;AACvD,UAAU,QAAQ,CAAC,cAAc,CAAA;AACjC,UAAU,kBAAkB;AAC5B,UAAU,IAAI;AACd,SAAS;AACT,MAAM;AACN,IAAI;AACJ,EAAE;;AAEF;AACA,EAAE,MAAM,2BAAA,GAA8B,MAAM,CAAC,8BAA8B,CAAC;AAC5E,EAAE,IAAI,2BAAA,IAA+B,OAAO,2BAAA,KAAgC,QAAQ,EAAE;AACtF,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAA,IAAK,MAAM,CAAC,OAAO;AACjD,MAAM,2BAAA;AACN,KAAK,EAAE;AACP,MAAM,IAAI,OAAO,QAAQ,CAAC,cAAc,CAAA,KAAM,UAAU,EAAE;AAC1D,QAAQ,QAAQ,CAAC,cAAc,CAAA,GAAI,oBAAoB;AACvD,UAAU,QAAQ,CAAC,cAAc,CAAA;AACjC,UAAU,kBAAkB;AAC5B,UAAU,IAAI;AACd,SAAS;AACT,MAAM;AACN,IAAI;AACJ,EAAE;;AAEF;AACA,EAAE,MAAM,iBAAA,GAAoB,MAAM,CAAC,oBAAoB,CAAC;AACxD,EAAE,IAAI,iBAAA,IAAqB,OAAO,iBAAA,KAAsB,QAAQ,EAAE;AAClE,IAAI,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAA,IAAK,MAAM,CAAC,OAAO,CAAC,iBAAA,EAA6D,EAAE;AAC/G,MAAM,IAAI,OAAO,MAAM,CAAC,SAAS,CAAA,KAAM,UAAU,EAAE;AACnD,QAAQ,MAAM,CAAC,SAAS,CAAA,GAAI,oBAAoB,CAAC,MAAM,CAAC,SAAS,CAAA,GAAiB,gBAAgB,EAAE,IAAI,CAAC;AACzG,MAAM;AACN,IAAI;AACJ,EAAE;AACF;;;;"}
+{"version":3,"file":"handlers.js","sources":["../../../../src/integrations/mcp-server/handlers.ts"],"sourcesContent":["/**\n * Handler method wrapping for MCP server instrumentation\n *\n * Provides automatic error capture and span correlation for tool, resource,\n * and prompt handlers.\n */\n\nimport { DEBUG_BUILD } from '../../debug-build';\nimport { debug } from '../../utils/debug-logger';\nimport { fill } from '../../utils/object';\nimport { captureError } from './errorCapture';\nimport type { MCPHandler, MCPServerInstance } from './types';\n\n/**\n * Generic function to wrap MCP server method handlers\n * @internal\n * @param serverInstance - MCP server instance\n * @param methodName - Method name to wrap (tool, resource, prompt)\n */\nfunction wrapMethodHandler(serverInstance: MCPServerInstance, methodName: keyof MCPServerInstance): void {\n  fill(serverInstance, methodName, originalMethod => {\n    return function (this: MCPServerInstance, name: string, ...args: unknown[]) {\n      const handler = args[args.length - 1];\n\n      if (typeof handler !== 'function') {\n        return (originalMethod as (...args: unknown[]) => unknown).call(this, name, ...args);\n      }\n\n      const wrappedHandler = createWrappedHandler(handler as MCPHandler, methodName, name);\n      return (originalMethod as (...args: unknown[]) => unknown).call(this, name, ...args.slice(0, -1), wrappedHandler);\n    };\n  });\n}\n\n/**\n * Creates a wrapped handler with span correlation and error capture\n * @internal\n * @param originalHandler - Original handler function\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n * @returns Wrapped handler function\n */\nfunction createWrappedHandler(originalHandler: MCPHandler, methodName: keyof MCPServerInstance, handlerName: string) {\n  return function (this: unknown, ...handlerArgs: unknown[]): unknown {\n    try {\n      return createErrorCapturingHandler.call(this, originalHandler, methodName, handlerName, handlerArgs);\n    } catch (error) {\n      DEBUG_BUILD && debug.warn('MCP handler wrapping failed:', error);\n      return originalHandler.apply(this, handlerArgs);\n    }\n  };\n}\n\n/**\n * Creates an error-capturing wrapper for handler execution\n * @internal\n * @param originalHandler - Original handler function\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n * @param handlerArgs - Handler arguments\n * @param extraHandlerData - Additional handler context\n * @returns Handler execution result\n */\nfunction createErrorCapturingHandler(\n  this: MCPServerInstance,\n  originalHandler: MCPHandler,\n  methodName: keyof MCPServerInstance,\n  handlerName: string,\n  handlerArgs: unknown[],\n): unknown {\n  try {\n    const result = originalHandler.apply(this, handlerArgs);\n\n    if (result && typeof result === 'object' && typeof (result as { then?: unknown }).then === 'function') {\n      return Promise.resolve(result).catch(error => {\n        captureHandlerError(error, methodName, handlerName);\n        throw error;\n      });\n    }\n\n    return result;\n  } catch (error) {\n    captureHandlerError(error as Error, methodName, handlerName);\n    throw error;\n  }\n}\n\n/**\n * Captures handler execution errors based on handler type\n * @internal\n * @param error - Error to capture\n * @param methodName - MCP method name\n * @param handlerName - Handler identifier\n */\nfunction captureHandlerError(error: Error, methodName: keyof MCPServerInstance, handlerName: string): void {\n  try {\n    const extraData: Record<string, unknown> = {};\n\n    if (methodName === 'tool' || methodName === 'registerTool') {\n      extraData.tool_name = handlerName;\n\n      if (\n        error.name === 'ProtocolValidationError' ||\n        error.message.includes('validation') ||\n        error.message.includes('protocol')\n      ) {\n        captureError(error, 'validation', extraData);\n      } else if (\n        error.name === 'ServerTimeoutError' ||\n        error.message.includes('timed out') ||\n        error.message.includes('timeout')\n      ) {\n        captureError(error, 'timeout', extraData);\n      } else {\n        captureError(error, 'tool_execution', extraData);\n      }\n    } else if (methodName === 'resource' || methodName === 'registerResource') {\n      extraData.resource_uri = handlerName;\n      captureError(error, 'resource_execution', extraData);\n    } else if (methodName === 'prompt' || methodName === 'registerPrompt') {\n      extraData.prompt_name = handlerName;\n      captureError(error, 'prompt_execution', extraData);\n    }\n  } catch (_captureErr) {\n    // noop\n  }\n}\n\n/**\n * Wraps tool handlers to associate them with request spans.\n * Instruments both `tool` (legacy API) and `registerTool` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapToolHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.tool === 'function') wrapMethodHandler(serverInstance, 'tool');\n  if (typeof serverInstance.registerTool === 'function') wrapMethodHandler(serverInstance, 'registerTool');\n}\n\n/**\n * Wraps resource handlers to associate them with request spans.\n * Instruments both `resource` (legacy API) and `registerResource` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapResourceHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.resource === 'function') wrapMethodHandler(serverInstance, 'resource');\n  if (typeof serverInstance.registerResource === 'function') wrapMethodHandler(serverInstance, 'registerResource');\n}\n\n/**\n * Wraps prompt handlers to associate them with request spans.\n * Instruments both `prompt` (legacy API) and `registerPrompt` (new API) if present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapPromptHandlers(serverInstance: MCPServerInstance): void {\n  if (typeof serverInstance.prompt === 'function') wrapMethodHandler(serverInstance, 'prompt');\n  if (typeof serverInstance.registerPrompt === 'function') wrapMethodHandler(serverInstance, 'registerPrompt');\n}\n\n/**\n * Wraps all MCP handler types for span correlation.\n * Supports both the legacy API (`tool`, `resource`, `prompt`) and the newer API\n * (`registerTool`, `registerResource`, `registerPrompt`), instrumenting whichever methods are present.\n * @param serverInstance - MCP server instance\n */\nexport function wrapAllMCPHandlers(serverInstance: MCPServerInstance): void {\n  wrapToolHandlers(serverInstance);\n  wrapResourceHandlers(serverInstance);\n  wrapPromptHandlers(serverInstance);\n}\n\n/**\n * Retroactively wraps handlers on tools, resources, and prompts that were registered\n * before `wrapMcpServerWithSentry` was called.\n *\n * The MCP SDK stores registered entries in private maps and invokes them via the entry's\n * own property at call time — `executor` for tools, `readCallback` for resources, and\n * `handler` for prompts. Replacing those properties\n * in-place is therefore equivalent to having wrapped the original registration call.\n *\n * NOTE: This intentionally accesses private MCP SDK internals (`_registeredTools` etc.).\n * The properties and their shapes are verified against @modelcontextprotocol/sdk source:\n * https://github.com/modelcontextprotocol/typescript-sdk/blob/2c0c481cb9dbfd15c8613f765c940a5f5bace94d/packages/server/src/server/mcp.ts#L304\n * When upgrading the MCP SDK, re-verify that these internal maps and their callable\n * properties still exist and are invoked directly (not captured by closure at registration).\n * All access is defensive — if a property is absent or not a function we skip silently.\n * @internal\n */\nexport function wrapExistingHandlers(serverInstance: MCPServerInstance): void {\n  const server = serverInstance as unknown as Record<string, unknown>;\n\n  // Tools: MCP SDK calls registeredTool.executor (generated from handler at registration time)\n  const registeredTools = server['_registeredTools'];\n  if (registeredTools && typeof registeredTools === 'object') {\n    for (const [name, tool] of Object.entries(registeredTools as Record<string, Record<string, unknown>>)) {\n      if (typeof tool['executor'] === 'function') {\n        tool['executor'] = createWrappedHandler(tool['executor'] as MCPHandler, 'registerTool', name);\n      }\n    }\n  }\n\n  // Resources: MCP SDK calls registeredResource.readCallback\n  const registeredResources = server['_registeredResources'];\n  if (registeredResources && typeof registeredResources === 'object') {\n    for (const [name, resource] of Object.entries(registeredResources as Record<string, Record<string, unknown>>)) {\n      if (typeof resource['readCallback'] === 'function') {\n        resource['readCallback'] = createWrappedHandler(\n          resource['readCallback'] as MCPHandler,\n          'registerResource',\n          name,\n        );\n      }\n    }\n  }\n\n  // Resource templates: MCP SDK calls registeredResourceTemplate.readCallback\n  const registeredResourceTemplates = server['_registeredResourceTemplates'];\n  if (registeredResourceTemplates && typeof registeredResourceTemplates === 'object') {\n    for (const [name, template] of Object.entries(\n      registeredResourceTemplates as Record<string, Record<string, unknown>>,\n    )) {\n      if (typeof template['readCallback'] === 'function') {\n        template['readCallback'] = createWrappedHandler(\n          template['readCallback'] as MCPHandler,\n          'registerResource',\n          name,\n        );\n      }\n    }\n  }\n\n  // Prompts: MCP SDK calls registeredPrompt.handler\n  const registeredPrompts = server['_registeredPrompts'];\n  if (registeredPrompts && typeof registeredPrompts === 'object') {\n    for (const [name, prompt] of Object.entries(registeredPrompts as Record<string, Record<string, unknown>>)) {\n      if (typeof prompt['handler'] === 'function') {\n        prompt['handler'] = createWrappedHandler(prompt['handler'] as MCPHandler, 'registerPrompt', name);\n      }\n    }\n  }\n}\n"],"names":[],"mappings":";;;;;AAmBA,SAAS,iBAAA,CAAkB,gBAAmC,UAAA,EAA2C;AACvG,EAAA,IAAA,CAAK,cAAA,EAAgB,YAAY,CAAA,cAAA,KAAkB;AACjD,IAAA,OAAO,SAAmC,SAAiB,IAAA,EAAiB;AAC1E,MAAA,MAAM,OAAA,GAAU,IAAA,CAAK,IAAA,CAAK,MAAA,GAAS,CAAC,CAAA;AAEpC,MAAA,IAAI,OAAO,YAAY,UAAA,EAAY;AACjC,QAAA,OAAQ,cAAA,CAAmD,IAAA,CAAK,IAAA,EAAM,IAAA,EAAM,GAAG,IAAI,CAAA;AAAA,MACrF;AAEA,MAAA,MAAM,cAAA,GAAiB,oBAAA,CAAqB,OAAA,EAAuB,UAAA,EAAY,IAAI,CAAA;AACnF,MAAA,OAAQ,cAAA,CAAmD,IAAA,CAAK,IAAA,EAAM,IAAA,EAAM,GAAG,KAAK,KAAA,CAAM,CAAA,EAAG,EAAE,CAAA,EAAG,cAAc,CAAA;AAAA,IAClH,CAAA;AAAA,EACF,CAAC,CAAA;AACH;AAUA,SAAS,oBAAA,CAAqB,eAAA,EAA6B,UAAA,EAAqC,WAAA,EAAqB;AACnH,EAAA,OAAO,YAA4B,WAAA,EAAiC;AAClE,IAAA,IAAI;AACF,MAAA,OAAO,4BAA4B,IAAA,CAAK,IAAA,EAAM,eAAA,EAAiB,UAAA,EAAY,aAAa,WAAW,CAAA;AAAA,IACrG,SAAS,KAAA,EAAO;AACd,MAAA,WAAA,IAAe,KAAA,CAAM,IAAA,CAAK,8BAAA,EAAgC,KAAK,CAAA;AAC/D,MAAA,OAAO,eAAA,CAAgB,KAAA,CAAM,IAAA,EAAM,WAAW,CAAA;AAAA,IAChD;AAAA,EACF,CAAA;AACF;AAYA,SAAS,2BAAA,CAEP,eAAA,EACA,UAAA,EACA,WAAA,EACA,WAAA,EACS;AACT,EAAA,IAAI;AACF,IAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,KAAA,CAAM,IAAA,EAAM,WAAW,CAAA;AAEtD,IAAA,IAAI,UAAU,OAAO,MAAA,KAAW,YAAY,OAAQ,MAAA,CAA8B,SAAS,UAAA,EAAY;AACrG,MAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,MAAM,CAAA,CAAE,MAAM,CAAA,KAAA,KAAS;AAC5C,QAAA,mBAAA,CAAoB,KAAA,EAAO,YAAY,WAAW,CAAA;AAClD,QAAA,MAAM,KAAA;AAAA,MACR,CAAC,CAAA;AAAA,IACH;AAEA,IAAA,OAAO,MAAA;AAAA,EACT,SAAS,KAAA,EAAO;AACd,IAAA,mBAAA,CAAoB,KAAA,EAAgB,YAAY,WAAW,CAAA;AAC3D,IAAA,MAAM,KAAA;AAAA,EACR;AACF;AASA,SAAS,mBAAA,CAAoB,KAAA,EAAc,UAAA,EAAqC,WAAA,EAA2B;AACzG,EAAA,IAAI;AACF,IAAA,MAAM,YAAqC,EAAC;AAE5C,IAAA,IAAI,UAAA,KAAe,MAAA,IAAU,UAAA,KAAe,cAAA,EAAgB;AAC1D,MAAA,SAAA,CAAU,SAAA,GAAY,WAAA;AAEtB,MAAA,IACE,KAAA,CAAM,IAAA,KAAS,yBAAA,IACf,KAAA,CAAM,OAAA,CAAQ,QAAA,CAAS,YAAY,CAAA,IACnC,KAAA,CAAM,OAAA,CAAQ,QAAA,CAAS,UAAU,CAAA,EACjC;AACA,QAAA,YAAA,CAAa,KAAA,EAAO,cAAc,SAAS,CAAA;AAAA,MAC7C,CAAA,MAAA,IACE,KAAA,CAAM,IAAA,KAAS,oBAAA,IACf,KAAA,CAAM,OAAA,CAAQ,QAAA,CAAS,WAAW,CAAA,IAClC,KAAA,CAAM,OAAA,CAAQ,QAAA,CAAS,SAAS,CAAA,EAChC;AACA,QAAA,YAAA,CAAa,KAAA,EAAO,WAAW,SAAS,CAAA;AAAA,MAC1C,CAAA,MAAO;AACL,QAAA,YAAA,CAAa,KAAA,EAAO,kBAAkB,SAAS,CAAA;AAAA,MACjD;AAAA,IACF,CAAA,MAAA,IAAW,UAAA,KAAe,UAAA,IAAc,UAAA,KAAe,kBAAA,EAAoB;AACzE,MAAA,SAAA,CAAU,YAAA,GAAe,WAAA;AACzB,MAAA,YAAA,CAAa,KAAA,EAAO,sBAAsB,SAAS,CAAA;AAAA,IACrD,CAAA,MAAA,IAAW,UAAA,KAAe,QAAA,IAAY,UAAA,KAAe,gBAAA,EAAkB;AACrE,MAAA,SAAA,CAAU,WAAA,GAAc,WAAA;AACxB,MAAA,YAAA,CAAa,KAAA,EAAO,oBAAoB,SAAS,CAAA;AAAA,IACnD;AAAA,EACF,SAAS,WAAA,EAAa;AAAA,EAEtB;AACF;AAOO,SAAS,iBAAiB,cAAA,EAAyC;AACxE,EAAA,IAAI,OAAO,cAAA,CAAe,IAAA,KAAS,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,MAAM,CAAA;AACvF,EAAA,IAAI,OAAO,cAAA,CAAe,YAAA,KAAiB,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,cAAc,CAAA;AACzG;AAOO,SAAS,qBAAqB,cAAA,EAAyC;AAC5E,EAAA,IAAI,OAAO,cAAA,CAAe,QAAA,KAAa,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,UAAU,CAAA;AAC/F,EAAA,IAAI,OAAO,cAAA,CAAe,gBAAA,KAAqB,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,kBAAkB,CAAA;AACjH;AAOO,SAAS,mBAAmB,cAAA,EAAyC;AAC1E,EAAA,IAAI,OAAO,cAAA,CAAe,MAAA,KAAW,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,QAAQ,CAAA;AAC3F,EAAA,IAAI,OAAO,cAAA,CAAe,cAAA,KAAmB,UAAA,EAAY,iBAAA,CAAkB,gBAAgB,gBAAgB,CAAA;AAC7G;AAQO,SAAS,mBAAmB,cAAA,EAAyC;AAC1E,EAAA,gBAAA,CAAiB,cAAc,CAAA;AAC/B,EAAA,oBAAA,CAAqB,cAAc,CAAA;AACnC,EAAA,kBAAA,CAAmB,cAAc,CAAA;AACnC;AAmBO,SAAS,qBAAqB,cAAA,EAAyC;AAC5E,EAAA,MAAM,MAAA,GAAS,cAAA;AAGf,EAAA,MAAM,eAAA,GAAkB,OAAO,kBAAkB,CAAA;AACjD,EAAA,IAAI,eAAA,IAAmB,OAAO,eAAA,KAAoB,QAAA,EAAU;AAC1D,IAAA,KAAA,MAAW,CAAC,IAAA,EAAM,IAAI,KAAK,MAAA,CAAO,OAAA,CAAQ,eAA0D,CAAA,EAAG;AACrG,MAAA,IAAI,OAAO,IAAA,CAAK,UAAU,CAAA,KAAM,UAAA,EAAY;AAC1C,QAAA,IAAA,CAAK,UAAU,CAAA,GAAI,oBAAA,CAAqB,KAAK,UAAU,CAAA,EAAiB,gBAAgB,IAAI,CAAA;AAAA,MAC9F;AAAA,IACF;AAAA,EACF;AAGA,EAAA,MAAM,mBAAA,GAAsB,OAAO,sBAAsB,CAAA;AACzD,EAAA,IAAI,mBAAA,IAAuB,OAAO,mBAAA,KAAwB,QAAA,EAAU;AAClE,IAAA,KAAA,MAAW,CAAC,IAAA,EAAM,QAAQ,KAAK,MAAA,CAAO,OAAA,CAAQ,mBAA8D,CAAA,EAAG;AAC7G,MAAA,IAAI,OAAO,QAAA,CAAS,cAAc,CAAA,KAAM,UAAA,EAAY;AAClD,QAAA,QAAA,CAAS,cAAc,CAAA,GAAI,oBAAA;AAAA,UACzB,SAAS,cAAc,CAAA;AAAA,UACvB,kBAAA;AAAA,UACA;AAAA,SACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAGA,EAAA,MAAM,2BAAA,GAA8B,OAAO,8BAA8B,CAAA;AACzE,EAAA,IAAI,2BAAA,IAA+B,OAAO,2BAAA,KAAgC,QAAA,EAAU;AAClF,IAAA,KAAA,MAAW,CAAC,IAAA,EAAM,QAAQ,CAAA,IAAK,MAAA,CAAO,OAAA;AAAA,MACpC;AAAA,KACF,EAAG;AACD,MAAA,IAAI,OAAO,QAAA,CAAS,cAAc,CAAA,KAAM,UAAA,EAAY;AAClD,QAAA,QAAA,CAAS,cAAc,CAAA,GAAI,oBAAA;AAAA,UACzB,SAAS,cAAc,CAAA;AAAA,UACvB,kBAAA;AAAA,UACA;AAAA,SACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAGA,EAAA,MAAM,iBAAA,GAAoB,OAAO,oBAAoB,CAAA;AACrD,EAAA,IAAI,iBAAA,IAAqB,OAAO,iBAAA,KAAsB,QAAA,EAAU;AAC9D,IAAA,KAAA,MAAW,CAAC,IAAA,EAAM,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,iBAA4D,CAAA,EAAG;AACzG,MAAA,IAAI,OAAO,MAAA,CAAO,SAAS,CAAA,KAAM,UAAA,EAAY;AAC3C,QAAA,MAAA,CAAO,SAAS,CAAA,GAAI,oBAAA,CAAqB,OAAO,SAAS,CAAA,EAAiB,kBAAkB,IAAI,CAAA;AAAA,MAClG;AAAA,IACF;AAAA,EACF;AACF;;;;"}

package/build/esm/integrations/mcp-server/index.js

@@ -4,90 +4,37 @@ import { wrapAllMCPHandlers, wrapExistingHandlers } from './handlers.js';
 import { wrapTransportOnMessage, wrapTransportSend, wrapTransportOnClose, wrapTransportError } from './transport.js';
 import { validateMcpServerInstance } from './validation.js';
 
-/**
- * Tracks wrapped MCP server instances to prevent double-wrapping
- * @internal
- */
-const wrappedMcpServerInstances = new WeakSet();
-
-/**
- * Wraps a MCP Server instance from the `@modelcontextprotocol/sdk` package with Sentry instrumentation.
- *
- * Compatible with versions `^1.9.0` of the `@modelcontextprotocol/sdk` package (legacy `tool`/`resource`/`prompt` API)
- * and versions that expose the newer `registerTool`/`registerResource`/`registerPrompt` API (introduced in 1.x, sole API in 2.x).
- * Automatically instruments transport methods and handler functions for comprehensive monitoring.
- *
- * Both call orderings are supported: wrapping before or after registering tools, resources,
- * and prompts. Sentry patches the registration methods for future handlers and retroactively
- * wraps any already-registered ones. Wrapping at construction time is recommended by
- * convention (consistent with other SDK integrations), but is not required.
- *
- * @example
- * ```typescript
- * import * as Sentry from '@sentry/core';
- * import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
- * import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
- *
- * // Wrap first, then register tools — this is the correct order
- * const server = Sentry.wrapMcpServerWithSentry(
- *   new McpServer({ name: "my-server", version: "1.0.0" })
- * );
- *
- * server.registerTool('my-tool', schema, handler);
- *
- * // Explicitly control input/output capture
- * const server = Sentry.wrapMcpServerWithSentry(
- *   new McpServer({ name: "my-server", version: "1.0.0" }),
- *   { recordInputs: true, recordOutputs: false }
- * );
- *
- * const transport = new StreamableHTTPServerTransport();
- * await server.connect(transport);
- * ```
- *
- * @param mcpServerInstance - MCP server instance to instrument
- * @param options - Optional configuration for recording inputs and outputs
- * @returns Instrumented server instance (same reference)
- */
+const wrappedMcpServerInstances = /* @__PURE__ */ new WeakSet();
 function wrapMcpServerWithSentry(mcpServerInstance, options) {
   if (wrappedMcpServerInstances.has(mcpServerInstance)) {
     return mcpServerInstance;
   }
-
   if (!validateMcpServerInstance(mcpServerInstance)) {
     return mcpServerInstance;
   }
-
-  const serverInstance = mcpServerInstance ;
+  const serverInstance = mcpServerInstance;
   const client = getClient();
   const sendDefaultPii = Boolean(client?.getOptions().sendDefaultPii);
-
   const resolvedOptions = {
     recordInputs: options?.recordInputs ?? sendDefaultPii,
-    recordOutputs: options?.recordOutputs ?? sendDefaultPii,
+    recordOutputs: options?.recordOutputs ?? sendDefaultPii
   };
-
-  fill(serverInstance, 'connect', originalConnect => {
-    return async function ( transport, ...restArgs) {
-      const result = await (originalConnect ).call(
+  fill(serverInstance, "connect", (originalConnect) => {
+    return async function(transport, ...restArgs) {
+      const result = await originalConnect.call(
         this,
         transport,
-        ...restArgs,
+        ...restArgs
       );
-
       wrapTransportOnMessage(transport, resolvedOptions);
       wrapTransportSend(transport, resolvedOptions);
       wrapTransportOnClose(transport);
       wrapTransportError(transport);
-
       return result;
     };
   });
-
   wrapAllMCPHandlers(serverInstance);
-
   wrapExistingHandlers(serverInstance);
-
   wrappedMcpServerInstances.add(mcpServerInstance);
   return mcpServerInstance;
 }