apollo-agent 0.2.2

Local-first Rust AI agent runtime — Telegram-first, trait-driven, SurrealDB + RocksDB state layer.
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

//! Edit tool — surgical text replacement in files (like OpenClaw's Edit).
//! Finds exact text and replaces it, preserving the rest of the file.

use async_trait::async_trait;
use serde::Deserialize;
use std::path::PathBuf;

use super::sandbox::resolve_workspace_existing_path;
use super::traits::*;

pub struct EditTool {
    workspace: PathBuf,
}

impl EditTool {
    pub fn new(workspace: PathBuf) -> Self {
        Self { workspace }
    }
}

#[derive(Deserialize)]
struct EditArgs {
    /// File path (relative to workspace or absolute)
    #[serde(alias = "file_path")]
    path: String,
    /// Exact text to find (must match exactly including whitespace)
    #[serde(alias = "oldText")]
    old_string: String,
    /// New text to replace with
    #[serde(alias = "newText")]
    new_string: String,
}

#[async_trait]
impl Tool for EditTool {
    fn name(&self) -> &str {
        "Edit"
    }

    fn spec(&self) -> ToolSpec {
        ToolSpec {
            name: "Edit".to_string(),
            description: "Edit a file by replacing exact text. The old_string must match exactly (including whitespace). Use this for precise, surgical edits.".to_string(),
            parameters: serde_json::json!({
                "type": "object",
                "properties": {
                    "path": {
                        "type": "string",
                        "description": "Path to the file to edit (relative to workspace or absolute)"
                    },
                    "old_string": {
                        "type": "string",
                        "description": "Exact text to find and replace (must match exactly)"
                    },
                    "new_string": {
                        "type": "string",
                        "description": "New text to replace the old text with"
                    }
                },
                "required": ["path", "old_string", "new_string"]
            }),
        }
    }

    async fn execute(&self, arguments: &str) -> anyhow::Result<ToolResult> {
        let args: EditArgs = serde_json::from_str(arguments)?;

        let full_path = match resolve_workspace_existing_path(&self.workspace, &args.path) {
            Ok(path) => path,
            Err(err) => return Ok(ToolResult::error(err.to_string())),
        };

        let content = match tokio::fs::read_to_string(&full_path).await {
            Ok(c) => c,
            Err(e) => {
                return Ok(ToolResult::error(format!(
                    "Cannot read file '{}': {}",
                    args.path, e
                )))
            }
        };

        // Find and replace
        if !content.contains(&args.old_string) {
            return Ok(ToolResult::error(format!(
                "Could not find the exact text in {}. The old_string must match exactly including all whitespace and newlines.",
                args.path
            )));
        }

        let new_content = content.replacen(&args.old_string, &args.new_string, 1);

        // Write back
        match tokio::fs::write(&full_path, &new_content).await {
            Ok(_) => Ok(ToolResult::success(format!(
                "Successfully replaced text in {}",
                args.path
            ))),
            Err(e) => Ok(ToolResult::error(format!("Failed to write: {}", e))),
        }
    }
}