pi-harness-delegate 0.1.0 → 0.2.0

package/extensions/harnesses/amp.ts

@@ -1,137 +1,144 @@
 import { execFile } from 'node:child_process';
 import { promisify } from 'node:util';
-import type { Harness, BuildArgsOpts, NormalizedPermission, ParseOutcome, ParseState, StreamedResult } from './types.ts';
+import type {
+  BuildArgsOpts,
+  Harness,
+  NormalizedPermission,
+  ParseOutcome,
+  ParseState,
+  StreamedResult,
+} from './types.ts';
 
 const execFileAsync = promisify(execFile);
 
 function isRecord(v: unknown): v is Record<string, unknown> {
-	return typeof v === 'object' && v !== null && !Array.isArray(v);
+  return typeof v === 'object' && v !== null && !Array.isArray(v);
 }
 
 const PERMISSION_MAP: Record<NormalizedPermission, string> = {
-	readonly: 'read-only',
-	edit: 'workspace',
-	danger: 'danger',
+  readonly: 'read-only',
+  edit: 'workspace',
+  danger: 'danger',
 };
 
 function extractAmpText(o: Record<string, unknown>): string | undefined {
-	if (typeof o.text === 'string') return o.text;
-	if (typeof o.output === 'string') return o.output;
-	if (typeof o.delta === 'string') return o.delta;
-	if (typeof o.content === 'string') return o.content;
-	return undefined;
+  if (typeof o.text === 'string') return o.text;
+  if (typeof o.output === 'string') return o.output;
+  if (typeof o.delta === 'string') return o.delta;
+  if (typeof o.content === 'string') return o.content;
+  return undefined;
 }
 
 export function parseAmpLine(line: string, state: ParseState): ParseOutcome {
-	let o: unknown;
-	try {
-		o = JSON.parse(line);
-	} catch {
-		if (line.trim().length > 0) return { streamedText: line + '\n', activities: [] };
-		return { activities: [] };
-	}
-	if (!isRecord(o)) return { activities: [] };
-	const activities: ParseOutcome['activities'] = [];
-	let streamedText: string | undefined;
-	const typeStr = typeof o.type === 'string' ? o.type : '';
+  let o: unknown;
+  try {
+    o = JSON.parse(line);
+  } catch {
+    if (line.trim().length > 0) return { streamedText: line + '\n', activities: [] };
+    return { activities: [] };
+  }
+  if (!isRecord(o)) return { activities: [] };
+  const activities: ParseOutcome['activities'] = [];
+  let streamedText: string | undefined;
+  const typeStr = typeof o.type === 'string' ? o.type : '';
 
-	if (typeStr.includes('tool') || o.type === 'tool_use') {
-		const name = typeof o.name === 'string' ? o.name : 'tool';
-		if (typeStr.includes('start') || o.type === 'tool_use') {
-			activities.push({ kind: 'tool_start', name });
-			if (isRecord(o.input)) activities.push({ kind: 'tool_input', name, input: o.input as Record<string, unknown> });
-		} else {
-			activities.push({ kind: 'tool_result', isError: o.is_error === true });
-		}
-	}
-	if (typeStr.includes('thinking')) activities.push({ kind: 'thinking', chars: 10 });
+  if (typeStr.includes('tool') || o.type === 'tool_use') {
+    const name = typeof o.name === 'string' ? o.name : 'tool';
+    if (typeStr.includes('start') || o.type === 'tool_use') {
+      activities.push({ kind: 'tool_start', name });
+      if (isRecord(o.input)) activities.push({ kind: 'tool_input', name, input: o.input as Record<string, unknown> });
+    } else {
+      activities.push({ kind: 'tool_result', isError: o.is_error === true });
+    }
+  }
+  if (typeStr.includes('thinking')) activities.push({ kind: 'thinking', chars: 10 });
 
-	const text = extractAmpText(o);
-	if (text && !typeStr.includes('tool')) streamedText = text;
+  const text = extractAmpText(o);
+  if (text && !typeStr.includes('tool')) streamedText = text;
 
-	if (o.type === 'result' || o.type === 'done' || o.type === 'completed') {
-		const usage = isRecord(o.usage) ? o.usage : null;
-		const result: StreamedResult = {
-			result: typeof o.result === 'string' ? o.result : state.streamedText + (streamedText ?? ''),
-			isError: o.is_error === true,
-			numTurns: typeof o.num_turns === 'number' ? o.num_turns : 0,
-			totalCostUsd: typeof o.total_cost_usd === 'number' ? o.total_cost_usd : 0,
-			sessionId: typeof o.session_id === 'string' ? o.session_id : null,
-			stopReason: typeof o.stop_reason === 'string' ? o.stop_reason : null,
-			permissionDenials: [],
-			durationMs: typeof o.duration_ms === 'number' ? o.duration_ms : null,
-			durationApiMs: null,
-			ttftMs: null,
-			model: typeof o.model === 'string' ? o.model : null,
-			contextWindow: null,
-			maxOutputTokens: null,
-			usage: usage
-				? {
-						inputTokens: typeof usage.input_tokens === 'number' ? usage.input_tokens : 0,
-						outputTokens: typeof usage.output_tokens === 'number' ? usage.output_tokens : 0,
-						cacheCreationInputTokens: 0,
-						cacheReadInputTokens: 0,
-					}
-				: null,
-		};
-		return { activities, streamedText, result };
-	}
-	return { activities, streamedText };
+  if (o.type === 'result' || o.type === 'done' || o.type === 'completed') {
+    const usage = isRecord(o.usage) ? o.usage : null;
+    const result: StreamedResult = {
+      result: typeof o.result === 'string' ? o.result : state.streamedText + (streamedText ?? ''),
+      isError: o.is_error === true,
+      numTurns: typeof o.num_turns === 'number' ? o.num_turns : 0,
+      totalCostUsd: typeof o.total_cost_usd === 'number' ? o.total_cost_usd : 0,
+      sessionId: typeof o.session_id === 'string' ? o.session_id : null,
+      stopReason: typeof o.stop_reason === 'string' ? o.stop_reason : null,
+      permissionDenials: [],
+      durationMs: typeof o.duration_ms === 'number' ? o.duration_ms : null,
+      durationApiMs: null,
+      ttftMs: null,
+      model: typeof o.model === 'string' ? o.model : null,
+      contextWindow: null,
+      maxOutputTokens: null,
+      usage: usage
+        ? {
+            inputTokens: typeof usage.input_tokens === 'number' ? usage.input_tokens : 0,
+            outputTokens: typeof usage.output_tokens === 'number' ? usage.output_tokens : 0,
+            cacheCreationInputTokens: 0,
+            cacheReadInputTokens: 0,
+          }
+        : null,
+    };
+    return { activities, streamedText, result };
+  }
+  return { activities, streamedText };
 }
 
 export const ampHarness: Harness = {
-	name: 'amp',
-	displayName: 'Amp',
-	binary: 'amp',
-	async detect() {
-		try {
-			const { stdout } = await execFileAsync('amp', ['--version'], { timeout: 5000 });
-			return { ok: true, version: stdout.trim() };
-		} catch {
-			try {
-				const { stdout } = await execFileAsync('omp', ['--version'], { timeout: 5000 });
-				return { ok: true, version: stdout.trim() };
-			} catch {
-				return { ok: false, hint: 'Install Amp: https://ampcode.com' };
-			}
-		}
-	},
-	buildArgs(opts: BuildArgsOpts): string[] {
-		const perm = opts.nativePermission ?? PERMISSION_MAP[opts.permission] ?? 'workspace';
-		const args = ['--output', 'jsonl', opts.prompt, '--permission', perm];
-		if (opts.model) args.push('--model', opts.model);
-		if (opts.resumeSessionId) args.push('--resume', opts.resumeSessionId);
-		for (const dir of opts.addDirs ?? []) args.push('--add-dir', dir);
-		return args;
-	},
-	parseLine(line: string, state: ParseState): ParseOutcome {
-		return parseAmpLine(line, state);
-	},
-	extractResult(state: ParseState): StreamedResult | null {
-		if (state.result) return state.result;
-		if (state.streamedText.trim().length > 0) {
-			return {
-				result: state.streamedText,
-				isError: false,
-				numTurns: 1,
-				totalCostUsd: 0,
-				sessionId: null,
-				stopReason: null,
-				permissionDenials: [],
-				durationMs: null,
-				durationApiMs: null,
-				ttftMs: null,
-				model: null,
-				contextWindow: null,
-				maxOutputTokens: null,
-				usage: null,
-			};
-		}
-		return null;
-	},
-	permissionMap: {
-		readonly: ['read-only'],
-		edit: ['workspace'],
-		danger: ['danger'],
-	},
+  name: 'amp',
+  displayName: 'Amp',
+  binary: 'amp',
+  async detect() {
+    try {
+      const { stdout } = await execFileAsync('amp', ['--version'], { timeout: 5000 });
+      return { ok: true, version: stdout.trim() };
+    } catch {
+      try {
+        const { stdout } = await execFileAsync('omp', ['--version'], { timeout: 5000 });
+        return { ok: true, version: stdout.trim() };
+      } catch {
+        return { ok: false, hint: 'Install Amp: https://ampcode.com' };
+      }
+    }
+  },
+  buildArgs(opts: BuildArgsOpts): string[] {
+    const perm = opts.nativePermission ?? PERMISSION_MAP[opts.permission] ?? 'workspace';
+    const args = ['--output', 'jsonl', opts.prompt, '--permission', perm];
+    if (opts.model) args.push('--model', opts.model);
+    if (opts.resumeSessionId) args.push('--resume', opts.resumeSessionId);
+    for (const dir of opts.addDirs ?? []) args.push('--add-dir', dir);
+    return args;
+  },
+  parseLine(line: string, state: ParseState): ParseOutcome {
+    return parseAmpLine(line, state);
+  },
+  extractResult(state: ParseState): StreamedResult | null {
+    if (state.result) return state.result;
+    if (state.streamedText.trim().length > 0) {
+      return {
+        result: state.streamedText,
+        isError: false,
+        numTurns: 1,
+        totalCostUsd: 0,
+        sessionId: null,
+        stopReason: null,
+        permissionDenials: [],
+        durationMs: null,
+        durationApiMs: null,
+        ttftMs: null,
+        model: null,
+        contextWindow: null,
+        maxOutputTokens: null,
+        usage: null,
+      };
+    }
+    return null;
+  },
+  permissionMap: {
+    readonly: ['read-only'],
+    edit: ['workspace'],
+    danger: ['danger'],
+  },
 };

