defect-tools 0.1.0-alpha.6

//! `edit_file` tool: exact string replacement.
//!
//! Edit tool — applies a patch to an existing file.

use std::io;
use std::path::PathBuf;
use std::pin::Pin;
use std::sync::Arc;

use agent_client_protocol_schema::{
    Content, ContentBlock, Diff, TextContent, ToolCallContent, ToolCallLocation,
    ToolCallUpdateFields, ToolKind,
};
use defect_agent::error::BoxError;
use defect_agent::fs::{FsBackend, FsError};
use defect_agent::tool::{
    SafetyClass, Tool, ToolCallDescription, ToolContext, ToolError, ToolEvent, ToolSchema,
    ToolStream,
};
use futures::future::BoxFuture;
use futures::stream;
use serde::{Deserialize, Serialize};
use serde_json::json;

use super::replacer::{EditOutcome, replace};

pub struct EditFileTool {
    schema: ToolSchema,
}

impl EditFileTool {
    pub fn new() -> Self {
        Self {
            schema: ToolSchema {
                name: "edit_file".to_string(),
                description: "Replace a string in a UTF-8 text file. \
                              Prefers an exact match; if that fails it falls back to \
                              progressively looser matching (ignoring leading/trailing \
                              whitespace, indentation, and line-ending differences). \
                              Fails if `old_string` cannot be located, or if the match is \
                              not unique unless `replace_all` is true. \
                              Path must be inside the workspace root."
                    .to_string(),
                input_schema: json!({
                    "type": "object",
                    "properties": {
                        "path": {
                            "type": "string",
                            "description": "Absolute path or path relative to the session cwd."
                        },
                        "old_string": {
                            "type": "string",
                            "description": "Exact text to replace. Must match a unique substring \
                                            unless `replace_all` is true. Empty string is rejected."
                        },
                        "new_string": {
                            "type": "string",
                            "description": "Replacement text. Must differ from old_string."
                        },
                        "replace_all": {
                            "type": "boolean",
                            "description": "When true, replace every occurrence; when false (default), \
                                            require old_string to appear exactly once.",
                            "default": false
                        }
                    },
                    "required": ["path", "old_string", "new_string"]
                }),
            },
        }
    }
}

impl Default for EditFileTool {
    fn default() -> Self {
        Self::new()
    }
}

#[derive(Debug, Deserialize)]
struct EditArgs {
    path: String,
    old_string: String,
    new_string: String,
    #[serde(default)]
    replace_all: bool,
}

#[derive(Debug, Serialize)]
struct EditFileOutput {
    matches_replaced: u32,
    bytes_before: u64,
    bytes_after: u64,
    /// Which match level succeeded: `"exact"` for the strict path, or the name of the
    /// fallback matcher (e.g. `"line_trimmed"`). Surfaced so fuzzy matches are observable
    /// rather than silent.
    matched_strategy: &'static str,
}

impl Tool for EditFileTool {
    fn schema(&self) -> &ToolSchema {
        &self.schema
    }

    fn safety_hint(&self, _args: &serde_json::Value) -> SafetyClass {
        SafetyClass::Mutating
    }

    fn describe<'a>(
        &'a self,
        args: &'a serde_json::Value,
        _ctx: ToolContext<'a>,
    ) -> BoxFuture<'a, ToolCallDescription> {
        Box::pin(async move {
            let path = args.get("path").and_then(|v| v.as_str()).unwrap_or("");
            let title = if path.is_empty() {
                "Edit".to_string()
            } else {
                format!("Edit {path}")
            };
            let mut fields = ToolCallUpdateFields::default();
            fields.title = Some(title);
            fields.kind = Some(ToolKind::Edit);
            if !path.is_empty() {
                fields.locations = Some(vec![ToolCallLocation::new(PathBuf::from(path))]);
            }
            ToolCallDescription { fields }
        })
    }

    fn execute(&self, args: serde_json::Value, ctx: ToolContext<'_>) -> ToolStream {
        let cancel = ctx.cancel.clone();
        let fs = ctx.fs.clone();
        let fut = async move { run_edit(args, cancel, fs).await };
        let s: Pin<Box<dyn futures::Stream<Item = ToolEvent> + Send>> = Box::pin(stream::once(fut));
        s
    }
}

