Merge pull request #97 from wafflestudio/feature/tauri-ui2

feat(ui): add session statistics and flamechart timeline visualization

Author: tantara

src-tauri/src/commands/session.rs

@@ -120,15 +120,20 @@ pub async fn get_session_detail(
         tool_operations.len()
     );
 
+    // Create a map of tool_operation_id -> tool_operation for efficient lookup
+    log::debug!("Building tool operation lookup map");
+    let tool_op_by_id: std::collections::HashMap<_, _> = tool_operations
+        .into_iter()
+        .map(|op| (op.id, op))
+        .collect();
+
     // Create a map of message_id -> tool_operation
-    log::debug!("Building tool operation map");
+    log::debug!("Building message -> tool operation map");
     let mut tool_op_map = std::collections::HashMap::new();
-    for tool_op in tool_operations {
-        // Find the message that references this tool operation
-        for msg in &response.messages {
-            if msg.tool_operation_id == Some(tool_op.id) {
+    for msg in &response.messages {
+        if let Some(tool_op_id) = msg.tool_operation_id {
+            if let Some(tool_op) = tool_op_by_id.get(&tool_op_id) {
                 tool_op_map.insert(msg.id, tool_op.clone());
-                break;
             }
         }
     }
@@ -159,6 +164,8 @@ pub async fn get_session_detail(
                     timestamp: op.timestamp.to_rfc3339(),
                     success: op.success,
                     result_summary: op.result_summary.clone(),
+                    raw_input: op.raw_input.clone(),
+                    raw_result: op.raw_result.clone(),
                     file_metadata: op.file_metadata.as_ref().map(|fm| FileMetadataItem {
                         file_path: fm.file_path.clone(),
                         file_extension: fm.file_extension.clone(),

src-tauri/src/dto.rs

@@ -51,6 +51,8 @@ pub struct ToolOperationItem {
     pub result_summary: Option<String>,
     pub file_metadata: Option<FileMetadataItem>,
     pub bash_metadata: Option<serde_json::Value>,
+    pub raw_input: Option<serde_json::Value>,
+    pub raw_result: Option<serde_json::Value>,
 }
 
 #[derive(Debug, Serialize, Deserialize)]

src/parsers/claude_code.rs

@@ -265,6 +265,12 @@ impl ClaudeCodeParser {
                         }
                     }
 
+                    // Skip messages with no meaningful content and no tools
+                    if content == "[No content]" && tool_uses.is_empty() && tool_results.is_empty()
+                    {
+                        continue;
+                    }
+
                     let mut message = Message::new(session_id, role, content, timestamp, sequence);
 
                     message.id = message_id;

src/parsers/codex.rs

@@ -316,6 +316,11 @@ impl CodexParser {
                 content.clone()
             };
 
+            // Skip messages with no meaningful content and no tools
+            if content == "[No content]" {
+                continue;
+            }
+
             // Generate a deterministic UUID for the message
             let message_id = self.generate_uuid_from_string(&format!("{session_id}-msg-{index}"));
 

tests/unit/test_codex_parser.rs

@@ -150,9 +150,8 @@ async fn test_codex_parser_empty_content() -> Result<()> {
     assert!(result.is_ok());
     let (session, messages) = result.unwrap();
 
-    assert_eq!(session.message_count, 1);
-    // Empty content should be replaced with "[No content]"
-    assert_eq!(messages[0].content, "[No content]");
+    // Empty content should be discarded so 0
+    assert_eq!(session.message_count, 0);
 
     Ok(())
 }

ui-react/package.json

@@ -60,12 +60,14 @@
     "input-otp": "1.4.1",
     "lucide-react": "^0.553.0",
     "next-themes": "^0.4.6",
+    "plotly.js": "^3.3.0",
     "react": "^19.1.1",
     "react-day-picker": "9.8.0",
     "react-dom": "^19.1.1",
     "react-hook-form": "^7.60.0",
     "react-hotkeys-hook": "^5.2.1",
     "react-markdown": "^10.1.0",
+    "react-plotly.js": "^2.6.0",
     "react-resizable-panels": "^2.1.7",
     "recharts": "2.15.4",
     "remark-gfm": "^4.0.1",
@@ -79,6 +81,7 @@
     "@biomejs/biome": "^2.3.4",
     "@eslint/js": "^9.36.0",
     "@tailwindcss/postcss": "^4.1.17",
+    "@types/plotly.js": "^3.0.8",
     "@types/react": "^19.1.16",
     "@types/react-dom": "^19.1.9",
     "@vitejs/plugin-react": "^5.0.4",