package/extensions/harnesses/claude.ts

@@ -1,129 +1,154 @@
 import { execFile } from 'node:child_process';
 import { promisify } from 'node:util';
-import type { Harness, BuildArgsOpts, NormalizedPermission, ParseOutcome, ParseState, StreamedResult } from './types.ts';
+import type {
+  BuildArgsOpts,
+  Harness,
+  NormalizedPermission,
+  ParseOutcome,
+  ParseState,
+  StreamedResult,
+} from './types.ts';
 
 const execFileAsync = promisify(execFile);
 
 function isRecord(v: unknown): v is Record<string, unknown> {
-	return typeof v === 'object' && v !== null && !Array.isArray(v);
+  return typeof v === 'object' && v !== null && !Array.isArray(v);
 }
 
 const PERMISSION_MAP: Record<NormalizedPermission, string> = {
-	readonly: 'plan',
-	edit: 'acceptEdits',
-	danger: 'bypassPermissions',
+  readonly: 'plan',
+  edit: 'acceptEdits',
+  danger: 'bypassPermissions',
 };
 
 export function parseClaudeLine(line: string, state: ParseState): ParseOutcome {
-	let o: unknown;
-	try {
-		o = JSON.parse(line);
-	} catch {
-		return { activities: [] };
-	}
-	if (!isRecord(o)) return { activities: [] };
-	const activities: ParseOutcome['activities'] = [];
-	let streamedText: string | undefined;
+  let o: unknown;
+  try {
+    o = JSON.parse(line);
+  } catch {
+    return { activities: [] };
+  }
+  if (!isRecord(o)) return { activities: [] };
+  const activities: ParseOutcome['activities'] = [];
+  let streamedText: string | undefined;
 
-	if (o.type === 'stream_event' && isRecord(o.event)) {
-		const ev = o.event;
-		const delta = isRecord(ev.delta) ? ev.delta : undefined;
-		if (ev.type === 'content_block_delta' && delta?.type === 'text_delta' && typeof delta.text === 'string') {
-			streamedText = delta.text;
-		} else if (ev.type === 'content_block_delta' && delta?.type === 'thinking_delta' && typeof delta.thinking === 'string') {
-			activities.push({ kind: 'thinking', chars: delta.thinking.length });
-		} else if (ev.type === 'content_block_start' && isRecord(ev.content_block)) {
-			const cb = ev.content_block;
-			if (cb.type === 'tool_use' && typeof cb.name === 'string') {
-				activities.push({ kind: 'tool_start', name: cb.name });
-			}
-		}
-	} else if (o.type === 'assistant' && isRecord(o.message)) {
-		for (const block of Array.isArray(o.message.content) ? o.message.content : []) {
-			if (isRecord(block) && block.type === 'tool_use' && typeof block.name === 'string') {
-				activities.push({ kind: 'tool_input', name: block.name, input: isRecord(block.input) ? block.input : {} });
-			}
-		}
-	} else if (o.type === 'user' && isRecord(o.message)) {
-		for (const block of Array.isArray(o.message.content) ? o.message.content : []) {
-			if (isRecord(block) && block.type === 'tool_result') {
-				activities.push({ kind: 'tool_result', isError: block.is_error === true });
-			}
-		}
-	} else if (o.type === 'result') {
-		const u = isRecord(o.usage) ? o.usage : null;
-		let model: string | null = null;
-		let contextWindow: number | null = null;
-		let maxOutputTokens: number | null = null;
-		if (isRecord(o.modelUsage)) {
-			const first = Object.entries(o.modelUsage)[0]?.[1];
-			const firstKey = Object.keys(o.modelUsage)[0];
-			if (firstKey) model = firstKey;
-			if (isRecord(first)) {
-				if (typeof first.contextWindow === 'number') contextWindow = first.contextWindow;
-				if (typeof first.maxOutputTokens === 'number') maxOutputTokens = first.maxOutputTokens;
-			}
-		}
-		const result: StreamedResult = {
-			result: typeof o.result === 'string' ? o.result : state.streamedText,
-			isError: o.is_error === true,
-			numTurns: typeof o.num_turns === 'number' ? o.num_turns : 0,
-			totalCostUsd: typeof o.total_cost_usd === 'number' ? o.total_cost_usd : 0,
-			sessionId: typeof o.session_id === 'string' ? o.session_id : null,
-			stopReason: typeof o.stop_reason === 'string' ? o.stop_reason : null,
-			permissionDenials: Array.isArray(o.permission_denials) ? o.permission_denials : [],
-			durationMs: typeof o.duration_ms === 'number' ? o.duration_ms : null,
-			durationApiMs: typeof o.duration_api_ms === 'number' ? o.duration_api_ms : null,
-			ttftMs: typeof o.ttft_ms === 'number' ? o.ttft_ms : null,
-			model,
-			contextWindow,
-			maxOutputTokens,
-			usage: u
-				? {
-						inputTokens: typeof u.input_tokens === 'number' ? u.input_tokens : 0,
-						outputTokens: typeof u.output_tokens === 'number' ? u.output_tokens : 0,
-						cacheCreationInputTokens: typeof u.cache_creation_input_tokens === 'number' ? u.cache_creation_input_tokens : 0,
-						cacheReadInputTokens: typeof u.cache_read_input_tokens === 'number' ? u.cache_read_input_tokens : 0,
-					}
-				: null,
-		};
-		return { activities, streamedText, result };
-	}
+  if (o.type === 'stream_event' && isRecord(o.event)) {
+    const ev = o.event;
+    const delta = isRecord(ev.delta) ? ev.delta : undefined;
+    if (ev.type === 'content_block_delta' && delta?.type === 'text_delta' && typeof delta.text === 'string') {
+      streamedText = delta.text;
+    } else if (
+      ev.type === 'content_block_delta' &&
+      delta?.type === 'thinking_delta' &&
+      typeof delta.thinking === 'string'
+    ) {
+      activities.push({ kind: 'thinking', chars: delta.thinking.length });
+    } else if (ev.type === 'content_block_start' && isRecord(ev.content_block)) {
+      const cb = ev.content_block;
+      if (cb.type === 'tool_use' && typeof cb.name === 'string') {
+        activities.push({ kind: 'tool_start', name: cb.name });
+      }
+    }
+  } else if (o.type === 'assistant' && isRecord(o.message)) {
+    for (const block of Array.isArray(o.message.content) ? o.message.content : []) {
+      if (isRecord(block) && block.type === 'tool_use' && typeof block.name === 'string') {
+        activities.push({
+          kind: 'tool_input',
+          name: block.name,
+          input: isRecord(block.input) ? block.input : {},
+        });
+      }
+    }
+  } else if (o.type === 'user' && isRecord(o.message)) {
+    for (const block of Array.isArray(o.message.content) ? o.message.content : []) {
+      if (isRecord(block) && block.type === 'tool_result') {
+        activities.push({ kind: 'tool_result', isError: block.is_error === true });
+      }
+    }
+  } else if (o.type === 'result') {
+    const u = isRecord(o.usage) ? o.usage : null;
+    let model: string | null = null;
+    let contextWindow: number | null = null;
+    let maxOutputTokens: number | null = null;
+    if (isRecord(o.modelUsage)) {
+      const first = Object.entries(o.modelUsage)[0]?.[1];
+      const firstKey = Object.keys(o.modelUsage)[0];
+      if (firstKey) model = firstKey;
+      if (isRecord(first)) {
+        if (typeof first.contextWindow === 'number') contextWindow = first.contextWindow;
+        if (typeof first.maxOutputTokens === 'number') maxOutputTokens = first.maxOutputTokens;
+      }
+    }
+    const result: StreamedResult = {
+      result: typeof o.result === 'string' ? o.result : state.streamedText,
+      isError: o.is_error === true,
+      numTurns: typeof o.num_turns === 'number' ? o.num_turns : 0,
+      totalCostUsd: typeof o.total_cost_usd === 'number' ? o.total_cost_usd : 0,
+      sessionId: typeof o.session_id === 'string' ? o.session_id : null,
+      stopReason: typeof o.stop_reason === 'string' ? o.stop_reason : null,
+      permissionDenials: Array.isArray(o.permission_denials) ? o.permission_denials : [],
+      durationMs: typeof o.duration_ms === 'number' ? o.duration_ms : null,
+      durationApiMs: typeof o.duration_api_ms === 'number' ? o.duration_api_ms : null,
+      ttftMs: typeof o.ttft_ms === 'number' ? o.ttft_ms : null,
+      model,
+      contextWindow,
+      maxOutputTokens,
+      usage: u
+        ? {
+            inputTokens: typeof u.input_tokens === 'number' ? u.input_tokens : 0,
+            outputTokens: typeof u.output_tokens === 'number' ? u.output_tokens : 0,
+            cacheCreationInputTokens:
+              typeof u.cache_creation_input_tokens === 'number' ? u.cache_creation_input_tokens : 0,
+            cacheReadInputTokens: typeof u.cache_read_input_tokens === 'number' ? u.cache_read_input_tokens : 0,
+          }
+        : null,
+    };
+    return { activities, streamedText, result };
+  }
 
-	return { activities, streamedText };
+  return { activities, streamedText };
 }
 
 export const claudeHarness: Harness = {
-	name: 'claude',
-	displayName: 'Claude Code',
-	binary: 'claude',
-	async detect() {
-		try {
-			const { stdout } = await execFileAsync('claude', ['--version'], { timeout: 5000 });
-			return { ok: true, version: stdout.trim() };
-		} catch {
-			return { ok: false, hint: 'Install Claude Code: https://docs.anthropic.com/en/docs/claude-code' };
-		}
-	},
-	buildArgs(opts: BuildArgsOpts): string[] {
-		const mode = opts.nativePermission ?? PERMISSION_MAP[opts.permission] ?? 'acceptEdits';
-		const args = ['-p', opts.prompt, '--output-format', 'stream-json', '--verbose', '--include-partial-messages', '--permission-mode', mode];
-		if (opts.resumeSessionId) args.push('--resume', opts.resumeSessionId);
-		else args.push('--no-session-persistence');
-		if (opts.model) args.push('--model', opts.model);
-		if (opts.maxBudgetUsd !== undefined) args.push('--max-budget-usd', String(opts.maxBudgetUsd));
-		for (const dir of opts.addDirs ?? []) args.push('--add-dir', dir);
-		return args;
-	},
-	parseLine(line: string, state: ParseState): ParseOutcome {
-		return parseClaudeLine(line, state);
-	},
-	extractResult(state: ParseState): StreamedResult | null {
-		return state.result;
-	},
-	permissionMap: {
-		readonly: ['plan'],
-		edit: ['acceptEdits'],
-		danger: ['bypassPermissions'],
-	},
+  name: 'claude',
+  displayName: 'Claude Code',
+  binary: 'claude',
+  async detect() {
+    try {
+      const { stdout } = await execFileAsync('claude', ['--version'], { timeout: 5000 });
+      return { ok: true, version: stdout.trim() };
+    } catch {
+      return { ok: false, hint: 'Install Claude Code: https://docs.anthropic.com/en/docs/claude-code' };
+    }
+  },
+  buildArgs(opts: BuildArgsOpts): string[] {
+    const mode = opts.nativePermission ?? PERMISSION_MAP[opts.permission] ?? 'acceptEdits';
+    const args = [
+      '-p',
+      opts.prompt,
+      '--output-format',
+      'stream-json',
+      '--verbose',
+      '--include-partial-messages',
+      '--permission-mode',
+      mode,
+    ];
+    if (opts.resumeSessionId) args.push('--resume', opts.resumeSessionId);
+    else args.push('--no-session-persistence');
+    if (opts.model) args.push('--model', opts.model);
+    if (opts.maxBudgetUsd !== undefined) args.push('--max-budget-usd', String(opts.maxBudgetUsd));
+    for (const dir of opts.addDirs ?? []) args.push('--add-dir', dir);
+    return args;
+  },
+  parseLine(line: string, state: ParseState): ParseOutcome {
+    return parseClaudeLine(line, state);
+  },
+  extractResult(state: ParseState): StreamedResult | null {
+    return state.result;
+  },
+  permissionMap: {
+    readonly: ['plan'],
+    edit: ['acceptEdits'],
+    danger: ['bypassPermissions'],
+  },
 };