@assistant-ui/react-o11y 0.0.46 → 0.0.47

package/dist/resources/SpanResource.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SpanResource.d.ts","sourceRoot":"","sources":["../../src/resources/SpanResource.tsx"],"names":[],"mappings":"AAAA,0BAAuB;AAIvB,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGzE,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;IACvD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1B,CAAC;AAoQF,eAAO,MAAM,YAAY;WAlEhB,QAAQ,EAAE;GAkEkC,CAAC"}
+{"version":3,"file":"SpanResource.d.ts","sourceRoot":"","sources":["../../src/resources/SpanResource.tsx"],"names":[],"mappings":"AAAA,0BAAuB;AAIvB,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGzE,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,SAAS,CAAC;IACvD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1B,CAAC;AAoQF,eAAO,MAAM,YAAY;WAnEhB,QAAQ,EAAE;GAmEkC,CAAC"}

package/dist/resources/SpanResource.js

@@ -76,19 +76,9 @@ function enrichSpans(rawSpans) {
 	if (min === Infinity) min = Date.now();
 	if (max === -Infinity) max = Date.now();
 	if (max === min) max = min + 100;
-	return {
-		allSpans,
-		parents,
-		timeRange: {
-			min,
-			max
-		}
-	};
-}
-function buildFlatList(allSpans, parents, collapsedIds) {
 	const children = /* @__PURE__ */ new Map();
 	for (const span of allSpans.values()) {
-		const parent = parents.get(span.id) ?? null;
+		const parent = span.parentSpanId;
 		let group = children.get(parent);
 		if (!group) {
 			group = [];
@@ -97,6 +87,16 @@ function buildFlatList(allSpans, parents, collapsedIds) {
 		group.push(span);
 	}
 	for (const group of children.values()) group.sort((a, b) => a.startedAt - b.startedAt);
+	return {
+		allSpans,
+		children,
+		timeRange: {
+			min,
+			max
+		}
+	};
+}
+function buildFlatList(children, collapsedIds) {
 	const result = [];
 	const stack = [];
 	const roots = children.get(null) ?? [];
@@ -104,10 +104,10 @@ function buildFlatList(allSpans, parents, collapsedIds) {
 	while (stack.length > 0) {
 		const span = stack.pop();
 		const isCollapsed = collapsedIds.has(span.id);
-		result.push({
+		result.push(isCollapsed ? {
 			...span,
 			isCollapsed
-		});
+		} : span);
 		if (!isCollapsed) {
 			const kids = children.get(span.id) ?? [];
 			for (let index = kids.length - 1; index >= 0; index--) stack.push(kids[index]);
@@ -133,7 +133,7 @@ const useSpanChildResource = ({ span, timeRange, onToggleCollapse }) => {
 };
 const SpanChildResource = resource(useSpanChildResource);
 const useSpanResource = (t0) => {
-	const $ = c(20);
+	const $ = c(21);
 	const { spans } = t0;
 	let t1;
 	if ($[0] !== spans) {
@@ -141,19 +141,18 @@ const useSpanResource = (t0) => {
 		$[0] = spans;
 		$[1] = t1;
 	} else t1 = $[1];
-	const { allSpans, parents, timeRange } = t1;
+	const { allSpans, children, timeRange } = t1;
 	const [collapsedIds, setCollapsedIds] = useState(_temp);
 	let t2;
-	if ($[2] !== allSpans || $[3] !== collapsedIds || $[4] !== parents) {
-		t2 = buildFlatList(allSpans, parents, collapsedIds);
-		$[2] = allSpans;
+	if ($[2] !== children || $[3] !== collapsedIds) {
+		t2 = buildFlatList(children, collapsedIds);
+		$[2] = children;
 		$[3] = collapsedIds;
-		$[4] = parents;
-		$[5] = t2;
-	} else t2 = $[5];
+		$[4] = t2;
+	} else t2 = $[4];
 	const visibleSpans = t2;
 	let t3;
-	if ($[6] === Symbol.for("react.memo_cache_sentinel")) {
+	if ($[5] === Symbol.for("react.memo_cache_sentinel")) {
 		t3 = (spanId) => {
 			setCollapsedIds((prev) => {
 				const next = new Set(prev);
@@ -162,22 +161,24 @@ const useSpanResource = (t0) => {
 				return next;
 			});
 		};
-		$[6] = t3;
-	} else t3 = $[6];
+		$[5] = t3;
+	} else t3 = $[5];
 	const toggleCollapse = t3;
 	let t4;
-	if ($[7] !== timeRange || $[8] !== visibleSpans) {
+	if ($[6] !== allSpans || $[7] !== timeRange || $[8] !== visibleSpans) {
 		let t5;
-		if ($[10] !== timeRange) {
+		if ($[10] !== allSpans || $[11] !== timeRange) {
 			t5 = (span) => withKey(span.id, SpanChildResource({
 				span,
 				timeRange,
 				onToggleCollapse: toggleCollapse
-			}));
-			$[10] = timeRange;
-			$[11] = t5;
-		} else t5 = $[11];
+			}), [allSpans, span.isCollapsed]);
+			$[10] = allSpans;
+			$[11] = timeRange;
+			$[12] = t5;
+		} else t5 = $[12];
 		t4 = visibleSpans.map(t5);
+		$[6] = allSpans;
 		$[7] = timeRange;
 		$[8] = visibleSpans;
 		$[9] = t4;
@@ -186,7 +187,7 @@ const useSpanResource = (t0) => {
 	const t5 = timeRange.max - timeRange.min;
 	const t6 = lookup.state.length > 0;
 	let t7;
-	if ($[12] !== lookup.state || $[13] !== t5 || $[14] !== t6 || $[15] !== timeRange) {
+	if ($[13] !== lookup.state || $[14] !== t5 || $[15] !== t6 || $[16] !== timeRange) {
 		const rootState = {
 			id: "__root__",
 			parentSpanId: null,
@@ -203,23 +204,23 @@ const useSpanResource = (t0) => {
 			timeRange
 		};
 		t7 = () => rootState;
-		$[12] = lookup.state;
-		$[13] = t5;
-		$[14] = t6;
-		$[15] = timeRange;
-		$[16] = t7;
-	} else t7 = $[16];
+		$[13] = lookup.state;
+		$[14] = t5;
+		$[15] = t6;
+		$[16] = timeRange;
+		$[17] = t7;
+	} else t7 = $[17];
 	let t8;
-	if ($[17] !== lookup.get || $[18] !== t7) {
+	if ($[18] !== lookup.get || $[19] !== t7) {
 		t8 = {
 			getState: t7,
 			child: lookup.get,
 			toggleCollapse: _temp2
 		};
-		$[17] = lookup.get;
-		$[18] = t7;
-		$[19] = t8;
-	} else t8 = $[19];
+		$[18] = lookup.get;
+		$[19] = t7;
+		$[20] = t8;
+	} else t8 = $[20];
 	return t8;
 };
 const SpanResource = resource(useSpanResource);

package/dist/resources/SpanResource.js.map

@@ -1 +1 @@
-{"version":3,"file":"SpanResource.js","names":["useMemo","useState","resource","withKey","useClientLookup","ClientOutput","SpanItemState","SpanState","SpanData","id","parentSpanId","name","type","status","startedAt","endedAt","latencyMs","calculateDepth","spanId","parents","Map","cache","cached","get","undefined","path","current","depth","currentDepth","push","index","length","set","normalizeSpanParents","spanMap","span","values","has","resolved","Set","keys","pathIds","add","enrichSpans","rawSpans","allSpans","timeRange","min","max","depthCache","childrenCount","Infinity","hasChildren","isCollapsed","end","Date","now","buildFlatList","collapsedIds","children","parent","group","sort","a","b","result","stack","roots","pop","kids","useSpanChildResource","onToggleCollapse","state","getState","child","Error","toggleCollapse","SpanChildResource","useSpanResource","t0","$","_c","spans","t1","setCollapsedIds","_temp","t2","visibleSpans","t3","Symbol","for","prev","next","delete","t4","t5","map","lookup","t6","t7","rootState","t8","_temp2","SpanResource","_c"],"sources":["../../src/resources/SpanResource.tsx"],"sourcesContent":["import \"../o11y-scope\";\n\nimport { useMemo, useState } from \"react\";\nimport { resource, withKey } from \"@assistant-ui/tap\";\nimport { useClientLookup, type ClientOutput } from \"@assistant-ui/store\";\nimport type { SpanItemState, SpanState } from \"../o11y-scope\";\n\nexport type SpanData = {\n  id: string;\n  parentSpanId: string | null;\n  name: string;\n  type: string;\n  status: \"running\" | \"completed\" | \"failed\" | \"skipped\";\n  startedAt: number;\n  endedAt: number | null;\n  latencyMs: number | null;\n};\n\nfunction calculateDepth(\n  spanId: string,\n  parents: Map<string, string | null>,\n  cache: Map<string, number>,\n): number {\n  const cached = cache.get(spanId);\n  if (cached !== undefined) return cached;\n\n  const path: string[] = [];\n  let current: string | null = spanId;\n  let depth = -1;\n\n  while (current !== null) {\n    const currentDepth = cache.get(current);\n    if (currentDepth !== undefined) {\n      depth = currentDepth;\n      break;\n    }\n\n    path.push(current);\n    current = parents.get(current) ?? null;\n  }\n\n  for (let index = path.length - 1; index >= 0; index--) {\n    depth += 1;\n    cache.set(path[index]!, depth);\n  }\n\n  return depth;\n}\n\nfunction normalizeSpanParents(\n  spanMap: Map<string, SpanData>,\n): Map<string, string | null> {\n  const parents = new Map<string, string | null>();\n  for (const span of spanMap.values()) {\n    parents.set(\n      span.id,\n      span.parentSpanId && spanMap.has(span.parentSpanId)\n        ? span.parentSpanId\n        : null,\n    );\n  }\n\n  const resolved = new Set<string>();\n  for (const spanId of spanMap.keys()) {\n    if (resolved.has(spanId)) continue;\n\n    const path: string[] = [];\n    const pathIds = new Set<string>();\n    let current: string | null = spanId;\n\n    while (current !== null && !resolved.has(current)) {\n      if (pathIds.has(current)) {\n        parents.set(current, null);\n        break;\n      }\n\n      path.push(current);\n      pathIds.add(current);\n      current = parents.get(current) ?? null;\n    }\n\n    for (const id of path) resolved.add(id);\n  }\n\n  return parents;\n}\n\nfunction enrichSpans(rawSpans: SpanData[]): {\n  allSpans: Map<string, SpanItemState>;\n  parents: Map<string, string | null>;\n  timeRange: { min: number; max: number };\n} {\n  const spanMap = new Map<string, SpanData>();\n  for (const span of rawSpans) {\n    spanMap.set(span.id, span);\n  }\n\n  const parents = normalizeSpanParents(spanMap);\n  const depthCache = new Map<string, number>();\n  const childrenCount = new Map<string, number>();\n  for (const span of rawSpans) {\n    const parentSpanId = parents.get(span.id);\n    if (parentSpanId) {\n      childrenCount.set(\n        parentSpanId,\n        (childrenCount.get(parentSpanId) ?? 0) + 1,\n      );\n    }\n  }\n\n  const allSpans = new Map<string, SpanItemState>();\n  let min = Infinity;\n  let max = -Infinity;\n\n  for (const span of rawSpans) {\n    const depth = calculateDepth(span.id, parents, depthCache);\n    const hasChildren = (childrenCount.get(span.id) ?? 0) > 0;\n    allSpans.set(span.id, {\n      ...span,\n      parentSpanId: parents.get(span.id) ?? null,\n      depth,\n      hasChildren,\n      isCollapsed: false,\n    });\n\n    if (span.startedAt < min) min = span.startedAt;\n    const end = span.endedAt ?? Date.now();\n    if (end > max) max = end;\n  }\n\n  if (min === Infinity) min = Date.now();\n  if (max === -Infinity) max = Date.now();\n  if (max === min) max = min + 100;\n\n  return { allSpans, parents, timeRange: { min, max } };\n}\n\nfunction buildFlatList(\n  allSpans: Map<string, SpanItemState>,\n  parents: Map<string, string | null>,\n  collapsedIds: Set<string>,\n): SpanItemState[] {\n  const children = new Map<string | null, SpanItemState[]>();\n  for (const span of allSpans.values()) {\n    const parent = parents.get(span.id) ?? null;\n    let group = children.get(parent);\n    if (!group) {\n      group = [];\n      children.set(parent, group);\n    }\n    group.push(span);\n  }\n\n  for (const group of children.values()) {\n    group.sort((a, b) => a.startedAt - b.startedAt);\n  }\n\n  const result: SpanItemState[] = [];\n  const stack: SpanItemState[] = [];\n  const roots = children.get(null) ?? [];\n  for (let index = roots.length - 1; index >= 0; index--) {\n    stack.push(roots[index]!);\n  }\n\n  while (stack.length > 0) {\n    const span = stack.pop()!;\n    const isCollapsed = collapsedIds.has(span.id);\n    result.push({ ...span, isCollapsed });\n\n    if (!isCollapsed) {\n      const kids = children.get(span.id) ?? [];\n      for (let index = kids.length - 1; index >= 0; index--) {\n        stack.push(kids[index]!);\n      }\n    }\n  }\n\n  return result;\n}\n\nconst useSpanChildResource = ({\n  span,\n  timeRange,\n  onToggleCollapse,\n}: {\n  span: SpanItemState;\n  timeRange: { min: number; max: number };\n  onToggleCollapse: (spanId: string) => void;\n}): ClientOutput<\"span\"> => {\n  const state: SpanState = {\n    ...span,\n    children: [],\n    timeRange,\n  };\n  return {\n    getState: () => state,\n    child: () => {\n      throw new Error(\"child spans do not have children in flat mode\");\n    },\n    toggleCollapse: () => {\n      onToggleCollapse(span.id);\n    },\n  };\n};\n\nconst SpanChildResource = resource(useSpanChildResource);\n\nconst useSpanResource = ({\n  spans,\n}: {\n  spans: SpanData[];\n}): ClientOutput<\"span\"> => {\n  const { allSpans, parents, timeRange } = useMemo(\n    () => enrichSpans(spans),\n    [spans],\n  );\n\n  const [collapsedIds, setCollapsedIds] = useState<Set<string>>(\n    () => new Set(),\n  );\n\n  const visibleSpans = useMemo(\n    () => buildFlatList(allSpans, parents, collapsedIds),\n    [allSpans, parents, collapsedIds],\n  );\n\n  const toggleCollapse = (spanId: string) => {\n    setCollapsedIds((prev) => {\n      const next = new Set(prev);\n      if (next.has(spanId)) {\n        next.delete(spanId);\n      } else {\n        next.add(spanId);\n      }\n      return next;\n    });\n  };\n\n  const lookup = useClientLookup(\n    visibleSpans.map((span) =>\n      withKey(\n        span.id,\n        SpanChildResource({\n          span,\n          timeRange,\n          onToggleCollapse: toggleCollapse,\n        }),\n      ),\n    ),\n  );\n\n  const rootState: SpanState = {\n    id: \"__root__\",\n    parentSpanId: null,\n    name: \"\",\n    type: \"root\",\n    status: \"completed\",\n    startedAt: timeRange.min,\n    endedAt: timeRange.max,\n    latencyMs: timeRange.max - timeRange.min,\n    depth: -1,\n    hasChildren: lookup.state.length > 0,\n    isCollapsed: false,\n    children: lookup.state,\n    timeRange,\n  };\n\n  return {\n    getState: () => rootState,\n    child: lookup.get,\n    toggleCollapse: () => {\n      // Root span collapse is a no-op\n    },\n  };\n};\n\nexport const SpanResource = resource(useSpanResource);\n"],"mappings":";;;;;AAkBA,SAASiB,eACPC,QACAC,SACAE,OACQ;CACR,MAAMC,SAASD,MAAME,IAAIL,MAAM;CAC/B,IAAII,WAAWE,KAAAA,GAAW,OAAOF;CAEjC,MAAMG,OAAiB,CAAA;CACvB,IAAIC,UAAyBR;CAC7B,IAAIS,QAAQ;CAEZ,OAAOD,YAAY,MAAM;EACvB,MAAME,eAAeP,MAAME,IAAIG,OAAO;EACtC,IAAIE,iBAAiBJ,KAAAA,GAAW;GAC9BG,QAAQC;GACR;EACF;EAEAH,KAAKI,KAAKH,OAAO;EACjBA,UAAUP,QAAQI,IAAIG,OAAO,KAAK;CACpC;CAEA,KAAK,IAAII,QAAQL,KAAKM,SAAS,GAAGD,SAAS,GAAGA,SAAS;EACrDH,SAAS;EACTN,MAAMW,IAAIP,KAAKK,QAASH,KAAK;CAC/B;CAEA,OAAOA;AACT;AAEA,SAASM,qBACPC,SAC4B;CAC5B,MAAMf,0BAAU,IAAIC,IAA2B;CAC/C,KAAK,MAAMe,QAAQD,QAAQE,OAAO,GAChCjB,QAAQa,IACNG,KAAK1B,IACL0B,KAAKzB,gBAAgBwB,QAAQG,IAAIF,KAAKzB,YAAY,IAC9CyB,KAAKzB,eACL,IACN;CAGF,MAAM4B,2BAAW,IAAIC,IAAY;CACjC,KAAK,MAAMrB,UAAUgB,QAAQM,KAAK,GAAG;EACnC,IAAIF,SAASD,IAAInB,MAAM,GAAG;EAE1B,MAAMO,OAAiB,CAAA;EACvB,MAAMgB,0BAAU,IAAIF,IAAY;EAChC,IAAIb,UAAyBR;EAE7B,OAAOQ,YAAY,QAAQ,CAACY,SAASD,IAAIX,OAAO,GAAG;GACjD,IAAIe,QAAQJ,IAAIX,OAAO,GAAG;IACxBP,QAAQa,IAAIN,SAAS,IAAI;IACzB;GACF;GAEAD,KAAKI,KAAKH,OAAO;GACjBe,QAAQC,IAAIhB,OAAO;GACnBA,UAAUP,QAAQI,IAAIG,OAAO,KAAK;EACpC;EAEA,KAAK,MAAMjB,MAAMgB,MAAMa,SAASI,IAAIjC,EAAE;CACxC;CAEA,OAAOU;AACT;AAEA,SAASwB,YAAYC,UAInB;CACA,MAAMV,0BAAU,IAAId,IAAsB;CAC1C,KAAK,MAAMe,QAAQS,UACjBV,QAAQF,IAAIG,KAAK1B,IAAI0B,IAAI;CAG3B,MAAMhB,UAAUc,qBAAqBC,OAAO;CAC5C,MAAMe,6BAAa,IAAI7B,IAAoB;CAC3C,MAAM8B,gCAAgB,IAAI9B,IAAoB;CAC9C,KAAK,MAAMe,QAAQS,UAAU;EAC3B,MAAMlC,eAAeS,QAAQI,IAAIY,KAAK1B,EAAE;EACxC,IAAIC,cACFwC,cAAclB,IACZtB,eACCwC,cAAc3B,IAAIb,YAAY,KAAK,KAAK,CAC3C;CAEJ;CAEA,MAAMmC,2BAAW,IAAIzB,IAA2B;CAChD,IAAI2B,MAAMI;CACV,IAAIH,MAAM;CAEV,KAAK,MAAMb,QAAQS,UAAU;EAC3B,MAAMjB,QAAQV,eAAekB,KAAK1B,IAAIU,SAAS8B,UAAU;EACzD,MAAMG,eAAeF,cAAc3B,IAAIY,KAAK1B,EAAE,KAAK,KAAK;EACxDoC,SAASb,IAAIG,KAAK1B,IAAI;GACpB,GAAG0B;GACHzB,cAAcS,QAAQI,IAAIY,KAAK1B,EAAE,KAAK;GACtCkB;GACAyB;GACAC,aAAa;EACf,CAAC;EAED,IAAIlB,KAAKrB,YAAYiC,KAAKA,MAAMZ,KAAKrB;EACrC,MAAMwC,MAAMnB,KAAKpB,WAAWwC,KAAKC,IAAI;EACrC,IAAIF,MAAMN,KAAKA,MAAMM;CACvB;CAEA,IAAIP,QAAQI,UAAUJ,MAAMQ,KAAKC,IAAI;CACrC,IAAIR,QAAQ,WAAWA,MAAMO,KAAKC,IAAI;CACtC,IAAIR,QAAQD,KAAKC,MAAMD,MAAM;CAE7B,OAAO;EAAEF;EAAU1B;EAAS2B,WAAW;GAAEC;GAAKC;EAAI;CAAE;AACtD;AAEA,SAASS,cACPZ,UACA1B,SACAuC,cACiB;CACjB,MAAMC,2BAAW,IAAIvC,IAAoC;CACzD,KAAK,MAAMe,QAAQU,SAAST,OAAO,GAAG;EACpC,MAAMwB,SAASzC,QAAQI,IAAIY,KAAK1B,EAAE,KAAK;EACvC,IAAIoD,QAAQF,SAASpC,IAAIqC,MAAM;EAC/B,IAAI,CAACC,OAAO;GACVA,QAAQ,CAAA;GACRF,SAAS3B,IAAI4B,QAAQC,KAAK;EAC5B;EACAA,MAAMhC,KAAKM,IAAI;CACjB;CAEA,KAAK,MAAM0B,SAASF,SAASvB,OAAO,GAClCyB,MAAMC,MAAMC,GAAGC,MAAMD,EAAEjD,YAAYkD,EAAElD,SAAS;CAGhD,MAAMmD,SAA0B,CAAA;CAChC,MAAMC,QAAyB,CAAA;CAC/B,MAAMC,QAAQR,SAASpC,IAAI,IAAI,KAAK,CAAA;CACpC,KAAK,IAAIO,QAAQqC,MAAMpC,SAAS,GAAGD,SAAS,GAAGA,SAC7CoC,MAAMrC,KAAKsC,MAAMrC,MAAO;CAG1B,OAAOoC,MAAMnC,SAAS,GAAG;EACvB,MAAMI,OAAO+B,MAAME,IAAI;EACvB,MAAMf,cAAcK,aAAarB,IAAIF,KAAK1B,EAAE;EAC5CwD,OAAOpC,KAAK;GAAE,GAAGM;GAAMkB;EAAY,CAAC;EAEpC,IAAI,CAACA,aAAa;GAChB,MAAMgB,OAAOV,SAASpC,IAAIY,KAAK1B,EAAE,KAAK,CAAA;GACtC,KAAK,IAAIqB,QAAQuC,KAAKtC,SAAS,GAAGD,SAAS,GAAGA,SAC5CoC,MAAMrC,KAAKwC,KAAKvC,MAAO;EAE3B;CACF;CAEA,OAAOmC;AACT;AAEA,MAAMK,wBAAwB,EAC5BnC,MACAW,WACAyB,uBAK0B;CAC1B,MAAMC,QAAmB;EACvB,GAAGrC;EACHwB,UAAU,CAAA;EACVb;CACF;CACA,OAAO;EACL2B,gBAAgBD;EAChBE,aAAa;GACX,MAAM,IAAIC,MAAM,+CAA+C;EACjE;EACAC,sBAAsB;GACpBL,iBAAiBpC,KAAK1B,EAAE;EAC1B;CACF;AACF;AAEA,MAAMoE,oBAAoB3E,SAASoE,oBAAoB;AAEvD,MAAMQ,mBAAkBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAC,UAAAH;CAIxB,IAAAI;CAAA,IAAAH,EAAA,OAAAE,OAAA;EAESC,KAAAxC,YAAYuC,KAAK;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CAD1B,MAAA,EAAAnC,UAAA1B,SAAA2B,cACQqC;CAIR,MAAA,CAAAzB,cAAA0B,mBAAwCnF,SACtCoF,KACF;CAAE,IAAAC;CAAA,IAAAN,EAAA,OAAAnC,YAAAmC,EAAA,OAAAtB,gBAAAsB,EAAA,OAAA7D,SAAA;EAGMmE,KAAA7B,cAAcZ,UAAU1B,SAASuC,YAAY;EAACsB,EAAA,KAAAnC;EAAAmC,EAAA,KAAAtB;EAAAsB,EAAA,KAAA7D;EAAA6D,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CADtD,MAAAO,eACQD;CAEN,IAAAE;CAAA,IAAAR,EAAA,OAAAS,OAAAC,IAAA,2BAAA,GAAA;EAEqBF,MAAAtE,WAAA;GACrBkE,iBAAgBO,SAAA;IACd,MAAAC,OAAa,IAAIrD,IAAIoD,IAAI;IACzB,IAAIC,KAAIvD,IAAKnB,MAAM,GACjB0E,KAAIC,OAAQ3E,MAAM;SAElB0E,KAAIlD,IAAKxB,MAAM;IAChB,OACM0E;GAAI,CACZ;EAAC;EACHZ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAVD,MAAAJ,iBAAuBY;CAUrB,IAAAM;CAAA,IAAAd,EAAA,OAAAlC,aAAAkC,EAAA,OAAAO,cAAA;EAAA,IAAAQ;EAAA,IAAAf,EAAA,QAAAlC,WAAA;GAGiBiD,MAAA5D,SACfhC,QACEgC,KAAI1B,IACJoE,kBAAkB;IAAA1C;IAAAW;IAAAyB,kBAGEK;GACpB,CAAC,CACH;GAACI,EAAA,MAAAlC;GAAAkC,EAAA,MAAAe;EAAA,OAAAA,KAAAf,EAAA;EARHc,KAAAP,aAAYS,IAAKD,EASjB;EAACf,EAAA,KAAAlC;EAAAkC,EAAA,KAAAO;EAAAP,EAAA,KAAAc;CAAA,OAAAA,KAAAd,EAAA;CAVH,MAAAiB,SAAe7F,gBACb0F,EAUF;CAUa,MAAAC,KAAAjD,UAASE,MAAOF,UAASC;CAEvB,MAAAmD,KAAAD,OAAMzB,MAAMzC,SAAU;CAAC,IAAAoE;CAAA,IAAAnB,EAAA,QAAAiB,OAAAzB,SAAAQ,EAAA,QAAAe,MAAAf,EAAA,QAAAkB,MAAAlB,EAAA,QAAAlC,WAAA;EAVtC,MAAAsD,YAA6B;GAAA3F,IACvB;GAAUC,cACA;GAAIC,MACZ;GAAEC,MACF;GAAMC,QACJ;GAAWC,WACRgC,UAASC;GAAIhC,SACf+B,UAASE;GAAIhC,WACX+E;GAA6BpE,OACjC;GAAEyB,aACI8C;GAAuB7C,aACvB;GAAKM,UACRsC,OAAMzB;GAAM1B;EAExB;EAGYqD,WAAMC;EAASpB,EAAA,MAAAiB,OAAAzB;EAAAQ,EAAA,MAAAe;EAAAf,EAAA,MAAAkB;EAAAlB,EAAA,MAAAlC;EAAAkC,EAAA,MAAAmB;CAAA,OAAAA,KAAAnB,EAAA;CAAA,IAAAqB;CAAA,IAAArB,EAAA,QAAAiB,OAAA1E,OAAAyD,EAAA,QAAAmB,IAAA;EADpBE,KAAA;GAAA5B,UACK0B;GAAezB,OAClBuB,OAAM1E;GAAIqD,gBACD0B;EAGlB;EAACtB,EAAA,MAAAiB,OAAA1E;EAAAyD,EAAA,MAAAmB;EAAAnB,EAAA,MAAAqB;CAAA,OAAAA,KAAArB,EAAA;CAAA,OANMqB;AAMN;AAGH,MAAaE,eAAerG,SAAS4E,eAAe;AArE5B,SAAAO,QAAA;CAAA,uBAWd,IAAI9C,IAAI;AAAC;AAXK,SAAA+D,SAAA,CAAA"}
+{"version":3,"file":"SpanResource.js","names":["useMemo","useState","resource","withKey","useClientLookup","ClientOutput","SpanItemState","SpanState","SpanData","id","parentSpanId","name","type","status","startedAt","endedAt","latencyMs","calculateDepth","spanId","parents","Map","cache","cached","get","undefined","path","current","depth","currentDepth","push","index","length","set","normalizeSpanParents","spanMap","span","values","has","resolved","Set","keys","pathIds","add","enrichSpans","rawSpans","allSpans","children","timeRange","min","max","depthCache","childrenCount","Infinity","hasChildren","isCollapsed","end","Date","now","parent","group","sort","a","b","buildFlatList","collapsedIds","result","stack","roots","pop","kids","useSpanChildResource","onToggleCollapse","state","getState","child","Error","toggleCollapse","SpanChildResource","useSpanResource","t0","$","_c","spans","t1","setCollapsedIds","_temp","t2","visibleSpans","t3","Symbol","for","prev","next","delete","t4","t5","map","lookup","t6","t7","rootState","t8","_temp2","SpanResource","_c"],"sources":["../../src/resources/SpanResource.tsx"],"sourcesContent":["import \"../o11y-scope\";\n\nimport { useMemo, useState } from \"react\";\nimport { resource, withKey } from \"@assistant-ui/tap\";\nimport { useClientLookup, type ClientOutput } from \"@assistant-ui/store\";\nimport type { SpanItemState, SpanState } from \"../o11y-scope\";\n\nexport type SpanData = {\n  id: string;\n  parentSpanId: string | null;\n  name: string;\n  type: string;\n  status: \"running\" | \"completed\" | \"failed\" | \"skipped\";\n  startedAt: number;\n  endedAt: number | null;\n  latencyMs: number | null;\n};\n\nfunction calculateDepth(\n  spanId: string,\n  parents: Map<string, string | null>,\n  cache: Map<string, number>,\n): number {\n  const cached = cache.get(spanId);\n  if (cached !== undefined) return cached;\n\n  const path: string[] = [];\n  let current: string | null = spanId;\n  let depth = -1;\n\n  while (current !== null) {\n    const currentDepth = cache.get(current);\n    if (currentDepth !== undefined) {\n      depth = currentDepth;\n      break;\n    }\n\n    path.push(current);\n    current = parents.get(current) ?? null;\n  }\n\n  for (let index = path.length - 1; index >= 0; index--) {\n    depth += 1;\n    cache.set(path[index]!, depth);\n  }\n\n  return depth;\n}\n\nfunction normalizeSpanParents(\n  spanMap: Map<string, SpanData>,\n): Map<string, string | null> {\n  const parents = new Map<string, string | null>();\n  for (const span of spanMap.values()) {\n    parents.set(\n      span.id,\n      span.parentSpanId && spanMap.has(span.parentSpanId)\n        ? span.parentSpanId\n        : null,\n    );\n  }\n\n  const resolved = new Set<string>();\n  for (const spanId of spanMap.keys()) {\n    if (resolved.has(spanId)) continue;\n\n    const path: string[] = [];\n    const pathIds = new Set<string>();\n    let current: string | null = spanId;\n\n    while (current !== null && !resolved.has(current)) {\n      if (pathIds.has(current)) {\n        parents.set(current, null);\n        break;\n      }\n\n      path.push(current);\n      pathIds.add(current);\n      current = parents.get(current) ?? null;\n    }\n\n    for (const id of path) resolved.add(id);\n  }\n\n  return parents;\n}\n\nfunction enrichSpans(rawSpans: SpanData[]): {\n  allSpans: Map<string, SpanItemState>;\n  children: Map<string | null, SpanItemState[]>;\n  timeRange: { min: number; max: number };\n} {\n  const spanMap = new Map<string, SpanData>();\n  for (const span of rawSpans) {\n    spanMap.set(span.id, span);\n  }\n\n  const parents = normalizeSpanParents(spanMap);\n  const depthCache = new Map<string, number>();\n  const childrenCount = new Map<string, number>();\n  for (const span of rawSpans) {\n    const parentSpanId = parents.get(span.id);\n    if (parentSpanId) {\n      childrenCount.set(\n        parentSpanId,\n        (childrenCount.get(parentSpanId) ?? 0) + 1,\n      );\n    }\n  }\n\n  const allSpans = new Map<string, SpanItemState>();\n  let min = Infinity;\n  let max = -Infinity;\n\n  for (const span of rawSpans) {\n    const depth = calculateDepth(span.id, parents, depthCache);\n    const hasChildren = (childrenCount.get(span.id) ?? 0) > 0;\n    allSpans.set(span.id, {\n      ...span,\n      parentSpanId: parents.get(span.id) ?? null,\n      depth,\n      hasChildren,\n      isCollapsed: false,\n    });\n\n    if (span.startedAt < min) min = span.startedAt;\n    const end = span.endedAt ?? Date.now();\n    if (end > max) max = end;\n  }\n\n  if (min === Infinity) min = Date.now();\n  if (max === -Infinity) max = Date.now();\n  if (max === min) max = min + 100;\n\n  const children = new Map<string | null, SpanItemState[]>();\n  for (const span of allSpans.values()) {\n    const parent = span.parentSpanId;\n    let group = children.get(parent);\n    if (!group) {\n      group = [];\n      children.set(parent, group);\n    }\n    group.push(span);\n  }\n\n  for (const group of children.values()) {\n    group.sort((a, b) => a.startedAt - b.startedAt);\n  }\n\n  return { allSpans, children, timeRange: { min, max } };\n}\n\nfunction buildFlatList(\n  children: Map<string | null, SpanItemState[]>,\n  collapsedIds: Set<string>,\n): SpanItemState[] {\n  const result: SpanItemState[] = [];\n  const stack: SpanItemState[] = [];\n  const roots = children.get(null) ?? [];\n  for (let index = roots.length - 1; index >= 0; index--) {\n    stack.push(roots[index]!);\n  }\n\n  while (stack.length > 0) {\n    const span = stack.pop()!;\n    const isCollapsed = collapsedIds.has(span.id);\n    result.push(isCollapsed ? { ...span, isCollapsed } : span);\n\n    if (!isCollapsed) {\n      const kids = children.get(span.id) ?? [];\n      for (let index = kids.length - 1; index >= 0; index--) {\n        stack.push(kids[index]!);\n      }\n    }\n  }\n\n  return result;\n}\n\nconst useSpanChildResource = ({\n  span,\n  timeRange,\n  onToggleCollapse,\n}: {\n  span: SpanItemState;\n  timeRange: { min: number; max: number };\n  onToggleCollapse: (spanId: string) => void;\n}): ClientOutput<\"span\"> => {\n  const state: SpanState = {\n    ...span,\n    children: [],\n    timeRange,\n  };\n  return {\n    getState: () => state,\n    child: () => {\n      throw new Error(\"child spans do not have children in flat mode\");\n    },\n    toggleCollapse: () => {\n      onToggleCollapse(span.id);\n    },\n  };\n};\n\nconst SpanChildResource = resource(useSpanChildResource);\n\nconst useSpanResource = ({\n  spans,\n}: {\n  spans: SpanData[];\n}): ClientOutput<\"span\"> => {\n  const { allSpans, children, timeRange } = useMemo(\n    () => enrichSpans(spans),\n    [spans],\n  );\n\n  const [collapsedIds, setCollapsedIds] = useState<Set<string>>(\n    () => new Set(),\n  );\n\n  const visibleSpans = useMemo(\n    () => buildFlatList(children, collapsedIds),\n    [children, collapsedIds],\n  );\n\n  const toggleCollapse = (spanId: string) => {\n    setCollapsedIds((prev) => {\n      const next = new Set(prev);\n      if (next.has(spanId)) {\n        next.delete(spanId);\n      } else {\n        next.add(spanId);\n      }\n      return next;\n    });\n  };\n\n  const lookup = useClientLookup(\n    visibleSpans.map((span) =>\n      withKey(\n        span.id,\n        SpanChildResource({\n          span,\n          timeRange,\n          onToggleCollapse: toggleCollapse,\n        }),\n        [allSpans, span.isCollapsed],\n      ),\n    ),\n  );\n\n  const rootState: SpanState = {\n    id: \"__root__\",\n    parentSpanId: null,\n    name: \"\",\n    type: \"root\",\n    status: \"completed\",\n    startedAt: timeRange.min,\n    endedAt: timeRange.max,\n    latencyMs: timeRange.max - timeRange.min,\n    depth: -1,\n    hasChildren: lookup.state.length > 0,\n    isCollapsed: false,\n    children: lookup.state,\n    timeRange,\n  };\n\n  return {\n    getState: () => rootState,\n    child: lookup.get,\n    toggleCollapse: () => {\n      // Root span collapse is a no-op\n    },\n  };\n};\n\nexport const SpanResource = resource(useSpanResource);\n"],"mappings":";;;;;AAkBA,SAASiB,eACPC,QACAC,SACAE,OACQ;CACR,MAAMC,SAASD,MAAME,IAAIL,MAAM;CAC/B,IAAII,WAAWE,KAAAA,GAAW,OAAOF;CAEjC,MAAMG,OAAiB,CAAA;CACvB,IAAIC,UAAyBR;CAC7B,IAAIS,QAAQ;CAEZ,OAAOD,YAAY,MAAM;EACvB,MAAME,eAAeP,MAAME,IAAIG,OAAO;EACtC,IAAIE,iBAAiBJ,KAAAA,GAAW;GAC9BG,QAAQC;GACR;EACF;EAEAH,KAAKI,KAAKH,OAAO;EACjBA,UAAUP,QAAQI,IAAIG,OAAO,KAAK;CACpC;CAEA,KAAK,IAAII,QAAQL,KAAKM,SAAS,GAAGD,SAAS,GAAGA,SAAS;EACrDH,SAAS;EACTN,MAAMW,IAAIP,KAAKK,QAASH,KAAK;CAC/B;CAEA,OAAOA;AACT;AAEA,SAASM,qBACPC,SAC4B;CAC5B,MAAMf,0BAAU,IAAIC,IAA2B;CAC/C,KAAK,MAAMe,QAAQD,QAAQE,OAAO,GAChCjB,QAAQa,IACNG,KAAK1B,IACL0B,KAAKzB,gBAAgBwB,QAAQG,IAAIF,KAAKzB,YAAY,IAC9CyB,KAAKzB,eACL,IACN;CAGF,MAAM4B,2BAAW,IAAIC,IAAY;CACjC,KAAK,MAAMrB,UAAUgB,QAAQM,KAAK,GAAG;EACnC,IAAIF,SAASD,IAAInB,MAAM,GAAG;EAE1B,MAAMO,OAAiB,CAAA;EACvB,MAAMgB,0BAAU,IAAIF,IAAY;EAChC,IAAIb,UAAyBR;EAE7B,OAAOQ,YAAY,QAAQ,CAACY,SAASD,IAAIX,OAAO,GAAG;GACjD,IAAIe,QAAQJ,IAAIX,OAAO,GAAG;IACxBP,QAAQa,IAAIN,SAAS,IAAI;IACzB;GACF;GAEAD,KAAKI,KAAKH,OAAO;GACjBe,QAAQC,IAAIhB,OAAO;GACnBA,UAAUP,QAAQI,IAAIG,OAAO,KAAK;EACpC;EAEA,KAAK,MAAMjB,MAAMgB,MAAMa,SAASI,IAAIjC,EAAE;CACxC;CAEA,OAAOU;AACT;AAEA,SAASwB,YAAYC,UAInB;CACA,MAAMV,0BAAU,IAAId,IAAsB;CAC1C,KAAK,MAAMe,QAAQS,UACjBV,QAAQF,IAAIG,KAAK1B,IAAI0B,IAAI;CAG3B,MAAMhB,UAAUc,qBAAqBC,OAAO;CAC5C,MAAMgB,6BAAa,IAAI9B,IAAoB;CAC3C,MAAM+B,gCAAgB,IAAI/B,IAAoB;CAC9C,KAAK,MAAMe,QAAQS,UAAU;EAC3B,MAAMlC,eAAeS,QAAQI,IAAIY,KAAK1B,EAAE;EACxC,IAAIC,cACFyC,cAAcnB,IACZtB,eACCyC,cAAc5B,IAAIb,YAAY,KAAK,KAAK,CAC3C;CAEJ;CAEA,MAAMmC,2BAAW,IAAIzB,IAA2B;CAChD,IAAI4B,MAAMI;CACV,IAAIH,MAAM;CAEV,KAAK,MAAMd,QAAQS,UAAU;EAC3B,MAAMjB,QAAQV,eAAekB,KAAK1B,IAAIU,SAAS+B,UAAU;EACzD,MAAMG,eAAeF,cAAc5B,IAAIY,KAAK1B,EAAE,KAAK,KAAK;EACxDoC,SAASb,IAAIG,KAAK1B,IAAI;GACpB,GAAG0B;GACHzB,cAAcS,QAAQI,IAAIY,KAAK1B,EAAE,KAAK;GACtCkB;GACA0B;GACAC,aAAa;EACf,CAAC;EAED,IAAInB,KAAKrB,YAAYkC,KAAKA,MAAMb,KAAKrB;EACrC,MAAMyC,MAAMpB,KAAKpB,WAAWyC,KAAKC,IAAI;EACrC,IAAIF,MAAMN,KAAKA,MAAMM;CACvB;CAEA,IAAIP,QAAQI,UAAUJ,MAAMQ,KAAKC,IAAI;CACrC,IAAIR,QAAQ,WAAWA,MAAMO,KAAKC,IAAI;CACtC,IAAIR,QAAQD,KAAKC,MAAMD,MAAM;CAE7B,MAAMF,2BAAW,IAAI1B,IAAoC;CACzD,KAAK,MAAMe,QAAQU,SAAST,OAAO,GAAG;EACpC,MAAMsB,SAASvB,KAAKzB;EACpB,IAAIiD,QAAQb,SAASvB,IAAImC,MAAM;EAC/B,IAAI,CAACC,OAAO;GACVA,QAAQ,CAAA;GACRb,SAASd,IAAI0B,QAAQC,KAAK;EAC5B;EACAA,MAAM9B,KAAKM,IAAI;CACjB;CAEA,KAAK,MAAMwB,SAASb,SAASV,OAAO,GAClCuB,MAAMC,MAAMC,GAAGC,MAAMD,EAAE/C,YAAYgD,EAAEhD,SAAS;CAGhD,OAAO;EAAE+B;EAAUC;EAAUC,WAAW;GAAEC;GAAKC;EAAI;CAAE;AACvD;AAEA,SAASc,cACPjB,UACAkB,cACiB;CACjB,MAAMC,SAA0B,CAAA;CAChC,MAAMC,QAAyB,CAAA;CAC/B,MAAMC,QAAQrB,SAASvB,IAAI,IAAI,KAAK,CAAA;CACpC,KAAK,IAAIO,QAAQqC,MAAMpC,SAAS,GAAGD,SAAS,GAAGA,SAC7CoC,MAAMrC,KAAKsC,MAAMrC,MAAO;CAG1B,OAAOoC,MAAMnC,SAAS,GAAG;EACvB,MAAMI,OAAO+B,MAAME,IAAI;EACvB,MAAMd,cAAcU,aAAa3B,IAAIF,KAAK1B,EAAE;EAC5CwD,OAAOpC,KAAKyB,cAAc;GAAE,GAAGnB;GAAMmB;EAAY,IAAInB,IAAI;EAEzD,IAAI,CAACmB,aAAa;GAChB,MAAMe,OAAOvB,SAASvB,IAAIY,KAAK1B,EAAE,KAAK,CAAA;GACtC,KAAK,IAAIqB,QAAQuC,KAAKtC,SAAS,GAAGD,SAAS,GAAGA,SAC5CoC,MAAMrC,KAAKwC,KAAKvC,MAAO;EAE3B;CACF;CAEA,OAAOmC;AACT;AAEA,MAAMK,wBAAwB,EAC5BnC,MACAY,WACAwB,uBAK0B;CAC1B,MAAMC,QAAmB;EACvB,GAAGrC;EACHW,UAAU,CAAA;EACVC;CACF;CACA,OAAO;EACL0B,gBAAgBD;EAChBE,aAAa;GACX,MAAM,IAAIC,MAAM,+CAA+C;EACjE;EACAC,sBAAsB;GACpBL,iBAAiBpC,KAAK1B,EAAE;EAC1B;CACF;AACF;AAEA,MAAMoE,oBAAoB3E,SAASoE,oBAAoB;AAEvD,MAAMQ,mBAAkBC,OAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAAC,MAAA,EAAAC,UAAAH;CAIxB,IAAAI;CAAA,IAAAH,EAAA,OAAAE,OAAA;EAESC,KAAAxC,YAAYuC,KAAK;EAACF,EAAA,KAAAE;EAAAF,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CAD1B,MAAA,EAAAnC,UAAAC,UAAAC,cACQoC;CAIR,MAAA,CAAAnB,cAAAoB,mBAAwCnF,SACtCoF,KACF;CAAE,IAAAC;CAAA,IAAAN,EAAA,OAAAlC,YAAAkC,EAAA,OAAAhB,cAAA;EAGMsB,KAAAvB,cAAcjB,UAAUkB,YAAY;EAACgB,EAAA,KAAAlC;EAAAkC,EAAA,KAAAhB;EAAAgB,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAD7C,MAAAO,eACQD;CAEN,IAAAE;CAAA,IAAAR,EAAA,OAAAS,OAAAC,IAAA,2BAAA,GAAA;EAEqBF,MAAAtE,WAAA;GACrBkE,iBAAgBO,SAAA;IACd,MAAAC,OAAa,IAAIrD,IAAIoD,IAAI;IACzB,IAAIC,KAAIvD,IAAKnB,MAAM,GACjB0E,KAAIC,OAAQ3E,MAAM;SAElB0E,KAAIlD,IAAKxB,MAAM;IAChB,OACM0E;GAAI,CACZ;EAAC;EACHZ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAVD,MAAAJ,iBAAuBY;CAUrB,IAAAM;CAAA,IAAAd,EAAA,OAAAnC,YAAAmC,EAAA,OAAAjC,aAAAiC,EAAA,OAAAO,cAAA;EAAA,IAAAQ;EAAA,IAAAf,EAAA,QAAAnC,YAAAmC,EAAA,QAAAjC,WAAA;GAGiBgD,MAAA5D,SACfhC,QACEgC,KAAI1B,IACJoE,kBAAkB;IAAA1C;IAAAY;IAAAwB,kBAGEK;GACpB,CAAC,GACD,CAAC/B,UAAUV,KAAImB,WAAY,CAC7B;GAAC0B,EAAA,MAAAnC;GAAAmC,EAAA,MAAAjC;GAAAiC,EAAA,MAAAe;EAAA,OAAAA,KAAAf,EAAA;EATHc,KAAAP,aAAYS,IAAKD,EAUjB;EAACf,EAAA,KAAAnC;EAAAmC,EAAA,KAAAjC;EAAAiC,EAAA,KAAAO;EAAAP,EAAA,KAAAc;CAAA,OAAAA,KAAAd,EAAA;CAXH,MAAAiB,SAAe7F,gBACb0F,EAWF;CAUa,MAAAC,KAAAhD,UAASE,MAAOF,UAASC;CAEvB,MAAAkD,KAAAD,OAAMzB,MAAMzC,SAAU;CAAC,IAAAoE;CAAA,IAAAnB,EAAA,QAAAiB,OAAAzB,SAAAQ,EAAA,QAAAe,MAAAf,EAAA,QAAAkB,MAAAlB,EAAA,QAAAjC,WAAA;EAVtC,MAAAqD,YAA6B;GAAA3F,IACvB;GAAUC,cACA;GAAIC,MACZ;GAAEC,MACF;GAAMC,QACJ;GAAWC,WACRiC,UAASC;GAAIjC,SACfgC,UAASE;GAAIjC,WACX+E;GAA6BpE,OACjC;GAAE0B,aACI6C;GAAuB5C,aACvB;GAAKR,UACRmD,OAAMzB;GAAMzB;EAExB;EAGYoD,WAAMC;EAASpB,EAAA,MAAAiB,OAAAzB;EAAAQ,EAAA,MAAAe;EAAAf,EAAA,MAAAkB;EAAAlB,EAAA,MAAAjC;EAAAiC,EAAA,MAAAmB;CAAA,OAAAA,KAAAnB,EAAA;CAAA,IAAAqB;CAAA,IAAArB,EAAA,QAAAiB,OAAA1E,OAAAyD,EAAA,QAAAmB,IAAA;EADpBE,KAAA;GAAA5B,UACK0B;GAAezB,OAClBuB,OAAM1E;GAAIqD,gBACD0B;EAGlB;EAACtB,EAAA,MAAAiB,OAAA1E;EAAAyD,EAAA,MAAAmB;EAAAnB,EAAA,MAAAqB;CAAA,OAAAA,KAAArB,EAAA;CAAA,OANMqB;AAMN;AAGH,MAAaE,eAAerG,SAAS4E,eAAe;AAtE5B,SAAAO,QAAA;CAAA,uBAWd,IAAI9C,IAAI;AAAC;AAXK,SAAA+D,SAAA,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-o11y",
-  "version": "0.0.46",
+  "version": "0.0.47",
   "description": "Observability primitives for @assistant-ui",
   "keywords": [
     "observability",
@@ -28,9 +28,9 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/store": "^0.3.17",
-    "@assistant-ui/tap": "^0.9.21",
-    "@radix-ui/react-primitive": "^2.1.10"
+    "@assistant-ui/store": "^0.3.18",
+    "@assistant-ui/tap": "^0.9.22",
+    "@radix-ui/react-primitive": "^2.1.11"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -46,6 +46,7 @@
     "@assistant-ui/x-react18": "0.0.0",
     "@types/react": "^19.3.0",
     "@types/react-dom": "^19.3.0",
+    "jsdom": "^30.1.2",
     "react": "^19.3.0",
     "react-dom": "^19.3.0",
     "vitest": "^5.0.3"
@@ -66,6 +67,7 @@
   "scripts": {
     "build": "aui-build",
     "test": "vitest run",
+    "test:coverage": "vitest run --coverage --coverage.reporter=text --coverage.reporter=html --coverage.reporter=json-summary",
     "typecheck": "tsc --noEmit",
     "test:peer-react18": "vitest run --config vitest.peer-react18.config.ts",
     "test:types:peer-react18": "aui-react18-types"

package/src/resources/SpanResource.test.tsx

@@ -1,6 +1,11 @@
+// @vitest-environment jsdom
+
+import { act } from "react";
+import { createRoot } from "react-dom/client";
 import { renderToStaticMarkup } from "react-dom/server";
 import { describe, expect, it } from "vitest";
-import { AuiProvider, useAui } from "@assistant-ui/store";
+import { useResource } from "@assistant-ui/tap";
+import { AuiProvider, useAui, type ClientOutput } from "@assistant-ui/store";
 import * as SpanPrimitive from "../primitives/span";
 import { SpanResource, type SpanData } from "./SpanResource";
 
@@ -122,6 +127,59 @@ describe("SpanResource", () => {
     expect(spanIds).toEqual(["root", "earlier", "nested", "later"]);
   });
 
+  it("keeps sibling resources and flat list order when a span is toggled", () => {
+    const spans = [
+      createSpan("root", null),
+      createSpan("later", "root", 2),
+      createSpan("earlier", "root", 1),
+      createSpan("nested", "earlier", 3),
+    ];
+    let spanResource: ClientOutput<"span"> | undefined;
+    const container = document.createElement("div");
+    const root = createRoot(container);
+
+    const Fixture = () => {
+      spanResource = useResource(SpanResource({ spans }));
+      return null;
+    };
+
+    try {
+      act(() => root.render(<Fixture />));
+      const initial = spanResource!.getState().children;
+      expect(initial.map((span) => span.id)).toEqual([
+        "root",
+        "earlier",
+        "nested",
+        "later",
+      ]);
+
+      act(() => spanResource!.child({ key: "earlier" }).toggleCollapse());
+      const collapsed = spanResource!.getState().children;
+      expect(collapsed.map((span) => span.id)).toEqual([
+        "root",
+        "earlier",
+        "later",
+      ]);
+      expect(collapsed[0]).toBe(initial[0]);
+      expect(collapsed[2]).toBe(initial[3]);
+      expect(collapsed[1]?.isCollapsed).toBe(true);
+
+      act(() => spanResource!.child({ key: "earlier" }).toggleCollapse());
+      const expanded = spanResource!.getState().children;
+      expect(expanded.map((span) => span.id)).toEqual([
+        "root",
+        "earlier",
+        "nested",
+        "later",
+      ]);
+      expect(expanded[0]).toBe(initial[0]);
+      expect(expanded[3]).toBe(initial[3]);
+      expect(expanded[1]?.isCollapsed).toBe(false);
+    } finally {
+      act(() => root.unmount());
+    }
+  });
+
   it("renders deeply nested parent chains without overflowing the stack", () => {
     const spans = Array.from({ length: 20_000 }, (_, index) =>
       createSpan(`span-${index}`, index === 0 ? null : `span-${index - 1}`),

package/src/resources/SpanResource.tsx

@@ -87,7 +87,7 @@ function normalizeSpanParents(
 
 function enrichSpans(rawSpans: SpanData[]): {
   allSpans: Map<string, SpanItemState>;
-  parents: Map<string, string | null>;
+  children: Map<string | null, SpanItemState[]>;
   timeRange: { min: number; max: number };
 } {
   const spanMap = new Map<string, SpanData>();
@@ -132,17 +132,9 @@ function enrichSpans(rawSpans: SpanData[]): {
   if (max === -Infinity) max = Date.now();
   if (max === min) max = min + 100;
 
-  return { allSpans, parents, timeRange: { min, max } };
-}
-
-function buildFlatList(
-  allSpans: Map<string, SpanItemState>,
-  parents: Map<string, string | null>,
-  collapsedIds: Set<string>,
-): SpanItemState[] {
   const children = new Map<string | null, SpanItemState[]>();
   for (const span of allSpans.values()) {
-    const parent = parents.get(span.id) ?? null;
+    const parent = span.parentSpanId;
     let group = children.get(parent);
     if (!group) {
       group = [];
@@ -155,6 +147,13 @@ function buildFlatList(
     group.sort((a, b) => a.startedAt - b.startedAt);
   }
 
+  return { allSpans, children, timeRange: { min, max } };
+}
+
+function buildFlatList(
+  children: Map<string | null, SpanItemState[]>,
+  collapsedIds: Set<string>,
+): SpanItemState[] {
   const result: SpanItemState[] = [];
   const stack: SpanItemState[] = [];
   const roots = children.get(null) ?? [];
@@ -165,7 +164,7 @@ function buildFlatList(
   while (stack.length > 0) {
     const span = stack.pop()!;
     const isCollapsed = collapsedIds.has(span.id);
-    result.push({ ...span, isCollapsed });
+    result.push(isCollapsed ? { ...span, isCollapsed } : span);
 
     if (!isCollapsed) {
       const kids = children.get(span.id) ?? [];
@@ -210,7 +209,7 @@ const useSpanResource = ({
 }: {
   spans: SpanData[];
 }): ClientOutput<"span"> => {
-  const { allSpans, parents, timeRange } = useMemo(
+  const { allSpans, children, timeRange } = useMemo(
     () => enrichSpans(spans),
     [spans],
   );
@@ -220,8 +219,8 @@ const useSpanResource = ({
   );
 
   const visibleSpans = useMemo(
-    () => buildFlatList(allSpans, parents, collapsedIds),
-    [allSpans, parents, collapsedIds],
+    () => buildFlatList(children, collapsedIds),
+    [children, collapsedIds],
   );
 
   const toggleCollapse = (spanId: string) => {
@@ -245,6 +244,7 @@ const useSpanResource = ({
           timeRange,
           onToggleCollapse: toggleCollapse,
         }),
+        [allSpans, span.isCollapsed],
       ),
     ),
   );