feat(diagram-engine): style markers + XML→tree reverse parser

Groundwork for a declarative diagram engine where the model declares nesting and
the engine computes every coordinate, instead of the model emitting raw mxCell XML.

The design keeps the canvas as the SINGLE source of truth: the node tree is never
persisted, it is re-derived from the current canvas XML whenever needed. A user's
manual edits are therefore an input to the next re-layout, not a second copy of
the state that has to be reconciled.

Two behaviours this relies on, both verified against the real embedded editor with
a Playwright mouse drag (reading the editor's own autosave payload):

  1. An AWS group stencil WITHOUT container=1 does not get a dragged shape
     reparented — parent stays "1" and geometry stays absolute. WITH container=1
     it does: parent becomes the frame, geometry becomes parent-relative. So the
     engine must stamp container=1 on every container it emits.
  2. draw.io preserves style keys it does not understand, and resolves a duplicate
     key last-wins. So dai_* markers survive a user edit, and container=1 can be
     appended to a catalog style without first parsing out an existing value —
     which matters because the AWS catalog is inconsistent about it (group_vpc,
     group_region, group_subnet ship without it; group_account ships with it).

markers.ts  — dai_kind / dai_dir / dai_gap / dai_cols / dai_pin, and the container
              token normalisation.
types.ts    — the node tree contract, plus a `foreign` bucket so cells the engine
              does not understand (user annotations, imported shapes) round-trip
              verbatim rather than being destroyed by a re-layout.
parse.ts    — XML → tree. Handles all four icon encodings (resIcon=, bare shape=,
              shape=image data URI, grIcon=), resolves nesting from parent with a
              geometry fallback for frames that lack container=1, recovers layout
              direction from the marker or infers it from child positions, and
              survives cycles, compressed files and multi-page decks.

75 unit tests, including a round-trip against real output from the reference
project's build_vpc.mjs.

One finding worth recording: the reference project's "phantom" node (a wrapper that
participates in layout but emits no cell) makes the round-trip lossy by
construction. In build_vpc.mjs a phantom erased a container's col direction — its
children were reparented onto the grandparent, leaving a 2-D arrangement the parser
can only read as a grid. 26 of the reference project's 31 examples use phantoms, so
our engine needs an invisible-but-real container instead. Tracked separately.
This commit is contained in:
dayuan.jiang
2026-08-09 11:35:32 +09:00
parent fd758b9e87
commit 8765dfb96c
6 changed files with 1829 additions and 0 deletions

View File

