mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-12 20:49:53 +08:00
dayuan.jiang
fc8ffc5655
- index.ts: registerTool() does the session check, try/catch and the
write queue for all 15 tools; fail()/ok(), syncFromBrowser(),
commitDocument() and describePages() replace the copied blocks; drop
pickPageSelector, currentSession.version, the input ?? {} defaults;
add_page reuses prepareNewModel() split out of prepareNewDiagram
- xml-validation.ts: drop the hand-written syntax checks that saxes
already reports, MAX_XML_SIZE and the stale header
- http-server.ts: one pendingExport object, no unread response fields,
no CORS/OPTIONS branch for same-origin requests, fewer exports
- diagram-operations parseNewCell(), addHistory returns void,
normalizeToMxfile without host, foldCells without layer, expandStyles
removed, unused preview.css rule removed, no .d.ts output
- xml-examples.ts styleRules() shares the Styles paragraphs with the
web prompt
- tests: setupFiles installs the DOM polyfill once, server-wiring uses
the SDK client, mcp-core.test.ts drops the cases MCP tests already
cover
210 lines
8.4 KiB
TypeScript
210 lines
8.4 KiB
TypeScript
/**
|
|
* Tests for prepareNewDiagram's truncation check: output that stops inside an
|
|
* mxCell gets a clear error instead of one about the wrapper tags the model
|
|
* never sent.
|
|
*/
|
|
|
|
import { describe, expect, it } from "vitest"
|
|
import {
|
|
isMxCellXmlComplete,
|
|
prepareNewDiagram,
|
|
truncatedCellError,
|
|
} from "../src/new-diagram.ts"
|
|
|
|
describe("isMxCellXmlComplete", () => {
|
|
it("returns false for empty/null input", () => {
|
|
expect(isMxCellXmlComplete("")).toBe(false)
|
|
expect(isMxCellXmlComplete(null)).toBe(false)
|
|
expect(isMxCellXmlComplete(undefined)).toBe(false)
|
|
})
|
|
|
|
it("returns true for self-closing mxCell", () => {
|
|
const xml =
|
|
'<mxCell id="2" value="Hello" style="rounded=1;" vertex="1" parent="1"/>'
|
|
expect(isMxCellXmlComplete(xml)).toBe(true)
|
|
// A raw "<" in a value (escaped later by the auto-fix)
|
|
expect(
|
|
isMxCellXmlComplete(
|
|
'<mxCell id="3" value="<b>Title</b>" style="text;html=1;" vertex="1" parent="1"/>',
|
|
),
|
|
).toBe(true)
|
|
})
|
|
|
|
it("returns true for mxCell with closing tag", () => {
|
|
const xml = `<mxCell id="2" value="Hello" vertex="1" parent="1">
|
|
<mxGeometry x="100" y="100" width="120" height="60" as="geometry"/>
|
|
</mxCell>`
|
|
expect(isMxCellXmlComplete(xml)).toBe(true)
|
|
})
|
|
|
|
it("returns false for truncated mxCell", () => {
|
|
const xml =
|
|
'<mxCell id="2" value="Hello" style="rounded=1;" vertex="1" parent'
|
|
expect(isMxCellXmlComplete(xml)).toBe(false)
|
|
})
|
|
|
|
it("returns false for mxCell with unclosed geometry", () => {
|
|
const xml = `<mxCell id="2" value="Hello" vertex="1" parent="1">
|
|
<mxGeometry x="100" y="100" width="120"`
|
|
expect(isMxCellXmlComplete(xml)).toBe(false)
|
|
})
|
|
|
|
it("returns false when output stops after a child of an open mxCell", () => {
|
|
const xml = `<mxCell id="2" value="A" vertex="1" parent="1">
|
|
<mxGeometry x="0" y="0" width="80" height="40" as="geometry"/>
|
|
</mxCell>
|
|
<mxCell id="3" value="B" vertex="1" parent="1">
|
|
<mxGeometry x="100" y="0" width="80" height="40" as="geometry"/>`
|
|
expect(isMxCellXmlComplete(xml)).toBe(false)
|
|
})
|
|
|
|
it("returns false when output stops after </mxGeometry> of an open mxCell", () => {
|
|
const xml = `<mxCell id="e1" edge="1" parent="1" source="2" target="3">
|
|
<mxGeometry relative="1" as="geometry">
|
|
<mxPoint x="10" y="10" as="sourcePoint"/>
|
|
</mxGeometry>`
|
|
expect(isMxCellXmlComplete(xml)).toBe(false)
|
|
})
|
|
|
|
it("returns true for a self-closing last mxCell with > in its value", () => {
|
|
const xml = `<mxCell id="2" value="A" vertex="1" parent="1">
|
|
<mxGeometry as="geometry"/>
|
|
</mxCell>
|
|
<mxCell id="3" value="A -> B" vertex="1" parent="1"/></root>`
|
|
expect(isMxCellXmlComplete(xml)).toBe(true)
|
|
})
|
|
|
|
it("returns true for multiple complete mxCells", () => {
|
|
const xml = `<mxCell id="2" value="A" vertex="1" parent="1"/>
|
|
<mxCell id="3" value="B" vertex="1" parent="1"/>`
|
|
expect(isMxCellXmlComplete(xml)).toBe(true)
|
|
})
|
|
})
|
|
|
|
describe("prepareNewDiagram with cut-off output", () => {
|
|
const A = `<mxCell id="2" value="A" vertex="1" parent="1"><mxGeometry x="0" y="0" width="80" height="40" as="geometry"/></mxCell>`
|
|
|
|
it("rejects a cell cut off inside an attribute", () => {
|
|
const cut = `${A}<mxCell id="3" value="B" vertex="1" par`
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).toContain("cut off")
|
|
// The tail is quoted so the model knows where to resume
|
|
expect(out.error).toContain('<mxCell id="3" value="B" vertex="1" par')
|
|
expect(out.error).toContain("edit_diagram add")
|
|
})
|
|
|
|
it("rejects a cell cut off after its geometry", () => {
|
|
const cut = `${A}<mxCell id="3" value="B" vertex="1" parent="1"><mxGeometry x="200" y="0" width="80" height="40" as="geometry"/>`
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).toContain("cut off")
|
|
expect(out.error).not.toContain("</root>")
|
|
})
|
|
|
|
it("quotes at most the last 300 characters", () => {
|
|
const cut = `${A}<mxCell id="3" value="${"x".repeat(400)}`
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).toContain(cut.slice(-300))
|
|
expect(out.error).not.toContain(cut.slice(-301))
|
|
})
|
|
|
|
it("does not report complete cells followed by an XML comment as cut", () => {
|
|
const cells = `${A}<!-- end of the diagram -->`
|
|
expect(truncatedCellError(cells)).toBeNull()
|
|
expect(prepareNewDiagram(cells).ok).toBe(true)
|
|
})
|
|
|
|
it("leaves a closing tag the auto-fix repairs to the auto-fix", () => {
|
|
// </mxCell/> is a typo, not a cut
|
|
const typo = A.replace("</mxCell>", "</mxCell/>")
|
|
expect(truncatedCellError(typo)).toBeNull()
|
|
expect(prepareNewDiagram(typo).ok).toBe(true)
|
|
})
|
|
|
|
it("leaves input without any cell to the validator", () => {
|
|
for (const input of ["", "Here is the diagram"]) {
|
|
expect(truncatedCellError(input)).toBeNull()
|
|
const out = prepareNewDiagram(input)
|
|
expect(out.ok).toBe(false)
|
|
if (!out.ok) expect(out.error).not.toContain("cut off")
|
|
}
|
|
})
|
|
|
|
it("never checks a full <mxfile>", () => {
|
|
const cut = `<mxfile><diagram id="p1" name="Page-1"><mxGraphModel><root><mxCell id="0"/><mxCell id="1" parent="0"/><mxCell id="2" value="A" vertex="1" par`
|
|
expect(truncatedCellError(cut)).toBeNull()
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).not.toContain("cut off")
|
|
})
|
|
})
|
|
|
|
describe("truncation check with named styles and compact cells", () => {
|
|
const BLUE =
|
|
'<mxStyle name="blue" value="fillColor=#dae8fc;strokeColor=#6c8ebf;"/>\n'
|
|
const A =
|
|
'<mxCell id="2" value="A" style="blue;" x="0" y="0" w="80" h="40"/>\n'
|
|
|
|
it("accepts complete compact cells after the definitions and expands them", () => {
|
|
const out = prepareNewDiagram(
|
|
`${BLUE}${A}<mxCell id="3" value="B" style="blue;" x="120" y="0" w="80" h="40"/>`,
|
|
)
|
|
expect(out.ok).toBe(true)
|
|
if (!out.ok) return
|
|
expect(out.xml).toContain("fillColor=#dae8fc")
|
|
expect(out.xml).toContain("<mxGeometry")
|
|
expect(out.xml).not.toContain("<mxStyle")
|
|
})
|
|
|
|
it("reports a cut inside a compact cell that follows the definitions", () => {
|
|
const cut = `${BLUE}${A}<mxCell id="3" value="B" style="blue;" x="120" y="0" w="8`
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).toContain("cut off")
|
|
expect(out.error).toContain(
|
|
'<mxCell id="3" value="B" style="blue;" x="120" y="0" w="8',
|
|
)
|
|
expect(out.error).toContain("edit_diagram add")
|
|
})
|
|
|
|
it("reports output that stops right after the definitions and the first cells", () => {
|
|
const cut = `${BLUE}${A}<mxCell id="3" value="B" style="blue;"`
|
|
expect(isMxCellXmlComplete(cut)).toBe(false)
|
|
expect(truncatedCellError(cut)).toContain("cut off")
|
|
})
|
|
|
|
it("does not treat a compact cell's quote slip as a cut", () => {
|
|
const slip = `${A}<mxCell id="3" value="B" x="120" y="0" w="80" h="40/>`
|
|
expect(isMxCellXmlComplete(slip)).toBe(true)
|
|
expect(truncatedCellError(slip)).toBeNull()
|
|
const out = prepareNewDiagram(slip)
|
|
expect(out.ok).toBe(true)
|
|
if (out.ok) expect(out.xml).toContain('height="40"')
|
|
})
|
|
|
|
it("leaves definitions without any cell to the style check", () => {
|
|
expect(truncatedCellError(BLUE)).toBeNull()
|
|
const out = prepareNewDiagram(BLUE)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).not.toContain("cut off")
|
|
expect(out.error).toContain("no cells")
|
|
})
|
|
|
|
it("leaves an unclosed definition to the style check", () => {
|
|
const cut = `${BLUE}<mxStyle name="flow" value="edgeSt`
|
|
const out = prepareNewDiagram(cut)
|
|
expect(out.ok).toBe(false)
|
|
if (out.ok) return
|
|
expect(out.error).not.toContain("cut off")
|
|
expect(out.error).toContain("not closed")
|
|
})
|
|
})
|