o11y 260.4.0 → 260.8.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) {
@@ -1684,13 +1711,17 @@ class InstrumentationImpl {
                 noRerooting = true;
                 context = {
                     rootId,
-                    isRootActivitySampled: options.instrumentationContext.isRootActivitySampled
+                    isRootActivitySampled: options.instrumentationContext.isRootActivitySampled,
+                    isRootActivityExternalSampled: options.instrumentationContext.isRootActivityExternalSampled
                 };
             }
             else {
                 context = this._nextGen.getDefaultInstrumentationContext();
             }
-            const activity = new ActivityImpl(name, context.rootId, this._onActivityStoppedCallback, this._onActivityErrorCallback, maxActivityDurationMsecs, options === null || options === void 0 ? void 0 : options.id, context.isRootActivitySampled, options === null || options === void 0 ? void 0 : options.perfStartOverride, noRerooting, options === null || options === void 0 ? void 0 : options.isExternal, parentActivity);
+            const isActivitySampled = (options === null || options === void 0 ? void 0 : options.isExternal)
+                ? context.isRootActivityExternalSampled
+                : context.isRootActivitySampled;
+            const activity = new ActivityImpl(name, context.rootId, this._onActivityStoppedCallback, this._onActivityErrorCallback, maxActivityDurationMsecs, options === null || options === void 0 ? void 0 : options.id, isActivitySampled, options === null || options === void 0 ? void 0 : options.perfStartOverride, noRerooting, options === null || options === void 0 ? void 0 : options.isExternal, parentActivity);
             return activity;
         }, activityNoOp, { op: 'startActivity' });
     }
@@ -1884,6 +1915,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';
 
@@ -1939,9 +1974,17 @@ class RootActivityHistoryImpl {
 }
 
 class RootActivityImpl extends ActivityImpl {
-    constructor(name, onStopped, onError, rootId, isSampled, startPerfTime, generatedRootIdLength) {
+    constructor(name, onStopped, onError, rootId, isSampled, startPerfTime, generatedRootIdLength, isExternalSampled) {
         super(name, undefined, onStopped, onError, undefined, rootId || utility.generateUniqueId(generatedRootIdLength !== null && generatedRootIdLength !== void 0 ? generatedRootIdLength : 32), isSampled, startPerfTime);
         this._isRoot = true;
+        this._isExternalSampled = isExternalSampled;
+    }
+    get isExternalSampled() {
+        var _a;
+        return (_a = this._isExternalSampled) !== null && _a !== void 0 ? _a : false;
+    }
+    set isExternalSampled(value) {
+        this._isExternalSampled = value;
     }
     get preRootId() {
         return this._preRootId;
@@ -2006,13 +2049,13 @@ class AppInstrumentationImpl extends InstrumentationImpl {
         this._rootActivityHistory = new RootActivityHistoryImpl();
         this._onRootActivityStoppedCallback = this._handleRootActivityStop.bind(this);
     }
-    startRootActivity(name, rootId, isSampled, options) {
+    startRootActivity(name, rootId, isSampled, options, isExternalActivitySampled) {
         return this._safety.tryCatchReturn(() => {
             utility.requireArgument(name, 'name');
             if ((options === null || options === void 0 ? void 0 : options.id) !== undefined) {
                 throw new Error('Cannot specify id for root activity, use rootId argument instead');
             }
-            const newRootActivity = new RootActivityImpl(name, this._onRootActivityStoppedCallback, this._onActivityErrorCallback, rootId, isSampled, options === null || options === void 0 ? void 0 : options.perfStartOverride, this._nextGen.preferredRootIdLength);
+            const newRootActivity = new RootActivityImpl(name, this._onRootActivityStoppedCallback, this._onActivityErrorCallback, rootId, isSampled, options === null || options === void 0 ? void 0 : options.perfStartOverride, this._nextGen.preferredRootIdLength, isExternalActivitySampled);
             if (!this._allowMulti) {
                 const entry = {
                     id: newRootActivity.getId(),
@@ -2058,6 +2101,10 @@ class AppInstrumentationImpl extends InstrumentationImpl {
         var _a;
         return (_a = this._singleRootActivity) === null || _a === void 0 ? void 0 : _a.getIsSampled();
     }
+    isSingleRootActivityExternalSampled() {
+        var _a, _b;
+        return (_b = (_a = this._singleRootActivity) === null || _a === void 0 ? void 0 : _a.isExternalSampled) !== null && _b !== void 0 ? _b : false;
+    }
     _initMetrics() {
         return new MetricsImpl(this.name, () => this.name);
     }
@@ -2397,6 +2444,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
@@ -2532,10 +2580,11 @@ class NextgenImpl {
         return this._autoClickTracker;
     }
     getDefaultInstrumentationContext() {
-        var _a, _b;
+        var _a, _b, _c;
         return {
             rootId: (_a = this._appInstr) === null || _a === void 0 ? void 0 : _a.getSingleRootActivityId(),
-            isRootActivitySampled: (_b = this._appInstr) === null || _b === void 0 ? void 0 : _b.isSingleRootActivitySampled()
+            isRootActivitySampled: (_b = this._appInstr) === null || _b === void 0 ? void 0 : _b.isSingleRootActivitySampled(),
+            isRootActivityExternalSampled: (_c = this._appInstr) === null || _c === void 0 ? void 0 : _c.isSingleRootActivityExternalSampled()
         };
     }
     activateClickTracker() {
@@ -2751,7 +2800,7 @@ class ConsoleCollector {
     }
 }
 
-const version = '260.4.0';
+const version = '260.8.0';
 
 const idleDetector = new IdleDetectorImpl({
     logThreshold: 300,