mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-13 18:32:12 +03:00
feat(dashboard): orchestrationToFlow — deterministic 3-layer layout + stable fitKey
This commit is contained in:
@@ -0,0 +1,40 @@
|
||||
/** OrchSnapshot → @xyflow nodes/edges with a deterministic shallow 3-layer layout. Pure. */
|
||||
import type { Edge, Node } from "@xyflow/react";
|
||||
import { edgeStyle } from "@/shared/components/flow/edgeStyles";
|
||||
import type { OrchNodeKind, OrchSnapshot } from "./orchestrationTypes";
|
||||
|
||||
const LAYER_Y: Record<OrchNodeKind, number> = {
|
||||
orchestrator: 0, source: 150, work: 320, overflow: 320, activity: 470,
|
||||
};
|
||||
const X_GAP = 260;
|
||||
|
||||
export function orchestrationToFlow(snap: OrchSnapshot): { nodes: Node[]; edges: Edge[]; fitKey: string } {
|
||||
const byLayer = new Map<number, string[]>();
|
||||
for (const n of [...snap.nodes].sort((a, b) => a.id.localeCompare(b.id))) {
|
||||
const y = LAYER_Y[n.kind];
|
||||
const ids = byLayer.get(y) ?? [];
|
||||
ids.push(n.id);
|
||||
byLayer.set(y, ids);
|
||||
}
|
||||
const pos = new Map<string, { x: number; y: number }>();
|
||||
for (const [y, ids] of byLayer) {
|
||||
const width = (ids.length - 1) * X_GAP;
|
||||
ids.forEach((id, i) => pos.set(id, { x: i * X_GAP - width / 2, y }));
|
||||
}
|
||||
|
||||
const stateOf = new Map(snap.nodes.map((n) => [n.id, n.state]));
|
||||
const nodes: Node[] = snap.nodes.map((n) => ({
|
||||
id: n.id, type: n.kind, position: pos.get(n.id)!, data: n as unknown as Record<string, unknown>,
|
||||
}));
|
||||
const edges: Edge[] = snap.edges.map((e) => {
|
||||
const target = stateOf.get(e.to);
|
||||
const style = edgeStyle(e.active, false, target === "failed", target === "succeeded");
|
||||
return {
|
||||
id: e.id, source: e.from, target: e.to, animated: e.active,
|
||||
style: e.kind === "mirror" ? { ...style, strokeDasharray: "6 4" } : style,
|
||||
};
|
||||
});
|
||||
|
||||
const fitKey = snap.nodes.filter((n) => n.kind === "work").map((n) => n.id).sort().join("|");
|
||||
return { nodes, edges, fitKey };
|
||||
}
|
||||
45
tests/unit/ui/orchestrationToFlow.test.ts
Normal file
45
tests/unit/ui/orchestrationToFlow.test.ts
Normal file
@@ -0,0 +1,45 @@
|
||||
/** Run: node --import tsx/esm --test tests/unit/ui/orchestrationToFlow.test.ts */
|
||||
import { describe, it } from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
import { orchestrationToFlow } from "../../../src/app/(dashboard)/dashboard/orchestration/model/orchestrationToFlow.ts";
|
||||
import type { OrchSnapshot } from "../../../src/app/(dashboard)/dashboard/orchestration/model/orchestrationTypes.ts";
|
||||
|
||||
const snap: OrchSnapshot = {
|
||||
nodes: [
|
||||
{ id: "orchestrator", kind: "orchestrator", label: "OmniRoute" },
|
||||
{ id: "source:a2a", kind: "source", source: "a2a", label: "A2A" },
|
||||
{ id: "a2a:t1", kind: "work", source: "a2a", state: "running", label: "smart-routing" },
|
||||
{ id: "a2a:t2", kind: "work", source: "a2a", state: "failed", label: "cost-analysis" },
|
||||
],
|
||||
edges: [
|
||||
{ id: "e1", from: "orchestrator", to: "source:a2a", kind: "owns", active: false },
|
||||
{ id: "e2", from: "source:a2a", to: "a2a:t1", kind: "owns", active: true },
|
||||
{ id: "e3", from: "source:a2a", to: "a2a:t2", kind: "owns", active: false },
|
||||
],
|
||||
sources: [], generatedAt: "2026-08-30T12:00:00Z",
|
||||
};
|
||||
|
||||
describe("orchestrationToFlow", () => {
|
||||
it("puts each kind on its own Y layer and is deterministic", () => {
|
||||
const a = orchestrationToFlow(snap);
|
||||
const b = orchestrationToFlow(snap);
|
||||
assert.deepEqual(a.nodes.map((n) => n.position), b.nodes.map((n) => n.position));
|
||||
const ys = new Map(a.nodes.map((n) => [n.id, n.position.y]));
|
||||
assert.equal(ys.get("orchestrator"), 0);
|
||||
assert.equal(ys.get("source:a2a"), 150);
|
||||
assert.equal(ys.get("a2a:t1"), 320);
|
||||
});
|
||||
it("active edge is animated; edge to failed work is red", () => {
|
||||
const { edges } = orchestrationToFlow(snap);
|
||||
assert.equal(edges.find((e) => e.id === "e2")?.animated, true);
|
||||
const failedEdge = edges.find((e) => e.id === "e3");
|
||||
assert.equal((failedEdge?.style as { stroke?: string })?.stroke, "#ef4444");
|
||||
});
|
||||
it("fitKey only tracks the set of work ids", () => {
|
||||
const k1 = orchestrationToFlow(snap).fitKey;
|
||||
const stateChanged = { ...snap, nodes: snap.nodes.map((n) => n.id === "a2a:t1" ? { ...n, state: "succeeded" as const } : n) };
|
||||
assert.equal(orchestrationToFlow(stateChanged).fitKey, k1);
|
||||
const nodeRemoved = { ...snap, nodes: snap.nodes.filter((n) => n.id !== "a2a:t2"), edges: snap.edges.filter((e) => e.to !== "a2a:t2") };
|
||||
assert.notEqual(orchestrationToFlow(nodeRemoved).fitKey, k1);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user