Skip to main content

call_coding_clis/output/
parse.rs

1use super::model::{Event, ToolCall, ToolResult, Transcript, Usage};
2use crate::invoke::RunnerKind;
3
4pub fn parse_transcript(raw: &str, schema: &str) -> Transcript {
5    transcript_from_parsed(crate::json_output::parse_json_output(raw, schema))
6}
7
8pub fn parse_transcript_for_runner(raw: &str, runner: RunnerKind) -> Option<Transcript> {
9    schema_name_for_runner(runner).map(|schema| parse_transcript(raw, schema))
10}
11
12pub fn schema_name_for_runner(runner: RunnerKind) -> Option<&'static str> {
13    match runner {
14        RunnerKind::OpenCode => Some("opencode"),
15        RunnerKind::Claude => Some("claude-code"),
16        RunnerKind::Codex => Some("codex"),
17        RunnerKind::Kimi => Some("kimi"),
18        RunnerKind::Cursor => Some("cursor-agent"),
19        RunnerKind::Gemini => Some("gemini"),
20        RunnerKind::Pi => Some("pi"),
21        RunnerKind::RooCode | RunnerKind::Crush => None,
22    }
23}
24
25pub fn transcript_from_parsed(parsed: crate::json_output::ParsedJsonOutput) -> Transcript {
26    Transcript {
27        events: parsed.events.into_iter().map(convert_event).collect(),
28        final_text: parsed.final_text,
29        session_id: (!parsed.session_id.is_empty()).then_some(parsed.session_id),
30        usage: Usage {
31            counts: parsed.usage,
32            cost_usd: parsed.cost_usd,
33            duration_ms: parsed.duration_ms,
34        },
35        error: (!parsed.error.is_empty()).then_some(parsed.error),
36        unknown_json_lines: parsed.unknown_json_lines,
37    }
38}
39
40fn convert_event(event: crate::json_output::JsonEvent) -> Event {
41    if let Some(tool_call) = event.tool_call {
42        return Event::ToolCall(ToolCall {
43            id: tool_call.id,
44            name: tool_call.name,
45            arguments: tool_call.arguments,
46        });
47    }
48    if let Some(tool_result) = event.tool_result {
49        return Event::ToolResult(ToolResult {
50            tool_call_id: tool_result.tool_call_id,
51            content: tool_result.content,
52            is_error: tool_result.is_error,
53        });
54    }
55    if !event.thinking.is_empty() {
56        return Event::Thinking(event.thinking);
57    }
58    match event.event_type.as_str() {
59        "error" => Event::Error(event.text),
60        "raw_unknown_json" => Event::RawUnknownJson(event.text),
61        _ => Event::Text(event.text),
62    }
63}