upstream-tldraw/apps/examples/e2e-device/helpers/app.ts

import type { Editor } from 'tldraw'

declare const editor: Editor

const NATIVE_CONTEXT = 'NATIVE_APP'

/**
 * Open the `/end-to-end` example (which exposes `window.editor`), wait for the
 * canvas, and reset to a known state: no animations, zoom at 1, no shapes.
 */
export async function openEditor() {
	const webContext = await currentWebContext()
	await browser.switchContext(webContext)

	if (!(await browser.getUrl()).includes('end-to-end')) {
		await browser.url(`${browser.options.baseUrl}/end-to-end`)
	}
	await browser.$('.tl-canvas').waitForExist({ timeout: 30_000 })

	await browser.execute(() => {
		editor.user.updateUserPreferences({ animationSpeed: 0 })
		editor.selectAll().deleteShapes(editor.getSelectedShapeIds())
		editor.setCurrentTool('select')
		editor.resetZoom()
	})
}

/** Read the current zoom level from the live editor. */
export function getZoom(): Promise<number> {
	return browser.execute(() => editor.getZoomLevel())
}

/** Read the currently selected shape ids. */
export function getSelectedShapeIds(): Promise<string[]> {
	return browser.execute(() => editor.getSelectedShapeIds())
}

/** The webview context id changes per platform/run; find it rather than hardcode. */
export async function currentWebContext(): Promise<string> {
	const contexts = (await browser.getContexts()) as string[]
	const web = contexts.find((c) => c !== NATIVE_CONTEXT)
	if (!web) throw new Error(`No web context found. Available: ${contexts.join(', ')}`)
	return web
}