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::RooCode | RunnerKind::Crush => None,
21    }
22}
23
24pub fn transcript_from_parsed(parsed: crate::json_output::ParsedJsonOutput) -> Transcript {
25    Transcript {
26        events: parsed.events.into_iter().map(convert_event).collect(),
27        final_text: parsed.final_text,
28        session_id: (!parsed.session_id.is_empty()).then_some(parsed.session_id),
29        usage: Usage {
30            counts: parsed.usage,
31            cost_usd: parsed.cost_usd,
32            duration_ms: parsed.duration_ms,
33        },
34        error: (!parsed.error.is_empty()).then_some(parsed.error),
35        unknown_json_lines: parsed.unknown_json_lines,
36    }
37}
38
39fn convert_event(event: crate::json_output::JsonEvent) -> Event {
40    if let Some(tool_call) = event.tool_call {
41        return Event::ToolCall(ToolCall {
42            id: tool_call.id,
43            name: tool_call.name,
44            arguments: tool_call.arguments,
45        });
46    }
47    if let Some(tool_result) = event.tool_result {
48        return Event::ToolResult(ToolResult {
49            tool_call_id: tool_result.tool_call_id,
50            content: tool_result.content,
51            is_error: tool_result.is_error,
52        });
53    }
54    if !event.thinking.is_empty() {
55        return Event::Thinking(event.thinking);
56    }
57    match event.event_type.as_str() {
58        "error" => Event::Error(event.text),
59        "raw_unknown_json" => Event::RawUnknownJson(event.text),
60        _ => Event::Text(event.text),
61    }
62}