@obsrviq/tracker 0.6.0 → 0.7.1

package/dist/index.js

@@ -792,6 +792,83 @@ function tokenBucket(rate2, burst) {
   };
 }
 
+// src/trace.ts
+function randomHex(bytes) {
+  const arr = new Uint8Array(bytes);
+  if (typeof crypto !== "undefined" && typeof crypto.getRandomValues === "function") {
+    crypto.getRandomValues(arr);
+  } else {
+    for (let i = 0; i < bytes; i++) arr[i] = Math.random() * 256 | 0;
+  }
+  let s = "";
+  for (let i = 0; i < bytes; i++) s += arr[i].toString(16).padStart(2, "0");
+  return s;
+}
+function newTrace() {
+  const traceId = randomHex(16);
+  const spanId = randomHex(8);
+  return { traceId, spanId, header: `00-${traceId}-${spanId}-01` };
+}
+function traceTarget(rawUrl, cfg) {
+  if (!cfg.tracePropagation) return false;
+  let u;
+  try {
+    u = new URL(rawUrl, typeof location !== "undefined" ? location.href : void 0);
+  } catch {
+    return false;
+  }
+  if (u.protocol !== "http:" && u.protocol !== "https:") return false;
+  if (typeof location !== "undefined" && u.origin === location.origin) return true;
+  return hostAllowed(u.hostname, cfg.traceHosts);
+}
+function hostAllowed(hostname, hosts) {
+  const h = hostname.toLowerCase();
+  for (const raw of hosts) {
+    let entry = String(raw).trim().toLowerCase();
+    if (!entry) continue;
+    if (entry.includes("://")) {
+      try {
+        entry = new URL(entry).hostname;
+      } catch {
+      }
+    }
+    entry = entry.replace(/^\*?\./, "");
+    if (h === entry || h.endsWith("." + entry)) return true;
+  }
+  return false;
+}
+function traceIdOf(header) {
+  if (!header) return null;
+  const parts = header.split("-");
+  return parts.length >= 2 && /^[0-9a-f]{32}$/.test(parts[1]) ? parts[1] : null;
+}
+function existingFetchTraceparent(input, init) {
+  const hasInitHeaders = !!init && "headers" in init && init.headers != null;
+  const src = hasInitHeaders ? init.headers : input instanceof Request ? input.headers : void 0;
+  if (!src) return null;
+  try {
+    return new Headers(src).get("traceparent");
+  } catch {
+    return null;
+  }
+}
+function applyFetchTrace(input, init) {
+  const existing = existingFetchTraceparent(input, init);
+  if (existing) {
+    return { input, init, traceId: traceIdOf(existing) ?? "" };
+  }
+  const tr = newTrace();
+  const hasInitHeaders = !!init && "headers" in init && init.headers != null;
+  if (input instanceof Request && !hasInitHeaders) {
+    const h2 = new Headers(input.headers);
+    h2.set("traceparent", tr.header);
+    return { input: new Request(input, { headers: h2 }), init, traceId: tr.traceId };
+  }
+  const h = new Headers(init?.headers ?? void 0);
+  h.set("traceparent", tr.header);
+  return { input, init: { ...init || {}, headers: h }, traceId: tr.traceId };
+}
+
 // src/network.ts
 var BODY_CAP = 4 * 1024;
 var NET_RATE = 50;
@@ -807,6 +884,7 @@ var HEADER_ALLOW = /* @__PURE__ */ new Set([
   "content-encoding",
   "x-request-id",
   "x-correlation-id",
+  "traceparent",
   "server",
   "vary"
 ]);
@@ -1041,6 +1119,16 @@ function wrapFetch(ctx, tr, emitNet) {
     const perfStart = perfNow();
     const method = (init?.method || (input instanceof Request ? input.method : "GET")).toUpperCase();
     const url = redactUrl(rawUrl);
+    let traceId;
+    if (traceTarget(rawUrl, ctx.config)) {
+      try {
+        const inj = applyFetchTrace(input, init);
+        input = inj.input;
+        init = inj.init;
+        traceId = inj.traceId || void 0;
+      } catch {
+      }
+    }
     const reqHeaders = pickHeaders(
       ctx.config.redactHeaders,
       headerEntries(init?.headers ?? (input instanceof Request ? input.headers : void 0))
@@ -1063,6 +1151,7 @@ function wrapFetch(ctx, tr, emitNet) {
           ok: false,
           resourceType: "fetch",
           requestHeaders: reqHeaders,
+          traceId,
           timing,
           ...partial
         };
@@ -1144,6 +1233,19 @@ function wrapXhr(ctx, tr, emitNet) {
       meta.startT = ctx.clock.now();
       meta.perfStart = perfNow();
       if (ctx.config.captureNetworkBodies) meta.body = stringifyBody(body);
+      if (traceTarget(meta.rawUrl, ctx.config)) {
+        try {
+          const appSet = meta.reqHeaders["traceparent"];
+          if (appSet) {
+            meta.traceId = traceIdOf(appSet) ?? void 0;
+          } else {
+            const tr2 = newTrace();
+            this.setRequestHeader("traceparent", tr2.header);
+            meta.traceId = tr2.traceId;
+          }
+        } catch {
+        }
+      }
       const finish = (error) => {
         const endT = ctx.clock.now();
         const base = {
@@ -1178,6 +1280,7 @@ function wrapXhr(ctx, tr, emitNet) {
             ok: status >= 200 && status < 400,
             resourceType: "xhr",
             requestHeaders: meta.reqHeaders,
+            traceId: meta.traceId,
             responseHeaders: respHeaders,
             responseSize,
             requestBody: ctx.config.captureNetworkBodies ? bodyPreview(meta.body, void 0) : void 0,
@@ -2179,6 +2282,8 @@ var DEFAULTS = {
   noPrivacy: false,
   captureNetworkBodies: false,
   redactHeaders: ["authorization", "cookie", "set-cookie"],
+  tracePropagation: false,
+  traceHosts: [],
   sampleRate: 1,
   recordCanvas: false,
   endpoint: "https://in.lumera.app",