feat(server): refactor handlers, router, state management, and memory tools

server/src/handlers/logs.rs

@@ -1,11 +1,11 @@
 use crate::router::McpTool;
 use crate::state::MemoryState;
-use crate::tools::{WatchProcessLogsTool, GetRecentLogsTool};
+use crate::tools::{GetRecentLogsTool, WatchProcessLogsTool};
 use async_trait::async_trait;
 use serde_json::Value;
-use std::sync::Arc;
-use std::io::{Read, Seek, SeekFrom};
 use std::fs::File;
+use std::io::{Read, Seek, SeekFrom};
+use std::sync::Arc;
 
 pub struct WatchProcessLogsHandler;
 
@@ -23,9 +23,13 @@ impl McpTool for WatchProcessLogsHandler {
     }
 
     async fn execute(&self, args: Value, _state: Arc<MemoryState>) -> crate::error::Result<String> {
-        let tool_args: WatchProcessLogsTool = serde_json::from_value(args).map_err(|e| e.to_string())?;
+        let tool_args: WatchProcessLogsTool =
+            serde_json::from_value(args).map_err(|e| e.to_string())?;
         if !std::path::Path::new(&tool_args.file_path).exists() {
-            return Err(crate::error::AppError::Internal(format!("File does not exist: {}", tool_args.file_path)));
+            return Err(crate::error::AppError::Internal(format!(
+                "File does not exist: {}",
+                tool_args.file_path
+            )));
         }
         Ok(format!("Started watching logs for {}", tool_args.file_path))
     }
@@ -47,25 +51,30 @@ impl McpTool for GetRecentLogsHandler {
     }
 
     async fn execute(&self, args: Value, _state: Arc<MemoryState>) -> crate::error::Result<String> {
-        let tool_args: GetRecentLogsTool = serde_json::from_value(args).map_err(|e| e.to_string())?;
-        
+        let tool_args: GetRecentLogsTool =
+            serde_json::from_value(args).map_err(|e| e.to_string())?;
+
         let result = tokio::task::spawn_blocking(move || -> crate::error::Result<String> {
-            let mut file = File::open(&tool_args.file_path).map_err(|e| crate::error::AppError::Internal(format!("Failed to open file: {}", e)))?;
+            let mut file = File::open(&tool_args.file_path).map_err(|e| {
+                crate::error::AppError::Internal(format!("Failed to open file: {}", e))
+            })?;
             let len = file.metadata().map_err(|e| e.to_string())?.len();
-            
+
             let read_size = std::cmp::min(16 * 1024, len);
-            file.seek(SeekFrom::End(-(read_size as i64))).map_err(|e| e.to_string())?;
-            
+            file.seek(SeekFrom::End(-(read_size as i64)))
+                .map_err(|e| e.to_string())?;
+
             let mut buffer = String::new();
-            file.read_to_string(&mut buffer).map_err(|e| e.to_string())?;
-            
+            file.read_to_string(&mut buffer)
+                .map_err(|e| e.to_string())?;
+
             let lines: Vec<&str> = buffer.lines().collect();
             let recent_lines = if lines.len() > 100 {
                 lines[lines.len() - 100..].join("\n")
             } else {
                 buffer
             };
-            
+
             Ok(recent_lines)
         })
         .await
@@ -78,9 +87,9 @@ impl McpTool for GetRecentLogsHandler {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use tempfile::tempdir;
-    use std::sync::Arc;
     use serde_json::json;
+    use std::sync::Arc;
+    use tempfile::tempdir;
 
     #[tokio::test]
     async fn test_watch_process_logs() {
@@ -95,7 +104,8 @@ mod tests {
             "file_path": log_file.to_str().unwrap()
         });
 
-        let result = handler.execute(args, state)
+        let result = handler
+            .execute(args, state)
             .await
             .map_err(|e| format!("Failed to watch logs: {}", e))
             .unwrap();
@@ -115,7 +125,8 @@ mod tests {
             "file_path": log_file.to_str().unwrap()
         });
 
-        let result = handler.execute(args, state)
+        let result = handler
+            .execute(args, state)
             .await
             .map_err(|e| format!("Failed to get recent logs: {}", e))
             .unwrap();