1use 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
19fn 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
29pub 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
45pub 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 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
78fn 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 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
146fn 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 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
199fn 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
237fn 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
274fn 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
283fn 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
319fn 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 for (idx, raw) in convo.preamble.iter().enumerate() {
331 events.push(preamble_to_event(idx, raw));
332 }
333
334 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 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 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 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 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
456fn 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
487fn 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
529pub(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
548fn 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 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
602fn 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 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
639fn 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
676impl 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#[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 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 let mut updated_turn: Option<Turn> = None;
769
770 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 let mut updated = (**turn).clone();
780 merge_tool_results(std::slice::from_mut(&mut updated), msg);
781 updated_turn = Some(updated.clone());
782 **turn = updated;
785 break;
786 }
787 }
788
789 if let Some(turn) = updated_turn {
790 events.push(WatcherEvent::TurnUpdated(Box::new(turn)));
791 }
792 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
809pub fn to_view(convo: &Conversation) -> ConversationView {
817 conversation_to_view(convo)
818}
819
820pub fn to_turn(entry: &ConversationEntry) -> Option<Turn> {
826 entry_to_turn(entry)
827}
828
829#[cfg(test)]
832mod tests {
833 use super::*;
834 use crate::PathResolver;
835 use std::fs;
836 use tempfile::TempDir;
837
838 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 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 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 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 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 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 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 assert_eq!(view.turns.len(), 5);
1044
1045 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 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 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 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 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 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 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 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); 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 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 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 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 assert_eq!(events.len(), 4);
1349
1350 assert!(matches!(&events[0], WatcherEvent::Turn(t) if t.role == Role::User));
1352
1353 assert!(matches!(&events[1], WatcherEvent::Turn(t) if t.role == Role::Assistant));
1355
1356 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 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 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 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 let events1 = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1403 assert_eq!(events1.len(), 2);
1404 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 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 let events2 = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1421 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 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 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 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 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 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 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 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 #[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 assert_eq!(
1545 view.turns[1].tool_uses[0].category,
1546 Some(ToolCategory::FileRead)
1547 );
1548 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 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 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 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 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 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()); 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 for turn in &view.turns {
1732 assert!(turn.delegations.is_empty());
1733 }
1734 }
1735
1736 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 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 let entries_b = [
1753 r#"{"uuid":"b0","type":"user","timestamp":"2024-01-01T01:00:00Z","sessionId":"session-a","message":{"role":"user","content":"Continue the fix"}}"#,
1755 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 let view = ConversationProvider::load_conversation(&provider, "/test/project", "session-a")
1771 .unwrap();
1772
1773 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 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 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 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 assert_eq!(metas.len(), 1);
1832 assert_eq!(metas[0].id, "session-a");
1833
1834 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 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 let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1866 assert_eq!(events.len(), 1);
1867 assert!(matches!(&events[0], WatcherEvent::Turn(_)));
1868
1869 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 let events = toolpath_convo::ConversationWatcher::poll(&mut watcher).unwrap();
1878
1879 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 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 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 let meta_a =
1917 ConversationProvider::load_metadata(&provider, "/test/project", "session-a").unwrap();
1918 assert_eq!(meta_a.id, "session-a");
1919 assert_eq!(meta_a.message_count, 5);
1921 assert!(meta_a.predecessor.is_none());
1923 assert!(meta_a.successor.is_none());
1924
1925 let meta_b =
1927 ConversationProvider::load_metadata(&provider, "/test/project", "session-b").unwrap();
1928 assert_eq!(meta_b.id, "session-a"); assert_eq!(meta_b.message_count, 5);
1930 assert!(meta_b.predecessor.is_none());
1931 assert!(meta_b.successor.is_none());
1932 }
1933}