agent-device 0.4.1 → 0.5.0

package/src/daemon/session-store.ts

@@ -1,298 +0,0 @@
-import fs from 'node:fs';
-import os from 'node:os';
-import path from 'node:path';
-import type { CommandFlags } from '../core/dispatch.ts';
-import type { SessionAction, SessionState } from './types.ts';
-import { inferFillText } from './action-utils.ts';
-
-export class SessionStore {
-  private readonly sessions = new Map<string, SessionState>();
-  private readonly sessionsDir: string;
-
-  constructor(sessionsDir: string) {
-    this.sessionsDir = sessionsDir;
-  }
-
-  get(name: string): SessionState | undefined {
-    return this.sessions.get(name);
-  }
-
-  has(name: string): boolean {
-    return this.sessions.has(name);
-  }
-
-  set(name: string, session: SessionState): void {
-    this.sessions.set(name, session);
-  }
-
-  delete(name: string): boolean {
-    return this.sessions.delete(name);
-  }
-
-  values(): IterableIterator<SessionState> {
-    return this.sessions.values();
-  }
-
-  toArray(): SessionState[] {
-    return Array.from(this.sessions.values());
-  }
-
-  recordAction(
-    session: SessionState,
-    entry: {
-      command: string;
-      positionals: string[];
-      flags: CommandFlags;
-      result?: Record<string, unknown>;
-    },
-  ): void {
-    if (entry.flags?.noRecord) return;
-    if (entry.flags?.saveScript) {
-      session.recordSession = true;
-      if (typeof entry.flags.saveScript === 'string') {
-        session.saveScriptPath = SessionStore.expandHome(entry.flags.saveScript);
-      }
-    }
-    session.actions.push({
-      ts: Date.now(),
-      command: entry.command,
-      positionals: entry.positionals,
-      flags: sanitizeFlags(entry.flags),
-      result: entry.result,
-    });
-  }
-
-  writeSessionLog(session: SessionState): void {
-    try {
-      if (!session.recordSession) return;
-      const scriptPath = this.resolveScriptPath(session);
-      const scriptDir = path.dirname(scriptPath);
-      if (!fs.existsSync(scriptDir)) fs.mkdirSync(scriptDir, { recursive: true });
-      const script = formatScript(session, this.buildOptimizedActions(session));
-      fs.writeFileSync(scriptPath, script);
-    } catch {
-      // ignore
-    }
-  }
-
-  defaultTracePath(session: SessionState): string {
-    const safeName = session.name.replace(/[^a-zA-Z0-9._-]/g, '_');
-    const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
-    return path.join(this.sessionsDir, `${safeName}-${timestamp}.trace.log`);
-  }
-
-  static expandHome(filePath: string): string {
-    if (filePath.startsWith('~/')) {
-      return path.join(os.homedir(), filePath.slice(2));
-    }
-    return path.resolve(filePath);
-  }
-
-  private resolveScriptPath(session: SessionState): string {
-    if (session.saveScriptPath) {
-      return SessionStore.expandHome(session.saveScriptPath);
-    }
-    if (!fs.existsSync(this.sessionsDir)) fs.mkdirSync(this.sessionsDir, { recursive: true });
-    const safeName = session.name.replace(/[^a-zA-Z0-9._-]/g, '_');
-    const timestamp = new Date(session.createdAt).toISOString().replace(/[:.]/g, '-');
-    return path.join(this.sessionsDir, `${safeName}-${timestamp}.ad`);
-  }
-
-  private buildOptimizedActions(session: SessionState): SessionAction[] {
-    const optimized: SessionAction[] = [];
-    for (const action of session.actions) {
-      if (action.command === 'snapshot') {
-        continue;
-      }
-      const selectorChain =
-        Array.isArray(action.result?.selectorChain) &&
-        action.result?.selectorChain.every((entry) => typeof entry === 'string')
-          ? (action.result.selectorChain as string[])
-          : [];
-      if (selectorChain.length > 0 && (action.command === 'click' || action.command === 'fill' || action.command === 'get')) {
-        const selectorExpr = selectorChain.join(' || ');
-        if (action.command === 'click') {
-          optimized.push({
-            ...action,
-            positionals: [selectorExpr],
-          });
-          continue;
-        }
-        if (action.command === 'fill') {
-          const text = inferFillText(action);
-          if (text.length > 0) {
-            optimized.push({
-              ...action,
-              positionals: [selectorExpr, text],
-            });
-            continue;
-          }
-        }
-        if (action.command === 'get') {
-          const sub = action.positionals?.[0];
-          if (sub === 'text' || sub === 'attrs') {
-            optimized.push({
-              ...action,
-              positionals: [sub, selectorExpr],
-            });
-            continue;
-          }
-        }
-      }
-      if (action.command === 'click' || action.command === 'fill' || action.command === 'get') {
-        const refLabel = action.result?.refLabel;
-        if (typeof refLabel === 'string' && refLabel.trim().length > 0) {
-          optimized.push({
-            ts: action.ts,
-            command: 'snapshot',
-            positionals: [],
-            flags: {
-              platform: session.device.platform,
-              snapshotInteractiveOnly: true,
-              snapshotCompact: true,
-              snapshotScope: refLabel.trim(),
-            },
-            result: { scope: refLabel.trim() },
-          });
-        }
-      }
-      optimized.push(action);
-    }
-    return optimized;
-  }
-}
-
-function sanitizeFlags(flags: CommandFlags | undefined): SessionAction['flags'] {
-  if (!flags) return {};
-  const {
-    platform,
-    device,
-    udid,
-    serial,
-    out,
-    verbose,
-    snapshotInteractiveOnly,
-    snapshotCompact,
-    snapshotDepth,
-    snapshotScope,
-    snapshotRaw,
-    snapshotBackend,
-    appsMetadata,
-    relaunch,
-    saveScript,
-    noRecord,
-  } = flags;
-  return {
-    platform,
-    device,
-    udid,
-    serial,
-    out,
-    verbose,
-    snapshotInteractiveOnly,
-    snapshotCompact,
-    snapshotDepth,
-    snapshotScope,
-    snapshotRaw,
-    snapshotBackend,
-    appsMetadata,
-    relaunch,
-    saveScript,
-    noRecord,
-  };
-}
-
-function formatScript(session: SessionState, actions: SessionAction[]): string {
-  const lines: string[] = [];
-  const deviceLabel = session.device.name.replace(/"/g, '\\"');
-  const kind = session.device.kind ? ` kind=${session.device.kind}` : '';
-  const theme = 'unknown';
-  lines.push(`context platform=${session.device.platform} device="${deviceLabel}"${kind} theme=${theme}`);
-  for (const action of actions) {
-    if (action.flags?.noRecord) continue;
-    lines.push(formatActionLine(action));
-  }
-  return `${lines.join('\n')}\n`;
-}
-
-function formatActionLine(action: SessionAction): string {
-  const parts: string[] = [action.command];
-  if (action.command === 'click') {
-    const ref = action.positionals?.[0];
-    if (ref) {
-      parts.push(formatArg(ref));
-      if (ref.startsWith('@')) {
-        const refLabel = action.result?.refLabel;
-        if (typeof refLabel === 'string' && refLabel.trim().length > 0) {
-          parts.push(formatArg(refLabel));
-        }
-      }
-      return parts.join(' ');
-    }
-  }
-  if (action.command === 'fill') {
-    const ref = action.positionals?.[0];
-    if (ref && ref.startsWith('@')) {
-      parts.push(formatArg(ref));
-      const refLabel = action.result?.refLabel;
-      const text = action.positionals.slice(1).join(' ');
-      if (typeof refLabel === 'string' && refLabel.trim().length > 0) {
-        parts.push(formatArg(refLabel));
-      }
-      if (text) {
-        parts.push(formatArg(text));
-      }
-      return parts.join(' ');
-    }
-  }
-  if (action.command === 'get') {
-    const sub = action.positionals?.[0];
-    const ref = action.positionals?.[1];
-    if (sub && ref) {
-      parts.push(formatArg(sub));
-      parts.push(formatArg(ref));
-      if (ref.startsWith('@')) {
-        const refLabel = action.result?.refLabel;
-        if (typeof refLabel === 'string' && refLabel.trim().length > 0) {
-          parts.push(formatArg(refLabel));
-        }
-      }
-      return parts.join(' ');
-    }
-  }
-  if (action.command === 'snapshot') {
-    if (action.flags?.snapshotInteractiveOnly) parts.push('-i');
-    if (action.flags?.snapshotCompact) parts.push('-c');
-    if (typeof action.flags?.snapshotDepth === 'number') {
-      parts.push('-d', String(action.flags.snapshotDepth));
-    }
-    if (action.flags?.snapshotScope) {
-      parts.push('-s', formatArg(action.flags.snapshotScope));
-    }
-    if (action.flags?.snapshotRaw) parts.push('--raw');
-    if (action.flags?.snapshotBackend) {
-      parts.push(`--backend`, action.flags.snapshotBackend);
-    }
-    return parts.join(' ');
-  }
-  if (action.command === 'open') {
-    for (const positional of action.positionals ?? []) {
-      parts.push(formatArg(positional));
-    }
-    if (action.flags?.relaunch) {
-      parts.push('--relaunch');
-    }
-    return parts.join(' ');
-  }
-  for (const positional of action.positionals ?? []) {
-    parts.push(formatArg(positional));
-  }
-  return parts.join(' ');
-}
-
-function formatArg(value: string): string {
-  const trimmed = value.trim();
-  if (trimmed.startsWith('@')) return trimmed;
-  if (/^-?\d+(\.\d+)?$/.test(trimmed)) return trimmed;
-  return JSON.stringify(trimmed);
-}

package/src/daemon/snapshot-processing.ts

@@ -1,131 +0,0 @@
-import type { RawSnapshotNode, SnapshotState } from '../utils/snapshot.ts';
-
-export function findNodeByLabel(nodes: SnapshotState['nodes'], label: string) {
-  const query = label.toLowerCase();
-  return (
-    nodes.find((node) => {
-      const labelValue = (node.label ?? '').toLowerCase();
-      const valueValue = (node.value ?? '').toLowerCase();
-      const idValue = (node.identifier ?? '').toLowerCase();
-      return labelValue.includes(query) || valueValue.includes(query) || idValue.includes(query);
-    }) ?? null
-  );
-}
-
-export function resolveRefLabel(
-  node: SnapshotState['nodes'][number],
-  nodes: SnapshotState['nodes'],
-): string | undefined {
-  const primary = [node.label, node.value, node.identifier]
-    .map((value) => (typeof value === 'string' ? value.trim() : ''))
-    .find((value) => value && value.length > 0);
-  if (primary && isMeaningfulLabel(primary)) return primary;
-  const fallback = findNearestMeaningfulLabel(node, nodes);
-  return fallback ?? (primary && isMeaningfulLabel(primary) ? primary : undefined);
-}
-
-function isMeaningfulLabel(value: string): boolean {
-  const trimmed = value.trim();
-  if (!trimmed) return false;
-  if (/^(true|false)$/i.test(trimmed)) return false;
-  if (/^\d+$/.test(trimmed)) return false;
-  return true;
-}
-
-function findNearestMeaningfulLabel(
-  target: SnapshotState['nodes'][number],
-  nodes: SnapshotState['nodes'],
-): string | undefined {
-  if (!target.rect) return undefined;
-  const targetY = target.rect.y + target.rect.height / 2;
-  let best: { label: string; distance: number } | null = null;
-  for (const node of nodes) {
-    if (!node.rect) continue;
-    const label = [node.label, node.value, node.identifier]
-      .map((value) => (typeof value === 'string' ? value.trim() : ''))
-      .find((value) => value && value.length > 0);
-    if (!label || !isMeaningfulLabel(label)) continue;
-    const nodeY = node.rect.y + node.rect.height / 2;
-    const distance = Math.abs(nodeY - targetY);
-    if (!best || distance < best.distance) {
-      best = { label, distance };
-    }
-  }
-  return best?.label;
-}
-
-export function pruneGroupNodes(nodes: RawSnapshotNode[]): RawSnapshotNode[] {
-  const skippedDepths: number[] = [];
-  const result: RawSnapshotNode[] = [];
-  for (const node of nodes) {
-    const depth = node.depth ?? 0;
-    while (skippedDepths.length > 0 && depth <= skippedDepths[skippedDepths.length - 1]) {
-      skippedDepths.pop();
-    }
-    const type = normalizeType(node.type ?? '');
-    const labelCandidate = [node.label, node.value, node.identifier]
-      .map((value) => (typeof value === 'string' ? value.trim() : ''))
-      .find((value) => value && value.length > 0);
-    const hasMeaningfulLabel = labelCandidate ? isMeaningfulLabel(labelCandidate) : false;
-    if ((type === 'group' || type === 'ioscontentgroup') && !hasMeaningfulLabel) {
-      skippedDepths.push(depth);
-      continue;
-    }
-    const adjustedDepth = Math.max(0, depth - skippedDepths.length);
-    result.push({ ...node, depth: adjustedDepth });
-  }
-  return result;
-}
-
-export function normalizeType(type: string): string {
-  let value = type.trim().replace(/XCUIElementType/gi, '').toLowerCase();
-  if (value.startsWith('ax')) {
-    value = value.replace(/^ax/, '');
-  }
-  const lastSeparator = Math.max(value.lastIndexOf('.'), value.lastIndexOf('/'));
-  if (lastSeparator !== -1) {
-    value = value.slice(lastSeparator + 1);
-  }
-  return value;
-}
-
-export function isFillableType(type: string, platform: 'ios' | 'android'): boolean {
-  const normalized = normalizeType(type);
-  if (!normalized) return true;
-  if (platform === 'android') {
-    return normalized.includes('edittext') || normalized.includes('autocompletetextview');
-  }
-  return (
-    normalized.includes('textfield') ||
-    normalized.includes('securetextfield') ||
-    normalized.includes('searchfield') ||
-    normalized.includes('textview') ||
-    normalized.includes('textarea') ||
-    normalized === 'search'
-  );
-}
-
-export function findNearestHittableAncestor(
-  nodes: SnapshotState['nodes'],
-  node: SnapshotState['nodes'][number],
-): SnapshotState['nodes'][number] | null {
-  if (node.hittable) return node;
-  let current = node;
-  const visited = new Set<string>();
-  while (current.parentIndex !== undefined) {
-    if (visited.has(current.ref)) break;
-    visited.add(current.ref);
-    const parent = nodes[current.parentIndex];
-    if (!parent) break;
-    if (parent.hittable) return parent;
-    current = parent;
-  }
-  return null;
-}
-
-export function extractNodeText(node: SnapshotState['nodes'][number]): string {
-  const candidates = [node.label, node.value, node.identifier]
-    .map((value) => (typeof value === 'string' ? value.trim() : ''))
-    .filter((value) => value.length > 0);
-  return candidates[0] ?? '';
-}

package/src/daemon/types.ts

@@ -1,56 +0,0 @@
-import type { CommandFlags } from '../core/dispatch.ts';
-import type { DeviceInfo } from '../utils/device.ts';
-import type { ExecResult } from '../utils/exec.ts';
-import type { SnapshotState } from '../utils/snapshot.ts';
-
-export type DaemonRequest = {
-  token: string;
-  session: string;
-  command: string;
-  positionals: string[];
-  flags?: CommandFlags;
-};
-
-export type DaemonResponse =
-  | { ok: true; data?: Record<string, unknown> }
-  | { ok: false; error: { code: string; message: string; details?: Record<string, unknown> } };
-
-export type SessionState = {
-  name: string;
-  device: DeviceInfo;
-  createdAt: number;
-  appBundleId?: string;
-  appName?: string;
-  snapshot?: SnapshotState;
-  trace?: {
-    outPath: string;
-    startedAt: number;
-  };
-  recordSession?: boolean;
-  saveScriptPath?: string;
-  actions: SessionAction[];
-  recording?: {
-    platform: 'ios' | 'android';
-    outPath: string;
-    remotePath?: string;
-    child: ReturnType<typeof import('node:child_process').spawn>;
-    wait: Promise<ExecResult>;
-  };
-};
-
-export type SessionAction = {
-  ts: number;
-  command: string;
-  positionals: string[];
-  flags: Partial<CommandFlags> & {
-    snapshotInteractiveOnly?: boolean;
-    snapshotCompact?: boolean;
-    snapshotDepth?: number;
-    snapshotScope?: string;
-    snapshotRaw?: boolean;
-    snapshotBackend?: 'ax' | 'xctest';
-    saveScript?: boolean | string;
-    noRecord?: boolean;
-  };
-  result?: Record<string, unknown>;
-};

package/src/daemon-client.ts

@@ -1,172 +0,0 @@
-import net from 'node:net';
-import fs from 'node:fs';
-import os from 'node:os';
-import path from 'node:path';
-import { AppError } from './utils/errors.ts';
-import type { CommandFlags } from './core/dispatch.ts';
-import { runCmdDetached } from './utils/exec.ts';
-import { findProjectRoot, readVersion } from './utils/version.ts';
-import { stopProcessForTakeover } from './utils/process-identity.ts';
-
-export type DaemonRequest = {
-  token: string;
-  session: string;
-  command: string;
-  positionals: string[];
-  flags?: CommandFlags;
-};
-
-export type DaemonResponse =
-  | { ok: true; data?: Record<string, unknown> }
-  | { ok: false; error: { code: string; message: string; details?: Record<string, unknown> } };
-
-type DaemonInfo = {
-  port: number;
-  token: string;
-  pid: number;
-  version?: string;
-  processStartTime?: string;
-};
-
-const baseDir = path.join(os.homedir(), '.agent-device');
-const infoPath = path.join(baseDir, 'daemon.json');
-const REQUEST_TIMEOUT_MS = resolveDaemonRequestTimeoutMs();
-const DAEMON_STARTUP_TIMEOUT_MS = 5000;
-const DAEMON_TAKEOVER_TERM_TIMEOUT_MS = 3000;
-const DAEMON_TAKEOVER_KILL_TIMEOUT_MS = 1000;
-
-export async function sendToDaemon(req: Omit<DaemonRequest, 'token'>): Promise<DaemonResponse> {
-  const info = await ensureDaemon();
-  const request = { ...req, token: info.token };
-  return await sendRequest(info, request);
-}
-
-async function ensureDaemon(): Promise<DaemonInfo> {
-  const existing = readDaemonInfo();
-  const localVersion = readVersion();
-  const existingReachable = existing ? await canConnect(existing) : false;
-  if (existing && existing.version === localVersion && existingReachable) return existing;
-  if (existing && (existing.version !== localVersion || !existingReachable)) {
-    await stopDaemonProcessForTakeover(existing);
-    removeDaemonInfo();
-  }
-
-  await startDaemon();
-
-  const start = Date.now();
-  while (Date.now() - start < DAEMON_STARTUP_TIMEOUT_MS) {
-    const info = readDaemonInfo();
-    if (info && (await canConnect(info))) return info;
-    await new Promise((resolve) => setTimeout(resolve, 100));
-  }
-
-  throw new AppError('COMMAND_FAILED', 'Failed to start daemon', {
-    infoPath,
-    hint: 'Run pnpm build, or delete ~/.agent-device/daemon.json if stale.',
-  });
-}
-
-async function stopDaemonProcessForTakeover(info: DaemonInfo): Promise<void> {
-  await stopProcessForTakeover(info.pid, {
-    termTimeoutMs: DAEMON_TAKEOVER_TERM_TIMEOUT_MS,
-    killTimeoutMs: DAEMON_TAKEOVER_KILL_TIMEOUT_MS,
-    expectedStartTime: info.processStartTime,
-  });
-}
-
-function readDaemonInfo(): DaemonInfo | null {
-  if (!fs.existsSync(infoPath)) return null;
-  try {
-    const data = JSON.parse(fs.readFileSync(infoPath, 'utf8')) as DaemonInfo;
-    if (!data.port || !data.token) return null;
-    return {
-      ...data,
-      pid: Number.isInteger(data.pid) && data.pid > 0 ? data.pid : 0,
-    };
-  } catch {
-    return null;
-  }
-}
-
-function removeDaemonInfo(): void {
-  try {
-    if (fs.existsSync(infoPath)) fs.unlinkSync(infoPath);
-  } catch {
-    // Best-effort cleanup only; daemon can still overwrite this file on startup.
-  }
-}
-
-async function canConnect(info: DaemonInfo): Promise<boolean> {
-  return new Promise((resolve) => {
-    const socket = net.createConnection({ host: '127.0.0.1', port: info.port }, () => {
-      socket.destroy();
-      resolve(true);
-    });
-    socket.on('error', () => {
-      resolve(false);
-    });
-  });
-}
-
-async function startDaemon(): Promise<void> {
-  const root = findProjectRoot();
-  const distPath = path.join(root, 'dist', 'src', 'daemon.js');
-  const srcPath = path.join(root, 'src', 'daemon.ts');
-
-  const hasDist = fs.existsSync(distPath);
-  const hasSrc = fs.existsSync(srcPath);
-  if (!hasDist && !hasSrc) {
-    throw new AppError('COMMAND_FAILED', 'Daemon entry not found', { distPath, srcPath });
-  }
-  const runningFromSource = process.execArgv.includes('--experimental-strip-types');
-  const useSrc = runningFromSource ? hasSrc : !hasDist && hasSrc;
-  const args = useSrc ? ['--experimental-strip-types', srcPath] : [distPath];
-
-  runCmdDetached(process.execPath, args);
-}
-
-async function sendRequest(info: DaemonInfo, req: DaemonRequest): Promise<DaemonResponse> {
-  return new Promise((resolve, reject) => {
-    const socket = net.createConnection({ host: '127.0.0.1', port: info.port }, () => {
-      socket.write(`${JSON.stringify(req)}\n`);
-    });
-    const timeout = setTimeout(() => {
-      socket.destroy();
-      reject(
-        new AppError('COMMAND_FAILED', 'Daemon request timed out', { timeoutMs: REQUEST_TIMEOUT_MS }),
-      );
-    }, REQUEST_TIMEOUT_MS);
-
-    let buffer = '';
-    socket.setEncoding('utf8');
-    socket.on('data', (chunk) => {
-      buffer += chunk;
-      const idx = buffer.indexOf('\n');
-      if (idx === -1) return;
-      const line = buffer.slice(0, idx).trim();
-      if (!line) return;
-      try {
-        const response = JSON.parse(line) as DaemonResponse;
-        socket.end();
-        clearTimeout(timeout);
-        resolve(response);
-      } catch (err) {
-        clearTimeout(timeout);
-        reject(err);
-      }
-    });
-
-    socket.on('error', (err) => {
-      clearTimeout(timeout);
-      reject(err);
-    });
-  });
-}
-
-export function resolveDaemonRequestTimeoutMs(raw: string | undefined = process.env.AGENT_DEVICE_DAEMON_TIMEOUT_MS): number {
-  // iOS physical-device runner startup/build can exceed 60s, so use a safer default for daemon RPCs.
-  if (!raw) return 180000;
-  const parsed = Number(raw);
-  if (!Number.isFinite(parsed)) return 180000;
-  return Math.max(1000, Math.floor(parsed));
-}