perf: cache successful device readiness checks

Author: thymikee

src/daemon/__tests__/device-ready.test.ts

@@ -1,6 +1,77 @@
-import { test } from 'vitest';
+import { afterEach, beforeEach, expect, test, vi } from 'vitest';
 import assert from 'node:assert/strict';
-import { parseIosReadyPayload, resolveIosReadyHint } from '../device-ready.ts';
+import type { DeviceInfo } from '../../utils/device.ts';
+
+vi.mock('../../platforms/ios/index.ts', () => ({
+  ensureBootedSimulator: vi.fn(async () => {}),
+}));
+vi.mock('../../platforms/android/devices.ts', () => ({
+  waitForAndroidBoot: vi.fn(async () => {}),
+}));
+
+import { waitForAndroidBoot } from '../../platforms/android/devices.ts';
+import { ensureBootedSimulator } from '../../platforms/ios/index.ts';
+import { ANDROID_EMULATOR, IOS_SIMULATOR } from '../../__tests__/test-utils/index.ts';
+import {
+  clearDeviceReadyCacheForTests,
+  DEVICE_READY_CACHE_TTL_MS,
+  ensureDeviceReady,
+  parseIosReadyPayload,
+  resolveIosReadyHint,
+} from '../device-ready.ts';
+
+const mockEnsureBootedSimulator = vi.mocked(ensureBootedSimulator);
+const mockWaitForAndroidBoot = vi.mocked(waitForAndroidBoot);
+
+beforeEach(() => {
+  vi.useFakeTimers();
+  vi.setSystemTime(new Date('2026-04-01T10:00:00.000Z'));
+  clearDeviceReadyCacheForTests();
+  mockEnsureBootedSimulator.mockReset();
+  mockEnsureBootedSimulator.mockResolvedValue(undefined);
+  mockWaitForAndroidBoot.mockReset();
+  mockWaitForAndroidBoot.mockResolvedValue(undefined);
+});
+
+afterEach(() => {
+  clearDeviceReadyCacheForTests();
+  vi.useRealTimers();
+});
+
+test('ensureDeviceReady caches successful simulator readiness checks', async () => {
+  const device: DeviceInfo = { ...IOS_SIMULATOR, simulatorSetPath: '/tmp/simset-a' };
+
+  await ensureDeviceReady(device);
+  await ensureDeviceReady({ ...device });
+
+  expect(mockEnsureBootedSimulator).toHaveBeenCalledTimes(1);
+});
+
+test('ensureDeviceReady includes simulator set path in the cache key', async () => {
+  await ensureDeviceReady({ ...IOS_SIMULATOR, simulatorSetPath: '/tmp/simset-a' });
+  await ensureDeviceReady({ ...IOS_SIMULATOR, simulatorSetPath: '/tmp/simset-b' });
+
+  expect(mockEnsureBootedSimulator).toHaveBeenCalledTimes(2);
+});
+
+test('ensureDeviceReady expires cached readiness checks after the ttl', async () => {
+  await ensureDeviceReady(ANDROID_EMULATOR);
+  vi.setSystemTime(new Date(Date.now() + DEVICE_READY_CACHE_TTL_MS - 1));
+  await ensureDeviceReady({ ...ANDROID_EMULATOR });
+  vi.setSystemTime(new Date(Date.now() + 1));
+  await ensureDeviceReady({ ...ANDROID_EMULATOR });
+
+  expect(mockWaitForAndroidBoot).toHaveBeenCalledTimes(2);
+});
+
+test('ensureDeviceReady does not cache failed readiness checks', async () => {
+  mockEnsureBootedSimulator.mockRejectedValueOnce(new Error('boot failed'));
+
+  await expect(ensureDeviceReady(IOS_SIMULATOR)).rejects.toThrow('boot failed');
+  await ensureDeviceReady(IOS_SIMULATOR);
+
+  expect(mockEnsureBootedSimulator).toHaveBeenCalledTimes(2);
+});
 
 test('parseIosReadyPayload reads tunnelState from direct connectionProperties', () => {
   const parsed = parseIosReadyPayload({

src/daemon/device-ready.ts

@@ -13,25 +13,59 @@ const IOS_DEVICE_READY_TIMEOUT_MS = resolveTimeoutMs(
   1_000,
 );
 const IOS_DEVICE_READY_COMMAND_TIMEOUT_BUFFER_MS = 3_000;
+export const DEVICE_READY_CACHE_TTL_MS = 5_000;
+
+const readyCache = new Map<string, number>();
 
 export async function ensureDeviceReady(device: DeviceInfo): Promise<void> {
+  const cacheKey = deviceReadyCacheKey(device);
+  const cachedUntil = readyCache.get(cacheKey);
+  if (cachedUntil !== undefined) {
+    if (cachedUntil > Date.now()) {
+      return;
+    }
+    readyCache.delete(cacheKey);
+  }
+
   if (device.platform === 'ios') {
     if (device.kind === 'simulator') {
       const { ensureBootedSimulator } = await import('../platforms/ios/index.ts');
       await ensureBootedSimulator(device);
+      markDeviceReady(cacheKey);
       return;
     }
     if (device.kind === 'device') {
       await ensureIosDeviceReady(device.id);
+      markDeviceReady(cacheKey);
       return;
     }
   }
   if (device.platform === 'android') {
     const { waitForAndroidBoot } = await import('../platforms/android/devices.ts');
     await waitForAndroidBoot(device.id);
+    markDeviceReady(cacheKey);
   }
 }
 
+export function clearDeviceReadyCacheForTests(): void {
+  readyCache.clear();
+}
+
+function markDeviceReady(cacheKey: string): void {
+  readyCache.set(cacheKey, Date.now() + DEVICE_READY_CACHE_TTL_MS);
+}
+
+function deviceReadyCacheKey(device: DeviceInfo): string {
+  const simulatorSetPath = device.kind === 'simulator' ? (device.simulatorSetPath ?? '') : '';
+  return JSON.stringify([
+    device.platform,
+    device.kind,
+    device.id,
+    device.target ?? '',
+    simulatorSetPath,
+  ]);
+}
+
 async function ensureIosDeviceReady(deviceId: string): Promise<void> {
   const jsonPath = path.join(
     os.tmpdir(),