stablyai--orca/tools/win-update-e2e/window-watch.mjs

// Node wrapper around window-watch.ps1: start/stop the background window watch
// and parse its JSONL event log. Also runnable standalone as `--selftest`,
// which opens a real transient PowerShell window and asserts the watch caught
// it — so the harness's own instrument is testable without any installers.

import { existsSync, mkdtempSync, readFileSync, writeFileSync, rmSync } from 'node:fs'
import { tmpdir } from 'node:os'
import path from 'node:path'
import { assertWin32 } from './platform-guard.mjs'
import { runScriptFileJson, spawnScriptFile, runCommandSync } from './powershell-runner.mjs'

const HERE = import.meta.dirname
const WATCH_SCRIPT = path.join(HERE, 'window-watch.ps1')
const ENUM_SCRIPT = path.join(HERE, 'window-enum.ps1')

/**
 * Take a one-shot baseline snapshot of visible top-level windows and write it
 * to baselinePath. Returns the window array. Normalizes the PS 5.1 single-item
 * unwrap (windows may arrive as one object or as a nested array).
 */
export function captureBaseline(baselinePath) {
  assertWin32('window-watch baseline')
  const parsed = runScriptFileJson(ENUM_SCRIPT)
  const windows = normalizeWindows(parsed.windows)
  writeFileSync(baselinePath, JSON.stringify({ windows }, null, 2))
  return windows
}

function normalizeWindows(raw) {
  if (!raw) {
    return []
  }
  // PS 5.1 ConvertTo-Json can hand back a single object, a flat array, or (from
  // a stray comma operator) a one-element array wrapping the real array.
  if (Array.isArray(raw)) {
    if (raw.length === 1 && Array.isArray(raw[0])) {
      return raw[0]
    }
    return raw
  }
  return [raw]
}

/**
 * Start the background watch. Returns a handle with stop() -> events array.
 * The watch seeds from baselinePath and records new windows to outPath until
 * stop() (which drops a stop-file) or durationSec elapses.
 */
export function startWatch({ baselinePath, outPath, stopFile, durationSec = 600, pollMs = 500 }) {
  assertWin32('window-watch')
  const resolvedStopFile = stopFile ?? `${outPath}.stop`
  if (existsSync(resolvedStopFile)) {
    rmSync(resolvedStopFile)
  }

  const child = spawnScriptFile(WATCH_SCRIPT, [
    '-BaselinePath',
    baselinePath,
    '-OutPath',
    outPath,
    '-StopFile',
    resolvedStopFile,
    '-DurationSec',
    String(durationSec),
    '-PollMs',
    String(pollMs),
    '-EnumScript',
    ENUM_SCRIPT
  ])

  let stderr = ''
  child.stderr.on('data', (d) => {
    stderr += d.toString()
  })

  const exited = new Promise((resolve) => child.once('exit', resolve))

  return {
    process: child,
    stopFile: resolvedStopFile,
    outPath,
    async stop() {
      // Signal the loop to end, then wait for it to flush and exit. Fall back
      // to a hard kill if the loop is wedged so a stuck watch never hangs teardown.
      writeFileSync(resolvedStopFile, 'stop')
      const timer = setTimeout(() => child.kill(), 5000)
      await exited
      clearTimeout(timer)
      return { events: readEvents(outPath), stderr }
    }
  }
}

/** Parse a window-watch JSONL log into an array of event objects. */
export function readEvents(outPath) {
  if (!existsSync(outPath)) {
    return []
  }
  return readFileSync(outPath, 'utf8')
    .split('\n')
    .map((l) => l.trim())
    .filter(Boolean)
    .flatMap((line) => {
      try {
        return [JSON.parse(line)]
      } catch {
        return []
      }
    })
}

async function selftest() {
  assertWin32('window-watch --selftest')
  const dir = mkdtempSync(path.join(tmpdir(), 'orca-winwatch-selftest-'))
  const baselinePath = path.join(dir, 'baseline.json')
  const outPath = path.join(dir, 'events.jsonl')
  const canary = `ORCA-E2E-SELFTEST-${Date.now()}`

  console.log(`[selftest] baseline snapshot -> ${baselinePath}`)
  const baseline = captureBaseline(baselinePath)
  console.log(`[selftest] baseline captured ${baseline.length} visible windows`)

  const watch = startWatch({ baselinePath, outPath, durationSec: 20, pollMs: 300 })
  // Give the watch a moment to compile Add-Type and take its first poll before
  // the transient window appears, so the appearance is genuinely "new".
  await delay(1500)

  console.log(`[selftest] opening transient window titled ${canary}`)
  // Must be a real console window: Start-Process allocates one, whereas a
  // detached Node spawn gets DETACHED_PROCESS (no console at all) and would
  // never appear in enumeration. This mirrors how a daemon child that lacks
  // CREATE_NO_WINDOW allocates a fresh visible console — exactly the flash the
  // real harness must catch.
  const transientScript = path.join(dir, 'transient.ps1')
  writeFileSync(transientScript, `$host.UI.RawUI.WindowTitle='${canary}'; Start-Sleep -Seconds 5`)
  runCommandSync(
    `Start-Process -FilePath 'powershell.exe' -ArgumentList '-NoProfile','-File','${transientScript}'`
  )

  // Poll cycles at 300ms; 3.5s covers several polls while the window is alive.
  await delay(3500)
  const { events, stderr } = await watch.stop()
  rmSync(dir, { recursive: true, force: true })

  const caught = events.filter((e) => typeof e.title === 'string' && e.title.includes(canary))
  console.log(`[selftest] watch recorded ${events.length} new windows total`)
  if (stderr.trim()) {
    console.log(`[selftest] watch stderr:\n${stderr.trim()}`)
  }
  if (caught.length === 0) {
    console.error(
      `[selftest] FAIL: watch did not capture a window titled "${canary}". ` +
        `New windows seen: ${JSON.stringify(events.map((e) => e.title))}`
    )
    process.exitCode = 1
    return
  }
  console.log(`[selftest] PASS: caught canary window: ${JSON.stringify(caught[0])}`)
}

function delay(ms) {
  return new Promise((resolve) => setTimeout(resolve, ms))
}

if (process.argv[1] && path.resolve(process.argv[1]) === import.meta.filename) {
  if (process.argv.includes('--selftest')) {
    selftest().catch((err) => {
      console.error(err)
      process.exitCode = 1
    })
  } else {
    console.log('Usage: node window-watch.mjs --selftest')
  }
}