agent-device 0.4.1 → 0.5.0

package/src/daemon.ts

@@ -1,295 +0,0 @@
-import net from 'node:net';
-import fs from 'node:fs';
-import os from 'node:os';
-import path from 'node:path';
-import crypto from 'node:crypto';
-import { dispatchCommand, type CommandFlags } from './core/dispatch.ts';
-import { isCommandSupportedOnDevice } from './core/capabilities.ts';
-import { asAppError, AppError } from './utils/errors.ts';
-import { readVersion } from './utils/version.ts';
-import { stopAllIosRunnerSessions } from './platforms/ios/runner-client.ts';
-import type { DaemonRequest, DaemonResponse } from './daemon/types.ts';
-import { SessionStore } from './daemon/session-store.ts';
-import { contextFromFlags as contextFromFlagsWithLog, type DaemonCommandContext } from './daemon/context.ts';
-import { handleSessionCommands } from './daemon/handlers/session.ts';
-import { handleSnapshotCommands } from './daemon/handlers/snapshot.ts';
-import { handleFindCommands } from './daemon/handlers/find.ts';
-import { handleRecordTraceCommands } from './daemon/handlers/record-trace.ts';
-import { handleInteractionCommands } from './daemon/handlers/interaction.ts';
-import { assertSessionSelectorMatches } from './daemon/session-selector.ts';
-import { resolveEffectiveSessionName } from './daemon/session-routing.ts';
-import {
-  isAgentDeviceDaemonProcess,
-  readProcessStartTime,
-} from './utils/process-identity.ts';
-
-const baseDir = path.join(os.homedir(), '.agent-device');
-const infoPath = path.join(baseDir, 'daemon.json');
-const lockPath = path.join(baseDir, 'daemon.lock');
-const logPath = path.join(baseDir, 'daemon.log');
-const sessionsDir = path.join(baseDir, 'sessions');
-const sessionStore = new SessionStore(sessionsDir);
-const version = readVersion();
-const token = crypto.randomBytes(24).toString('hex');
-const selectorValidationExemptCommands = new Set(['session_list', 'devices']);
-
-type DaemonLockInfo = {
-  pid: number;
-  version: string;
-  startedAt: number;
-  processStartTime?: string;
-};
-
-const daemonProcessStartTime = readProcessStartTime(process.pid) ?? undefined;
-
-function contextFromFlags(
-  flags: CommandFlags | undefined,
-  appBundleId?: string,
-  traceLogPath?: string,
-): DaemonCommandContext {
-  return contextFromFlagsWithLog(logPath, flags, appBundleId, traceLogPath);
-}
-
-async function handleRequest(req: DaemonRequest): Promise<DaemonResponse> {
-  if (req.token !== token) {
-    return { ok: false, error: { code: 'UNAUTHORIZED', message: 'Invalid token' } };
-  }
-
-  const command = req.command;
-  const sessionName = resolveEffectiveSessionName(req, sessionStore);
-  const existingSession = sessionStore.get(sessionName);
-  if (existingSession && !selectorValidationExemptCommands.has(command)) {
-    assertSessionSelectorMatches(existingSession, req.flags);
-  }
-
-  const sessionResponse = await handleSessionCommands({
-    req,
-    sessionName,
-    logPath,
-    sessionStore,
-    invoke: handleRequest,
-  });
-  if (sessionResponse) return sessionResponse;
-
-  const snapshotResponse = await handleSnapshotCommands({
-    req,
-    sessionName,
-    logPath,
-    sessionStore,
-  });
-  if (snapshotResponse) return snapshotResponse;
-
-  const recordTraceResponse = await handleRecordTraceCommands({
-    req,
-    sessionName,
-    sessionStore,
-  });
-  if (recordTraceResponse) return recordTraceResponse;
-
-  const findResponse = await handleFindCommands({
-    req,
-    sessionName,
-    logPath,
-    sessionStore,
-    invoke: handleRequest,
-  });
-  if (findResponse) return findResponse;
-
-  const interactionResponse = await handleInteractionCommands({
-    req,
-    sessionName,
-    sessionStore,
-    contextFromFlags,
-  });
-  if (interactionResponse) return interactionResponse;
-
-
-  const session = sessionStore.get(sessionName);
-  if (!session) {
-    return {
-      ok: false,
-      error: { code: 'SESSION_NOT_FOUND', message: 'No active session. Run open first.' },
-    };
-  }
-
-  if (!isCommandSupportedOnDevice(command, session.device)) {
-    return {
-      ok: false,
-      error: { code: 'UNSUPPORTED_OPERATION', message: `${command} is not supported on this device` },
-    };
-  }
-
-  const data = await dispatchCommand(session.device, command, req.positionals ?? [], req.flags?.out, {
-    ...contextFromFlags(req.flags, session.appBundleId, session.trace?.outPath),
-  });
-  sessionStore.recordAction(session, {
-    command,
-    positionals: req.positionals ?? [],
-    flags: req.flags ?? {},
-    result: data ?? {},
-  });
-  return { ok: true, data: data ?? {} };
-}
-
-function writeInfo(port: number): void {
-  if (!fs.existsSync(baseDir)) fs.mkdirSync(baseDir, { recursive: true });
-  fs.writeFileSync(logPath, '');
-  fs.writeFileSync(
-    infoPath,
-    JSON.stringify({ port, token, pid: process.pid, version, processStartTime: daemonProcessStartTime }, null, 2),
-    {
-      mode: 0o600,
-    },
-  );
-}
-
-function removeInfo(): void {
-  if (fs.existsSync(infoPath)) fs.unlinkSync(infoPath);
-}
-
-function readLockInfo(): DaemonLockInfo | null {
-  if (!fs.existsSync(lockPath)) return null;
-  try {
-    const parsed = JSON.parse(fs.readFileSync(lockPath, 'utf8')) as DaemonLockInfo;
-    if (!Number.isInteger(parsed.pid) || parsed.pid <= 0) return null;
-    return parsed;
-  } catch {
-    return null;
-  }
-}
-
-function acquireDaemonLock(): boolean {
-  if (!fs.existsSync(baseDir)) fs.mkdirSync(baseDir, { recursive: true });
-  const lockData: DaemonLockInfo = {
-    pid: process.pid,
-    version,
-    startedAt: Date.now(),
-    processStartTime: daemonProcessStartTime,
-  };
-  const payload = JSON.stringify(lockData, null, 2);
-
-  const tryWriteLock = (): boolean => {
-    try {
-      fs.writeFileSync(lockPath, payload, { flag: 'wx', mode: 0o600 });
-      return true;
-    } catch (err) {
-      if ((err as NodeJS.ErrnoException).code === 'EEXIST') return false;
-      throw err;
-    }
-  };
-
-  if (tryWriteLock()) return true;
-  const existing = readLockInfo();
-  if (
-    existing?.pid
-    && existing.pid !== process.pid
-    && isAgentDeviceDaemonProcess(existing.pid, existing.processStartTime)
-  ) {
-    return false;
-  }
-  // Best-effort stale-lock cleanup: another process may win the race between unlink and re-create.
-  // We rely on the subsequent write with `wx` to enforce single-writer semantics.
-  try {
-    fs.unlinkSync(lockPath);
-  } catch {
-    // ignore
-  }
-  return tryWriteLock();
-}
-
-function releaseDaemonLock(): void {
-  const existing = readLockInfo();
-  if (existing && existing.pid !== process.pid) return;
-  try {
-    if (fs.existsSync(lockPath)) fs.unlinkSync(lockPath);
-  } catch {
-    // ignore
-  }
-}
-
-function start(): void {
-  if (!acquireDaemonLock()) {
-    process.stderr.write('Daemon lock is held by another process; exiting.\n');
-    process.exit(0);
-    return;
-  }
-
-  const server = net.createServer((socket) => {
-    let buffer = '';
-    socket.setEncoding('utf8');
-    socket.on('data', async (chunk) => {
-      buffer += chunk;
-      let idx = buffer.indexOf('\n');
-      while (idx !== -1) {
-        const line = buffer.slice(0, idx).trim();
-        buffer = buffer.slice(idx + 1);
-        if (line.length === 0) {
-          idx = buffer.indexOf('\n');
-          continue;
-        }
-        let response: DaemonResponse;
-        try {
-          const req = JSON.parse(line) as DaemonRequest;
-          response = await handleRequest(req);
-        } catch (err) {
-          const appErr = asAppError(err);
-          response = {
-            ok: false,
-            error: { code: appErr.code, message: appErr.message, details: appErr.details },
-          };
-        }
-        socket.write(`${JSON.stringify(response)}\n`);
-        idx = buffer.indexOf('\n');
-      }
-    });
-  });
-
-  server.listen(0, '127.0.0.1', () => {
-    const address = server.address();
-    if (typeof address === 'object' && address?.port) {
-      writeInfo(address.port);
-      process.stdout.write(`AGENT_DEVICE_DAEMON_PORT=${address.port}\n`);
-    }
-  });
-
-  let shuttingDown = false;
-  const closeServer = async (): Promise<void> => {
-    await new Promise<void>((resolve) => {
-      try {
-        server.close(() => resolve());
-      } catch {
-        resolve();
-      }
-    });
-  };
-  const shutdown = async () => {
-    if (shuttingDown) return;
-    shuttingDown = true;
-    await closeServer();
-    const sessionsToStop = sessionStore.toArray();
-    for (const session of sessionsToStop) {
-      sessionStore.writeSessionLog(session);
-    }
-    await stopAllIosRunnerSessions();
-    removeInfo();
-    releaseDaemonLock();
-    process.exit(0);
-  };
-
-  process.on('SIGINT', () => {
-    void shutdown();
-  });
-  process.on('SIGTERM', () => {
-    void shutdown();
-  });
-  process.on('SIGHUP', () => {
-    void shutdown();
-  });
-  process.on('uncaughtException', (err) => {
-    const appErr = err instanceof AppError ? err : asAppError(err);
-    process.stderr.write(`Daemon error: ${appErr.message}\n`);
-    void shutdown();
-  });
-}
-
-start();

