claude-rust-tools 1.3.0

Tool implementations for bash and file operations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use claude_rust_errors::{AppError, AppResult};
use claude_rust_types::{PermissionLevel, Tool};
use serde_json::{Value, json};
use tokio::process::Command;

pub struct GrepTool;

#[async_trait::async_trait]
impl Tool for GrepTool {
    fn name(&self) -> &str {
        "grep"
    }

    fn description(&self) -> &str {
        "Search file contents using regex. Uses ripgrep (rg) if available, falls back to grep."
    }

    fn input_schema(&self) -> Value {
        json!({
            "type": "object",
            "properties": {
                "pattern": {
                    "type": "string",
                    "description": "Regular expression pattern to search for"
                },
                "path": {
                    "type": "string",
                    "description": "File or directory to search in (defaults to current working directory)"
                },
                "glob": {
                    "type": "string",
                    "description": "Glob pattern to filter files (e.g. \"*.rs\", \"*.{ts,tsx}\")"
                }
            },
            "required": ["pattern"]
        })
    }

    fn permission_level(&self) -> PermissionLevel {
        PermissionLevel::ReadOnly
    }

    async fn execute(&self, input: Value) -> AppResult<String> {
        let pattern = input
            .get("pattern")
            .and_then(|v| v.as_str())
            .ok_or_else(|| AppError::Tool("missing 'pattern' field".into()))?;

        let search_path = input
            .get("path")
            .and_then(|v| v.as_str())
            .unwrap_or(".");

        let file_glob = input.get("glob").and_then(|v| v.as_str());

        tracing::info!(pattern, search_path, "grepping");

        // Try ripgrep first, fall back to grep
        let output = match try_ripgrep(pattern, search_path, file_glob).await {
            Ok(out) => out,
            Err(_) => try_grep(pattern, search_path).await?,
        };

        if output.is_empty() {
            return Ok("No matches found.".into());
        }

        let mut result = output;
        if result.len() > 100_000 {
            result.truncate(100_000);
            result.push_str("\n... (truncated)");
        }

        Ok(result)
    }
}

async fn try_ripgrep(
    pattern: &str,
    path: &str,
    file_glob: Option<&str>,
) -> AppResult<String> {
    let mut cmd = Command::new("rg");
    cmd.arg("-n").arg("--no-heading").arg(pattern);

    if let Some(g) = file_glob {
        cmd.arg("--glob").arg(g);
    }

    cmd.arg(path);

    let output = cmd
        .output()
        .await
        .map_err(|e| AppError::Tool(format!("rg not available: {e}")))?;

    // rg returns exit code 1 for no matches, 2 for errors
    if output.status.code() == Some(2) {
        return Err(AppError::Tool("ripgrep error".into()));
    }

    Ok(String::from_utf8_lossy(&output.stdout).to_string())
}

async fn try_grep(pattern: &str, path: &str) -> AppResult<String> {
    let output = Command::new("grep")
        .arg("-rn")
        .arg(pattern)
        .arg(path)
        .output()
        .await
        .map_err(|e| AppError::Tool(format!("grep failed: {e}")))?;

    Ok(String::from_utf8_lossy(&output.stdout).to_string())
}