@@ -0,0 +1,212 @@
import { describe, expect, it } from "vitest"
import {
hasMarkers,
isPinned,
readDir,
readIntMarker,
readKind,
readMarker,
stampContainer,
stampLeaf,
stripMarkers,
} from "@/lib/diagram-engine/markers"
// Real catalog styles from drawio-ai-kit's catalog/aws.json, verbatim. group_vpc
// ships WITHOUT container=1; group_account ships WITH it. The engine has to handle both.
const VPC_STYLE =
"sketch=0;outlineConnect=0;gradientColor=none;html=1;whiteSpace=wrap;fontSize=12;fontStyle=0;shape=mxgraph.aws4.group;grIcon=mxgraph.aws4.group_vpc;strokeColor=#879196;fillColor=none;verticalAlign=top;align=left;spacingLeft=30;fontColor=#879196;dashed=0;"
const ACCOUNT_STYLE =
"points=[[0,0],[0.25,0],[0.5,0]];outlineConnect=0;gradientColor=none;html=1;whiteSpace=wrap;fontSize=12;fontStyle=0;container=1;pointerEvents=0;collapsible=0;recursiveResize=0;shape=mxgraph.aws4.group;grIcon=mxgraph.aws4.group_account;strokeColor=#CD2264;fillColor=none;verticalAlign=top;align=left;spacingLeft=30;fontColor=#CD2264;dashed=0;"
describe("stampContainer", () => {
it("adds container tokens to a catalog style that lacks them", () => {
const s = stampContainer(VPC_STYLE, {
kind: "group",
dir: "col",
gap: 22,
})
expect(s).toContain("container=1;")
expect(s).toContain("pointerEvents=0;")
expect(s).toContain("collapsible=0;")
expect(s).toContain("recursiveResize=0;")
})
it("is safe on a style that already declares container=1", () => {
const s = stampContainer(ACCOUNT_STYLE, {
kind: "group",
dir: "row",
gap: 30,
})
// Duplicate keys are legal in draw.io and the LAST wins (verified in-browser),
// so appending a second container=1 keeps the shape a container.
expect(s.match(/container=1/g)?.length).toBe(2)
expect(readDir(s)).toBe("row")
})
it("records kind, dir and gap so the parser need not guess", () => {
const s = stampContainer(VPC_STYLE, {
kind: "group",
dir: "col",
gap: 22,
})
expect(readKind(s)).toBe("group")
expect(readDir(s)).toBe("col")
expect(readIntMarker(s, "dai_gap")).toBe(22)
})
it("records cols only for a grid", () => {
const grid = stampContainer(VPC_STYLE, {
kind: "grid",
dir: "grid",
gap: 14,
cols: 3,
})
expect(readIntMarker(grid, "dai_cols")).toBe(3)
const group = stampContainer(VPC_STYLE, {
kind: "group",
dir: "row",
gap: 14,
cols: 3,
})
expect(readIntMarker(group, "dai_cols")).toBeNull()
})
it("rounds a fractional gap", () => {
const s = stampContainer(VPC_STYLE, {
kind: "group",
dir: "col",
gap: 21.6,
})
expect(readIntMarker(s, "dai_gap")).toBe(22)
})
it("adds the missing separator when a style does not end in ;", () => {
const s = stampContainer("rounded=0;fillColor=#FFF", {
kind: "group",
dir: "row",
gap: 10,
})
expect(s).not.toContain("#FFFcontainer")
expect(s).toContain("#FFF;container=1;")
})
})
describe("readMarker duplicate handling", () => {
it("returns the LAST value, mirroring how draw.io resolves duplicate keys", () => {
// Verified in-browser: container=0;...;container=1; behaves as a container.
expect(readMarker("a=1;dai_dir=row;b=2;dai_dir=col;", "dai_dir")).toBe(
"col",
)
})
it("does not match a key that is only a suffix of another key", () => {
expect(readMarker("xdai_dir=row;", "dai_dir")).toBeNull()
})
it("returns null for an absent key", () => {
expect(readMarker(VPC_STYLE, "dai_dir")).toBeNull()
})
it("reads a marker at the very start of the style", () => {
expect(readMarker("dai_kind=box;rounded=0;", "dai_kind")).toBe("box")
})
})
describe("readKind / readDir reject unknown values", () => {
it("rejects a kind that is not in the union", () => {
expect(readKind("dai_kind=wormhole;")).toBeNull()
})
it("rejects a direction that is not in the union", () => {
expect(readDir("dai_dir=diagonal;")).toBeNull()
})
})
describe("readIntMarker", () => {
it("rejects a non-numeric value rather than returning NaN", () => {
expect(readIntMarker("dai_gap=wide;", "dai_gap")).toBeNull()
})
it("rejects a negative value", () => {
expect(readIntMarker("dai_gap=-5;", "dai_gap")).toBeNull()
})
it("accepts zero", () => {
expect(readIntMarker("dai_gap=0;", "dai_gap")).toBe(0)
})
})
describe("isPinned", () => {
it("is false when the marker is absent", () => {
expect(isPinned(VPC_STYLE)).toBe(false)
})
it("is true for the value the engine writes", () => {
expect(isPinned("dai_pin=1;")).toBe(true)
})
it("accepts what a user might hand-type in draw.io's Edit Style dialog", () => {
expect(isPinned("dai_pin=yes;")).toBe(true)
expect(isPinned("dai_pin=true;")).toBe(true)
})
it("treats 0 / false / empty as not pinned, so a user can unpin by editing", () => {
expect(isPinned("dai_pin=0;")).toBe(false)
expect(isPinned("dai_pin=false;")).toBe(false)
expect(isPinned("dai_pin=;")).toBe(false)
})
})
describe("stampLeaf", () => {
it("marks an icon", () => {
expect(readKind(stampLeaf("shape=mxgraph.aws4.ec2;", "icon"))).toBe(
"icon",
)
})
it("marks a box", () => {
expect(readKind(stampLeaf("rounded=0;", "box"))).toBe("box")
})
})
describe("stripMarkers", () => {
it("removes every dai_ token and keeps the rest intact", () => {
const stamped = stampContainer(VPC_STYLE, {
kind: "group",
dir: "col",
gap: 22,
})
const stripped = stripMarkers(stamped)
expect(stripped).not.toContain("dai_")
// the real style tokens survive
expect(stripped).toContain("grIcon=mxgraph.aws4.group_vpc")
expect(stripped).toContain("container=1")
})
it("leaves a marker-free style semantically unchanged", () => {
expect(stripMarkers(VPC_STYLE)).toBe(VPC_STYLE)
})
})
describe("hasMarkers", () => {
it("is false for a plain catalog style", () => {
expect(hasMarkers(VPC_STYLE)).toBe(false)
})
it("is true for engine output", () => {
expect(
hasMarkers(
stampContainer(VPC_STYLE, {
kind: "group",
dir: "row",
gap: 8,
}),
),
).toBe(true)
})
it("is not fooled by a key that merely contains dai_", () => {
expect(hasMarkers("mydai_dir=row;")).toBe(false)
})
})

