Skip to main content

toolpath_claude/
provider.rs

1//! Implementation of `toolpath-convo` traits for Claude conversations.
2//!
3//! Handles cross-entry tool result assembly: Claude's JSONL format writes
4//! tool invocations and their results as separate entries. This module
5//! pairs them by `tool_use_id` so consumers get complete `Turn` values
6//! with `ToolInvocation.result` populated.
7
8use std::collections::HashMap;
9
10use crate::ClaudeConvo;
11use crate::types::{Conversation, ConversationEntry, Message, MessageContent, MessageRole};
12#[cfg(any(feature = "watcher", test))]
13use toolpath_convo::WatcherEvent;
14use toolpath_convo::{
15    ConversationMeta, ConversationProvider, ConversationView, ConvoError, DelegatedWork,
16    EnvironmentSnapshot, Role, TokenUsage, ToolCategory, ToolInvocation, ToolResult, Turn,
17};
18
19// ── Conversion helpers ───────────────────────────────────────────────
20
21fn claude_role_to_role(role: &MessageRole) -> Role {
22    match role {
23        MessageRole::User => Role::User,
24        MessageRole::Assistant => Role::Assistant,
25        MessageRole::System => Role::System,
26    }
27}
28
29/// Classify a Claude Code tool into toolpath's category ontology.
30///
31/// Returns `None` for unrecognized tools. When Claude Code adds or
32/// renames tools, update this map.
33pub fn tool_category(name: &str) -> Option<ToolCategory> {
34    match name {
35        "Read" => Some(ToolCategory::FileRead),
36        "Glob" | "Grep" => Some(ToolCategory::FileSearch),
37        "Write" | "Edit" | "MultiEdit" | "NotebookEdit" => Some(ToolCategory::FileWrite),
38        "Bash" => Some(ToolCategory::Shell),
39        "WebFetch" | "WebSearch" => Some(ToolCategory::Network),
40        "Task" | "Agent" => Some(ToolCategory::Delegation),
41        _ => None,
42    }
43}
44
45/// Reverse of [`tool_category`]: pick Claude's native tool name for a
46/// given [`ToolCategory`], disambiguating by `args` shape where needed
47/// (e.g. `Edit` vs `Write`, `Glob` vs `Grep`).
48///
49/// Returns `None` when no Claude-canonical equivalent exists. Mirrors
50/// the `provider::native_name` helpers on opencode / codex / gemini /
51/// pi — projectors call it to surface cross-harness tool calls under
52/// the names Claude Code's UI knows how to render.
53pub fn native_name(category: ToolCategory, args: &serde_json::Value) -> Option<&'static str> {
54    let has = |k: &str| args.get(k).is_some();
55    match category {
56        ToolCategory::Shell => Some("Bash"),
57        ToolCategory::FileRead => Some("Read"),
58        ToolCategory::FileWrite => Some(if has("old_string") || has("oldString") {
59            "Edit"
60        } else {
61            "Write"
62        }),
63        ToolCategory::FileSearch => Some(
64            // Grep takes a regex `pattern` and often has output_mode/type
65            // hints; Glob takes a glob pattern. When ambiguous, default to
66            // Glob — its file-list rendering at least shows results.
67            if has("output_mode") || has("path_pattern") || has("type") {
68                "Grep"
69            } else {
70                "Glob"
71            },
72        ),
73        ToolCategory::Network => Some(if has("url") { "WebFetch" } else { "WebSearch" }),
74        ToolCategory::Delegation => Some("Task"),
75    }
76}
77
78/// Convert a single entry to a Turn without cross-entry assembly.
79/// Tool results within the same message are still matched.
80fn message_to_turn(entry: &ConversationEntry, msg: &Message) -> Turn {
81    let text = msg.text();
82
83    let thinking = msg.thinking().map(|parts| parts.join("\n"));
84
85    let tool_uses: Vec<ToolInvocation> = msg
86        .tool_uses()
87        .into_iter()
88        .map(|tu| {
89            let result = find_tool_result_in_parts(msg, tu.id);
90            let category = tool_category(tu.name);
91            ToolInvocation {
92                id: tu.id.to_string(),
93                name: tu.name.to_string(),
94                input: tu.input.clone(),
95                result,
96                category,
97            }
98        })
99        .collect();
100
101    let file_mutations = compute_file_mutations(&tool_uses, entry.cwd.as_deref());
102
103    let token_usage = msg.usage.as_ref().map(|u| TokenUsage {
104        input_tokens: u.input_tokens,
105        output_tokens: u.output_tokens,
106        cache_read_tokens: u.cache_read_input_tokens,
107        cache_write_tokens: u.cache_creation_input_tokens,
108        ..Default::default()
109    });
110
111    let environment = if entry.cwd.is_some() || entry.git_branch.is_some() {
112        Some(EnvironmentSnapshot {
113            working_dir: entry.cwd.clone(),
114            vcs_branch: entry.git_branch.clone(),
115            vcs_revision: None,
116        })
117    } else {
118        None
119    };
120
121    let delegations = extract_delegations(&tool_uses);
122
123    Turn {
124        id: entry.uuid.clone(),
125        parent_id: entry.parent_uuid.clone(),
126        // The API message ID (`msg_…`). Claude Code writes one JSONL line
127        // per content block, so several turns can share one group_id —
128        // and each repeats the message-level `usage`. Downstream accounting
129        // (sum_usage, derive_path) counts a message group once.
130        group_id: msg.id.clone(),
131        role: claude_role_to_role(&msg.role),
132        timestamp: entry.timestamp.clone(),
133        text,
134        thinking,
135        tool_uses,
136        model: msg.model.clone(),
137        stop_reason: msg.stop_reason.clone(),
138        token_usage,
139        attributed_token_usage: None,
140        environment,
141        delegations,
142        file_mutations,
143    }
144}
145
146/// For each file-write tool invocation in the turn, synthesize a unified
147/// diff via [`toolpath_convo::file_write_diff`] and pre-resolve the
148/// before-state for `Write` via `git show HEAD:<path>` (best-effort).
149/// Each mutation links back to its tool via `tool_id`.
150fn compute_file_mutations(
151    tool_uses: &[ToolInvocation],
152    cwd: Option<&str>,
153) -> Vec<toolpath_convo::FileMutation> {
154    let mut out = Vec::new();
155    for tu in tool_uses {
156        if tu.category != Some(ToolCategory::FileWrite) {
157            continue;
158        }
159        let Some(path) = extract_file_path_for_tool(&tu.input) else {
160            continue;
161        };
162        // Only `Write` carries whole-file content; consult git HEAD for
163        // its pre-image so the diff isn't addition-only. Other tools
164        // (Edit / MultiEdit / NotebookEdit) carry old_string/new_string
165        // pairs and don't need a before-state lookup.
166        let before_state = if tu.name == "Write" {
167            cwd.and_then(|c| git_head_content(c, &path))
168        } else {
169            None
170        };
171        let raw_diff =
172            toolpath_convo::file_write_diff(&tu.name, &tu.input, &path, before_state.as_deref());
173        let operation = match tu.name.as_str() {
174            "Write" => Some("add".to_string()),
175            "Edit" | "MultiEdit" | "NotebookEdit" => Some("update".to_string()),
176            _ => None,
177        };
178        let after = match tu.name.as_str() {
179            "Write" => tu
180                .input
181                .get("content")
182                .and_then(|v| v.as_str())
183                .map(|s| s.to_string()),
184            _ => None,
185        };
186        out.push(toolpath_convo::FileMutation {
187            path,
188            tool_id: Some(tu.id.clone()),
189            operation,
190            raw_diff,
191            before: before_state,
192            after,
193            rename_to: None,
194        });
195    }
196    out
197}
198
199/// Best-effort lookup of a file's contents at `HEAD` in the git repo
200/// rooted at `repo_dir` (or one of its ancestors). Shells out to `git
201/// show HEAD:<relative-path>`. Returns `None` when any of these hold:
202/// `repo_dir` isn't inside a git repo, `path` isn't tracked at `HEAD`,
203/// `git` isn't on `PATH`, or the command otherwise fails.
204fn git_head_content(repo_dir: &str, path: &str) -> Option<String> {
205    use std::path::Path as FsPath;
206    use std::process::Command;
207    let repo = FsPath::new(repo_dir);
208    let file = FsPath::new(path);
209    let rel = if file.is_absolute() {
210        file.strip_prefix(repo).ok()?.to_path_buf()
211    } else {
212        file.to_path_buf()
213    };
214    let rel_str = rel.to_string_lossy().replace('\\', "/");
215    let output = Command::new("git")
216        .arg("-C")
217        .arg(repo)
218        .arg("show")
219        .arg(format!("HEAD:{rel_str}"))
220        .output()
221        .ok()?;
222    if !output.status.success() {
223        return None;
224    }
225    String::from_utf8(output.stdout).ok()
226}
227
228fn extract_file_path_for_tool(input: &serde_json::Value) -> Option<String> {
229    for k in ["file_path", "path", "filename", "file"] {
230        if let Some(s) = input.get(k).and_then(|v| v.as_str()) {
231            return Some(s.to_string());
232        }
233    }
234    None
235}
236
237/// Extract delegation info from Task tool invocations.
238fn extract_delegations(tool_uses: &[ToolInvocation]) -> Vec<DelegatedWork> {
239    tool_uses
240        .iter()
241        .filter(|tu| tu.category == Some(ToolCategory::Delegation))
242        .map(|tu| DelegatedWork {
243            agent_id: tu.id.clone(),
244            prompt: tu
245                .input
246                .get("prompt")
247                .and_then(|v| v.as_str())
248                .unwrap_or("")
249                .to_string(),
250            turns: vec![],
251            result: tu.result.as_ref().map(|r| r.content.clone()),
252        })
253        .collect()
254}
255
256fn find_tool_result_in_parts(msg: &Message, tool_use_id: &str) -> Option<ToolResult> {
257    let parts = match &msg.content {
258        Some(MessageContent::Parts(parts)) => parts,
259        _ => return None,
260    };
261    parts.iter().find_map(|p| match p {
262        crate::types::ContentPart::ToolResult {
263            tool_use_id: id,
264            content,
265            is_error,
266        } if id == tool_use_id => Some(ToolResult {
267            content: content.text(),
268            is_error: *is_error,
269        }),
270        _ => None,
271    })
272}
273
274/// Returns true if this entry is a tool-result-only user message
275/// (no human-authored text, only tool_result parts).
276fn is_tool_result_only(entry: &ConversationEntry) -> bool {
277    let Some(msg) = &entry.message else {
278        return false;
279    };
280    msg.role == MessageRole::User && msg.text().is_empty() && !msg.tool_results().is_empty()
281}
282
283/// Merge tool results from a tool-result-only message into existing turns.
284///
285/// Matches by `tool_use_id` — scans backwards through turns to find the
286/// `ToolInvocation` with a matching `id` for each result. This handles
287/// cases where a single result entry carries results for tool uses from
288/// different assistant turns.
289///
290/// Returns true if any results were merged.
291fn merge_tool_results(turns: &mut [Turn], msg: &Message) -> bool {
292    let mut merged = false;
293    for tr in msg.tool_results() {
294        for turn in turns.iter_mut().rev() {
295            if let Some(invocation) = turn
296                .tool_uses
297                .iter_mut()
298                .find(|tu| tu.id == tr.tool_use_id && tu.result.is_none())
299            {
300                invocation.result = Some(ToolResult {
301                    content: tr.content.text(),
302                    is_error: tr.is_error,
303                });
304                merged = true;
305                break;
306            }
307        }
308    }
309    merged
310}
311
312fn entry_to_turn(entry: &ConversationEntry) -> Option<Turn> {
313    entry
314        .message
315        .as_ref()
316        .map(|msg| message_to_turn(entry, msg))
317}
318
319/// Convert a full conversation to a view with cross-entry tool result assembly.
320///
321/// Tool-result-only user entries are absorbed into the preceding assistant
322/// turn's `ToolInvocation.result` fields rather than emitted as separate turns.
323fn conversation_to_view(convo: &Conversation) -> ConversationView {
324    let mut turns: Vec<Turn> = Vec::new();
325    let mut events: Vec<toolpath_convo::ConversationEvent> = Vec::new();
326
327    // Headerless preamble lines (ai-title, last-prompt, queue-operation,
328    // permission-mode, file-history-snapshot, etc.) become events so they
329    // round-trip back to JSONL.
330    for (idx, raw) in convo.preamble.iter().enumerate() {
331        events.push(preamble_to_event(idx, raw));
332    }
333
334    // Map from "absorbed-or-skipped entry UUID" → "the previous
335    // turn-bearing entry's UUID". Used so that an assistant turn whose
336    // wire parentUuid points at a tool-result-only entry (or any other
337    // absorbed entry that didn't become a Turn) gets a Turn.parent_id
338    // that still maps onto a real Turn — keeping the IR's turn-to-turn
339    // chain intact for `derive_path`. The original UUID is preserved
340    // via the `tool_result_user` event.
341    let mut parent_rewrites: HashMap<String, String> = HashMap::new();
342    let mut last_turn_uuid: Option<String> = None;
343
344    for entry in &convo.entries {
345        let Some(msg) = &entry.message else {
346            // Message-less entries (attachments, snapshots) survive as
347            // events so the projector can re-emit them.
348            events.push(entry_to_event(entry));
349            if let Some(prev) = &last_turn_uuid {
350                parent_rewrites.insert(entry.uuid.clone(), prev.clone());
351            }
352            continue;
353        };
354
355        // Tool-result-only user entries get merged into the preceding
356        // assistant's tool_uses[i].result and dropped from the turn
357        // stream. The next assistant entry's wire parentUuid points at
358        // this entry; we record a rewrite so the IR's turn-to-turn chain
359        // stays connected. (The projector re-synthesizes the wire-level
360        // tool-result entries on the way out from tool_uses[i].result —
361        // their original UUIDs aren't preserved across the roundtrip,
362        // but the Claude UI walks the chain by parentUuid, not by
363        // specific UUIDs, so that's fine.)
364        if is_tool_result_only(entry) {
365            merge_tool_results(&mut turns, msg);
366            if let Some(prev) = &last_turn_uuid {
367                parent_rewrites.insert(entry.uuid.clone(), prev.clone());
368            }
369            continue;
370        }
371
372        let mut turn = message_to_turn(entry, msg);
373        if let Some(pid) = turn.parent_id.as_ref()
374            && let Some(real) = parent_rewrites.get(pid)
375        {
376            turn.parent_id = Some(real.clone());
377        }
378        last_turn_uuid = Some(turn.id.clone());
379        turns.push(turn);
380    }
381
382    canonicalize_message_usage(&mut turns);
383
384    // Re-derive delegation results now that tool results are merged
385    for turn in &mut turns {
386        for delegation in &mut turn.delegations {
387            if delegation.result.is_none()
388                && let Some(tu) = turn
389                    .tool_uses
390                    .iter()
391                    .find(|tu| tu.id == delegation.agent_id)
392            {
393                delegation.result = tu.result.as_ref().map(|r| r.content.clone());
394            }
395        }
396    }
397
398    let total_usage = sum_usage(&turns);
399    let files_changed = extract_files_changed(&turns);
400
401    // Pull path-level base/producer from the first entry that carries the
402    // metadata (Claude records cwd / git_branch / version on every
403    // conversational entry; the first one is the canonical "this is where
404    // we started").
405    let mut base = toolpath_convo::SessionBase::default();
406    let mut producer_version: Option<String> = None;
407    for entry in &convo.entries {
408        if base.working_dir.is_none()
409            && let Some(cwd) = &entry.cwd
410        {
411            base.working_dir = Some(cwd.clone());
412        }
413        if base.vcs_branch.is_none()
414            && let Some(b) = &entry.git_branch
415        {
416            base.vcs_branch = Some(b.clone());
417        }
418        if producer_version.is_none()
419            && let Some(v) = &entry.version
420        {
421            producer_version = Some(v.clone());
422        }
423        if base.working_dir.is_some() && base.vcs_branch.is_some() && producer_version.is_some() {
424            break;
425        }
426    }
427    let view_base = if base.working_dir.is_some()
428        || base.vcs_branch.is_some()
429        || base.vcs_revision.is_some()
430        || base.vcs_remote.is_some()
431    {
432        Some(base)
433    } else {
434        None
435    };
436    let producer = producer_version.map(|v| toolpath_convo::ProducerInfo {
437        name: "claude-code".into(),
438        version: Some(v),
439    });
440
441    ConversationView {
442        id: convo.session_id.clone(),
443        started_at: convo.started_at,
444        last_activity: convo.last_activity,
445        turns,
446        total_usage,
447        provider_id: Some("claude-code".into()),
448        files_changed,
449        session_ids: vec![],
450        events,
451        base: view_base,
452        producer,
453    }
454}
455
456/// Build an event from a headerless preamble JSON line (`ai-title`,
457/// `last-prompt`, `queue-operation`, `permission-mode`, `file-history-snapshot`,
458/// or anything else above `entries` in Claude's JSONL).
459///
460/// The whole line is preserved verbatim under `data["raw"]`; the projector
461/// dumps it straight back onto `convo.preamble`. We don't model the shape —
462/// a headerless line is identified by the presence of `data["raw"]`, not by
463/// an enumerated `type` list. `event_type` carries the line's `type`, purely
464/// informational.
465fn preamble_to_event(idx: usize, raw: &serde_json::Value) -> toolpath_convo::ConversationEvent {
466    let event_type = raw
467        .get("type")
468        .and_then(|v| v.as_str())
469        .unwrap_or("preamble")
470        .to_string();
471    let timestamp = raw
472        .get("timestamp")
473        .and_then(|v| v.as_str())
474        .unwrap_or("")
475        .to_string();
476    let mut data: HashMap<String, serde_json::Value> = HashMap::new();
477    data.insert("raw".to_string(), raw.clone());
478    toolpath_convo::ConversationEvent {
479        id: format!("claude-preamble-{idx}"),
480        timestamp,
481        parent_id: None,
482        event_type,
483        data,
484    }
485}
486
487/// Build an event from a message-less ConversationEntry (attachment, snapshot).
488///
489/// Captures the entry's typed fields in `event.data` so the projector can
490/// reconstruct an equivalent entry. The flatten extras (e.g. an attachment's
491/// `attachment` payload) come along for the ride under `entry_extra`.
492fn entry_to_event(entry: &ConversationEntry) -> toolpath_convo::ConversationEvent {
493    let mut data = HashMap::new();
494    if let Some(v) = &entry.cwd {
495        data.insert("cwd".into(), serde_json::Value::String(v.clone()));
496    }
497    if let Some(v) = &entry.git_branch {
498        data.insert("git_branch".into(), serde_json::Value::String(v.clone()));
499    }
500    if let Some(v) = &entry.version {
501        data.insert("version".into(), serde_json::Value::String(v.clone()));
502    }
503    if let Some(v) = &entry.user_type {
504        data.insert("user_type".into(), serde_json::Value::String(v.clone()));
505    }
506    if let Some(v) = &entry.message_id {
507        data.insert("message_id".into(), serde_json::Value::String(v.clone()));
508    }
509    if let Some(v) = &entry.tool_use_result {
510        data.insert("tool_use_result".into(), v.clone());
511    }
512    if let Some(v) = &entry.snapshot {
513        data.insert("snapshot".into(), v.clone());
514    }
515    if !entry.extra.is_empty()
516        && let Ok(value) = serde_json::to_value(&entry.extra)
517    {
518        data.insert("entry_extra".into(), value);
519    }
520    toolpath_convo::ConversationEvent {
521        id: entry.uuid.clone(),
522        timestamp: entry.timestamp.clone(),
523        parent_id: entry.parent_uuid.clone(),
524        event_type: entry.entry_type.clone(),
525        data,
526    }
527}
528
529/// Field-wise maximum of two usage tuples. `None` is "absent", not 0, so a
530/// field present in only one operand survives.
531pub(crate) fn max_usage(a: &TokenUsage, b: &TokenUsage) -> TokenUsage {
532    fn m(x: Option<u32>, y: Option<u32>) -> Option<u32> {
533        match (x, y) {
534            (Some(a), Some(b)) => Some(a.max(b)),
535            (Some(v), None) | (None, Some(v)) => Some(v),
536            (None, None) => None,
537        }
538    }
539    TokenUsage {
540        input_tokens: m(a.input_tokens, b.input_tokens),
541        output_tokens: m(a.output_tokens, b.output_tokens),
542        cache_read_tokens: m(a.cache_read_tokens, b.cache_read_tokens),
543        cache_write_tokens: m(a.cache_write_tokens, b.cache_write_tokens),
544        ..Default::default()
545    }
546}
547
548/// Canonicalize message-level accounting for split messages.
549///
550/// Claude Code writes one JSONL line per content block of an assistant API
551/// message, each stamped with `message.usage`. That `usage` is a **streaming
552/// snapshot**, not a per-line bill: per the Anthropic streaming API,
553/// `message_start` seeds `output_tokens` near zero and each `message_delta`
554/// reports the running **cumulative** total, with the final value being the
555/// message total. So across a split message's lines, `input`/`cache` are
556/// constant and `output_tokens` climbs to the total on the final line —
557/// confirmed across every session sampled (~27% of multi-line messages vary;
558/// the rest repeat one value stamped after generation). The intermediate
559/// values are flush-time snapshots, **not** per-content-block costs (a real
560/// prose block routinely shows `output_tokens: 1`), so we do not derive
561/// per-step attribution from them, and — the format being undocumented — we
562/// do not trust line order.
563///
564/// For each consecutive `group_id` run this sets `token_usage` on the run's
565/// **final** turn to the field-wise **maximum** across the run (the message
566/// total — never under-counts whatever the stream order) and clears it from
567/// the others, so summing `token_usage` over turns yields session totals.
568fn canonicalize_message_usage(turns: &mut [Turn]) {
569    let mut i = 0;
570    while i < turns.len() {
571        let Some(mid) = turns[i].group_id.clone() else {
572            i += 1;
573            continue;
574        };
575        let mut j = i;
576        while j < turns.len() && turns[j].group_id.as_deref() == Some(mid.as_str()) {
577            j += 1;
578        }
579
580        // Message total = field-wise max across the run (the final streaming
581        // snapshot, found without trusting line order).
582        let mut total: Option<TokenUsage> = None;
583        for t in &turns[i..j] {
584            if let Some(u) = &t.token_usage {
585                total = Some(match total {
586                    Some(acc) => max_usage(&acc, u),
587                    None => u.clone(),
588                });
589            }
590        }
591
592        for t in &mut turns[i..j] {
593            t.token_usage = None;
594        }
595        if let Some(total) = total {
596            turns[j - 1].token_usage = Some(total);
597        }
598        i = j;
599    }
600}
601
602/// Sum token usage across all turns.
603fn sum_usage(turns: &[Turn]) -> Option<TokenUsage> {
604    let mut total = TokenUsage::default();
605    let mut any = false;
606    for (idx, turn) in turns.iter().enumerate() {
607        // Turns split from one provider message all repeat that message's
608        // usage; count it once, on the run's last turn.
609        if let Some(mid) = &turn.group_id
610            && turns
611                .get(idx + 1)
612                .is_some_and(|next| next.group_id.as_ref() == Some(mid))
613        {
614            continue;
615        }
616        if let Some(u) = &turn.token_usage {
617            any = true;
618            total.input_tokens =
619                Some(total.input_tokens.unwrap_or(0) + u.input_tokens.unwrap_or(0));
620            total.output_tokens =
621                Some(total.output_tokens.unwrap_or(0) + u.output_tokens.unwrap_or(0));
622            total.cache_read_tokens = match (total.cache_read_tokens, u.cache_read_tokens) {
623                (Some(a), Some(b)) => Some(a + b),
624                (Some(a), None) => Some(a),
625                (None, Some(b)) => Some(b),
626                (None, None) => None,
627            };
628            total.cache_write_tokens = match (total.cache_write_tokens, u.cache_write_tokens) {
629                (Some(a), Some(b)) => Some(a + b),
630                (Some(a), None) => Some(a),
631                (None, Some(b)) => Some(b),
632                (None, None) => None,
633            };
634        }
635    }
636    if any { Some(total) } else { None }
637}
638
639/// Extract deduplicated file paths from file-write tool invocations.
640fn extract_files_changed(turns: &[Turn]) -> Vec<String> {
641    let mut seen = std::collections::HashSet::new();
642    let mut files = Vec::new();
643    for turn in turns {
644        for tool_use in &turn.tool_uses {
645            if tool_use.category == Some(ToolCategory::FileWrite)
646                && let Some(path) = tool_use.input.get("file_path").and_then(|v| v.as_str())
647                && seen.insert(path.to_string())
648            {
649                files.push(path.to_string());
650            }
651        }
652    }
653    files
654}
655
656#[cfg(any(feature = "watcher", test))]
657fn entry_to_watcher_event(entry: &ConversationEntry) -> WatcherEvent {
658    match entry_to_turn(entry) {
659        Some(turn) => WatcherEvent::Turn(Box::new(turn)),
660        None => {
661            let mut data = serde_json::json!({
662                "uuid": entry.uuid,
663                "timestamp": entry.timestamp,
664            });
665            if !entry.extra.is_empty() {
666                data["claude"] = serde_json::to_value(&entry.extra).unwrap_or_default();
667            }
668            WatcherEvent::Progress {
669                kind: entry.entry_type.clone(),
670                data,
671            }
672        }
673    }
674}
675
676// ── ConversationProvider for ClaudeConvo ──────────────────────────────
677
678impl ConversationProvider for ClaudeConvo {
679    fn list_conversations(&self, project: &str) -> toolpath_convo::Result<Vec<String>> {
680        crate::ClaudeConvo::list_conversations(self, project)
681            .map_err(|e| ConvoError::Provider(e.to_string()))
682    }
683
684    fn load_conversation(
685        &self,
686        project: &str,
687        conversation_id: &str,
688    ) -> toolpath_convo::Result<ConversationView> {
689        let convo = self
690            .read_conversation(project, conversation_id)
691            .map_err(|e| ConvoError::Provider(e.to_string()))?;
692        let mut view = conversation_to_view(&convo);
693        view.session_ids = convo.session_ids.clone();
694        Ok(view)
695    }
696
697    fn load_metadata(
698        &self,
699        project: &str,
700        conversation_id: &str,
701    ) -> toolpath_convo::Result<ConversationMeta> {
702        let meta = self
703            .read_conversation_metadata(project, conversation_id)
704            .map_err(|e| ConvoError::Provider(e.to_string()))?;
705
706        Ok(ConversationMeta {
707            id: meta.session_id,
708            started_at: meta.started_at,
709            last_activity: meta.last_activity,
710            message_count: meta.message_count,
711            file_path: Some(meta.file_path),
712            predecessor: None,
713            successor: None,
714        })
715    }
716
717    fn list_metadata(&self, project: &str) -> toolpath_convo::Result<Vec<ConversationMeta>> {
718        let metas = self
719            .list_conversation_metadata(project)
720            .map_err(|e| ConvoError::Provider(e.to_string()))?;
721
722        Ok(metas
723            .into_iter()
724            .map(|m| ConversationMeta {
725                id: m.session_id,
726                started_at: m.started_at,
727                last_activity: m.last_activity,
728                message_count: m.message_count,
729                file_path: Some(m.file_path),
730                predecessor: None,
731                successor: None,
732            })
733            .collect())
734    }
735}
736
737// ── ConversationWatcher with eager emit + TurnUpdated ────────────────
738
739#[cfg(feature = "watcher")]
740impl toolpath_convo::ConversationWatcher for crate::watcher::ConversationWatcher {
741    fn poll(&mut self) -> toolpath_convo::Result<Vec<WatcherEvent>> {
742        let entries = crate::watcher::ConversationWatcher::poll(self)
743            .map_err(|e| ConvoError::Provider(e.to_string()))?;
744
745        let mut events: Vec<WatcherEvent> = Vec::new();
746
747        // Check for session rotations and prepend Progress events
748        for (from, to) in self.take_pending_rotations() {
749            events.push(WatcherEvent::Progress {
750                kind: "session_rotated".into(),
751                data: serde_json::json!({
752                    "from": from,
753                    "to": to,
754                }),
755            });
756        }
757
758        for entry in &entries {
759            let Some(msg) = &entry.message else {
760                events.push(entry_to_watcher_event(entry));
761                continue;
762            };
763
764            if is_tool_result_only(entry) {
765                // Find matching turns in previously emitted events and in
766                // our assembled state, merge results, emit TurnUpdated.
767                // Walk events in reverse to find the turn to update.
768                let mut updated_turn: Option<Turn> = None;
769
770                // Search backwards through events emitted this poll cycle
771                for event in events.iter_mut().rev() {
772                    if let WatcherEvent::Turn(turn) | WatcherEvent::TurnUpdated(turn) = event
773                        && turn.tool_uses.iter().any(|tu| {
774                            tu.result.is_none()
775                                && msg.tool_results().iter().any(|tr| tr.tool_use_id == tu.id)
776                        })
777                    {
778                        // Merge results into this turn
779                        let mut updated = (**turn).clone();
780                        merge_tool_results(std::slice::from_mut(&mut updated), msg);
781                        updated_turn = Some(updated.clone());
782                        // Also update the existing event in-place so later
783                        // result entries can find the right state
784                        **turn = updated;
785                        break;
786                    }
787                }
788
789                if let Some(turn) = updated_turn {
790                    events.push(WatcherEvent::TurnUpdated(Box::new(turn)));
791                }
792                // If no matching turn found, the tool-result-only entry
793                // is silently dropped (the matching turn was emitted in a
794                // prior poll cycle and can't be updated from here).
795                continue;
796            }
797
798            events.push(entry_to_watcher_event(entry));
799        }
800
801        Ok(events)
802    }
803
804    fn seen_count(&self) -> usize {
805        crate::watcher::ConversationWatcher::seen_count(self)
806    }
807}
808
809// ── Public re-exports for convenience ────────────────────────────────
810
811/// Convert a Claude [`Conversation`] directly into a [`ConversationView`].
812///
813/// This performs cross-entry tool result assembly: tool-result-only user
814/// entries are merged into the preceding assistant turn rather than emitted
815/// as separate turns.
816pub fn to_view(convo: &Conversation) -> ConversationView {
817    conversation_to_view(convo)
818}
819
820/// Convert a single Claude [`ConversationEntry`] into a [`Turn`], if it
821/// contains a message.
822///
823/// Note: this does *not* perform cross-entry assembly. For assembled
824/// results, use [`to_view`] instead.
825pub fn to_turn(entry: &ConversationEntry) -> Option<Turn> {
826    entry_to_turn(entry)
827}
828
829// ── Tests ────────────────────────────────────────────────────────────
830
831#[cfg(test)]
832mod tests {
833    use super::*;
834    use crate::PathResolver;
835    use std::fs;
836    use tempfile::TempDir;
837
838    /// One assistant turn carrying a cumulative usage snapshot (only
839    /// output varies across a split, so input/cache are fixed here).
840    fn grp_turn(id: &str, mid: &str, output: u32) -> Turn {
841        let mut t = message_turn_stub(id);
842        t.group_id = Some(mid.into());
843        t.token_usage = Some(TokenUsage {
844            input_tokens: Some(6),
845            output_tokens: Some(output),
846            cache_read_tokens: Some(14_842),
847            cache_write_tokens: Some(429_831),
848            ..Default::default()
849        });
850        t
851    }
852
853    fn message_turn_stub(id: &str) -> Turn {
854        Turn {
855            id: id.into(),
856            parent_id: None,
857            group_id: None,
858            role: Role::Assistant,
859            timestamp: "2024-01-01T00:00:00Z".into(),
860            text: String::new(),
861            thinking: None,
862            tool_uses: vec![],
863            model: None,
864            stop_reason: None,
865            token_usage: None,
866            attributed_token_usage: None,
867            environment: None,
868            delegations: vec![],
869            file_mutations: vec![],
870        }
871    }
872
873    #[test]
874    fn canonicalize_streamed_group_keeps_total_only_on_final_turn() {
875        // Streaming snapshots climb 55 -> 164 across two lines of one
876        // message. The final turn carries the message total (the final
877        // snapshot); earlier turns carry nothing. The intermediate snapshot
878        // (55) is NOT per-block attribution — it's where generation happened
879        // to be when the line was flushed — so we never record it.
880        let mut turns = vec![grp_turn("t1", "msg_A", 55), grp_turn("t2", "msg_A", 164)];
881        canonicalize_message_usage(&mut turns);
882
883        assert!(turns[0].token_usage.is_none(), "total only on final turn");
884        assert_eq!(
885            turns[1].token_usage.as_ref().unwrap().output_tokens,
886            Some(164)
887        );
888        assert_eq!(turns[1].token_usage.as_ref().unwrap().input_tokens, Some(6));
889        for t in &turns {
890            assert!(
891                t.attributed_token_usage.is_none(),
892                "Claude per-line snapshots are not per-step attribution"
893            );
894        }
895    }
896
897    #[test]
898    fn canonicalize_does_not_trust_line_order() {
899        // Defensive: the complete total arrives FIRST (out of order). We
900        // must still report 164 as the message total — the field-wise max,
901        // not the last line's snapshot.
902        let mut turns = vec![grp_turn("t1", "msg_A", 164), grp_turn("t2", "msg_A", 55)];
903        canonicalize_message_usage(&mut turns);
904
905        assert_eq!(
906            turns[1].token_usage.as_ref().unwrap().output_tokens,
907            Some(164),
908            "field-wise max, not the last line"
909        );
910    }
911
912    #[test]
913    fn canonicalize_collapses_repeated_total_to_one_turn() {
914        // Byte-identical lines (the ~73% case): the total lands once, on the
915        // final turn; no attribution either way.
916        let mut turns = vec![
917            grp_turn("t1", "msg_A", 997),
918            grp_turn("t2", "msg_A", 997),
919            grp_turn("t3", "msg_A", 997),
920        ];
921        canonicalize_message_usage(&mut turns);
922
923        assert!(turns[0].token_usage.is_none());
924        assert!(turns[1].token_usage.is_none());
925        assert_eq!(
926            turns[2].token_usage.as_ref().unwrap().output_tokens,
927            Some(997)
928        );
929        for t in &turns {
930            assert!(t.attributed_token_usage.is_none());
931        }
932    }
933
934    fn setup_provider() -> (TempDir, ClaudeConvo) {
935        let temp = TempDir::new().unwrap();
936        let claude_dir = temp.path().join(".claude");
937        let project_dir = claude_dir.join("projects/-test-project");
938        fs::create_dir_all(&project_dir).unwrap();
939
940        let entries = [
941            r#"{"uuid":"uuid-1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Fix the bug"}}"#,
942            r#"{"uuid":"uuid-2","type":"assistant","parentUuid":"uuid-1","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"I'll fix that."},{"type":"thinking","thinking":"The bug is in auth"},{"type":"tool_use","id":"t1","name":"Read","input":{"file_path":"src/main.rs"}}],"model":"claude-opus-4-6","stop_reason":"tool_use","usage":{"input_tokens":100,"output_tokens":50}}}"#,
943            r#"{"uuid":"uuid-3","type":"user","parentUuid":"uuid-2","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"fn main() { println!(\"hello\"); }","is_error":false}]}}"#,
944            r#"{"uuid":"uuid-4","type":"assistant","parentUuid":"uuid-3","timestamp":"2024-01-01T00:00:03Z","message":{"role":"assistant","content":[{"type":"text","text":"I see the issue. Let me fix it."},{"type":"tool_use","id":"t2","name":"Edit","input":{"file_path":"src/main.rs","old_string":"hello","new_string":"fixed"}}],"model":"claude-opus-4-6","stop_reason":"tool_use","usage":{"input_tokens":200,"output_tokens":100}}}"#,
945            r#"{"uuid":"uuid-5","type":"user","parentUuid":"uuid-4","timestamp":"2024-01-01T00:00:04Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t2","content":"File written successfully","is_error":false}]}}"#,
946            r#"{"uuid":"uuid-6","type":"assistant","parentUuid":"uuid-5","timestamp":"2024-01-01T00:00:05Z","message":{"role":"assistant","content":"Done! The bug is fixed.","model":"claude-opus-4-6","stop_reason":"end_turn"}}"#,
947            r#"{"uuid":"uuid-7","type":"user","parentUuid":"uuid-6","timestamp":"2024-01-01T00:00:06Z","message":{"role":"user","content":"Thanks!"}}"#,
948        ];
949        fs::write(project_dir.join("session-1.jsonl"), entries.join("\n")).unwrap();
950
951        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
952        (temp, ClaudeConvo::with_resolver(resolver))
953    }
954
955    /// A session whose first assistant API message is split across three
956    /// JSONL lines (text, then one per tool_use) — the on-disk shape Claude
957    /// Code writes. Each line repeats the same `message.id` and the full
958    /// message-level `usage`, followed by a singleton assistant message.
959    fn setup_split_message_provider() -> (TempDir, ClaudeConvo) {
960        let temp = TempDir::new().unwrap();
961        let claude_dir = temp.path().join(".claude");
962        let project_dir = claude_dir.join("projects/-test-project");
963        fs::create_dir_all(&project_dir).unwrap();
964
965        let usage_a = r#"{"input_tokens":6,"output_tokens":997,"cache_read_input_tokens":14842,"cache_creation_input_tokens":429831}"#;
966        let entries = [
967            r#"{"uuid":"uuid-1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Fix the bug"}}"#.to_string(),
968            format!(
969                r#"{{"uuid":"uuid-2","type":"assistant","parentUuid":"uuid-1","timestamp":"2024-01-01T00:00:01Z","message":{{"id":"msg_A","role":"assistant","content":[{{"type":"text","text":"Working on it."}}],"model":"claude-opus-4-7","stop_reason":null,"usage":{usage_a}}}}}"#
970            ),
971            format!(
972                r#"{{"uuid":"uuid-3","type":"assistant","parentUuid":"uuid-2","timestamp":"2024-01-01T00:00:02Z","message":{{"id":"msg_A","role":"assistant","content":[{{"type":"tool_use","id":"t1","name":"Read","input":{{"file_path":"a.rs"}}}}],"model":"claude-opus-4-7","stop_reason":null,"usage":{usage_a}}}}}"#
973            ),
974            format!(
975                r#"{{"uuid":"uuid-4","type":"assistant","parentUuid":"uuid-3","timestamp":"2024-01-01T00:00:03Z","message":{{"id":"msg_A","role":"assistant","content":[{{"type":"tool_use","id":"t2","name":"Read","input":{{"file_path":"b.rs"}}}}],"model":"claude-opus-4-7","stop_reason":"tool_use","usage":{usage_a}}}}}"#
976            ),
977            r#"{"uuid":"uuid-5","type":"assistant","parentUuid":"uuid-4","timestamp":"2024-01-01T00:00:04Z","message":{"id":"msg_B","role":"assistant","content":[{"type":"text","text":"Done."}],"model":"claude-opus-4-7","stop_reason":"end_turn","usage":{"input_tokens":5,"output_tokens":11}}}"#.to_string(),
978        ];
979        fs::write(project_dir.join("session-2.jsonl"), entries.join("\n")).unwrap();
980
981        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
982        (temp, ClaudeConvo::with_resolver(resolver))
983    }
984
985    #[test]
986    fn test_split_message_turns_share_group_id() {
987        let (_temp, provider) = setup_split_message_provider();
988        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-2")
989            .unwrap();
990
991        assert_eq!(view.turns.len(), 5);
992        assert!(view.turns[0].group_id.is_none(), "user lines carry no ID");
993        for turn in &view.turns[1..=3] {
994            assert_eq!(turn.group_id.as_deref(), Some("msg_A"));
995        }
996        assert_eq!(view.turns[4].group_id.as_deref(), Some("msg_B"));
997    }
998
999    #[test]
1000    fn test_view_usage_is_canonical_total_on_group_final_turn() {
1001        // IR contract: `Turn.token_usage` always means "the message's
1002        // total" and appears only on the message's final turn. The wire
1003        // repeats the total on every line of a split; the view must not.
1004        let (_temp, provider) = setup_split_message_provider();
1005        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-2")
1006            .unwrap();
1007
1008        assert!(view.turns[1].token_usage.is_none());
1009        assert!(view.turns[2].token_usage.is_none());
1010        assert_eq!(
1011            view.turns[3].token_usage.as_ref().unwrap().output_tokens,
1012            Some(997)
1013        );
1014        assert_eq!(
1015            view.turns[4].token_usage.as_ref().unwrap().output_tokens,
1016            Some(11)
1017        );
1018    }
1019
1020    #[test]
1021    fn test_total_usage_counts_each_message_once() {
1022        let (_temp, provider) = setup_split_message_provider();
1023        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-2")
1024            .unwrap();
1025
1026        // msg_A's usage appears on three lines but is one API message;
1027        // totals must be msg_A + msg_B, not 3×msg_A + msg_B.
1028        let total = view.total_usage.as_ref().unwrap();
1029        assert_eq!(total.output_tokens, Some(997 + 11));
1030        assert_eq!(total.input_tokens, Some(6 + 5));
1031        assert_eq!(total.cache_read_tokens, Some(14_842));
1032        assert_eq!(total.cache_write_tokens, Some(429_831));
1033    }
1034
1035    #[test]
1036    fn test_load_conversation_assembles_tool_results() {
1037        let (_temp, provider) = setup_provider();
1038        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1039            .unwrap();
1040
1041        assert_eq!(view.id, "session-1");
1042        // 7 entries collapse to 5 turns (2 tool-result-only entries absorbed)
1043        assert_eq!(view.turns.len(), 5);
1044
1045        // Turn 0: user "Fix the bug"
1046        assert_eq!(view.turns[0].role, Role::User);
1047        assert_eq!(view.turns[0].text, "Fix the bug");
1048        assert!(view.turns[0].parent_id.is_none());
1049
1050        // Turn 1: assistant with tool use + assembled result
1051        assert_eq!(view.turns[1].role, Role::Assistant);
1052        assert_eq!(view.turns[1].text, "I'll fix that.");
1053        assert_eq!(
1054            view.turns[1].thinking.as_deref(),
1055            Some("The bug is in auth")
1056        );
1057        assert_eq!(view.turns[1].tool_uses.len(), 1);
1058        assert_eq!(view.turns[1].tool_uses[0].name, "Read");
1059        assert_eq!(view.turns[1].tool_uses[0].id, "t1");
1060        // Key assertion: result is populated from the next entry
1061        let result = view.turns[1].tool_uses[0].result.as_ref().unwrap();
1062        assert!(!result.is_error);
1063        assert!(result.content.contains("fn main()"));
1064        assert_eq!(view.turns[1].model.as_deref(), Some("claude-opus-4-6"));
1065        assert_eq!(view.turns[1].stop_reason.as_deref(), Some("tool_use"));
1066        assert_eq!(view.turns[1].parent_id.as_deref(), Some("uuid-1"));
1067
1068        // Token usage
1069        let usage = view.turns[1].token_usage.as_ref().unwrap();
1070        assert_eq!(usage.input_tokens, Some(100));
1071        assert_eq!(usage.output_tokens, Some(50));
1072
1073        // Turn 2: second assistant with tool use + assembled result
1074        assert_eq!(view.turns[2].role, Role::Assistant);
1075        assert_eq!(view.turns[2].text, "I see the issue. Let me fix it.");
1076        assert_eq!(view.turns[2].tool_uses[0].name, "Edit");
1077        let result2 = view.turns[2].tool_uses[0].result.as_ref().unwrap();
1078        assert_eq!(result2.content, "File written successfully");
1079
1080        // Turn 3: final assistant (no tools)
1081        assert_eq!(view.turns[3].role, Role::Assistant);
1082        assert_eq!(view.turns[3].text, "Done! The bug is fixed.");
1083        assert!(view.turns[3].tool_uses.is_empty());
1084
1085        // Turn 4: user "Thanks!"
1086        assert_eq!(view.turns[4].role, Role::User);
1087        assert_eq!(view.turns[4].text, "Thanks!");
1088    }
1089
1090    #[test]
1091    fn test_no_phantom_empty_turns() {
1092        let (_temp, provider) = setup_provider();
1093        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1094            .unwrap();
1095
1096        // No turns should have empty text with User role (phantom turns)
1097        for turn in &view.turns {
1098            if turn.role == Role::User {
1099                assert!(
1100                    !turn.text.is_empty(),
1101                    "Found phantom empty user turn: {:?}",
1102                    turn.id
1103                );
1104            }
1105        }
1106    }
1107
1108    #[test]
1109    fn test_tool_result_error_flag() {
1110        let temp = TempDir::new().unwrap();
1111        let claude_dir = temp.path().join(".claude");
1112        let project_dir = claude_dir.join("projects/-test-project");
1113        fs::create_dir_all(&project_dir).unwrap();
1114
1115        let entries = [
1116            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Read a file"}}"#,
1117            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Reading..."},{"type":"tool_use","id":"t1","name":"Read","input":{"path":"/nonexistent"}}],"stop_reason":"tool_use"}}"#,
1118            r#"{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"File not found","is_error":true}]}}"#,
1119        ];
1120        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1121
1122        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1123        let provider = ClaudeConvo::with_resolver(resolver);
1124        let view =
1125            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1126
1127        assert_eq!(view.turns.len(), 2); // user + assistant (tool-result absorbed)
1128        let result = view.turns[1].tool_uses[0].result.as_ref().unwrap();
1129        assert!(result.is_error);
1130        assert_eq!(result.content, "File not found");
1131    }
1132
1133    #[test]
1134    fn test_multiple_tool_uses_single_result_entry() {
1135        let temp = TempDir::new().unwrap();
1136        let claude_dir = temp.path().join(".claude");
1137        let project_dir = claude_dir.join("projects/-test-project");
1138        fs::create_dir_all(&project_dir).unwrap();
1139
1140        let entries = [
1141            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Check two files"}}"#,
1142            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Reading both..."},{"type":"tool_use","id":"t1","name":"Read","input":{"path":"a.rs"}},{"type":"tool_use","id":"t2","name":"Read","input":{"path":"b.rs"}}]}}"#,
1143            r#"{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"file a contents","is_error":false},{"type":"tool_result","tool_use_id":"t2","content":"file b contents","is_error":false}]}}"#,
1144        ];
1145        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1146
1147        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1148        let provider = ClaudeConvo::with_resolver(resolver);
1149        let view =
1150            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1151
1152        assert_eq!(view.turns.len(), 2);
1153        assert_eq!(view.turns[1].tool_uses.len(), 2);
1154
1155        let r1 = view.turns[1].tool_uses[0].result.as_ref().unwrap();
1156        assert_eq!(r1.content, "file a contents");
1157
1158        let r2 = view.turns[1].tool_uses[1].result.as_ref().unwrap();
1159        assert_eq!(r2.content, "file b contents");
1160    }
1161
1162    #[test]
1163    fn test_conversation_without_tool_use_unchanged() {
1164        let temp = TempDir::new().unwrap();
1165        let claude_dir = temp.path().join(".claude");
1166        let project_dir = claude_dir.join("projects/-test-project");
1167        fs::create_dir_all(&project_dir).unwrap();
1168
1169        let entries = [
1170            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Hello"}}"#,
1171            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":"Hi there!"}}"#,
1172        ];
1173        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1174
1175        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1176        let provider = ClaudeConvo::with_resolver(resolver);
1177        let view =
1178            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1179
1180        assert_eq!(view.turns.len(), 2);
1181        assert_eq!(view.turns[0].text, "Hello");
1182        assert_eq!(view.turns[1].text, "Hi there!");
1183    }
1184
1185    #[test]
1186    fn test_assistant_turn_without_result_has_none() {
1187        // Tool use at end of conversation with no result entry
1188        let temp = TempDir::new().unwrap();
1189        let claude_dir = temp.path().join(".claude");
1190        let project_dir = claude_dir.join("projects/-test-project");
1191        fs::create_dir_all(&project_dir).unwrap();
1192
1193        let entries = [
1194            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Read a file"}}"#,
1195            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Reading..."},{"type":"tool_use","id":"t1","name":"Read","input":{"path":"test.rs"}}]}}"#,
1196        ];
1197        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1198
1199        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1200        let provider = ClaudeConvo::with_resolver(resolver);
1201        let view =
1202            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1203
1204        assert_eq!(view.turns.len(), 2);
1205        assert!(view.turns[1].tool_uses[0].result.is_none());
1206    }
1207
1208    #[test]
1209    fn test_list_conversations() {
1210        let (_temp, provider) = setup_provider();
1211        let ids = ConversationProvider::list_conversations(&provider, "/test/project").unwrap();
1212        assert_eq!(ids, vec!["session-1"]);
1213    }
1214
1215    #[test]
1216    fn test_load_metadata() {
1217        let (_temp, provider) = setup_provider();
1218        let meta =
1219            ConversationProvider::load_metadata(&provider, "/test/project", "session-1").unwrap();
1220        assert_eq!(meta.id, "session-1");
1221        assert_eq!(meta.message_count, 7);
1222        assert!(meta.file_path.is_some());
1223    }
1224
1225    #[test]
1226    fn test_list_metadata() {
1227        let (_temp, provider) = setup_provider();
1228        let metas = ConversationProvider::list_metadata(&provider, "/test/project").unwrap();
1229        assert_eq!(metas.len(), 1);
1230        assert_eq!(metas[0].id, "session-1");
1231    }
1232
1233    #[test]
1234    fn test_to_view() {
1235        let (_temp, manager) = setup_provider();
1236        let convo = manager
1237            .read_conversation("/test/project", "session-1")
1238            .unwrap();
1239        let view = to_view(&convo);
1240        assert_eq!(view.turns.len(), 5);
1241        assert_eq!(view.title(20).unwrap(), "Fix the bug");
1242    }
1243
1244    #[test]
1245    fn test_to_turn_with_message() {
1246        let entry: ConversationEntry = serde_json::from_str(
1247            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"hello"}}"#,
1248        )
1249        .unwrap();
1250        let turn = to_turn(&entry).unwrap();
1251        assert_eq!(turn.id, "u1");
1252        assert_eq!(turn.text, "hello");
1253        assert_eq!(turn.role, Role::User);
1254    }
1255
1256    #[test]
1257    fn test_to_turn_without_message() {
1258        let entry: ConversationEntry = serde_json::from_str(
1259            r#"{"uuid":"u1","type":"progress","timestamp":"2024-01-01T00:00:00Z"}"#,
1260        )
1261        .unwrap();
1262        assert!(to_turn(&entry).is_none());
1263    }
1264
1265    #[test]
1266    fn test_entry_to_watcher_event_turn() {
1267        let entry: ConversationEntry = serde_json::from_str(
1268            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"hi"}}"#,
1269        )
1270        .unwrap();
1271        let event = entry_to_watcher_event(&entry);
1272        assert!(matches!(event, WatcherEvent::Turn(_)));
1273    }
1274
1275    #[test]
1276    fn test_entry_to_watcher_event_progress() {
1277        let entry: ConversationEntry = serde_json::from_str(
1278            r#"{"uuid":"u1","type":"progress","timestamp":"2024-01-01T00:00:00Z"}"#,
1279        )
1280        .unwrap();
1281        let event = entry_to_watcher_event(&entry);
1282        assert!(matches!(event, WatcherEvent::Progress { .. }));
1283    }
1284
1285    #[cfg(feature = "watcher")]
1286    #[test]
1287    fn test_watcher_trait_basic() {
1288        let temp = TempDir::new().unwrap();
1289        let claude_dir = temp.path().join(".claude");
1290        let project_dir = claude_dir.join("projects/-test-project");
1291        fs::create_dir_all(&project_dir).unwrap();
1292
1293        let entries = [
1294            r#"{"uuid":"uuid-1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Hello"}}"#,
1295            r#"{"uuid":"uuid-2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":"Hi"}}"#,
1296        ];
1297        fs::write(project_dir.join("session-1.jsonl"), entries.join("\n")).unwrap();
1298
1299        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1300        let manager = ClaudeConvo::with_resolver(resolver);
1301
1302        let mut watcher = crate::watcher::ConversationWatcher::new(
1303            manager,
1304            "/test/project".to_string(),
1305            "session-1".to_string(),
1306        );
1307
1308        // Use the trait explicitly (inherent poll returns ConversationEntry)
1309        let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1310        assert_eq!(events.len(), 2);
1311        assert!(matches!(&events[0], WatcherEvent::Turn(t) if t.role == Role::User));
1312        assert!(matches!(&events[1], WatcherEvent::Turn(t) if t.role == Role::Assistant));
1313        assert_eq!(toolpath_convo::ConversationWatcher::seen_count(&watcher), 2);
1314
1315        // Second poll returns nothing
1316        let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1317        assert!(events.is_empty());
1318    }
1319
1320    #[cfg(feature = "watcher")]
1321    #[test]
1322    fn test_watcher_trait_assembles_tool_results() {
1323        let temp = TempDir::new().unwrap();
1324        let claude_dir = temp.path().join(".claude");
1325        let project_dir = claude_dir.join("projects/-test-project");
1326        fs::create_dir_all(&project_dir).unwrap();
1327
1328        let entries = [
1329            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Read the file"}}"#,
1330            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Reading..."},{"type":"tool_use","id":"t1","name":"Read","input":{"path":"test.rs"}}]}}"#,
1331            r#"{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"fn main() {}","is_error":false}]}}"#,
1332            r#"{"uuid":"u4","type":"assistant","timestamp":"2024-01-01T00:00:03Z","message":{"role":"assistant","content":"Done!"}}"#,
1333        ];
1334        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1335
1336        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1337        let manager = ClaudeConvo::with_resolver(resolver);
1338
1339        let mut watcher = crate::watcher::ConversationWatcher::new(
1340            manager,
1341            "/test/project".to_string(),
1342            "s1".to_string(),
1343        );
1344
1345        let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1346
1347        // Should get: Turn(user), Turn(assistant), TurnUpdated(assistant), Turn(assistant)
1348        assert_eq!(events.len(), 4);
1349
1350        // First: user turn
1351        assert!(matches!(&events[0], WatcherEvent::Turn(t) if t.role == Role::User));
1352
1353        // Second: assistant turn emitted eagerly (result may not be populated yet in the event)
1354        assert!(matches!(&events[1], WatcherEvent::Turn(t) if t.role == Role::Assistant));
1355
1356        // Third: TurnUpdated with results merged
1357        match &events[2] {
1358            WatcherEvent::TurnUpdated(turn) => {
1359                assert_eq!(turn.id, "u2");
1360                assert_eq!(turn.tool_uses.len(), 1);
1361                let result = turn.tool_uses[0].result.as_ref().unwrap();
1362                assert_eq!(result.content, "fn main() {}");
1363                assert!(!result.is_error);
1364            }
1365            other => panic!("Expected TurnUpdated, got {:?}", other),
1366        }
1367
1368        // Fourth: final assistant turn
1369        assert!(matches!(&events[3], WatcherEvent::Turn(t) if t.text == "Done!"));
1370    }
1371
1372    #[cfg(feature = "watcher")]
1373    #[test]
1374    fn test_watcher_trait_incremental_tool_results() {
1375        // Simulate tool results arriving in a different poll cycle than the tool use
1376        let temp = TempDir::new().unwrap();
1377        let claude_dir = temp.path().join(".claude");
1378        let project_dir = claude_dir.join("projects/-test-project");
1379        fs::create_dir_all(&project_dir).unwrap();
1380
1381        // Start with just the user message and assistant tool use
1382        let entries_phase1 = [
1383            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Read file"}}"#,
1384            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Reading..."},{"type":"tool_use","id":"t1","name":"Read","input":{"path":"test.rs"}}]}}"#,
1385        ];
1386        fs::write(
1387            project_dir.join("s1.jsonl"),
1388            entries_phase1.join("\n") + "\n",
1389        )
1390        .unwrap();
1391
1392        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1393        let manager = ClaudeConvo::with_resolver(resolver);
1394
1395        let mut watcher = crate::watcher::ConversationWatcher::new(
1396            manager,
1397            "/test/project".to_string(),
1398            "s1".to_string(),
1399        );
1400
1401        // First poll: get user + assistant turns
1402        let events1 = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1403        assert_eq!(events1.len(), 2);
1404        // Assistant turn emitted eagerly with result: None
1405        if let WatcherEvent::Turn(t) = &events1[1] {
1406            assert!(t.tool_uses[0].result.is_none());
1407        } else {
1408            panic!("Expected Turn");
1409        }
1410
1411        // Now append the tool result entry
1412        use std::io::Write;
1413        let mut file = fs::OpenOptions::new()
1414            .append(true)
1415            .open(project_dir.join("s1.jsonl"))
1416            .unwrap();
1417        writeln!(file, r#"{{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{{"role":"user","content":[{{"type":"tool_result","tool_use_id":"t1","content":"fn main() {{}}","is_error":false}}]}}}}"#).unwrap();
1418
1419        // Second poll: tool-result-only entry arrives
1420        let events2 = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1421        // The tool-result-only entry can't find its matching turn in this poll
1422        // cycle (it was emitted in the previous one), so it's silently absorbed.
1423        // This is a known limitation of the eager-emit approach for cross-poll
1424        // boundaries — the batch path (to_view) handles this correctly.
1425        // Consumers needing full fidelity across poll boundaries should
1426        // periodically do a full load_conversation.
1427        assert!(events2.is_empty() || events2.iter().all(|e| !matches!(e, WatcherEvent::Turn(_))));
1428    }
1429
1430    #[test]
1431    fn test_merge_tool_results_by_id() {
1432        // Verify that merge matches by tool_use_id, not position
1433        let mut turns = vec![Turn {
1434            id: "t1".into(),
1435            parent_id: None,
1436            group_id: None,
1437            role: Role::Assistant,
1438            timestamp: "2024-01-01T00:00:00Z".into(),
1439            text: "test".into(),
1440            thinking: None,
1441            tool_uses: vec![
1442                ToolInvocation {
1443                    id: "tool-a".into(),
1444                    name: "Read".into(),
1445                    input: serde_json::json!({}),
1446                    result: None,
1447                    category: Some(ToolCategory::FileRead),
1448                },
1449                ToolInvocation {
1450                    id: "tool-b".into(),
1451                    name: "Write".into(),
1452                    input: serde_json::json!({}),
1453                    result: None,
1454                    category: Some(ToolCategory::FileWrite),
1455                },
1456            ],
1457            model: None,
1458            stop_reason: None,
1459            token_usage: None,
1460            attributed_token_usage: None,
1461            environment: None,
1462            delegations: vec![],
1463            file_mutations: Vec::new(),
1464        }];
1465
1466        // Create a message with results in reversed order
1467        let msg: Message = serde_json::from_str(
1468            r#"{"role":"user","content":[{"type":"tool_result","tool_use_id":"tool-b","content":"write result","is_error":false},{"type":"tool_result","tool_use_id":"tool-a","content":"read result","is_error":true}]}"#,
1469        )
1470        .unwrap();
1471
1472        let merged = merge_tool_results(&mut turns, &msg);
1473        assert!(merged);
1474
1475        // Results should match by ID regardless of order
1476        assert_eq!(
1477            turns[0].tool_uses[0].result.as_ref().unwrap().content,
1478            "read result"
1479        );
1480        assert!(turns[0].tool_uses[0].result.as_ref().unwrap().is_error);
1481
1482        assert_eq!(
1483            turns[0].tool_uses[1].result.as_ref().unwrap().content,
1484            "write result"
1485        );
1486        assert!(!turns[0].tool_uses[1].result.as_ref().unwrap().is_error);
1487    }
1488
1489    #[test]
1490    fn test_is_tool_result_only() {
1491        // Tool-result-only entry
1492        let entry: ConversationEntry = serde_json::from_str(
1493            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"ok","is_error":false}]}}"#,
1494        )
1495        .unwrap();
1496        assert!(is_tool_result_only(&entry));
1497
1498        // Regular user entry with text
1499        let entry: ConversationEntry = serde_json::from_str(
1500            r#"{"uuid":"u2","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"hello"}}"#,
1501        )
1502        .unwrap();
1503        assert!(!is_tool_result_only(&entry));
1504
1505        // Entry without message
1506        let entry: ConversationEntry = serde_json::from_str(
1507            r#"{"uuid":"u3","type":"progress","timestamp":"2024-01-01T00:00:00Z"}"#,
1508        )
1509        .unwrap();
1510        assert!(!is_tool_result_only(&entry));
1511
1512        // Assistant entry (never tool-result-only)
1513        let entry: ConversationEntry = serde_json::from_str(
1514            r#"{"uuid":"u4","type":"assistant","timestamp":"2024-01-01T00:00:00Z","message":{"role":"assistant","content":"hi"}}"#,
1515        )
1516        .unwrap();
1517        assert!(!is_tool_result_only(&entry));
1518    }
1519
1520    // ── New enrichment tests ─────────────────────────────────────────
1521
1522    #[test]
1523    fn test_tool_category_mapping() {
1524        assert_eq!(tool_category("Read"), Some(ToolCategory::FileRead));
1525        assert_eq!(tool_category("Glob"), Some(ToolCategory::FileSearch));
1526        assert_eq!(tool_category("Grep"), Some(ToolCategory::FileSearch));
1527        assert_eq!(tool_category("Write"), Some(ToolCategory::FileWrite));
1528        assert_eq!(tool_category("Edit"), Some(ToolCategory::FileWrite));
1529        assert_eq!(tool_category("NotebookEdit"), Some(ToolCategory::FileWrite));
1530        assert_eq!(tool_category("Bash"), Some(ToolCategory::Shell));
1531        assert_eq!(tool_category("WebFetch"), Some(ToolCategory::Network));
1532        assert_eq!(tool_category("WebSearch"), Some(ToolCategory::Network));
1533        assert_eq!(tool_category("Task"), Some(ToolCategory::Delegation));
1534        assert_eq!(tool_category("UnknownTool"), None);
1535    }
1536
1537    #[test]
1538    fn test_turn_has_tool_category() {
1539        let (_temp, provider) = setup_provider();
1540        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1541            .unwrap();
1542
1543        // Turn 1 (assistant) has a Read tool
1544        assert_eq!(
1545            view.turns[1].tool_uses[0].category,
1546            Some(ToolCategory::FileRead)
1547        );
1548        // Turn 2 (assistant) has an Edit tool
1549        assert_eq!(
1550            view.turns[2].tool_uses[0].category,
1551            Some(ToolCategory::FileWrite)
1552        );
1553    }
1554
1555    #[test]
1556    fn test_environment_populated_from_entry() {
1557        let temp = TempDir::new().unwrap();
1558        let claude_dir = temp.path().join(".claude");
1559        let project_dir = claude_dir.join("projects/-test-project");
1560        fs::create_dir_all(&project_dir).unwrap();
1561
1562        let entries = [
1563            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","cwd":"/project/path","gitBranch":"feat/auth","message":{"role":"user","content":"Hello"}}"#,
1564            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":"Hi"}}"#,
1565        ];
1566        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1567
1568        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1569        let provider = ClaudeConvo::with_resolver(resolver);
1570        let view =
1571            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1572
1573        // User turn has environment (entry has cwd and gitBranch)
1574        let env = view.turns[0].environment.as_ref().unwrap();
1575        assert_eq!(env.working_dir.as_deref(), Some("/project/path"));
1576        assert_eq!(env.vcs_branch.as_deref(), Some("feat/auth"));
1577        assert!(env.vcs_revision.is_none());
1578
1579        // Assistant turn has no environment (entry has no cwd/gitBranch)
1580        assert!(view.turns[1].environment.is_none());
1581    }
1582
1583    #[test]
1584    fn test_cache_tokens_populated() {
1585        let temp = TempDir::new().unwrap();
1586        let claude_dir = temp.path().join(".claude");
1587        let project_dir = claude_dir.join("projects/-test-project");
1588        fs::create_dir_all(&project_dir).unwrap();
1589
1590        let entries = [
1591            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Hello"}}"#,
1592            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":"Hi","usage":{"input_tokens":100,"output_tokens":50,"cache_creation_input_tokens":200,"cache_read_input_tokens":500}}}"#,
1593        ];
1594        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1595
1596        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1597        let provider = ClaudeConvo::with_resolver(resolver);
1598        let view =
1599            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1600
1601        let usage = view.turns[1].token_usage.as_ref().unwrap();
1602        assert_eq!(usage.cache_read_tokens, Some(500));
1603        assert_eq!(usage.cache_write_tokens, Some(200));
1604    }
1605
1606    #[test]
1607    fn test_total_usage_aggregated() {
1608        let (_temp, provider) = setup_provider();
1609        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1610            .unwrap();
1611
1612        let total = view.total_usage.as_ref().unwrap();
1613        // Two assistant turns with usage: (100, 50) and (200, 100)
1614        assert_eq!(total.input_tokens, Some(300));
1615        assert_eq!(total.output_tokens, Some(150));
1616    }
1617
1618    #[test]
1619    fn test_provider_id_set() {
1620        let (_temp, provider) = setup_provider();
1621        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1622            .unwrap();
1623
1624        assert_eq!(view.provider_id.as_deref(), Some("claude-code"));
1625    }
1626
1627    #[test]
1628    fn test_files_changed_populated() {
1629        let temp = TempDir::new().unwrap();
1630        let claude_dir = temp.path().join(".claude");
1631        let project_dir = claude_dir.join("projects/-test-project");
1632        fs::create_dir_all(&project_dir).unwrap();
1633
1634        let entries = [
1635            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Edit files"}}"#,
1636            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Editing..."},{"type":"tool_use","id":"t1","name":"Write","input":{"file_path":"src/main.rs","content":"fn main() {}"}},{"type":"tool_use","id":"t2","name":"Edit","input":{"file_path":"src/lib.rs","old_string":"a","new_string":"b"}}]}}"#,
1637            r#"{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"t1","content":"ok","is_error":false},{"type":"tool_result","tool_use_id":"t2","content":"ok","is_error":false}]}}"#,
1638            r#"{"uuid":"u4","type":"assistant","timestamp":"2024-01-01T00:00:03Z","message":{"role":"assistant","content":[{"type":"text","text":"More edits..."},{"type":"tool_use","id":"t3","name":"Write","input":{"file_path":"src/main.rs","content":"updated"}}]}}"#,
1639        ];
1640        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1641
1642        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1643        let provider = ClaudeConvo::with_resolver(resolver);
1644        let view =
1645            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1646
1647        // Deduplicated, first-touch order: src/main.rs first, then src/lib.rs
1648        assert_eq!(view.files_changed, vec!["src/main.rs", "src/lib.rs"]);
1649    }
1650
1651    #[test]
1652    fn test_delegations_extracted() {
1653        let temp = TempDir::new().unwrap();
1654        let claude_dir = temp.path().join(".claude");
1655        let project_dir = claude_dir.join("projects/-test-project");
1656        fs::create_dir_all(&project_dir).unwrap();
1657
1658        let entries = [
1659            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","message":{"role":"user","content":"Search for bugs"}}"#,
1660            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","message":{"role":"assistant","content":[{"type":"text","text":"Delegating..."},{"type":"tool_use","id":"task-1","name":"Task","input":{"prompt":"Find the authentication bug","subagent_type":"Explore"}}]}}"#,
1661            r#"{"uuid":"u3","type":"user","timestamp":"2024-01-01T00:00:02Z","message":{"role":"user","content":[{"type":"tool_result","tool_use_id":"task-1","content":"Found the bug in auth.rs line 42","is_error":false}]}}"#,
1662        ];
1663        fs::write(project_dir.join("s1.jsonl"), entries.join("\n")).unwrap();
1664
1665        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1666        let provider = ClaudeConvo::with_resolver(resolver);
1667        let view =
1668            ConversationProvider::load_conversation(&provider, "/test/project", "s1").unwrap();
1669
1670        // Assistant turn should have one delegation
1671        assert_eq!(view.turns[1].delegations.len(), 1);
1672        let d = &view.turns[1].delegations[0];
1673        assert_eq!(d.agent_id, "task-1");
1674        assert_eq!(d.prompt, "Find the authentication bug");
1675        assert!(d.turns.is_empty()); // Sub-agent turns are in separate files
1676        // Result gets populated from tool result assembly
1677        assert_eq!(
1678            d.result.as_deref(),
1679            Some("Found the bug in auth.rs line 42")
1680        );
1681    }
1682
1683    #[test]
1684    fn test_progress_data_enriched_with_extras() {
1685        let entry: ConversationEntry = serde_json::from_str(
1686            r#"{"uuid":"u1","type":"progress","timestamp":"2024-01-01T00:00:00Z","data":{"type":"hook_progress","hookName":"pre-commit"}}"#,
1687        )
1688        .unwrap();
1689        let event = entry_to_watcher_event(&entry);
1690        match event {
1691            WatcherEvent::Progress { kind, data } => {
1692                assert_eq!(kind, "progress");
1693                assert_eq!(data["uuid"], "u1");
1694                assert_eq!(data["timestamp"], "2024-01-01T00:00:00Z");
1695                let claude = &data["claude"];
1696                assert_eq!(claude["data"]["type"], "hook_progress");
1697                assert_eq!(claude["data"]["hookName"], "pre-commit");
1698            }
1699            other => panic!(
1700                "Expected Progress, got {:?}",
1701                std::mem::discriminant(&other)
1702            ),
1703        }
1704    }
1705
1706    #[test]
1707    fn test_progress_data_no_claude_key_when_no_extras() {
1708        let entry: ConversationEntry = serde_json::from_str(
1709            r#"{"uuid":"u1","type":"progress","timestamp":"2024-01-01T00:00:00Z"}"#,
1710        )
1711        .unwrap();
1712        let event = entry_to_watcher_event(&entry);
1713        match event {
1714            WatcherEvent::Progress { data, .. } => {
1715                assert!(data.get("claude").is_none());
1716            }
1717            other => panic!(
1718                "Expected Progress, got {:?}",
1719                std::mem::discriminant(&other)
1720            ),
1721        }
1722    }
1723
1724    #[test]
1725    fn test_no_delegations_for_non_task_tools() {
1726        let (_temp, provider) = setup_provider();
1727        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-1")
1728            .unwrap();
1729
1730        // No turns should have delegations (none use Task tool)
1731        for turn in &view.turns {
1732            assert!(turn.delegations.is_empty());
1733        }
1734    }
1735
1736    // ── Session chain tests ─────────────────────────────────────────
1737
1738    fn setup_chained_provider() -> (TempDir, ClaudeConvo) {
1739        let temp = TempDir::new().unwrap();
1740        let claude_dir = temp.path().join(".claude");
1741        let project_dir = claude_dir.join("projects/-test-project");
1742        fs::create_dir_all(&project_dir).unwrap();
1743
1744        // Session A: original conversation
1745        let entries_a = [
1746            r#"{"uuid":"a1","type":"user","timestamp":"2024-01-01T00:00:00Z","sessionId":"session-a","message":{"role":"user","content":"Fix the bug"}}"#,
1747            r#"{"uuid":"a2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","sessionId":"session-a","message":{"role":"assistant","content":"I'll fix that.","model":"claude-opus-4-6","usage":{"input_tokens":100,"output_tokens":50}}}"#,
1748        ];
1749        fs::write(project_dir.join("session-a.jsonl"), entries_a.join("\n")).unwrap();
1750
1751        // Session B: continuation with bridge entry
1752        let entries_b = [
1753            // Bridge entry: session_id points back to session-a
1754            r#"{"uuid":"b0","type":"user","timestamp":"2024-01-01T01:00:00Z","sessionId":"session-a","message":{"role":"user","content":"Continue the fix"}}"#,
1755            // Real entries in session-b
1756            r#"{"uuid":"b1","type":"user","timestamp":"2024-01-01T01:00:01Z","sessionId":"session-b","message":{"role":"user","content":"What about the tests?"}}"#,
1757            r#"{"uuid":"b2","type":"assistant","timestamp":"2024-01-01T01:00:02Z","sessionId":"session-b","message":{"role":"assistant","content":"Tests pass now.","model":"claude-opus-4-6","usage":{"input_tokens":200,"output_tokens":100}}}"#,
1758        ];
1759        fs::write(project_dir.join("session-b.jsonl"), entries_b.join("\n")).unwrap();
1760
1761        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1762        (temp, ClaudeConvo::with_resolver(resolver))
1763    }
1764
1765    #[test]
1766    fn test_load_conversation_merges_chain() {
1767        let (_temp, provider) = setup_chained_provider();
1768
1769        // Load from session-a — should merge with session-b
1770        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-a")
1771            .unwrap();
1772
1773        // Should have turns from both segments (minus the bridge entry)
1774        // session-a: a1 (user), a2 (assistant)
1775        // session-b: b1 (user), b2 (assistant) — b0 is bridge, filtered
1776        assert_eq!(view.turns.len(), 4);
1777        assert_eq!(view.turns[0].text, "Fix the bug");
1778        assert_eq!(view.turns[1].text, "I'll fix that.");
1779        assert_eq!(view.turns[2].text, "What about the tests?");
1780        assert_eq!(view.turns[3].text, "Tests pass now.");
1781
1782        // Session IDs should be set
1783        assert_eq!(view.session_ids, vec!["session-a", "session-b"]);
1784    }
1785
1786    #[test]
1787    fn test_load_conversation_skips_bridge_entries() {
1788        let (_temp, provider) = setup_chained_provider();
1789
1790        let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-a")
1791            .unwrap();
1792
1793        // Bridge entry text "Continue the fix" should NOT appear
1794        for turn in &view.turns {
1795            assert_ne!(turn.text, "Continue the fix");
1796        }
1797    }
1798
1799    #[test]
1800    fn test_load_conversation_single_segment_unchanged() {
1801        let temp = TempDir::new().unwrap();
1802        let claude_dir = temp.path().join(".claude");
1803        let project_dir = claude_dir.join("projects/-test-project");
1804        fs::create_dir_all(&project_dir).unwrap();
1805
1806        let entries = [
1807            r#"{"uuid":"u1","type":"user","timestamp":"2024-01-01T00:00:00Z","sessionId":"solo","message":{"role":"user","content":"Hello"}}"#,
1808            r#"{"uuid":"u2","type":"assistant","timestamp":"2024-01-01T00:00:01Z","sessionId":"solo","message":{"role":"assistant","content":"Hi there!"}}"#,
1809        ];
1810        fs::write(project_dir.join("solo.jsonl"), entries.join("\n")).unwrap();
1811
1812        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1813        let provider = ClaudeConvo::with_resolver(resolver);
1814        let view =
1815            ConversationProvider::load_conversation(&provider, "/test/project", "solo").unwrap();
1816
1817        assert_eq!(view.turns.len(), 2);
1818        assert_eq!(view.turns[0].text, "Hello");
1819        assert_eq!(view.turns[1].text, "Hi there!");
1820        // Single segment — session_ids should be empty
1821        assert!(view.session_ids.is_empty());
1822    }
1823
1824    #[test]
1825    fn test_list_metadata_chain_transparent() {
1826        let (_temp, provider) = setup_chained_provider();
1827
1828        let metas = ConversationProvider::list_metadata(&provider, "/test/project").unwrap();
1829
1830        // Chain-default: only the chain head is returned
1831        assert_eq!(metas.len(), 1);
1832        assert_eq!(metas[0].id, "session-a");
1833
1834        // Chains are transparent — no predecessor/successor links
1835        assert!(metas[0].predecessor.is_none());
1836        assert!(metas[0].successor.is_none());
1837    }
1838
1839    #[cfg(feature = "watcher")]
1840    #[test]
1841    fn test_watcher_emits_rotation_progress() {
1842        let temp = TempDir::new().unwrap();
1843        let claude_dir = temp.path().join(".claude");
1844        let project_dir = claude_dir.join("projects/-test-project");
1845        fs::create_dir_all(&project_dir).unwrap();
1846
1847        // Session A
1848        let entry_a = r#"{"uuid":"a1","type":"user","timestamp":"2024-01-01T00:00:00Z","sessionId":"session-a","message":{"role":"user","content":"Hello"}}"#;
1849        fs::write(
1850            project_dir.join("session-a.jsonl"),
1851            format!("{}\n", entry_a),
1852        )
1853        .unwrap();
1854
1855        let resolver = PathResolver::new().with_claude_dir(&claude_dir);
1856        let manager = ClaudeConvo::with_resolver(resolver);
1857
1858        let mut watcher = crate::watcher::ConversationWatcher::new(
1859            manager,
1860            "/test/project".to_string(),
1861            "session-a".to_string(),
1862        );
1863
1864        // First poll via trait: consume session-a entries
1865        let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1866        assert_eq!(events.len(), 1);
1867        assert!(matches!(&events[0], WatcherEvent::Turn(_)));
1868
1869        // Create successor session-b
1870        let entries_b = [
1871            r#"{"uuid":"b0","type":"user","timestamp":"2024-01-01T01:00:00Z","sessionId":"session-a","message":{"role":"user","content":"Bridge"}}"#,
1872            r#"{"uuid":"b1","type":"user","timestamp":"2024-01-01T01:00:01Z","sessionId":"session-b","message":{"role":"user","content":"New"}}"#,
1873        ];
1874        fs::write(project_dir.join("session-b.jsonl"), entries_b.join("\n")).unwrap();
1875
1876        // Second poll via trait: should include rotation Progress event
1877        let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1878
1879        // First event: Progress(session_rotated) with from/to
1880        assert!(
1881            events.len() >= 2,
1882            "Expected Progress + Turn, got {} events",
1883            events.len()
1884        );
1885        match &events[0] {
1886            WatcherEvent::Progress { kind, data } => {
1887                assert_eq!(kind, "session_rotated");
1888                assert_eq!(data["from"], "session-a");
1889                assert_eq!(data["to"], "session-b");
1890            }
1891            other => panic!("Expected Progress, got {:?}", std::mem::discriminant(other)),
1892        }
1893
1894        // Second event: Turn for b1 (bridge entry b0 filtered out)
1895        match &events[1] {
1896            WatcherEvent::Turn(turn) => {
1897                assert_eq!(turn.id, "b1");
1898                assert_eq!(turn.text, "New");
1899            }
1900            other => panic!("Expected Turn(b1), got {:?}", std::mem::discriminant(other)),
1901        }
1902
1903        // No bridge entry should appear as a Turn
1904        for event in &events {
1905            if let WatcherEvent::Turn(t) = event {
1906                assert_ne!(t.id, "b0", "Bridge entry should not appear as a Turn");
1907            }
1908        }
1909    }
1910
1911    #[test]
1912    fn test_load_metadata_chain_transparent() {
1913        let (_temp, provider) = setup_chained_provider();
1914
1915        // Load from chain head — aggregated metadata
1916        let meta_a =
1917            ConversationProvider::load_metadata(&provider, "/test/project", "session-a").unwrap();
1918        assert_eq!(meta_a.id, "session-a");
1919        // Aggregated message count across both segments (2 + 3 = 5)
1920        assert_eq!(meta_a.message_count, 5);
1921        // Chains are transparent — no predecessor/successor links
1922        assert!(meta_a.predecessor.is_none());
1923        assert!(meta_a.successor.is_none());
1924
1925        // Load from a successor — still resolves the full chain
1926        let meta_b =
1927            ConversationProvider::load_metadata(&provider, "/test/project", "session-b").unwrap();
1928        assert_eq!(meta_b.id, "session-a"); // head of chain
1929        assert_eq!(meta_b.message_count, 5);
1930        assert!(meta_b.predecessor.is_none());
1931        assert!(meta_b.successor.is_none());
1932    }
1933}