orca/tests/e2e/helpers/terminal.ts

import type { Page } from '@stablyai/playwright-test'
import { expect } from '@stablyai/playwright-test'
import {
  getTerminalContent,
  readPaneIdentitySnapshot,
  resolveActiveTabId,
  type ActivePaneHookDescriptor,
  type PaneIdentitySnapshot
} from './terminal-pane-identity'
import {
  discoverActivePtyId as discoverActivePtyIdImpl,
  waitForActivePaneHookDescriptor as waitForActivePaneHookDescriptorImpl
} from './terminal-active-pane'
import {
  closeActiveTerminalPane as closeActiveTerminalPaneImpl,
  countVisibleTerminalPanes as countVisibleTerminalPanesImpl,
  execInTerminal as execInTerminalImpl,
  focusLastTerminalPane as focusLastTerminalPaneImpl,
  moveTerminalPaneByLeafId as moveTerminalPaneByLeafIdImpl,
  readTerminalPaneDomLeafOrder as readTerminalPaneDomLeafOrderImpl,
  sendToTerminal as sendToTerminalImpl,
  splitActiveTerminalPane as splitActiveTerminalPaneImpl,
  waitForActiveTerminalManager as waitForActiveTerminalManagerImpl,
  waitForPaneCount as waitForPaneCountImpl,
  waitForTerminalOutput as waitForTerminalOutputImpl
} from './terminal-pane-operations'

export {
  expectTerminalAccessibilityText,
  readTerminalAccessibilityText
} from './terminal-accessibility-tree'

export {
  getTerminalContent,
  readPaneIdentitySnapshot,
  resolveActiveTabId,
  type ActivePaneHookDescriptor,
  type PaneIdentitySnapshot
}

export const UUID_RE = /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/

// Why: typing-latency specs must type into xterm's helper textarea, not the
// page body — keyboard.type only reaches the PTY when that textarea has focus.
export async function focusActiveTerminalInput(page: Page): Promise<void> {
  await page.evaluate(() => {
    const state = window.__store?.getState()
    const worktreeId = state?.activeWorktreeId
    const tabId =
      state?.activeTabType === 'terminal'
        ? state.activeTabId
        : worktreeId
          ? (state?.activeTabIdByWorktree?.[worktreeId] ?? null)
          : null
    const manager = tabId ? window.__paneManagers?.get(tabId) : null
    const pane = manager?.getActivePane?.() ?? manager?.getPanes?.()[0] ?? null
    if (!state || !tabId || !pane) {
      throw new Error('No active terminal pane to focus')
    }
    state.setActiveTab(tabId)
    state.setActiveTabType('terminal')
    pane.terminal.focus()
    const textarea = pane.container.querySelector(
      '.xterm-helper-textarea'
    ) as HTMLTextAreaElement | null
    if (!textarea) {
      throw new Error('Active terminal has no xterm helper textarea')
    }
    textarea.focus()
  })
}

export async function waitForActivePanePtyId(page: Page, timeoutMs = 15_000): Promise<string> {
  let resolvedPtyId: string | null = null
  await expect
    .poll(
      async () => {
        const tabId = await resolveActiveTabId(page)
        if (!tabId) {
          return null
        }

        resolvedPtyId = await page.evaluate((tabId) => {
          const manager = window.__paneManagers?.get(tabId)
          const activePane = manager?.getActivePane?.() ?? manager?.getPanes?.()[0] ?? null
          return activePane?.container?.dataset?.ptyId ?? null
        }, tabId)
        return resolvedPtyId
      },
      {
        timeout: timeoutMs,
        message: 'Active terminal pane did not receive a PTY binding'
      }
    )
    .not.toBeNull()

  if (!resolvedPtyId) {
    throw new Error('waitForActivePanePtyId: active pane has no PTY binding')
  }
  return resolvedPtyId
}

export async function waitForPaneIdentitySnapshot(
  page: Page,
  paneCount: number
): Promise<PaneIdentitySnapshot> {
  let latestSnapshot: PaneIdentitySnapshot | null = null
  try {
    await expect
      .poll(
        async () => {
          latestSnapshot = await readPaneIdentitySnapshot(page)
          return Boolean(
            latestSnapshot &&
            latestSnapshot.panes.length === paneCount &&
            latestSnapshot.panes.every(
              (pane) =>
                UUID_RE.test(pane.leafId) &&
                pane.stablePaneId === pane.leafId &&
                pane.datasetLeafId === pane.leafId &&
                pane.ptyId !== null &&
                latestSnapshot?.ptyIdsByLeafId[pane.leafId] === pane.ptyId
            )
          )
        },
        {
          timeout: 15_000,
          message: 'Split terminal panes did not settle with UUID leaf-keyed PTY bindings'
        }
      )
      .toBe(true)
  } catch (error) {
    throw new Error(`Last pane identity snapshot: ${JSON.stringify(latestSnapshot)}`, {
      cause: error
    })
  }

  const snapshot = await readPaneIdentitySnapshot(page)
  if (!snapshot) {
    throw new Error('Pane identity snapshot disappeared after settling')
  }
  return snapshot
}

export function waitForActivePaneHookDescriptor(
  page: Page,
  timeoutMs = 15_000
): Promise<ActivePaneHookDescriptor> {
  return waitForActivePaneHookDescriptorImpl(page, timeoutMs)
}

export function discoverActivePtyId(page: Page): Promise<string> {
  return discoverActivePtyIdImpl(page)
}

export function readTerminalPaneDomLeafOrder(page: Page): Promise<string[]> {
  return readTerminalPaneDomLeafOrderImpl(page)
}

export function moveTerminalPaneByLeafId(
  page: Page,
  sourceLeafId: string,
  targetLeafId: string,
  zone: 'top' | 'bottom' | 'left' | 'right'
): Promise<void> {
  return moveTerminalPaneByLeafIdImpl(page, sourceLeafId, targetLeafId, zone)
}

export function sendToTerminal(page: Page, ptyId: string, text: string): Promise<void> {
  return sendToTerminalImpl(page, ptyId, text)
}

export function execInTerminal(page: Page, ptyId: string, command: string): Promise<void> {
  return execInTerminalImpl(page, ptyId, command)
}

export function waitForActiveTerminalManager(page: Page, timeoutMs = 30_000): Promise<void> {
  return waitForActiveTerminalManagerImpl(page, timeoutMs)
}

export function splitActiveTerminalPane(
  page: Page,
  direction: 'vertical' | 'horizontal'
): Promise<void> {
  return splitActiveTerminalPaneImpl(page, direction)
}

export function closeActiveTerminalPane(page: Page): Promise<void> {
  return closeActiveTerminalPaneImpl(page)
}

export function focusLastTerminalPane(page: Page): Promise<void> {
  return focusLastTerminalPaneImpl(page)
}

export function countVisibleTerminalPanes(page: Page): Promise<number> {
  return countVisibleTerminalPanesImpl(page)
}

export function waitForTerminalOutput(
  page: Page,
  expected: string,
  timeoutMs = 10_000,
  charLimit = 4000
): Promise<void> {
  return waitForTerminalOutputImpl(page, expected, timeoutMs, charLimit)
}

export function waitForPaneCount(
  page: Page,
  expectedCount: number,
  timeoutMs = 10_000
): Promise<void> {
  return waitForPaneCountImpl(page, expectedCount, timeoutMs)
}