change cursor shape depending on the mode; cargo fmt

Author: alexylon

README.md

@@ -89,8 +89,8 @@ sofos
 - `/resume` - Resume previous session
 - `/clear` - Clear conversation history
 - `/think [on|off]` - Toggle extended thinking (shows status if no arg)
-- `/s` - Safe mode (read-only, prompt: **`λ:`**)
-- `/n` - Normal mode (all tools, prompt: **`λ>`**)
+- `/s` - Safe mode (read-only, prompt: **`λ:`**, blinking underscore (`_`) cursor)
+- `/n` - Normal mode (all tools, prompt: **`λ>`**, default cursor)
 - `/exit`, `/quit`, `/q`, `Ctrl+D` - Exit with cost summary
 - `ESC` - Interrupt AI response
 

src/prompt.rs

@@ -1,3 +1,4 @@
+use crate::ui::{set_normal_mode_cursor_style, set_safe_mode_cursor_style};
 use colored::Colorize;
 use reedline::{Prompt, PromptEditMode, PromptHistorySearch};
 
@@ -12,6 +13,12 @@ impl ReplPrompt {
     }
 
     pub fn set_safe_mode(&mut self, safe_mode: bool) {
+        if safe_mode {
+            set_safe_mode_cursor_style().ok();
+        } else {
+            set_normal_mode_cursor_style().ok();
+        }
+
         self.safe_mode = safe_mode;
     }
 }

src/repl.rs

@@ -12,7 +12,7 @@ use crate::response_handler::ResponseHandler;
 use crate::session_state::SessionState;
 use crate::tools::image::{extract_image_references, ImageLoader, ImageReference};
 use crate::tools::ToolExecutor;
