import type { Page } from '@playwright/test'

export async function measureHitTesting(page: Page, run: () => Promise<void>) {
  const probe = await page.evaluateHandle(() => {
    const graph = window.openPencil?.getStore?.().graph
    if (!graph) throw new Error('OpenPencil graph not initialized')
    const original = graph.hitTest
    let calls = 0
    let totalMs = 0
    graph.hitTest = (...args) => {
      const start = performance.now()
      const result = original.apply(graph, args)
      totalMs += performance.now() - start
      calls++
      return result
    }
    return {
      read: () => ({ calls, totalMs }),
      dispose: () => {
        graph.hitTest = original
      }
    }
  })
  try {
    await run()
    return await probe.evaluate((probe) => probe.read())
  } finally {
    try {
      await probe.evaluate((probe) => probe.dispose())
    } finally {
      await probe.dispose()
    }
  }
}