Function lifecycleCold

services/cloud-agent-next/test/e2e/lifecycle.ts:92–169

(args: LifecycleArgs)

  `;
  try {
    await execFileAsync('docker', ['exec', containerId, 'bun', '-e', probe, sessionId], {
      timeout: 5_000,
    });
    return true;
  } catch {
    return false;
  }
}

async function findOwnedSandboxes(
  sessionId: string,
  kiloSessionId: string | undefined,
  knownIds: Set<string>
) {
  const candidates = sessionId.startsWith('workspace_')
    ? await listSandboxContainers()
    : await listSandboxesForAgentSession(sessionId);
  const matches: SandboxContainer[] = [];
  for (const container of candidates) {
    if (container.isProxy || knownIds.has(container.id)) continue;
    if (await sandboxOwnsSession(container.id, sessionId, kiloSessionId)) matches.push(container);
  }
  return matches;
}

export async function waitForOwnedSandbox(
  sessionId: string,
  kiloSessionId: string,
  knownIds: Set<string>,
  timeoutMs: number,
  signal?: AbortSignal
): Promise<SandboxContainer | null> {
  const deadline = Date.now() + timeoutMs;
  while (Date.now() < deadline) {
    // Docker discovery is not abortable mid-exec, so the signal is honoured
    // between polls rather than during one.
    if (signal?.aborted) return null;
    const matches = await findOwnedSandboxes(sessionId, kiloSessionId, knownIds);
    if (matches.length > 1) {
      throw new Error(`Multiple containers match ${sessionId}; refusing ambiguous ownership`);
    }
    if (matches[0]) return matches[0];
    await new Promise(resolve => setTimeout(resolve, 500));
  }
  return null;
}

/**
 * Single-shot owned-container discovery for the current session, used by the
 * `sessionSandbox.currentContainer` capability. Unlike `waitForOwnedSandbox` it
 * excludes nothing, because it observes a container that is expected to already
 * exist; it still refuses ambiguous ownership.
 */
export async function currentOwnedSandbox(
  sessionId: string,
  kiloSessionId: string
): Promise<SandboxContainer | null> {
  const matches = await findOwnedSandboxes(sessionId, kiloSessionId, new Set());