@dunx/http 2.4.0 → 3.0.0

package/dist/chunk-25g22350.js

@@ -0,0 +1,58 @@
+// @bun
+// src/server/trace-context.ts
+var TRACEPARENT_HEADER = "traceparent";
+var TRACESTATE_HEADER = "tracestate";
+var HEX_32 = /^[0-9a-f]{32}$/;
+var HEX_16 = /^[0-9a-f]{16}$/;
+var HEX_2 = /^[0-9a-f]{2}$/;
+var ZERO_TRACE = "0".repeat(32);
+var ZERO_SPAN = "0".repeat(16);
+var SAMPLED = 1;
+var TRACE = Symbol.for("dunx.http.trace");
+var mintSpanId = () => Buffer.from(crypto.getRandomValues(new Uint8Array(8))).toString("hex");
+
+class TraceContext {
+  static adopt(req, requestId) {
+    const inbound = TraceContext.#parse(req.headers.get(TRACEPARENT_HEADER));
+    const state = req.headers.get(TRACESTATE_HEADER);
+    const trace = {
+      traceId: inbound?.traceId ?? requestId.replaceAll("-", ""),
+      spanId: mintSpanId(),
+      ...inbound === undefined ? {} : { parentSpanId: inbound.spanId },
+      flags: inbound?.flags ?? "01",
+      ...inbound !== undefined && state !== null ? { state } : {}
+    };
+    req[TRACE] = trace;
+    return trace;
+  }
+  static of(req) {
+    return req[TRACE];
+  }
+  static header(trace) {
+    return `00-${trace.traceId}-${trace.spanId}-${trace.flags}`;
+  }
+  static sampled(trace) {
+    return (Number.parseInt(trace.flags, 16) & SAMPLED) === SAMPLED;
+  }
+  static #parse(header) {
+    if (header === null)
+      return;
+    const parts = header.split("-");
+    if (parts.length < 4)
+      return;
+    const [version, traceId, spanId, flags] = parts;
+    if (!HEX_2.test(version) || version === "ff")
+      return;
+    if (version === "00" && parts.length !== 4)
+      return;
+    if (!HEX_32.test(traceId) || traceId === ZERO_TRACE)
+      return;
+    if (!HEX_16.test(spanId) || spanId === ZERO_SPAN)
+      return;
+    if (!HEX_2.test(flags))
+      return;
+    return { traceId, spanId, flags };
+  }
+}
+
+export { TRACEPARENT_HEADER, TRACESTATE_HEADER, TraceContext };