@wrongstack/core 0.313.0 → 0.313.1

package/dist/chunk-HA4ZYA6T.js

@@ -0,0 +1,178 @@
+// src/observability/network-telemetry.ts
+import { AsyncLocalStorage } from "node:async_hooks";
+import { createHash, randomUUID } from "node:crypto";
+import { channel } from "node:diagnostics_channel";
+var storage = new AsyncLocalStorage();
+var requests = /* @__PURE__ */ new WeakMap();
+var users = 0;
+var unsubscribe;
+function runWithNetworkTelemetry(context, run) {
+  return storage.run(context, run);
+}
+function startNetworkTelemetryMonitor() {
+  users++;
+  if (!unsubscribe) unsubscribe = subscribe();
+  let stopped = false;
+  return () => {
+    if (stopped) return;
+    stopped = true;
+    users--;
+    if (users <= 0) {
+      unsubscribe?.();
+      unsubscribe = void 0;
+      users = 0;
+    }
+  };
+}
+function subscribe() {
+  const create = channel("undici:request:create");
+  const headers = channel("undici:request:headers");
+  const error = channel("undici:request:error");
+  const onCreate = (message) => {
+    const context = storage.getStore();
+    const request = objectField(message, "request");
+    if (!context || !request) return;
+    const target = safeTarget(request);
+    const startedAt = (/* @__PURE__ */ new Date()).toISOString();
+    const requestBytes = numberField(request, "contentLength");
+    const state = {
+      ...context,
+      requestId: randomUUID(),
+      ...target,
+      ...requestBytes !== void 0 ? { requestBytes } : {},
+      startedAt,
+      startedNs: process.hrtime.bigint()
+    };
+    requests.set(request, state);
+    context.events.emit("network.request.started", {
+      ...identity(state),
+      requestId: state.requestId,
+      method: state.method,
+      scheme: state.scheme,
+      serverAddress: state.serverAddress,
+      ...state.serverPort !== void 0 ? { serverPort: state.serverPort } : {},
+      pathHash: state.pathHash,
+      queryKeys: queryKeys(request),
+      ...state.requestBytes !== void 0 ? { requestBytes: state.requestBytes } : {},
+      startedAt
+    });
+  };
+  const onHeaders = (message) => {
+    const request = objectField(message, "request");
+    if (!request) return;
+    const state = requests.get(request);
+    if (!state) return;
+    requests.delete(request);
+    const response = objectField(message, "response");
+    const responseBytes = contentLength(response);
+    state.events.emit("network.request.completed", {
+      ...identity(state),
+      requestId: state.requestId,
+      method: state.method,
+      scheme: state.scheme,
+      serverAddress: state.serverAddress,
+      ...state.serverPort !== void 0 ? { serverPort: state.serverPort } : {},
+      pathHash: state.pathHash,
+      statusCode: numberField(response, "statusCode") ?? 0,
+      durationMs: elapsed(state.startedNs),
+      ...state.requestBytes !== void 0 ? { requestBytes: state.requestBytes } : {},
+      ...responseBytes !== void 0 ? { responseBytes } : {},
+      completedAt: (/* @__PURE__ */ new Date()).toISOString()
+    });
+  };
+  const onError = (message) => {
+    const request = objectField(message, "request");
+    if (!request) return;
+    const state = requests.get(request);
+    if (!state) return;
+    requests.delete(request);
+    const err = objectField(message, "error");
+    const errorCode = stringField(err, "code");
+    state.events.emit("network.request.failed", {
+      ...identity(state),
+      requestId: state.requestId,
+      method: state.method,
+      scheme: state.scheme,
+      serverAddress: state.serverAddress,
+      ...state.serverPort !== void 0 ? { serverPort: state.serverPort } : {},
+      pathHash: state.pathHash,
+      durationMs: elapsed(state.startedNs),
+      errorName: stringField(err, "name") ?? "Error",
+      ...errorCode ? { errorCode } : {},
+      failedAt: (/* @__PURE__ */ new Date()).toISOString()
+    });
+  };
+  create.subscribe(onCreate);
+  headers.subscribe(onHeaders);
+  error.subscribe(onError);
+  return () => {
+    create.unsubscribe(onCreate);
+    headers.unsubscribe(onHeaders);
+    error.unsubscribe(onError);
+  };
+}
+function identity(state) {
+  return {
+    sessionId: state.sessionId,
+    ...state.traceId ? { traceId: state.traceId } : {},
+    ...state.agentId ? { agentId: state.agentId } : {},
+    ...state.toolCallId ? { toolCallId: state.toolCallId } : {},
+    ...state.attemptId ? { attemptId: state.attemptId } : {},
+    initiator: state.initiator,
+    operationName: state.operationName
+  };
+}
+function safeTarget(request) {
+  const origin = String(request.origin ?? "");
+  const path = String(request.path ?? "");
+  let url;
+  try {
+    url = new URL(path, origin);
+  } catch {
+    url = new URL("http://invalid.local/");
+  }
+  return {
+    method: String(request.method ?? "GET").toUpperCase(),
+    scheme: url.protocol.replace(":", ""),
+    serverAddress: url.hostname,
+    ...url.port ? { serverPort: Number(url.port) } : {},
+    pathHash: hash(url.pathname)
+  };
+}
+function queryKeys(request) {
+  try {
+    return [...new URL(String(request.path ?? ""), String(request.origin ?? "")).searchParams.keys()].filter((v, i, a) => a.indexOf(v) === i).slice(0, 50);
+  } catch {
+    return [];
+  }
+}
+function contentLength(response) {
+  const raw = response?.headers;
+  if (!Array.isArray(raw)) return void 0;
+  for (let index = 0; index < raw.length - 1; index += 2) if (String(raw[index]).toLowerCase() === "content-length") {
+    const value = Number(String(raw[index + 1]));
+    return Number.isFinite(value) ? value : void 0;
+  }
+  return void 0;
+}
+function objectField(value, key) {
+  return value && typeof value === "object" && value[key] && typeof value[key] === "object" ? value[key] : void 0;
+}
+function stringField(value, key) {
+  return typeof value?.[key] === "string" ? value[key] : void 0;
+}
+function numberField(value, key) {
+  return typeof value?.[key] === "number" ? value[key] : void 0;
+}
+function elapsed(start) {
+  return Number(process.hrtime.bigint() - start) / 1e6;
+}
+function hash(value) {
+  return createHash("sha256").update(value).digest("hex");
+}
+
+export {
+  runWithNetworkTelemetry,
+  startNetworkTelemetryMonitor
+};
+//# sourceMappingURL=chunk-HA4ZYA6T.js.map