123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
type ExtraData = Record<string, null | boolean | string | number | undefined>;
interface CreateArgBase {	name: string;	extra?: ExtraData;}
interface NoParent extends CreateArgBase {	kind: "no-parent";	service: string;}
interface Skipped {	kind: "skipped";}
interface FromRequest extends CreateArgBase {	kind: "request";	parent: Request;	service: string;}
interface FromSpan extends CreateArgBase {	kind: "span";	parent: Span;}
export class Span {	spans: Span[] = []; // all the spans	service = ""; // the service the span is from	name = ""; // role of the span	startTime = 0; // when the span starts	duration?: number; // known when complete	spanId = "skipped"; // uniquely identifies the span	traceId = ""; // identifies which trace the span belongs to	parentId?: string; // the parent span that created this span	extra?: ExtraData; // extra data that might be helpful.
	constructor(arg: Skipped | NoParent | FromRequest | FromSpan) {		if (arg.kind === "skipped" || (arg.kind === "span" && arg.parent.spanId === "skipped")) return;		this.name = arg.name;		this.startTime = Date.now();		this.spanId = crypto.randomUUID();		this.extra = arg.extra;		if (arg.kind === "span") {			this.spans = arg.parent.spans;			this.service = arg.parent.service;			this.traceId = arg.parent.traceId;			this.parentId = arg.parent.spanId;		} else if (arg.kind === "request") {			this.spans = [];			this.service = arg.service;			this.traceId = arg.parent.headers.get("x-trace-id") ?? crypto.randomUUID();			this.parentId = arg.parent.headers.get("x-parent-id") ?? undefined;			// automatically add http attributes?		} else {			this.spans = [];			this.service = arg.service;			this.traceId = crypto.randomUUID();		}		this.spans.push(this);	}
	hcJSON() {		return {			time: new Date(this.startTime).toISOString(),			data: {				// https://docs.honeycomb.io/getting-data-in/tracing/send-trace-data/#opentelemetry				name: this.name,				"service.name": this.service,				duration_ms: this.duration,				"trace.span_id": this.spanId,				"trace.trace_id": this.traceId,				"trace.parent_id": this.parentId,				...this.extra,			},		};	}}
interface CreateArg extends CreateArgBase {	skip?: boolean;	service?: string;	parent?: Request | Span;}
export function create({ skip, name, service, parent, extra }: CreateArg): Span {	if (skip) return new Span({ kind: "skipped" });	if (parent instanceof Span) return new Span({ kind: "span", name, parent, extra });	if (!service) throw new Error("service must be provided when parent is not a span");	if (new URL(`http://localhost/${service}`).pathname.slice(1) !== service) {		console.warn("consider using a service name that is url friendly");	}	if (parent instanceof Request) {		return new Span({ kind: "request", name, service, parent, extra });	}	return new Span({ kind: "no-parent", name, service, extra });}
interface EventArg extends CreateArgBase {	parent: Span;}
export function event({ name, parent, extra }: EventArg) {	if (parent.spanId === "skipped") return;	new Span({		kind: "span",		name,		parent,		extra: { ...extra, "meta.annotation_type": "span_event" },	});}
export function link(parent: Span, to: Span) {	if (parent.spanId === "skipped") return;	if (to.spanId === "skipped") return;	new Span({		kind: "span",		name: "link",		parent,		extra: {			"meta.annotation_type": "link",			"trace.link.span_id": to.spanId,			"trace.link.trace_id": to.traceId,		},	});}
export function propagate(span: Span, headers: Headers) {	if (span.spanId === "skipped") return;	headers.set("x-trace-id", span.traceId);	headers.set("x-parent-id", span.spanId);}
export function extra(span: Span, extra: ExtraData) {	if (span.spanId === "skipped") return;	span.extra = { ...span.extra, ...extra };}
export function end(span: Span, extra?: ExtraData) {	if (span.spanId === "skipped") return;	span.duration = Date.now() - span.startTime;	span.extra = { ...span.extra, ...extra };}
interface HCReqArg {	apiKey: string;	span: Span;}
export function honeycombRequest({ apiKey, span }: HCReqArg) {	return new Request(`https://api.honeycomb.io/1/batch/${span.service}`, {		method: "POST",		headers: { "x-honeycomb-team": apiKey, "content-type": "application/json" },		body: JSON.stringify(span.spans.map((s) => s.hcJSON())),	});}
export async function sendToHoneycomb({ apiKey, span }: HCReqArg) {	if (!span.spans.length) return;	const res = await fetch(honeycombRequest({ apiKey, span }));	if (!res.ok) console.warn("failed to send spans to honeycomb");	return res;}