-use crate::ui::UI;
+use crate::ui::{set_safe_mode_cursor_style, UI};
 use colored::Colorize;
 use crossterm::event::{KeyCode, KeyModifiers};
 use reedline::{
@@ -98,6 +98,7 @@ impl Repl {
 
         if config.safe_mode {
             conversation.add_user_message(SAFE_MODE_MESSAGE.to_string());
+            set_safe_mode_cursor_style()?;
         }
 
         let commands: Vec<String> = crate::commands::COMMANDS

src/tools/codesearch.rs

@@ -14,17 +14,16 @@ impl CodeSearchTool {
         let env_override = std::env::var_os("SOFOS_RG_PATH").map(PathBuf::from);
 
         // Fallback search list for common macOS/Homebrew and Linux locations
-        let fallback_paths = [
-            "/opt/homebrew/bin/rg",
-            "/usr/local/bin/rg",
-            "/usr/bin/rg",
-        ];
+        let fallback_paths = ["/opt/homebrew/bin/rg", "/usr/local/bin/rg", "/usr/bin/rg"];
 
         let try_path = |p: &PathBuf| Command::new(p).arg("--version").output();
 
         if let Some(p) = env_override {
             if try_path(&p).is_ok() {
-                return Ok(Self { workspace, rg_path: p });
+                return Ok(Self {
+                    workspace,
+                    rg_path: p,
+                });
             }
         }
 

src/tools/image.rs

@@ -442,19 +442,19 @@ mod tests {
         assert_eq!(text, "check out please");
 
         // Test single-quoted path with spaces
-        let (text, refs) = extract_image_references(
-            "view '/home/user/my photos/vacation.jpg' now",
+        let (text, refs) = extract_image_references("view '/home/user/my photos/vacation.jpg' now");
+        assert_eq!(
+            refs.len(),
+            1,
+            "Should detect single-quoted path with spaces"
         );
-        assert_eq!(refs.len(), 1, "Should detect single-quoted path with spaces");
         assert!(
             matches!(&refs[0], ImageReference::LocalPath(p) if p == "/home/user/my photos/vacation.jpg")
         );
         assert_eq!(text, "view now");
 
         // Test path with spaces at the end
-        let (text, refs) = extract_image_references(
-            "\"/Users/alex/test/image file.png\"",
-        );
+        let (text, refs) = extract_image_references("\"/Users/alex/test/image file.png\"");
         assert_eq!(refs.len(), 1, "Should detect quoted path at end");
         assert!(
             matches!(&refs[0], ImageReference::LocalPath(p) if p == "/Users/alex/test/image file.png")
@@ -465,10 +465,13 @@ mod tests {
     #[test]
     fn test_extract_mixed_quoted_and_unquoted() {
         // Mix of quoted path and unquoted path
-        let (text, refs) = extract_image_references(
-            "compare \"file with space.png\" and simple.jpg",
+        let (text, refs) =
+            extract_image_references("compare \"file with space.png\" and simple.jpg");
+        assert_eq!(
+            refs.len(),
+            2,
+            "Should detect both quoted and unquoted paths"
         );
-        assert_eq!(refs.len(), 2, "Should detect both quoted and unquoted paths");
         assert!(matches!(&refs[0], ImageReference::LocalPath(p) if p == "file with space.png"));
         assert!(matches!(&refs[1], ImageReference::LocalPath(p) if p == "simple.jpg"));
         assert_eq!(text, "compare and");
@@ -477,9 +480,7 @@ mod tests {
     #[test]
     fn test_extract_quoted_non_image() {
         // Quoted text that's not an image should remain in text
-        let (text, refs) = extract_image_references(
-            "the title is \"Hello World\" and image.png",
-        );
+        let (text, refs) = extract_image_references("the title is \"Hello World\" and image.png");
         assert_eq!(refs.len(), 1, "Should only detect the actual image");
         assert!(matches!(&refs[0], ImageReference::LocalPath(p) if p == "image.png"));
         assert_eq!(text, "the title is \"Hello World\" and");
@@ -488,20 +489,21 @@ mod tests {
     #[test]
     fn test_extract_unclosed_quote() {
         // Unclosed quote should be treated as regular text
-        let (text, refs) = extract_image_references(
-            "this is \"unclosed quote and image.png",
-        );
+        let (text, refs) = extract_image_references("this is \"unclosed quote and image.png");
         // The unclosed quote should make everything after it part of the text
-        assert_eq!(refs.len(), 0, "Unclosed quote should prevent image detection");
+        assert_eq!(
+            refs.len(),
+            0,
+            "Unclosed quote should prevent image detection"
+        );
         assert!(text.contains("unclosed quote and image.png"));
     }
 
     #[test]
     fn test_extract_web_url_with_spaces_quoted() {
         // Web URLs with spaces (rare but possible)
-        let (text, refs) = extract_image_references(
-            "see \"https://example.com/my image.png\" please",
-        );
+        let (text, refs) =
+            extract_image_references("see \"https://example.com/my image.png\" please");
         assert_eq!(refs.len(), 1, "Should detect quoted URL with spaces");
         assert!(
             matches!(&refs[0], ImageReference::WebUrl(u) if u == "https://example.com/my image.png")
@@ -513,19 +515,21 @@ mod tests {
     fn test_user_reported_case() {
         // Exact case from user report: path with space but no quotes
         // This will NOT work without quotes - user needs to quote it
-        let (_text, refs) = extract_image_references(
-            "/Users/alex/test/sofos_allowed/test_r copy.png",
-        );
+        let (_text, refs) =
+            extract_image_references("/Users/alex/test/sofos_allowed/test_r copy.png");
         // Without quotes, this gets split into two words
         // Only "copy.png" would be detected as an image
         assert_eq!(refs.len(), 1, "Only the second part is detected as image");
         assert!(matches!(&refs[0], ImageReference::LocalPath(p) if p == "copy.png"));
 
         // With quotes, it should work
-        let (text, refs) = extract_image_references(
-            "\"/Users/alex/test/sofos_allowed/test_r copy.png\"",
+        let (text, refs) =
+            extract_image_references("\"/Users/alex/test/sofos_allowed/test_r copy.png\"");
+        assert_eq!(
+            refs.len(),
+            1,
+            "Quoted path should be detected as single image"
         );
-        assert_eq!(refs.len(), 1, "Quoted path should be detected as single image");
         assert!(
             matches!(&refs[0], ImageReference::LocalPath(p) if p == "/Users/alex/test/sofos_allowed/test_r copy.png")
         );

src/ui.rs

@@ -1,14 +1,15 @@
+use crate::history::DisplayMessage;
+use crate::syntax::SyntaxHighlighter;
 use colored::Colorize;
+use crossterm::cursor::SetCursorStyle;
 use crossterm::event::{self, Event, KeyCode, KeyEvent};
-use std::io::{self, Write};
+use crossterm::execute;
+use std::io::{self, stdout, Write};
 use std::sync::atomic::{AtomicBool, Ordering};
 use std::sync::Arc;
 use std::thread;
 use std::time::Duration;
 
-use crate::history::DisplayMessage;
-use crate::syntax::SyntaxHighlighter;
-
 /// Message severity levels for consistent UI feedback
 #[derive(Debug, Clone, Copy, PartialEq)]
 pub enum MessageSeverity {
@@ -516,3 +517,18 @@ impl UI {
         result.chars().rev().collect()
     }
 }
+
+fn set_cursor_style(style: SetCursorStyle) -> io::Result<()> {
+    let mut out = stdout();
+    execute!(out, style)?;
+    out.flush()?;
+    Ok(())
+}
+
+pub fn set_safe_mode_cursor_style() -> io::Result<()> {
+    set_cursor_style(SetCursorStyle::BlinkingUnderScore)
+}
+
+pub fn set_normal_mode_cursor_style() -> io::Result<()> {
+    set_cursor_style(SetCursorStyle::DefaultUserShape)
+}