package/src/platforms/__tests__/boot-diagnostics.test.ts

@@ -1,59 +0,0 @@
-import test from 'node:test';
-import assert from 'node:assert/strict';
-import { bootFailureHint, classifyBootFailure } from '../boot-diagnostics.ts';
-import { AppError } from '../../utils/errors.ts';
-
-test('classifyBootFailure maps timeout errors', () => {
-  const reason = classifyBootFailure({
-    message: 'bootstatus timed out after 120s',
-    context: { platform: 'ios', phase: 'boot' },
-  });
-  assert.equal(reason, 'IOS_BOOT_TIMEOUT');
-});
-
-test('classifyBootFailure maps adb offline errors', () => {
-  const reason = classifyBootFailure({
-    stderr: 'error: device offline',
-    context: { platform: 'android', phase: 'transport' },
-  });
-  assert.equal(reason, 'ADB_TRANSPORT_UNAVAILABLE');
-});
-
-test('classifyBootFailure maps tool missing from AppError code (android)', () => {
-  const reason = classifyBootFailure({
-    error: new AppError('TOOL_MISSING', 'adb not found in PATH'),
-    context: { platform: 'android', phase: 'transport' },
-  });
-  assert.equal(reason, 'ADB_TRANSPORT_UNAVAILABLE');
-});
-
-test('classifyBootFailure maps tool missing from AppError code (ios)', () => {
-  const reason = classifyBootFailure({
-    error: new AppError('TOOL_MISSING', 'xcrun not found in PATH'),
-    context: { platform: 'ios', phase: 'boot' },
-  });
-  assert.equal(reason, 'IOS_TOOL_MISSING');
-});
-
-test('classifyBootFailure reads stderr from AppError details', () => {
-  const reason = classifyBootFailure({
-    error: new AppError('COMMAND_FAILED', 'adb failed', {
-      stderr: 'error: device unauthorized',
-    }),
-    context: { platform: 'android', phase: 'transport' },
-  });
-  assert.equal(reason, 'ADB_TRANSPORT_UNAVAILABLE');
-});
-
-test('bootFailureHint returns actionable guidance', () => {
-  const hint = bootFailureHint('IOS_RUNNER_CONNECT_TIMEOUT');
-  assert.equal(hint.includes('xcodebuild logs'), true);
-});
-
-test('connect phase does not classify non-timeout errors as connect timeout', () => {
-  const reason = classifyBootFailure({
-    message: 'Runner returned malformed JSON payload',
-    context: { platform: 'ios', phase: 'connect' },
-  });
-  assert.equal(reason, 'BOOT_COMMAND_FAILED');
-});

