o11y 260.4.0 → 260.5.0

package/dist/modules/o11y/client/client.js

@@ -511,10 +511,18 @@ const headerSpanId = 'X-B3-SpanId';
 const headerSampled = 'X-B3-Sampled';
 const headerParentSpanId = 'X-B3-ParentSpanId';
 const headerRequestId = 'X-SFDC-Request-Id';
+const headerBaggage = 'baggage';
 const defaultActivityNameForFetch = 'fetch';
 const defaultActivityNameForXhrSend = 'xhr_send';
 const defaultTimingActivityName = 'perf-timing';
 const defaultFuzzyMapRange = 50;
+const userSettableHeaders = [
+    headerBaggage,
+    headerTraceId,
+    headerSpanId,
+    headerParentSpanId,
+    headerSampled
+];
 const o11yInfoForXHR = new WeakMap();
 class Tracing {
     constructor(_instr, _idleDetector) {
@@ -553,6 +561,18 @@ class Tracing {
         }
         return text;
     }
+    static setHeaders(headers) {
+        if (headers === undefined) {
+            Tracing._userHeaders = {};
+            return;
+        }
+        Object.entries(headers).forEach(([key]) => {
+            if (!userSettableHeaders.includes(key)) {
+                delete headers[key];
+            }
+        });
+        Tracing._userHeaders = headers;
+    }
     static getHeaders(traceId, spanId, isSampled, options) {
         utility.requireArgument(spanId, 'spanId', 'string');
         const useB3Headers = options === null || options === void 0 ? void 0 : options.useB3Headers;
@@ -668,6 +688,13 @@ class Tracing {
                 traceHeaders = undefined;
             }
         }
+        if (Tracing._userHeaders && Object.keys(Tracing._userHeaders).length > 0) {
+            traceHeaders = { ...traceHeaders };
+            Object.entries(Tracing._userHeaders).forEach(([key, value]) => {
+                traceHeaders[key] = value;
+            });
+            Tracing._userHeaders = undefined;
+        }
         return traceHeaders || {};
     }
     static _extractHeader(headerName, input, init) {
@@ -1884,6 +1911,10 @@ class InstrumentationImpl {
             act.stop((_c = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _c === void 0 ? void 0 : _c.schema, (_d = options === null || options === void 0 ? void 0 : options.stopPayload) === null || _d === void 0 ? void 0 : _d.payload, options === null || options === void 0 ? void 0 : options.stopOptions);
         }
     }
+    invokeNetworkCallWithHeaders(apiCall, params, headers) {
+        Tracing.setHeaders(headers);
+        return apiCall(...params);
+    }
 }
 InstrumentationImpl.defaultAppName = 'APP_NOT_REGISTERED';
 
@@ -2397,6 +2428,7 @@ class NextgenImpl {
             registerForLogPrompt: this._appInstr.registerForLogPrompt.bind(this._appInstr),
             activity: this._appInstr.activity.bind(this._appInstr),
             activityAsync: this._appInstr.activityAsync.bind(this._appInstr),
+            invokeNetworkCallWithHeaders: this._appInstr.invokeNetworkCallWithHeaders.bind(this._appInstr),
             startRootActivity: this._appInstr.startRootActivity.bind(this._appInstr),
             registerLogCollector: this._isSender ? noOpFunc : this.registerLogCollector.bind(this),
             registerMetricsCollector: this._isSender
@@ -2751,7 +2783,7 @@ class ConsoleCollector {
     }
 }
 
-const version = '260.4.0';
+const version = '260.5.0';
 
 const idleDetector = new IdleDetectorImpl({
     logThreshold: 300,