async fn run_edit(
    args: serde_json::Value,
    cancel: tokio_util::sync::CancellationToken,
    fs: Arc<dyn FsBackend>,
) -> ToolEvent {
    let parsed: EditArgs = match serde_json::from_value(args) {
        Ok(v) => v,
        Err(err) => return ToolEvent::Failed(ToolError::InvalidArgs(BoxError::new(err))),
    };

    if parsed.old_string.is_empty() {
        return ToolEvent::Failed(ToolError::InvalidArgs(BoxError::new(arg_err(
            "old_string must not be empty",
        ))));
    }
    if parsed.old_string == parsed.new_string {
        return ToolEvent::Failed(ToolError::InvalidArgs(BoxError::new(arg_err(
            "old_string and new_string must differ",
        ))));
    }

    let path = PathBuf::from(&parsed.path);

    let read_fut = fs.read_text(path.clone(), None, None);
    let old_content = tokio::select! {
        biased;
        () = cancel.cancelled() => return ToolEvent::Failed(ToolError::Canceled),
        r = read_fut => match r {
            Ok(t) => t,
            Err(e) => return ToolEvent::Failed(map_fs_err(e)),
        },
    };

    // Immediately after the read, capture a "read-time fingerprint" as a baseline. The
    // backend either uses mtime+size (`LocalFsBackend`) or falls back to re-reading the
    // full content hash (`AcpFsBackend`). Either scheme can be compared with the
    // "pre-write" fingerprint to detect concurrent external modifications during the
    // read→write window.
    //
    // On failure (rare, e.g. `NotPermitted`), drop this guard and proceed normally —
    // conflict detection is best-effort and should not block the main flow.
    let baseline_fp = fs.fingerprint(path.clone()).await.ok();

    // Match in the file's own line-ending space. The model almost always sends LF in
    // `old_string`; if the file is CRLF, an exact match would fail on every line. Convert
    // `old`/`new` to the file's dominant ending before matching so the replacer chain
    // operates on comparable bytes. (The backend re-normalizes on write regardless, but
    // that does not help the *match* step.)
    let crlf = is_crlf(&old_content);
    let old_norm = to_ending(&parsed.old_string, crlf);
    let new_norm = to_ending(&parsed.new_string, crlf);

    let (new_content, matches_replaced, matched_strategy) =
        match replace(&old_content, &old_norm, &new_norm, parsed.replace_all) {
            Ok(v) => v,
            Err(EditOutcome::NotFound) => {
                return ToolEvent::Failed(ToolError::InvalidArgs(BoxError::new(arg_err(
                    "old_string not found. It must match the file content; copy the exact \
                     text including whitespace and indentation, or re-read the file.",
                ))));
            }
            Err(EditOutcome::Ambiguous(n)) => {
                return ToolEvent::Failed(ToolError::InvalidArgs(BoxError::new(arg_err(
                    &format!(
                        "old_string matched {n} times; add unique surrounding context or set \
                         replace_all"
                    ),
                ))));
            }
        };

    let bytes_before = old_content.len() as u64;
    let bytes_after = new_content.len() as u64;

    // Conflict detection: re-fingerprint before writing and compare against the baseline.
    // If they differ, return [`FsError::Conflict`] — the LLM should re-read and re-edit
    // rather than overwrite.
    if let Some(baseline) = baseline_fp {
        match fs.fingerprint(path.clone()).await {
            Ok(current) if current != baseline => {
                return ToolEvent::Failed(map_fs_err(FsError::Conflict(path)));
            }
            // Don't block if the current fingerprint is unavailable — conflict
            // detection is best-effort.
            _ => {}
        }
    }

    let write_fut = fs.write_text(path.clone(), new_content.clone());
    tokio::select! {
        biased;
        () = cancel.cancelled() => return ToolEvent::Failed(ToolError::Canceled),
        r = write_fut => {
            if let Err(e) = r {
                return ToolEvent::Failed(map_fs_err(e));
            }
        }
    }

    let raw_output = serde_json::to_value(EditFileOutput {
        matches_replaced,
        bytes_before,
        bytes_after,
        matched_strategy,
    })
    .unwrap_or(serde_json::Value::Null);

    // When a non-exact matcher hit, the spliced replacement does not preserve the
    // region's original whitespace/indentation — make that observable to the model.
    let note = if matched_strategy == "exact" {
        format!("Replaced {matches_replaced} occurrence(s)")
    } else {
        format!(
            "Replaced {matches_replaced} occurrence(s) (matched via `{matched_strategy}` \
             fallback — exact text did not match; verify indentation/whitespace is correct)"
        )
    };

    let diff = Diff::new(path, new_content).old_text(Some(old_content));
    let mut fields = ToolCallUpdateFields::default();
    fields.content = Some(vec![
        ToolCallContent::Diff(diff),
        ToolCallContent::Content(Content::new(ContentBlock::Text(TextContent::new(note)))),
    ]);
    fields.raw_output = Some(raw_output);
    ToolEvent::Completed(fields)
}

/// Returns whether `text`'s dominant line ending is CRLF (more `\r\n` than lone `\n`).
fn is_crlf(text: &str) -> bool {
    let crlf = text.matches("\r\n").count();
    let lone_lf = text.matches('\n').count().saturating_sub(crlf);
    crlf > lone_lf
}

/// Converts `s` (assumed LF or mixed) to the target line ending. When `crlf` is true,
/// every `\n` becomes `\r\n` (after first collapsing any existing `\r\n` to avoid
/// doubling); when false, returns `s` unchanged (matching is done in LF space).
fn to_ending(s: &str, crlf: bool) -> String {
    if !crlf {
        return s.to_string();
    }
    s.replace("\r\n", "\n").replace('\n', "\r\n")
}

fn map_fs_err(e: FsError) -> ToolError {
    ToolError::Execution(BoxError::new(e))
}

fn arg_err(msg: &str) -> io::Error {
    io::Error::new(io::ErrorKind::InvalidInput, msg.to_string())
}