1import type { CodexRunResult, ToolCall, TokenUsage } from "./types.ts";
2 
3export function parseEventStream(raw: string): Omit<CodexRunResult, "rawLogPath" | "durationMs"> {
4  const lines = raw.split("\n").filter((l) => l.trim().length > 0);
5  let threadId: string | null = null;
6  let agentMessage: string | null = null;
7  let usage: TokenUsage | null = null;
8  const toolCallsById = new Map<string, ToolCall>();
9 
10  for (const line of lines) {
11    let event: any;
12    try {
13      event = JSON.parse(line);
14    } catch {
15      continue;
16    }
17    const type = event?.type;
18    if (type === "thread.started") {
19      threadId = event.thread_id ?? null;
20    } else if (type === "item.started" || type === "item.completed") {
21      const item = event.item;
22      if (!item?.id) continue;
23      const tc: ToolCall = {
24        id: item.id,
25        tool: deriveToolName(item),
26        status: item.status ?? (type === "item.completed" ? "completed" : "in_progress"),
27        command: item.command,
28      };
29      toolCallsById.set(item.id, tc);
30      if (item.type === "agent_message" && type === "item.completed") {
31        agentMessage = String(item.text ?? "");
32      }
33    } else if (type === "turn.completed") {
34      const u = event.usage;
35      if (u) {
36        usage = {
37          input: u.input_tokens ?? 0,
38          cached_input: u.cached_input_tokens ?? 0,
39          output: u.output_tokens ?? 0,
40          reasoning: u.reasoning_output_tokens ?? 0,
41        };
42      }
43    }
44  }
45 
46  return {
47    threadId,
48    toolCalls: Array.from(toolCallsById.values()),
49    agentMessage,
50    usage,
51  };
52}
53 
54function deriveToolName(item: any): string {
55  if (item.type === "command_execution") return "shell";
56  if (item.type === "agent_message") return "agent_message";
57  if (item.type === "image_gen" || item.type === "image_generation") return "image_gen";
58  if (typeof item.tool === "string") return item.tool;
59  return item.type ?? "unknown";
60}
61 
62export function hasImageGenInvocation(toolCalls: ToolCall[]): boolean {
63  return toolCalls.some((tc) => tc.tool === "image_gen");
64}
65