refactor(commands): read command inputs and output from caller hosts

src/agent-device-client.ts

@@ -162,7 +162,7 @@ export function createAgentDeviceClient(
     command: DaemonCommandName,
     options: InternalRequestOptions = {},
   ): Promise<T> => {
-    const request = await prepareDaemonCommandRequest(command, options);
+    const request = await prepareDaemonCommandRequest(command, options, host);
     return (await execute(
       request.command,
       request.positionals,

src/cli/commands/proxy.ts

@@ -106,7 +106,7 @@ export function renderProxyStartup(
   startup: ProxyStartup,
   options: { useColor?: boolean } = {},
 ): string {
-  const useColor = options.useColor ?? supportsColor();
+  const useColor = options.useColor ?? supportsColor(process.env);
   const checkmark = formatProxyOutputValue('✓', 'green', useColor);
   const proxyBaseUrl = formatProxyOutputValue(startup.proxyBaseUrl, 'cyan', useColor);
   const daemonBaseUrl = formatProxyOutputValue('<tunnel URL>', 'cyan', useColor);

src/cli/commands/screenshot.ts

@@ -52,7 +52,7 @@ export const diffCommand: ClientCommandHandler = async ({ positionals, flags, cl
       flags,
       env: host.env,
     });
-    await writeCommandOutput(flags, result, () => formatSnapshotDiffText(result));
+    await writeCommandOutput(flags, result, () => formatSnapshotDiffText(result, host.env));
     return true;
   }
 
@@ -104,7 +104,7 @@ export const diffCommand: ClientCommandHandler = async ({ positionals, flags, cl
   });
 
   await writeCommandOutput(flags, result, () =>
-    formatScreenshotDiffText(result as ScreenshotDiffResult),
+    formatScreenshotDiffText(result as ScreenshotDiffResult, host),
   );
   return true;
 };

src/cli/replay-test/progress.ts

@@ -130,7 +130,7 @@ function formatReplayTestLiveProgressLine(
 ): string {
   const title = event.title?.trim();
   const file = path.basename(event.file);
-  const useColor = supportsColor(process.stderr);
+  const useColor = supportsColor(process.env, process.stderr);
   const spinner = formatReplayTestProgressSpinner(spinnerFrame, { useColor });
   const shardSuffix = formatReplayTestProgressShardSuffix(event, { useColor });
   const stepSuffix = formatReplayTestLiveProgressStepSuffix(event, { useColor });
@@ -223,7 +223,7 @@ function replayTestProgressFailureContextLines(event: ReplayTestResult): string[
 }
 
 function formatReplayTestCaseSummaryLine(event: ReplayTestResult): string {
-  const useColor = supportsColor(process.stderr);
+  const useColor = supportsColor(process.env, process.stderr);
   const statusLabel = formatReplayTestProgressStatusLabel(event);
   const name = formatReplayTestProgressName(event);
   const shardSuffix = formatReplayTestProgressShardSuffix(event, { useColor });

src/cli/replay-test/reporters/default.ts

@@ -143,7 +143,7 @@ function renderWarningsSection(
 
 function formatReplayTestSummaryLine(data: ReplaySuiteResult, flakyCount: number): string {
   const durationMs = typeof data.durationMs === 'number' ? data.durationMs : undefined;
-  const useColor = supportsColor();
+  const useColor = supportsColor(process.env);
   const passed = formatReplaySummaryPassed(data.passed, { useColor });
   const total = formatReplaySummaryTotal(data.total, { useColor });
   const failedSuffix = data.failed > 0 ? `, ${data.failed} failed` : '';
@@ -169,7 +169,7 @@ function formatReplayDuration(durationMs: number, options: { useColor?: boolean
 }
 
 function replayFlakyStatusIcon(): string {
-  const useColor = supportsColor();
+  const useColor = supportsColor(process.env);
   return useColor ? colorize('✓', 'yellow') : '✓';
 }
 

src/commands/batch/projection.ts

@@ -1,3 +1,4 @@
+import type { HostEnvironment } from '@agent-device/kernel/source-value';
 import { PUBLIC_COMMANDS } from '@agent-device/command-registry/catalog';
 import {
   STRUCTURED_BATCH_COMMAND_NAMES,
@@ -27,23 +28,32 @@ type PrepareDaemonCommandRequest = (
   command: string,
   input: CommandInput,
   stepNumber: number,
+  host: HostEnvironment,
+) => Promise<DaemonCommandRequest>;
+
+type PrepareBatchStep = (
+  command: string,
+  input: CommandInput,
+  stepNumber: number,
 ) => Promise<DaemonCommandRequest>;
 
 export function createBatchDaemonWriter(
   prepareDaemonCommandRequest: PrepareDaemonCommandRequest,
 ): AsyncDaemonWriter {
-  return async (input) =>
+  return async (input, host) =>
     request(PUBLIC_COMMANDS.batch, [], {
       ...input,
-      batchSteps: await readBatchDaemonSteps(input.steps, prepareDaemonCommandRequest),
+      batchSteps: await readBatchDaemonSteps(input.steps, (command, input, stepNumber) =>
+        prepareDaemonCommandRequest(command, input, stepNumber, host),
+      ),
       batchOnError: input.onError,
       batchMaxSteps: input.maxSteps,
     });
 }
 
 async function readBatchDaemonSteps(
   steps: unknown,
-  prepareDaemonCommandRequest: PrepareDaemonCommandRequest,
+  prepareDaemonCommandRequest: PrepareBatchStep,
 ): Promise<DaemonBatchStep[]> {
   if (!Array.isArray(steps) || steps.length === 0) {
     throw new AppError('INVALID_ARGS', 'batch requires a non-empty steps array.');
@@ -60,7 +70,7 @@ async function readBatchDaemonSteps(
 async function readBatchDaemonStep(
   step: unknown,
   stepNumber: number,
-  prepareDaemonCommandRequest: PrepareDaemonCommandRequest,
+  prepareDaemonCommandRequest: PrepareBatchStep,
 ): Promise<DaemonBatchStep> {
   const record = readBatchStepRecord(step, stepNumber);
   const command = readBatchStepCommand(record, stepNumber);

src/commands/cli-grammar/types.ts

@@ -1,3 +1,4 @@
+import type { HostEnvironment } from '@agent-device/kernel/source-value';
 import type {
   CliFlags,
   CommandFlags,
@@ -97,6 +98,9 @@ export type DaemonWriter = (input: CommandInput) => DaemonCommandRequest;
  * `runFlow` includes on demand. The registry awaits whichever kind a command declares, so the
  * asynchrony stays inside the one writer that needs it.
  */
-export type AsyncDaemonWriter = (input: CommandInput) => Promise<DaemonCommandRequest>;
+export type AsyncDaemonWriter = (
+  input: CommandInput,
+  host: HostEnvironment,
+) => Promise<DaemonCommandRequest>;
 
 export type AnyDaemonWriter = DaemonWriter | AsyncDaemonWriter;

src/commands/command-projection.ts

@@ -1,3 +1,4 @@
+import type { HostEnvironment } from '@agent-device/kernel/source-value';
 import { createBatchDaemonWriter } from './batch/index.ts';
 import type { AnyDaemonWriter, CommandInput, DaemonCommandRequest } from './cli-grammar/types.ts';
 import { findCommandMetadata } from './command-metadata.ts';
@@ -17,6 +18,7 @@ async function prepareBatchDaemonCommandRequest(
   command: string,
   input: CommandInput,
   stepNumber: number,
+  host: HostEnvironment,
 ): Promise<DaemonCommandRequest> {
   const writer = (daemonWriters as Readonly<Record<string, AnyDaemonWriter>>)[command];
   if (!writer) {
@@ -31,6 +33,7 @@ async function prepareBatchDaemonCommandRequest(
       writer,
       metadata,
       metadata.readInput(input) as CommandInput,
+      host,
     );
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
@@ -46,22 +49,24 @@ async function prepareBatchDaemonCommandRequest(
 export async function prepareDaemonCommandRequest(
   command: DaemonCommandName,
   input: CommandInput,
+  host: HostEnvironment,
 ): Promise<DaemonCommandRequest> {
   const writer = daemonWriters[command];
   if (!writer) {
     throw new Error(`Missing daemon writer for command: ${command}`);
   }
   refuseRetiredCommonInput(input as Record<string, unknown>);
   const metadata = findCommandMetadata(command);
-  return await prepareRequestWithMetadataFlags(writer, metadata, input);
+  return await prepareRequestWithMetadataFlags(writer, metadata, input, host);
 }
 
 async function prepareRequestWithMetadataFlags(
   writer: AnyDaemonWriter,
   metadata: ReturnType<typeof findCommandMetadata>,
   input: CommandInput,
+  host: HostEnvironment,
 ): Promise<DaemonCommandRequest> {
-  const request = await writer(input);
+  const request = await writer(input, host);
   return {
     ...request,
     ...(metadata ? { metadataFlags: readMetadataCommandFlags(metadata, request.options) } : {}),

src/commands/output/color.test.ts

@@ -9,8 +9,8 @@ test('supportsColor uses the stream TTY state by default', () => {
   delete process.env.FORCE_COLOR;
   delete process.env.NO_COLOR;
   try {
-    assert.equal(supportsColor({ isTTY: true }), true);
-    assert.equal(supportsColor({ isTTY: false }), false);
+    assert.equal(supportsColor(process.env, { isTTY: true }), true);
+    assert.equal(supportsColor(process.env, { isTTY: false }), false);
   } finally {
     if (typeof originalForceColor === 'string') process.env.FORCE_COLOR = originalForceColor;
     else delete process.env.FORCE_COLOR;
@@ -25,14 +25,14 @@ test('supportsColor honors explicit environment overrides', () => {
   try {
     process.env.FORCE_COLOR = '0';
     delete process.env.NO_COLOR;
-    assert.equal(supportsColor({ isTTY: true }), false);
+    assert.equal(supportsColor(process.env, { isTTY: true }), false);
 
     process.env.FORCE_COLOR = '1';
-    assert.equal(supportsColor({ isTTY: false }), true);
+    assert.equal(supportsColor(process.env, { isTTY: false }), true);
 
     delete process.env.FORCE_COLOR;
     process.env.NO_COLOR = '1';
-    assert.equal(supportsColor({ isTTY: true }), false);
+    assert.equal(supportsColor(process.env, { isTTY: true }), false);
   } finally {
     if (typeof originalForceColor === 'string') process.env.FORCE_COLOR = originalForceColor;
     else delete process.env.FORCE_COLOR;

src/commands/output/color.ts

@@ -1,11 +1,12 @@
+import type { EnvMap } from '@agent-device/kernel/source-value';
 import { styleText } from 'node:util';
 
-export function supportsColor(stream: { isTTY?: boolean } = process.stdout): boolean {
-  const forceColor = process.env.FORCE_COLOR;
+export function supportsColor(env: EnvMap, stream: { isTTY?: boolean } = process.stdout): boolean {
+  const forceColor = env.FORCE_COLOR;
   if (typeof forceColor === 'string') {
     return forceColor !== '0';
   }
-  if (typeof process.env.NO_COLOR === 'string') {
+  if (typeof env.NO_COLOR === 'string') {
     return false;
   }
   return Boolean(stream.isTTY);