package/src/platforms/android/__tests__/index.test.ts

@@ -1,157 +0,0 @@
-import test from 'node:test';
-import assert from 'node:assert/strict';
-import { promises as fs } from 'node:fs';
-import os from 'node:os';
-import path from 'node:path';
-import { openAndroidApp, parseAndroidLaunchComponent, swipeAndroid } from '../index.ts';
-import type { DeviceInfo } from '../../../utils/device.ts';
-import { AppError } from '../../../utils/errors.ts';
-import { findBounds, parseUiHierarchy } from '../ui-hierarchy.ts';
-
-test('parseUiHierarchy reads double-quoted Android node attributes', () => {
-  const xml =
-    '<hierarchy><node class="android.widget.TextView" text="Hello" content-desc="Greeting" resource-id="com.demo:id/title" bounds="[10,20][110,60]" clickable="true" enabled="true"/></hierarchy>';
-
-  const result = parseUiHierarchy(xml, 800, { raw: true });
-  assert.equal(result.nodes.length, 1);
-  assert.equal(result.nodes[0].value, 'Hello');
-  assert.equal(result.nodes[0].label, 'Hello');
-  assert.equal(result.nodes[0].identifier, 'com.demo:id/title');
-  assert.deepEqual(result.nodes[0].rect, { x: 10, y: 20, width: 100, height: 40 });
-  assert.equal(result.nodes[0].hittable, true);
-  assert.equal(result.nodes[0].enabled, true);
-});
-
-test('parseUiHierarchy reads single-quoted Android node attributes', () => {
-  const xml =
-    "<hierarchy><node class='android.widget.TextView' text='Hello' content-desc='Greeting' resource-id='com.demo:id/title' bounds='[10,20][110,60]' clickable='true' enabled='true'/></hierarchy>";
-
-  const result = parseUiHierarchy(xml, 800, { raw: true });
-  assert.equal(result.nodes.length, 1);
-  assert.equal(result.nodes[0].value, 'Hello');
-  assert.equal(result.nodes[0].label, 'Hello');
-  assert.equal(result.nodes[0].identifier, 'com.demo:id/title');
-  assert.deepEqual(result.nodes[0].rect, { x: 10, y: 20, width: 100, height: 40 });
-  assert.equal(result.nodes[0].hittable, true);
-  assert.equal(result.nodes[0].enabled, true);
-});
-
-test('parseUiHierarchy supports mixed quote styles in one node', () => {
-  const xml =
-    '<hierarchy><node class="android.widget.TextView" text=\'Hello\' content-desc="Greeting" resource-id=\'com.demo:id/title\' bounds="[10,20][110,60]"/></hierarchy>';
-
-  const result = parseUiHierarchy(xml, 800, { raw: true });
-  assert.equal(result.nodes.length, 1);
-  assert.equal(result.nodes[0].value, 'Hello');
-  assert.equal(result.nodes[0].label, 'Hello');
-  assert.equal(result.nodes[0].identifier, 'com.demo:id/title');
-});
-
-test('findBounds supports single and double quoted attributes', () => {
-  const xml = [
-    '<hierarchy>',
-    '<node text="Nothing" content-desc="Irrelevant" bounds="[0,0][10,10]"/>',
-    "<node text='Target from single quote' content-desc='Alt single' bounds='[100,200][300,500]'/>",
-    '<node text="Target from double quote" content-desc="Alt double" bounds="[50,50][150,250]"/>',
-    '</hierarchy>',
-  ].join('');
-
-  assert.deepEqual(findBounds(xml, 'single quote'), { x: 200, y: 350 });
-  assert.deepEqual(findBounds(xml, 'alt double'), { x: 100, y: 150 });
-});
-
-test('parseUiHierarchy ignores attribute-name prefix spoofing', () => {
-  const xml =
-    "<hierarchy><node class='android.widget.TextView' hint-text='Spoofed' text='Actual' bounds='[10,20][110,60]'/></hierarchy>";
-
-  const result = parseUiHierarchy(xml, 800, { raw: true });
-  assert.equal(result.nodes.length, 1);
-  assert.equal(result.nodes[0].value, 'Actual');
-});
-
-test('findBounds ignores bounds-like fragments inside other attribute values', () => {
-  const xml = [
-    '<hierarchy>',
-    "<node text='Target' content-desc=\"metadata bounds='[900,900][1000,1000]'\" bounds='[100,200][300,500]'/>",
-    '</hierarchy>',
-  ].join('');
-
-  assert.deepEqual(findBounds(xml, 'target'), { x: 200, y: 350 });
-});
-
-test('parseAndroidLaunchComponent extracts final resolved component', () => {
-  const stdout = [
-    'priority=0 preferredOrder=0 match=0x108000 specificIndex=-1 isDefault=true',
-    'com.boatsgroup.boattrader/com.boatsgroup.boattrader.MainActivity',
-  ].join('\n');
-  assert.equal(
-    parseAndroidLaunchComponent(stdout),
-    'com.boatsgroup.boattrader/com.boatsgroup.boattrader.MainActivity',
-  );
-});
-
-test('parseAndroidLaunchComponent returns null when no component is present', () => {
-  const stdout = 'No activity found';
-  assert.equal(parseAndroidLaunchComponent(stdout), null);
-});
-
-test('openAndroidApp rejects activity override for deep link URLs', async () => {
-  const device: DeviceInfo = {
-    platform: 'android',
-    id: 'emulator-5554',
-    name: 'Pixel',
-    kind: 'emulator',
-    booted: true,
-  };
-
-  await assert.rejects(
-    () => openAndroidApp(device, '  https://example.com/path  ', '.MainActivity'),
-    (error: unknown) => {
-      assert.equal(error instanceof AppError, true);
-      assert.equal((error as AppError).code, 'INVALID_ARGS');
-      return true;
-    },
-  );
-});
-
-test('swipeAndroid invokes adb input swipe with duration', async () => {
-  const tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'agent-device-swipe-test-'));
-  const adbPath = path.join(tmpDir, 'adb');
-  const argsLogPath = path.join(tmpDir, 'args.log');
-  await fs.writeFile(
-    adbPath,
-    '#!/bin/sh\nprintf "%s\\n" "$@" > "$AGENT_DEVICE_TEST_ARGS_FILE"\nexit 0\n',
-    'utf8',
-  );
-  await fs.chmod(adbPath, 0o755);
-
-  const previousPath = process.env.PATH;
-  const previousArgsFile = process.env.AGENT_DEVICE_TEST_ARGS_FILE;
-  process.env.PATH = `${tmpDir}${path.delimiter}${previousPath ?? ''}`;
-  process.env.AGENT_DEVICE_TEST_ARGS_FILE = argsLogPath;
-
-  const device: DeviceInfo = {
-    platform: 'android',
-    id: 'emulator-5554',
-    name: 'Pixel',
-    kind: 'emulator',
-    booted: true,
-  };
-
-  try {
-    await swipeAndroid(device, 10, 20, 30, 40, 250);
-    const args = (await fs.readFile(argsLogPath, 'utf8'))
-      .trim()
-      .split('\n')
-      .filter(Boolean);
-    assert.deepEqual(args, ['-s', 'emulator-5554', 'shell', 'input', 'swipe', '10', '20', '30', '40', '250']);
-  } finally {
-    process.env.PATH = previousPath;
-    if (previousArgsFile === undefined) {
-      delete process.env.AGENT_DEVICE_TEST_ARGS_FILE;
-    } else {
-      process.env.AGENT_DEVICE_TEST_ARGS_FILE = previousArgsFile;
-    }
-    await fs.rm(tmpDir, { recursive: true, force: true });
-  }
-});