View File

@@ -0,0 +1,742 @@
import { readFileSync } from "node:fs"
import { join } from "node:path"
import { describe, expect, it } from "vitest"
import { stampContainer } from "@/lib/diagram-engine/markers"
import {
countPages,
extractPage,
parseDiagram,
} from "@/lib/diagram-engine/parse"
import {
type ContainerNode,
type DiagramNode,
findNode,
findParent,
isContainer,
walkTree,
} from "@/lib/diagram-engine/types"
/**
* Real output from drawio-ai-kit's examples/aws/build_vpc.mjs — the exact XML shape the
* parser has to handle. Structure declared by that script:
*
* root (phantom, row)
* ├── users (box)
* └── region (group_region, row)
* ├── vpc (group_vpc, col)
* │ ├── igw, alb (icons)
* │ └── azs (phantom, row)
* │ ├── az_a (group_availability_zone, col) → pub_a/app_a/db_a → nat_a/ec2_a/rds_a
* │ └── az_b (same, mirrored)
* └── reg_svc (plain frame, col) → waf, cw, s3
*
* The two phantoms emit no cell, so their children are reparented to the nearest
* visible ancestor: users/region become roots, az_a/az_b become children of vpc.
*/
const ENGINE_XML = readFileSync(
join(__dirname, "fixtures/engine-vpc-multiaz.drawio"),
"utf8",
)
/** Minimal page, built by hand so each test controls exactly one variable. */
function page(cells: string): string {
return `<mxfile host="app.diagrams.net"><diagram name="Page-1" id="p1"><mxGraphModel dx="1400" dy="900" grid="0" pageWidth="1200" pageHeight="800"><root><mxCell id="0"/><mxCell id="1" parent="0"/>${cells}</root></mxGraphModel></diagram></mxfile>`
}
function vtx(
id: string,
style: string,
geo: { x: number; y: number; w: number; h: number },
opts: { parent?: string; value?: string } = {},
): string {
return `<mxCell id="${id}" value="${opts.value ?? ""}" style="${style}" vertex="1" parent="${opts.parent ?? "1"}"><mxGeometry x="${geo.x}" y="${geo.y}" width="${geo.w}" height="${geo.h}" as="geometry"/></mxCell>`
}
const AWS_GROUP = (name: string) =>
`sketch=0;outlineConnect=0;gradientColor=none;html=1;whiteSpace=wrap;fontSize=12;shape=mxgraph.aws4.group;grIcon=mxgraph.aws4.${name};strokeColor=#879196;fillColor=none;verticalAlign=top;align=left;spacingLeft=30;`
const RES_ICON = (name: string) =>
`sketch=0;outlineConnect=0;fillColor=#ED7100;strokeColor=none;verticalLabelPosition=bottom;verticalAlign=top;align=center;html=1;aspect=fixed;shape=mxgraph.aws4.resourceIcon;resIcon=mxgraph.aws4.${name};`
const OWN_SHAPE_ICON = (name: string) =>
`sketch=0;outlineConnect=0;fillColor=#ED7100;strokeColor=none;verticalLabelPosition=bottom;verticalAlign=top;align=center;html=1;aspect=fixed;shape=mxgraph.aws4.${name};`
const IMAGE_ICON =
"sketch=0;html=1;outlineConnect=0;verticalLabelPosition=bottom;verticalAlign=top;align=center;fontColor=#232F3E;aspect=fixed;shape=image;image=data:image/png,iVBORw0KGgoAAAANSUhEUg==;"
const PLAIN_BOX =
"rounded=0;whiteSpace=wrap;html=1;fillColor=#DAE8FC;strokeColor=#6C8EBF;"
describe("extractPage", () => {
it("pulls the model body out of an mxfile", () => {
const p = extractPage(ENGINE_XML)
expect(p).toContain("<mxCell")
expect(p).not.toContain("<mxfile")
})
it("accepts a bare mxGraphModel", () => {
const bare = `<mxGraphModel><root><mxCell id="0"/><mxCell id="1" parent="0"/></root></mxGraphModel>`
expect(extractPage(bare)).toContain("<mxCell")
})
it("returns null for a compressed page rather than pretending it is empty", () => {
const compressed = `<mxfile><diagram name="Page-1" id="p1">7VvbcuI4EP0aHmfL8oXLY0gyu1O1t</diagram></mxfile>`
expect(extractPage(compressed)).toBeNull()
})
it("selects the requested page and clamps an out-of-range index", () => {
const two = `<mxfile><diagram name="A" id="a"><mxGraphModel><root><mxCell id="0"/><mxCell id="onlyA"/></root></mxGraphModel></diagram><diagram name="B" id="b"><mxGraphModel><root><mxCell id="0"/><mxCell id="onlyB"/></root></mxGraphModel></diagram></mxfile>`
expect(extractPage(two, 0)).toContain("onlyA")
expect(extractPage(two, 1)).toContain("onlyB")
expect(extractPage(two, 99)).toContain("onlyB")
})
})
describe("countPages", () => {
it("counts a multi-page deck", () => {
const two = `<mxfile><diagram name="A" id="a"></diagram><diagram name="B" id="b"></diagram></mxfile>`
expect(countPages(two)).toBe(2)
})
it("reports 1 for a single page", () => {
expect(countPages(ENGINE_XML)).toBe(1)
})
})
describe("parseDiagram on real engine output", () => {
const { tree, needsAdoption, warnings } = parseDiagram(ENGINE_XML)
it("recovers the nesting the build script declared", () => {
// az_a is a child of vpc (the `azs` phantom emitted no cell)
expect(findParent(tree, "az_a")?.id).toBe("vpc")
expect(findParent(tree, "vpc")?.id).toBe("region")
expect(findParent(tree, "ec2_a")?.id).toBe("app_a")
expect(findParent(tree, "app_a")?.id).toBe("az_a")
expect(findParent(tree, "rds_b")?.id).toBe("db_b")
})
it("puts users and region at the top level", () => {
const rootIds = tree.roots.map((r) => r.id).sort()
expect(rootIds).toEqual(["region", "users"])
})
it("classifies AWS group stencils as containers and keeps their stencil name", () => {
const vpc = findNode(tree, "vpc")
expect(isContainer(vpc as DiagramNode)).toBe(true)
expect((vpc as ContainerNode).gname).toBe("group_vpc")
expect((findNode(tree, "az_a") as ContainerNode).gname).toBe(
"group_availability_zone",
)
expect((findNode(tree, "pub_a") as ContainerNode).gname).toBe(
"group_subnet",
)
})
it("cannot recover a direction the phantom erased — the case that rules phantoms out", () => {
// The build script declared vpc as dir:"col" holding igw, alb, and a phantom
// wrapping the two AZ columns. A phantom emits NO cell, so its children were
// reparented onto vpc: igw(260,60) alb(260,164) az_a(24,268) az_b(309,268).
// That is a 2-D arrangement, so inference correctly reads "grid" — the original
// column structure is simply not in the XML any more.
//
// This is why our engine must not have phantoms: a wrapper that emits no cell
// makes the round-trip lossy by construction. See task #5.
expect(findNode(tree, "vpc")?.kind).toBe("grid")
expect(findParent(tree, "az_a")?.id).toBe("vpc")
expect(findNode(tree, "azs")).toBeNull()
})
it("classifies resourceIcon cells as icons and recovers their catalog name", () => {
const ec2 = findNode(tree, "ec2_a")
expect(ec2?.kind).toBe("icon")
expect(ec2 && "name" in ec2 ? ec2.name : null).toBe("ec2")
const nat = findNode(tree, "nat_a")
expect(nat && "name" in nat ? nat.name : null).toBe("nat_gateway")
})
it("classifies a plain rectangle as a box", () => {
expect(findNode(tree, "users")?.kind).toBe("box")
})
it("lifts the title out of the flow instead of leaving it as a node", () => {
expect(tree.title).toContain("VPC Multi-AZ 3-tier")
expect(findNode(tree, "__title")).toBeNull()
})
it("recovers every edge with its endpoints", () => {
expect(tree.links).toHaveLength(7)
const pairs = tree.links.map((l) => `${l.source}->${l.target}`)
expect(pairs).toContain("users->igw")
expect(pairs).toContain("alb->ec2_a")
expect(pairs).toContain("rds_a->rds_b")
})
it("splits a step number off the edge label", () => {
const first = tree.links.find(
(l) => l.source === "users" && l.target === "igw",
)
expect(first?.step).toBe(1)
expect(first?.label).toBe("HTTPS")
})
it("marks a dashed edge", () => {
const repl = tree.links.find(
(l) => l.source === "rds_a" && l.target === "rds_b",
)
expect(repl?.dashed).toBe(true)
expect(repl?.label).toBe("Multi-AZ replication")
})
it("flags the diagram as needing adoption — it carries no markers yet", () => {
// This fixture predates the marker scheme, so direction had to be inferred.
expect(needsAdoption).toBe(true)
})
it("parses without warnings on a well-formed single page", () => {
expect(warnings).toEqual([])
})
it("assigns every cell exactly once — no duplicates, nothing lost", () => {
const ids = [...walkTree(tree)].map((n) => n.id)
expect(new Set(ids).size).toBe(ids.length)
// 24 vertex cells in the fixture, minus __title which is lifted out of the flow
expect(ids).toHaveLength(23)
})
})
describe("layout inference (no markers present)", () => {
it("reads a row from children spread along x", () => {
const xml = page(
vtx("f", AWS_GROUP("group_vpc"), { x: 0, y: 0, w: 400, h: 120 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 20, y: 40, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"b",
RES_ICON("ec2"),
{ x: 120, y: 40, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"c",
RES_ICON("rds"),
{ x: 220, y: 40, w: 48, h: 48 },
{ parent: "f" },
),
)
const f = findNode(parseDiagram(xml).tree, "f") as ContainerNode
expect(f.kind).toBe("group")
expect((f as { dir?: string }).dir).toBe("row")
})
it("reads a column from children spread along y", () => {
const xml = page(
vtx("f", AWS_GROUP("group_vpc"), { x: 0, y: 0, w: 120, h: 400 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 30, y: 20, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"b",
RES_ICON("ec2"),
{ x: 30, y: 120, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"c",
RES_ICON("rds"),
{ x: 30, y: 220, w: 48, h: 48 },
{ parent: "f" },
),
)
const f = findNode(parseDiagram(xml).tree, "f") as ContainerNode
expect((f as { dir?: string }).dir).toBe("col")
})
it("recognises a 2-D arrangement as a grid rather than forcing row or column", () => {
const xml = page(
vtx("f", AWS_GROUP("group_vpc"), { x: 0, y: 0, w: 300, h: 300 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 20, y: 20, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"b",
RES_ICON("ec2"),
{ x: 140, y: 20, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"c",
RES_ICON("rds"),
{ x: 20, y: 140, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"d",
RES_ICON("sqs"),
{ x: 140, y: 140, w: 48, h: 48 },
{ parent: "f" },
),
)
const f = findNode(parseDiagram(xml).tree, "f")
expect(f?.kind).toBe("grid")
})
it("measures the gap between neighbours, not between their origins", () => {
// icons 48 wide at x=20,120,220 → edge-to-edge gap is 52
const xml = page(
vtx("f", AWS_GROUP("group_vpc"), { x: 0, y: 0, w: 400, h: 120 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 20, y: 40, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"b",
RES_ICON("ec2"),
{ x: 120, y: 40, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"c",
RES_ICON("rds"),
{ x: 220, y: 40, w: 48, h: 48 },
{ parent: "f" },
),
)
const f = findNode(parseDiagram(xml).tree, "f") as ContainerNode
expect((f as { gap?: number }).gap).toBe(52)
})
})
describe("markers take precedence over inference", () => {
it("believes dai_dir even when the geometry suggests otherwise", () => {
// children are laid out in a row, but the marker says col
const marked = stampContainer(AWS_GROUP("group_vpc"), {
kind: "group",
dir: "col",
gap: 33,
})
const xml = page(
vtx("f", marked, { x: 0, y: 0, w: 400, h: 120 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 20, y: 40, w: 48, h: 48 },
{ parent: "f" },
) +
vtx(
"b",
RES_ICON("ec2"),
{ x: 120, y: 40, w: 48, h: 48 },
{ parent: "f" },
),
)
const f = findNode(parseDiagram(xml).tree, "f") as ContainerNode
expect((f as { dir?: string }).dir).toBe("col")
expect((f as { gap?: number }).gap).toBe(33)
})
it("clears needsAdoption once any cell carries a marker", () => {
const marked = stampContainer(AWS_GROUP("group_vpc"), {
kind: "group",
dir: "row",
gap: 20,
})
const xml = page(vtx("f", marked, { x: 0, y: 0, w: 200, h: 100 }))
expect(parseDiagram(xml).needsAdoption).toBe(false)
})
it("recovers grid columns from dai_cols", () => {
const marked = stampContainer("rounded=0;html=1;", {
kind: "grid",
dir: "grid",
gap: 14,
cols: 3,
})
const xml = page(
vtx("g", marked, { x: 0, y: 0, w: 300, h: 200 }) +
vtx(
"a",
RES_ICON("s3"),
{ x: 10, y: 10, w: 48, h: 48 },
{ parent: "g" },
),
)
const g = findNode(parseDiagram(xml).tree, "g")
expect(g?.kind).toBe("grid")
expect((g as { cols?: number }).cols).toBe(3)
})
it("carries a user pin through to the node", () => {
const xml = page(
vtx("b", `${PLAIN_BOX}dai_kind=box;dai_pin=1;`, {
x: 10,
y: 10,
w: 120,
h: 60,
}),
)
const b = findNode(parseDiagram(xml).tree, "b")
expect((b as { pinned?: boolean }).pinned).toBe(true)
})
})
describe("icon classification covers all four encodings", () => {
it("handles resIcon=, bare shape=, and shape=image", () => {
// 554 AWS icons use resIcon; 429 use their own shape=; Azure/GCP use shape=image.
const xml = page(
vtx("res", RES_ICON("ec2"), { x: 0, y: 0, w: 48, h: 48 }) +
vtx("own", OWN_SHAPE_ICON("a1_instance"), {
x: 100,
y: 0,
w: 48,
h: 48,
}) +
vtx("img", IMAGE_ICON, { x: 200, y: 0, w: 48, h: 48 }),
)
const { tree } = parseDiagram(xml)
expect(findNode(tree, "res")?.kind).toBe("icon")
expect(findNode(tree, "own")?.kind).toBe("icon")
expect(findNode(tree, "img")?.kind).toBe("icon")
})
it("recovers the catalog name from a bare shape= icon", () => {
const xml = page(
vtx("own", OWN_SHAPE_ICON("a1_instance"), {
x: 0,
y: 0,
w: 48,
h: 48,
}),
)
const n = findNode(parseDiagram(xml).tree, "own")
expect(n && "name" in n ? n.name : null).toBe("a1_instance")
})
it("leaves the name empty for an embedded-image icon instead of inventing one", () => {
const xml = page(vtx("img", IMAGE_ICON, { x: 0, y: 0, w: 48, h: 48 }))
const n = findNode(parseDiagram(xml).tree, "img")
expect(n && "name" in n ? n.name : null).toBe("")
// ...but the verbatim style is kept, so it can be re-emitted unchanged
expect(n && "style" in n ? n.style : "").toContain("data:image/png")
})
})
describe("nesting when a frame lacks container=1", () => {
it("re-homes a cell that visually sits inside a loose frame", () => {
// draw.io will NOT reparent into a frame without container=1 (verified
// in-browser), so the user's drop leaves parent="1" while the shape is
// visually inside. Geometry is the truth here.
const xml = page(
vtx("loose", AWS_GROUP("group_vpc"), {
x: 100,
y: 100,
w: 400,
h: 300,
}) +
vtx("inside", RES_ICON("ec2"), {
x: 200,
y: 200,
w: 48,
h: 48,
}),
)
const { tree } = parseDiagram(xml)
expect(findParent(tree, "inside")?.id).toBe("loose")
})
it("picks the innermost frame when loose frames nest", () => {
const xml = page(
vtx("outer", AWS_GROUP("group_region"), {
x: 0,
y: 0,
w: 800,
h: 600,
}) +
vtx("inner", AWS_GROUP("group_vpc"), {
x: 100,
y: 100,
w: 300,
h: 200,
}) +
vtx("leaf", RES_ICON("ec2"), { x: 150, y: 150, w: 48, h: 48 }),
)
const { tree } = parseDiagram(xml)
expect(findParent(tree, "leaf")?.id).toBe("inner")
})
it("leaves a cell alone when it is outside every loose frame", () => {
const xml = page(
vtx("loose", AWS_GROUP("group_vpc"), {
x: 100,
y: 100,
w: 200,
h: 200,
}) +
vtx("outside", RES_ICON("ec2"), {
x: 600,
y: 600,
w: 48,
h: 48,
}),
)
const { tree } = parseDiagram(xml)
expect(findParent(tree, "outside")).toBeNull()
expect(tree.roots.map((r) => r.id)).toContain("outside")
})
it("does NOT override an explicit parent that draw.io maintained", () => {
// With container=1 the parent attribute is authoritative — even if a stale
// geometry would place the child inside a different frame.
const withContainer = stampContainer(AWS_GROUP("group_vpc"), {
kind: "group",
dir: "row",
gap: 20,
})
const xml = page(
vtx("real", withContainer, { x: 500, y: 0, w: 300, h: 200 }) +
vtx("loose", AWS_GROUP("group_region"), {
x: 0,
y: 0,
w: 400,
h: 400,
}) +
// parent says "real"; geometry (relative to real) lands it at 520,20
vtx(
"kid",
RES_ICON("ec2"),
{ x: 20, y: 20, w: 48, h: 48 },
{ parent: "real" },
),
)
const { tree } = parseDiagram(xml)
expect(findParent(tree, "kid")?.id).toBe("real")
})
})
describe("robustness", () => {
it("returns an empty tree with a warning for a compressed file", () => {
const compressed = `<mxfile><diagram name="Page-1" id="p1">7VvbcuI4EP0aHmfLGHN5DCQzu1O1t</diagram></mxfile>`
const r = parseDiagram(compressed)
expect(r.tree.roots).toEqual([])
expect(r.warnings[0]).toContain("compressed")
})
it("warns when it silently parsed only the first page of a deck", () => {
const two = `<mxfile><diagram name="A" id="a"><mxGraphModel><root><mxCell id="0"/><mxCell id="1" parent="0"/>${vtx("a", PLAIN_BOX, { x: 0, y: 0, w: 100, h: 50 })}</root></mxGraphModel></diagram><diagram name="B" id="b"><mxGraphModel><root><mxCell id="0"/><mxCell id="1" parent="0"/></root></mxGraphModel></diagram></mxfile>`
const r = parseDiagram(two)
expect(r.warnings.some((w) => w.includes("2 pages"))).toBe(true)
})
it("treats a cell whose parent does not exist as a root instead of dropping it", () => {
const xml = page(
vtx(
"orphan",
PLAIN_BOX,
{ x: 0, y: 0, w: 100, h: 50 },
{ parent: "ghost" },
),
)
const { tree } = parseDiagram(xml)
expect(tree.roots.map((r) => r.id)).toContain("orphan")
})
it("does not hang on a parent cycle", () => {
const cyclic = page(
vtx(
"a",
PLAIN_BOX,
{ x: 0, y: 0, w: 100, h: 50 },
{ parent: "b" },
) +
vtx(
"b",
PLAIN_BOX,
{ x: 0, y: 0, w: 100, h: 50 },
{ parent: "a" },
),
)
const r = parseDiagram(cyclic)
expect(r.warnings.some((w) => w.includes("cycle"))).toBe(true)
})
it("unescapes entities in labels", () => {
const xml = page(
vtx(
"b",
PLAIN_BOX,
{ x: 0, y: 0, w: 100, h: 50 },
{
value: "A &amp; B &lt;tag&gt; &quot;q&quot;",
},
),
)
const b = findNode(parseDiagram(xml).tree, "b")
expect((b as { label?: string }).label).toBe('A & B <tag> "q"')
})
it("keeps cells on the boundaries layer verbatim instead of restructuring them", () => {
const xml = page(
`<mxCell id="boundaries" value="Stack boundaries (locked)" parent="0" style="locked=1;"/>` +
vtx(
"cluster",
"rounded=0;dashed=1;fillColor=none;strokeColor=#ED7100;",
{
x: 10,
y: 10,
w: 200,
h: 100,
},
{ parent: "boundaries" },
),
)
const { tree } = parseDiagram(xml)
expect(tree.foreign.map((f) => f.id)).toContain("cluster")
expect(findNode(tree, "cluster")).toBeNull()
expect(tree.foreign[0].xml).toContain("strokeColor=#ED7100")
})
it("ignores an edge that is missing an endpoint", () => {
const xml = page(
vtx("a", PLAIN_BOX, { x: 0, y: 0, w: 100, h: 50 }) +
`<mxCell id="e1" style="edgeStyle=orthogonalEdgeStyle;" edge="1" parent="1" source="a"><mxGeometry relative="1" as="geometry"/></mxCell>`,
)
expect(parseDiagram(xml).tree.links).toHaveLength(0)
})
it("handles a self-closing mxCell", () => {
const xml = page(
`<mxCell id="bare" value="x" style="${PLAIN_BOX}" vertex="1" parent="1"/>`,
)
const { tree } = parseDiagram(xml)
expect(tree.roots.map((r) => r.id)).toContain("bare")
})
})
describe("container classification edge cases", () => {
it("treats a plain box that has children as a container", () => {
const xml = page(
vtx("frame", PLAIN_BOX, { x: 0, y: 0, w: 300, h: 200 }) +
vtx(
"kid",
RES_ICON("s3"),
{ x: 20, y: 20, w: 48, h: 48 },
{ parent: "frame" },
),
)
const f = findNode(parseDiagram(xml).tree, "frame")
expect(f?.kind).toBe("group")
expect(isContainer(f as DiagramNode)).toBe(true)
})
it("treats a childless container=1 frame as a container, not a box", () => {
const xml = page(
vtx("empty", `${PLAIN_BOX}container=1;`, {
x: 0,
y: 0,
w: 200,
h: 100,
}),
)
expect(findNode(parseDiagram(xml).tree, "empty")?.kind).toBe("group")
})
it("honours the last container= value when the key is duplicated", () => {
// Verified in-browser: draw.io resolves duplicate keys last-wins.
const xml = page(
vtx("c", `${PLAIN_BOX}container=1;container=0;`, {
x: 0,
y: 0,
w: 200,
h: 100,
}),
)
// last value is 0 → not a container, and it has no children → a box
expect(findNode(parseDiagram(xml).tree, "c")?.kind).toBe("box")
})
it("recognises a text cell as the title even without the __title id", () => {
const xml = page(
`<mxCell id="t9" value="My Diagram" style="text;html=1;align=center;fontSize=14;" vertex="1" parent="1"><mxGeometry x="0" y="24" width="800" height="30" as="geometry"/></mxCell>`,
)
const { tree } = parseDiagram(xml)
expect(tree.title).toBe("My Diagram")
expect(findNode(tree, "t9")).toBeNull()
})
it("keeps the first title when a diagram somehow has two", () => {
const xml = page(
`<mxCell id="t1" value="First" style="text;html=1;" vertex="1" parent="1"><mxGeometry x="0" y="0" width="100" height="30" as="geometry"/></mxCell>` +
`<mxCell id="t2" value="Second" style="text;html=1;" vertex="1" parent="1"><mxGeometry x="0" y="40" width="100" height="30" as="geometry"/></mxCell>`,
)
expect(parseDiagram(xml).tree.title).toBe("First")
})
})
describe("geometry resolution", () => {
it("resolves a nested cell's position to page coordinates", () => {
const withContainer = stampContainer(AWS_GROUP("group_vpc"), {
kind: "group",
dir: "row",
gap: 20,
})
const xml = page(
vtx("outer", withContainer, { x: 100, y: 200, w: 400, h: 300 }) +
vtx(
"kid",
`${RES_ICON("ec2")}dai_kind=icon;dai_pin=1;`,
{
x: 30,
y: 40,
w: 48,
h: 48,
},
{ parent: "outer" },
),
)
const kid = findNode(parseDiagram(xml).tree, "kid")
// 100+30, 200+40
expect((kid as { rect?: { x: number; y: number } }).rect).toEqual({
x: 130,
y: 240,
w: 48,
h: 48,
})
})
it("resolves through two levels of nesting", () => {
const c = stampContainer(AWS_GROUP("group_vpc"), {
kind: "group",
dir: "col",
gap: 10,
})
const xml = page(
vtx("l1", c, { x: 100, y: 100, w: 500, h: 400 }) +
vtx(
"l2",
c,
{ x: 50, y: 60, w: 300, h: 200 },
{ parent: "l1" },
) +
vtx(
"leaf",
`${RES_ICON("s3")}dai_kind=icon;dai_pin=1;`,
{
x: 10,
y: 20,
w: 48,
h: 48,
},
{ parent: "l2" },
),
)
const leaf = findNode(parseDiagram(xml).tree, "leaf")
// 100+50+10, 100+60+20
expect((leaf as { rect?: { x: number; y: number } }).rect?.x).toBe(160)
expect((leaf as { rect?: { x: number; y: number } }).rect?.y).toBe(180)
})
})

File diff suppressed because one or more lines are too long