mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-12 04:29:51 +08:00
feat(ui): canvas-first redesign with a self-hosted draw.io (#957)
The canvas now fills the window and the chat floats on its right. draw.io is served from our own origin (public/drawio, downloaded at build time and checked against the release's SHA-256), so the app can drive the editor directly: - every AI change becomes a version card with a thumbnail; compare, restore or undo it, and Ctrl+Z on the canvas takes an AI change back in one step - shapes the AI just changed are highlighted on the canvas - selected shapes appear as a chip and are sent to the model - draw.io's own simple UI: its toolbar, menus (auto layout, Mermaid, CSV import) and page tabs; the shape library starts closed - new start screen, session rename and switcher, open .drawio files - settings split into Models & keys, Appearance, Advanced and About - Bedrock in the model settings also takes an API key instead of access keys - phones switch between canvas and chat - dark mode uses draw.io's own dark mode, with neutral greys around it With an external draw.io (NEXT_PUBLIC_DRAWIO_BASE_URL) the features above that need same origin are off. The chat logic moved from the 1500-line chat panel into a chat engine plus small UI components; zustand holds canvas, versions, settings and UI state.
This commit is contained in:
@@ -30,7 +30,7 @@ export { expect }
|
||||
|
||||
/** Get the chat input textarea */
|
||||
export function getChatInput(page: Page) {
|
||||
return page.locator('textarea[aria-label="Chat input"]')
|
||||
return page.getByTestId("chat-input")
|
||||
}
|
||||
|
||||
/** Get the draw.io iframe */
|
||||
@@ -43,6 +43,18 @@ export function getIframeContent(page: Page) {
|
||||
return page.frameLocator("iframe")
|
||||
}
|
||||
|
||||
/** Open a .drawio file through the app's file picker */
|
||||
export async function openDrawioFile(page: Page, name: string, xml: string) {
|
||||
await page
|
||||
.locator('input[type="file"][accept^=".drawio"]')
|
||||
.first()
|
||||
.setInputFiles({
|
||||
name,
|
||||
mimeType: "application/xml",
|
||||
buffer: Buffer.from(xml),
|
||||
})
|
||||
}
|
||||
|
||||
/** Get the settings button */
|
||||
export function getSettingsButton(page: Page) {
|
||||
return page.locator('[data-testid="settings-button"]')
|
||||
@@ -60,20 +72,33 @@ export async function sendMessage(page: Page, message: string) {
|
||||
await chatInput.press("ControlOrMeta+Enter")
|
||||
}
|
||||
|
||||
/** Wait for diagram generation to complete */
|
||||
export async function waitForComplete(page: Page, timeout = 15000) {
|
||||
await expect(page.locator('text="Complete"')).toBeVisible({ timeout })
|
||||
/** Tool calls that finished (version cards and tool rows) */
|
||||
export function getCompletedTools(page: Page) {
|
||||
return page.locator('[data-tool-state="output-available"]')
|
||||
}
|
||||
|
||||
/** Wait for N "Complete" badges */
|
||||
/** Wait for diagram generation to complete */
|
||||
export async function waitForComplete(page: Page, timeout = 15000) {
|
||||
await expect(getCompletedTools(page).first()).toBeVisible({ timeout })
|
||||
}
|
||||
|
||||
/** Wait for N finished tool calls */
|
||||
export async function waitForCompleteCount(
|
||||
page: Page,
|
||||
count: number,
|
||||
timeout = 15000,
|
||||
) {
|
||||
await expect(page.locator('text="Complete"')).toHaveCount(count, {
|
||||
timeout,
|
||||
})
|
||||
await expect(getCompletedTools(page)).toHaveCount(count, { timeout })
|
||||
}
|
||||
|
||||
/** The file input behind the message box's "Add" menu */
|
||||
export function getAttachmentInput(page: Page) {
|
||||
return page.locator('[data-testid="attachment-input"]')
|
||||
}
|
||||
|
||||
/** The send button (icon only; found by its accessible name) */
|
||||
export function getSendButton(page: Page, name = "Send") {
|
||||
return page.getByRole("button", { name, exact: true })
|
||||
}
|
||||
|
||||
/** Wait for a specific text to appear */
|
||||
@@ -83,10 +108,19 @@ export async function waitForText(page: Page, text: string, timeout = 15000) {
|
||||
|
||||
/** Open settings dialog */
|
||||
export async function openSettings(page: Page) {
|
||||
await getSettingsButton(page).click()
|
||||
await getSettingsButton(page).first().click()
|
||||
await expect(page.locator('[role="dialog"]')).toBeVisible({ timeout: 5000 })
|
||||
}
|
||||
|
||||
/** Open settings on one of its pages: models, appearance, advanced, about */
|
||||
export async function openSettingsTab(
|
||||
page: Page,
|
||||
tab: "models" | "appearance" | "advanced" | "about",
|
||||
) {
|
||||
await openSettings(page)
|
||||
await page.locator(`[data-testid="settings-tab-${tab}"]`).click()
|
||||
}
|
||||
|
||||
// ============================================
|
||||
// Mock helpers
|
||||
// ============================================
|
||||
|
||||
Reference in New Issue
Block a user