#![allow(clippy::cast_possible_truncation, clippy::cast_sign_loss, clippy::cast_precision_loss, clippy::cast_possible_wrap)] //! Tool: `edit` — replace a substring in a file with a new string. use std::fs; use std::path::PathBuf; use serde_json::{json, Value}; use anyhow::{Result, anyhow}; use super::super::Tool; use super::super::ToolCtx; use super::super::resolve_path; use super::super::check_graduated_checks; use super::helpers::arg_str; /// Tool: replace text in a file. Requires the old string to be unique unless `replace_all` is true. pub struct Edit; impl Tool for Edit { fn name(&self) -> &'static str { "edit" } fn description(&self) -> &'static str { "Replace a string in a file with a new string. The old string must be unique unless replace_all is true." } fn parameters(&self) -> Value { json!({ "type": "object", "properties": { "path": { "type": "string", "description": "Path to the file to edit (relative to workspace root)" }, "old": { "type": "string", "description": "The exact text to replace" }, "new": { "type": "string", "description": "The replacement text" }, "replace_all": { "type": "boolean", "description": "Replace all occurrences instead of requiring uniqueness" }, "reason": { "type": "string", "description": "Reason for the change (must be non-empty)" } }, "required": ["path", "old", "new", "reason"] }) } /// Perform the in-file string replacement. /// /// Flow: validate args → resolve path → read file → count occurrences → /// replace one or all → write back → report byte delta (+ optional graduated checks). /// /// Why: requires a non-empty `reason` and a non-empty `old` string to prevent /// accidental identity edits. Enforces uniqueness unless `replace_all` is set. fn run(&self, ctx: &ToolCtx, args: &Value) -> Result { let rel = arg_str(args, "path")?; let old = arg_str(args, "old")?; let new_str = arg_str(args, "new")?; let reason = arg_str(args, "reason")?; if reason.trim().is_empty() { anyhow::bail!("reason must be a non-empty string"); } if old.is_empty() { anyhow::bail!("'old' must be a non-empty string; use 'write' to replace entire file contents"); } let check_matches = check_graduated_checks(&rel, &new_str, &ctx.graduated_checks); let replace_all = args.get("replace_all").and_then(serde_json::Value::as_bool).unwrap_or(false); let path: PathBuf = resolve_path(&ctx.workspaces, &rel)?; if !path.exists() { anyhow::bail!("file '{}' does not exist at resolved path {}", rel, path.display()); } if path.is_dir() { anyhow::bail!("'{rel}' is a directory, not a file"); } let content = fs::read_to_string(&path) .map_err(|e| anyhow!("failed to read '{rel}': {e}"))?; if !content.contains(&old) { anyhow::bail!("old string not found in '{rel}'"); } if !replace_all { let count = content.matches(&old).count(); if count > 1 { anyhow::bail!( "old string appears {count} times in '{rel}'. Set replace_all=true to replace all occurrences, or provide a more specific match." ); } } let new_content = if replace_all { content.replace(&old, &new_str) } else { content.replacen(&old, &new_str, 1) }; fs::write(&path, &new_content) .map_err(|e| anyhow!("failed to write '{rel}': {e}"))?; let bytes_diff = if new_content.len() > content.len() { new_content.len() - content.len() } else { content.len() - new_content.len() }; // Notify the LSP server of the on-disk change so diagnostics stay fresh. // Never fail the edit because of this — LSP errors are surfaced as a // trailing annotation on the success message instead. let lsp_note = if let Ok(mut lsp) = ctx.lsp_manager.lock() { lsp.did_change_file(&path); String::new() } else { String::new() }; if check_matches.is_empty() { Ok(format!("edited {} ({} byte delta){}", rel, bytes_diff as isize, lsp_note)) } else { Ok(format!("edited {} ({} byte delta). Graduated checks matched: {}{}", rel, bytes_diff as isize, check_matches.join(", "), lsp_note)) } } }