matrixcode_core/tools/
multi_edit.rs

1use anyhow::Result;
2use async_trait::async_trait;
3use serde_json::{Value, json};
4
5use super::{Tool, ToolDefinition};
6use crate::approval::RiskLevel;
7
8pub struct MultiEditTool;
9
10#[async_trait]
11impl Tool for MultiEditTool {
12    fn definition(&self) -> ToolDefinition {
13        ToolDefinition {
14            name: "multi_edit".to_string(),
15            description: "对单个文件应用多处精确字符串替换，一次性原子写入。\
16                 每个编辑必须在前序编辑后的文件状态中精确匹配一次。\
17                 若任一编辑失败，文件不会被修改。"
18                .to_string(),
19            parameters: json!({
20                "type": "object",
21                "properties": {
22                    "path": {
23                        "type": "string",
24                        "description": "要编辑的文件路径"
25                    },
26                    "edits": {
27                        "type": "array",
28                        "description": "有序的 {old_string, new_string} 替换列表",
29                        "items": {
30                            "type": "object",
31                            "properties": {
32                                "old_string": {"type": "string"},
33                                "new_string": {"type": "string"}
34                            },
35                            "required": ["old_string", "new_string"]
36                        }
37                    }
38                },
39                "required": ["path", "edits"]
40            }),
41            ..Default::default()
42        }
43    }
44
45    async fn execute(&self, params: Value) -> Result<String> {
46        let path = params["path"]
47            .as_str()
48            .ok_or_else(|| anyhow::anyhow!("missing 'path'"))?;
49        let edits = params["edits"]
50            .as_array()
51            .ok_or_else(|| anyhow::anyhow!("missing 'edits' array"))?;
52        if edits.is_empty() {
53            anyhow::bail!("'edits' must contain at least one entry");
54        }
55
56        // Show spinner while editing - RAII guard ensures cleanup on error
57        // let mut spinner = ToolSpinner::new(&format!("multi-editing {} ({} edits)", path, edits.len()));
58
59        let mut content = tokio::fs::read_to_string(path).await?;
60
61        for (idx, edit) in edits.iter().enumerate() {
62            let old_string = edit["old_string"]
63                .as_str()
64                .ok_or_else(|| anyhow::anyhow!("edit {}: missing 'old_string'", idx))?;
65            let new_string = edit["new_string"]
66                .as_str()
67                .ok_or_else(|| anyhow::anyhow!("edit {}: missing 'new_string'", idx))?;
68
69            if old_string.is_empty() {
70                // spinner.finish_error("empty old_string");
71                anyhow::bail!("edit {}: 'old_string' must not be empty", idx);
72            }
73
74            let count = content.matches(old_string).count();
75            if count == 0 {
76                // spinner.finish_error(&format!("edit {} not found", idx));
77                anyhow::bail!("edit {}: old_string not found", idx);
78            }
79            if count > 1 {
80                // spinner.finish_error(&format!("edit {} multiple matches", idx));
81                anyhow::bail!(
82                    "edit {}: old_string found {} times — must be unique",
83                    idx,
84                    count
85                );
86            }
87
88            content = content.replacen(old_string, new_string, 1);
89        }
90
91        tokio::fs::write(path, &content).await?;
92
93        // Return diff-style output
94        let mut diff = format!("Applied {} edit(s) to {}\n", edits.len(), path);
95        for (idx, edit) in edits.iter().enumerate() {
96            let old_string = edit["old_string"].as_str().unwrap_or("");
97            let new_string = edit["new_string"].as_str().unwrap_or("");
98            if edits.len() > 1 {
99                diff.push_str(&format!("edit {}:\n", idx + 1));
100            }
101            for line in old_string.lines().take(3) {
102                diff.push_str(&format!("- {}\n", line));
103            }
104            if old_string.lines().count() > 3 {
105                diff.push_str(&format!(
106                    "  ... ({} more lines removed)\n",
107                    old_string.lines().count() - 3
108                ));
109            }
110            for line in new_string.lines().take(3) {
111                diff.push_str(&format!("+ {}\n", line));
112            }
113            if new_string.lines().count() > 3 {
114                diff.push_str(&format!(
115                    "  ... ({} more lines added)\n",
116                    new_string.lines().count() - 3
117                ));
118            }
119        }
120        Ok(diff)
121    }
122
123    fn risk_level(&self) -> RiskLevel {
124        RiskLevel::Mutating
125    }
126}
matrixcode_core/tools/multi_edit.rs

matrixcode_core/tools/
multi_edit.rs
