agent-harness-rs 0.2.17

use std::path::{Path, PathBuf};
use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::{Arc, Mutex, OnceLock};
use std::time::{Duration, SystemTime, UNIX_EPOCH};

use async_trait::async_trait;
use serde_json::{json, Value};
use tokio::io::AsyncBufReadExt;
use tokio::process::Command;
use tokio_util::sync::CancellationToken;

use crate::shell_risk::{classify_shell_command, ShellRiskLevel};
use crate::tools::{
    builtin_tool_specs, ToolFailure, ToolFailureKind, ToolInvocation,
    ToolOutcome, ToolRuntime, ToolRuntimeError, ToolSpec, MAX_FS_GLOB_RESULTS, MAX_OUTPUT_BYTES,
};
use crate::tools::approval::{is_read_only, ApprovalGate};

/// Opaque event emitter: receives structured JSON events produced during tool
/// execution (e.g. `bash_stdout_line`). Use an `Arc<dyn Fn(Value) + ...>`
/// or provide a no-op with `Arc::new(|_| {})`.
pub type EmitFn = Arc<dyn Fn(Value) + Send + Sync + 'static>;

/// Configuration for `LocalToolRuntime`.
pub struct LocalToolConfig {
    /// Absolute path to the working directory for all tool operations.
    /// Relative paths inside tool arguments are resolved against this.
    /// Defaults to the process's `std::env::current_dir()` when `None`.
    pub cwd: Option<PathBuf>,
    /// Controls which tool calls are allowed and which tool specs are
    /// advertised to the model. Typically `YoloApproval`, `PlanApproval`,
    /// or a custom gate (e.g. `TauriApproval` in a desktop app).
    pub approval: Arc<dyn ApprovalGate>,
    /// Receives structured events produced during tool execution.
    /// Most callers forward these to a UI channel. Pass `Arc::new(|_| {})`
    /// to discard them.
    pub emit: EmitFn,
}

/// Tool runtime that executes bash, read, write, edit, glob and grep
/// directly on the local filesystem.
///
/// All tool output is capped at `MAX_TOOL_CHARS` characters so large
/// files and noisy commands don't blow out the model's context window.
/// Glob/grep skip dependency and build directories automatically.
#[derive(Clone)]
pub struct LocalToolRuntime {
    cwd: PathBuf,
    approval: Arc<dyn ApprovalGate>,
    emit: EmitFn,
}

impl LocalToolRuntime {
    pub fn new(config: LocalToolConfig) -> Self {
        let cwd = config.cwd
            .filter(|p| !p.as_os_str().is_empty())
            .or_else(|| std::env::current_dir().ok())
            .unwrap_or_else(|| PathBuf::from("/"));
        Self { cwd, approval: config.approval, emit: config.emit }
    }

    fn resolve(&self, path: &str) -> PathBuf {
        let p = Path::new(path);
        if p.is_absolute() { p.to_path_buf() } else { self.cwd.join(p) }
    }

    /// Gate that decides whether a tool invocation may proceed.
    ///
    /// * Hard-blocked bash commands are rejected in every mode.
    /// * Read-only bash / read / glob / grep pass without hitting the gate.
    /// * Everything else goes to `approval.approve()`.
    async fn gate(
        &self,
        inv: &ToolInvocation,
        cancel: Option<&CancellationToken>,
    ) -> Result<(), String> {
        if inv.name == "bash" {
            let cmd = inv.input.get("command").and_then(Value::as_str).unwrap_or("");
            let decision = classify_shell_command(cmd);
            match decision.level {
                ShellRiskLevel::Blocked => {
                    return Err(format!("命令在禁止清单上，已拒绝：{}", decision.reason));
                }
                ShellRiskLevel::SafeRead => return Ok(()),
                ShellRiskLevel::BoundedWrite
                    if self.approval.advertise_mutating_tools() =>
                {
                    return Ok(());
                }
                _ => {}
            }
        } else if is_read_only(&inv.name) {
            return Ok(());
        }

        // Non-read-only tool → delegate to the approval gate.
        let approved = if let Some(tok) = cancel {
            tokio::select! {
                biased;
                _ = tok.cancelled() => return Err("已取消".into()),
                result = self.approval.approve(inv) => result,
            }
        } else {
            self.approval.approve(inv).await
        };

        if approved { Ok(()) } else { Err("操作被拒绝".into()) }
    }
}

#[async_trait]
impl ToolRuntime for LocalToolRuntime {
    fn specs(&self) -> Vec<ToolSpec> {
        let all = builtin_tool_specs();
        if self.approval.advertise_mutating_tools() {
            all
        } else {
            all.into_iter().filter(|s| is_read_only(&s.name)).collect()
        }
    }

    async fn invoke(&self, inv: ToolInvocation) -> Result<ToolOutcome, ToolRuntimeError> {
        self.invoke_cancellable(inv, None).await
    }

    async fn invoke_cancellable(
        &self,
        inv: ToolInvocation,
        cancel: Option<&CancellationToken>,
    ) -> Result<ToolOutcome, ToolRuntimeError> {
        if let Err(reason) = self.gate(&inv, cancel).await {
            return Ok(ToolOutcome {
                output: Err(ToolFailure::new(ToolFailureKind::Denied, reason)),
                attachments: vec![],
            });
        }
        match inv.name.as_str() {
            "bash"  => bash_invoke(inv, cancel, &self.cwd, self.emit.clone()).await,
            "read"  => read_invoke(inv, self).await,
            "write" => write_invoke(inv, self).await,
            "edit"  => edit_invoke(inv, self).await,
            "glob"  => glob_invoke(inv, self).await,
            "grep"  => grep_invoke(inv, self).await,
            "web_fetch" => crate::tools::web_fetch::invoke(inv).await,
            other   => Err(ToolRuntimeError::UnknownTool(other.into())),
        }
    }
}

// ── bash ──────────────────────────────────────────────────────────────────────

fn epoch_ms() -> u64 {
    SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .unwrap_or_default()
        .as_millis() as u64
}

const CANCEL_TERMINATION_GRACE: Duration = Duration::from_millis(50);
const CHILD_WAIT_AFTER_KILL: Duration = Duration::from_millis(500);
const OUTPUT_DRAIN_TIMEOUT: Duration = Duration::from_millis(2_000);

enum BashCompletion {
    Exited(std::io::Result<std::process::ExitStatus>),
    SoftTimeout { total_ms: u64, silent_ms: u64 },
    HardTimeout,
    Cancelled,
}

async fn drain_output_tasks(
    mut stdout_task: tokio::task::JoinHandle<()>,
    mut stderr_task: tokio::task::JoinHandle<()>,
) {
    let drain = async {
        let _ = (&mut stdout_task).await;
        let _ = (&mut stderr_task).await;
    };

    if tokio::time::timeout(OUTPUT_DRAIN_TIMEOUT, drain).await.is_err() {
        stdout_task.abort();
        stderr_task.abort();
        let _ = stdout_task.await;
        let _ = stderr_task.await;
    }
}

#[cfg(unix)]
fn signal_process_group(process_group_id: u32, signal: libc::c_int) {
    let pgid = process_group_id as libc::pid_t;
    if pgid <= 0 {
        return;
    }

    let rc = unsafe { libc::killpg(pgid, signal) };
    if rc == 0 {
        return;
    }

    let err = std::io::Error::last_os_error();
    if err.raw_os_error() != Some(libc::ESRCH) {
        tracing::debug!(
            process_group_id,
            signal,
            error = %err,
            "failed to signal bash process group"
        );
    }
}

#[cfg(unix)]
async fn kill_process_group(process_group_id: u32, child: &mut tokio::process::Child) {
    signal_process_group(process_group_id, libc::SIGKILL);
    let _ = child.start_kill();
    let _ = tokio::time::timeout(CHILD_WAIT_AFTER_KILL, child.wait()).await;
}

#[cfg(not(unix))]
async fn kill_process_group(_: u32, child: &mut tokio::process::Child) {
    let _ = child.start_kill();
    let _ = tokio::time::timeout(CHILD_WAIT_AFTER_KILL, child.wait()).await;
}

#[cfg(unix)]
async fn terminate_process_group(process_group_id: u32, child: &mut tokio::process::Child) {
    signal_process_group(process_group_id, libc::SIGTERM);

    let child_exited =
        tokio::time::timeout(CANCEL_TERMINATION_GRACE, child.wait()).await.is_ok();

    // Match Codex's cancellation behavior: give the shell a short SIGTERM
    // cleanup window, then SIGKILL the group so TERM-ignoring descendants do
    // not survive after the parent exits.
    signal_process_group(process_group_id, libc::SIGKILL);
    if !child_exited {
        let _ = child.start_kill();
        let _ = tokio::time::timeout(CHILD_WAIT_AFTER_KILL, child.wait()).await;
    }
}

#[cfg(not(unix))]
async fn terminate_process_group(_: u32, child: &mut tokio::process::Child) {
    kill_process_group(0, child).await;
}

async fn bash_invoke(
    inv: ToolInvocation,
    cancel: Option<&CancellationToken>,
    cwd: &Path,
    emit: EmitFn,
) -> Result<ToolOutcome, ToolRuntimeError> {
    let command = req_str(&inv, "command")?;
    let id = &*inv.id;

    // Dual-layer timeout:
    //   soft_timeout_ms — no-output detector: if the process produces
    //     nothing for this long it is killed and the model is told to retry.
    //     Streaming output resets the clock, so a long build that prints
    //     progress will never hit this.
    //   timeout_ms — absolute hard ceiling, regardless of output activity.
    let soft_ms: u64 = inv.input.get("soft_timeout_ms")
        .and_then(|v| v.as_u64())
        .unwrap_or(10_000);
    let hard_ms: u64 = inv.input.get("timeout_ms")
        .and_then(|v| v.as_u64())
        .unwrap_or(120_000)
        .min(3_600_000);

    let last_out = Arc::new(AtomicU64::new(epoch_ms()));
    let stdout_buf = Arc::new(Mutex::new(String::new()));
    let stderr_buf = Arc::new(Mutex::new(String::new()));

    let shell = if Path::new("/bin/bash").exists() { "/bin/bash" } else { "/bin/sh" };
    let mut cmd = Command::new(shell);
    cmd.args(["-lc", command])
        .current_dir(cwd)
        .kill_on_drop(true)
        .stdout(std::process::Stdio::piped())
        .stderr(std::process::Stdio::piped());

    #[cfg(unix)]
    cmd.process_group(0);

    let mut child = cmd
        .spawn()
        .map_err(|e| ToolRuntimeError::Runtime(format!("spawn failed: {e}")))?;
    let child_pid = child.id();

    let raw_stdout = child.stdout.take().expect("stdout piped");
    let raw_stderr = child.stderr.take().expect("stderr piped");

    let act1 = last_out.clone();
    let emit_out = emit.clone();
    let stdout_acc = stdout_buf.clone();
    let stdout_task = tokio::spawn(async move {
        let mut lines = tokio::io::BufReader::new(raw_stdout).lines();
        while let Ok(Some(line)) = lines.next_line().await {
            emit_out(json!({ "type": "bash_stdout_line", "line": line, "stream": "stdout" }));
            act1.store(epoch_ms(), Ordering::Relaxed);
            if let Ok(mut acc) = stdout_acc.lock() {
                acc.push_str(&line);
                acc.push('\n');
            }
        }
    });

    let act2 = last_out.clone();
    let emit_err = emit.clone();
    let stderr_acc = stderr_buf.clone();
    let stderr_task = tokio::spawn(async move {
        let mut lines = tokio::io::BufReader::new(raw_stderr).lines();
        while let Ok(Some(line)) = lines.next_line().await {
            emit_err(json!({ "type": "bash_stdout_line", "line": line, "stream": "stderr" }));
            act2.store(epoch_ms(), Ordering::Relaxed);
            if let Ok(mut acc) = stderr_acc.lock() {
                acc.push_str(&line);
                acc.push('\n');
            }
        }
    });

    let watcher_ts = last_out.clone();
    let soft_watcher = async move {
        let start = epoch_ms();
        loop {
            tokio::time::sleep(Duration::from_millis(500)).await;
            let now = epoch_ms();
            if now.saturating_sub(start) >= soft_ms
                && now.saturating_sub(watcher_ts.load(Ordering::Relaxed)) >= soft_ms
            {
                return (now.saturating_sub(start), now.saturating_sub(watcher_ts.load(Ordering::Relaxed)));
            }
        }
    };

    let hard_timer = tokio::time::sleep(Duration::from_millis(hard_ms));
    let cancellation = async {
        if let Some(tok) = cancel {
            tok.cancelled().await;
        } else {
            std::future::pending::<()>().await;
        }
    };

    let timeout_outcome = |kind: &str, message: String| ToolOutcome {
        output: Ok(json!({
            "command": command,
            "shell": shell,
            "stdout": bound_output(stdout_buf.lock().map(|s| s.clone()).unwrap_or_default(), id, "stdout"),
            "stderr": bound_output(stderr_buf.lock().map(|s| s.clone()).unwrap_or_default(), id, "stderr"),
            "exit_code": null,
            "success": false,
            "timed_out": true,
            "timeout_kind": kind,
            "message": message,
        })),
        attachments: vec![],
    };
    let soft_err = |tot: u64, sil: u64| timeout_outcome(
        "soft",
        format!(
            "Command produced no output for {sil}ms (total {tot}ms). \
Retry with larger `soft_timeout_ms` or `timeout_ms` if it is expected to take longer."
        ),
    );
    let hard_err = || timeout_outcome(
        "hard",
        format!(
            "Command did not finish in {hard_ms}ms. Retry with a larger `timeout_ms` if it is expected to take longer."
        ),
    );

    let completion = tokio::select! {
        biased;
        _ = cancellation => BashCompletion::Cancelled,
        status = child.wait() => BashCompletion::Exited(status),
        (tot, sil) = soft_watcher => BashCompletion::SoftTimeout { total_ms: tot, silent_ms: sil },
        _ = hard_timer => BashCompletion::HardTimeout,
    };

    let status_result = match completion {
        BashCompletion::Exited(status) => status,
        BashCompletion::SoftTimeout { total_ms, silent_ms } => {
            if let Some(pid) = child_pid {
                kill_process_group(pid, &mut child).await;
            }
            drain_output_tasks(stdout_task, stderr_task).await;
            return Ok(soft_err(total_ms, silent_ms));
        }
        BashCompletion::HardTimeout => {
            if let Some(pid) = child_pid {
                kill_process_group(pid, &mut child).await;
            }
            drain_output_tasks(stdout_task, stderr_task).await;
            return Ok(hard_err());
        }
        BashCompletion::Cancelled => {
            if let Some(pid) = child_pid {
                terminate_process_group(pid, &mut child).await;
            }
            drain_output_tasks(stdout_task, stderr_task).await;
            return Ok(ToolOutcome {
                output: Err(ToolFailure::new(ToolFailureKind::Runtime, "cancelled")),
                attachments: vec![],
            });
        }
    };

    drain_output_tasks(stdout_task, stderr_task).await;
    let stdout = stdout_buf.lock().map(|s| s.clone()).unwrap_or_default();
    let stderr = stderr_buf.lock().map(|s| s.clone()).unwrap_or_default();

    let exit_code = status_result.map(|s| s.code().unwrap_or(-1)).unwrap_or(-1);

    Ok(ToolOutcome {
        output: Ok(json!({
            "command": command,
            "shell": shell,
            "stdout": bound_output(stdout, id, "stdout"),
            "stderr": bound_output(stderr, id, "stderr"),
            "exit_code": exit_code,
            "success": exit_code == 0,
        })),
        attachments: vec![],
    })
}

// ── read ──────────────────────────────────────────────────────────────────────

async fn read_invoke(inv: ToolInvocation, rt: &LocalToolRuntime) -> Result<ToolOutcome, ToolRuntimeError> {
    let path = req_str(&inv, "path")?;
    let resolved = rt.resolve(path);
    match tokio::fs::read_to_string(&resolved).await {
        Ok(content) => {
            let total = content.lines().count();
            let offset = inv.input.get("offset").and_then(Value::as_u64).unwrap_or(0) as usize;
            let limit = inv.input.get("limit").and_then(Value::as_u64)
                .map(|v| v.clamp(1, 2_000) as usize);
            let selected: Vec<&str> = match limit {
                Some(n) => content.lines().skip(offset).take(n).collect(),
                None    => content.lines().skip(offset).collect(),
            };
            let end = offset + selected.len();
            let text = if selected.is_empty() {
                String::new()
            } else {
                let mut t = selected.join("\n");
                if content.ends_with('\n') && end == total { t.push('\n'); }
                t
            };
            Ok(ToolOutcome {
                output: Ok(json!({
                    "path": resolved.to_string_lossy(),
                    "content": truncate(text),
                    "offset": offset,
                    "limit": limit,
                    "start_line": if selected.is_empty() { Value::Null } else { json!(offset + 1) },
                    "end_line": if selected.is_empty() { Value::Null } else { json!(end) },
                    "total_lines": total,
                    "truncated": limit.map(|n| offset + n < total).unwrap_or(false),
                })),
                attachments: vec![],
            })
        }
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::NotFound,
                format!("file not found: {}", resolved.display()))),
            attachments: vec![],
        }),
        Err(e) => Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::Runtime, format!("read error: {e}"))),
            attachments: vec![],
        }),
    }
}

// ── write ─────────────────────────────────────────────────────────────────────

async fn write_invoke(inv: ToolInvocation, rt: &LocalToolRuntime) -> Result<ToolOutcome, ToolRuntimeError> {
    let path = req_str(&inv, "path")?;
    let content = req_str(&inv, "content")?;
    let resolved = rt.resolve(path);
    if let Some(parent) = resolved.parent() {
        if !parent.as_os_str().is_empty() {
            tokio::fs::create_dir_all(parent).await
                .map_err(|e| ToolRuntimeError::Runtime(format!("mkdir: {e}")))?;
        }
    }
    tokio::fs::write(&resolved, content).await
        .map_err(|e| ToolRuntimeError::Runtime(format!("write error: {e}")))?;
    Ok(ToolOutcome {
        output: Ok(json!({ "path": resolved.to_string_lossy(), "written": true })),
        attachments: vec![],
    })
}

// ── edit ──────────────────────────────────────────────────────────────────────

async fn edit_invoke(inv: ToolInvocation, rt: &LocalToolRuntime) -> Result<ToolOutcome, ToolRuntimeError> {
    let path = req_str(&inv, "path")?;
    let old_string = req_str(&inv, "old_string")?;
    let new_string = inv.input.get("new_string").and_then(Value::as_str).unwrap_or("");
    let replace_all = inv.input.get("replace_all").and_then(Value::as_bool).unwrap_or(false);
    let resolved = rt.resolve(path);

    let content = match tokio::fs::read_to_string(&resolved).await {
        Ok(c) => c,
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::NotFound,
                format!("file not found: {}", resolved.display()))),
            attachments: vec![],
        }),
        Err(e) => return Err(ToolRuntimeError::Runtime(e.to_string())),
    };

    let occurrences = content.matches(old_string).count();
    if occurrences == 0 {
        return Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::InvalidInput,
                "Could not find old_string in the file. It must match exactly, including whitespace and indentation. Read the file again before retrying.".to_string())),
            attachments: vec![],
        });
    }
    if !replace_all && occurrences > 1 {
        return Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::InvalidInput,
                format!("Found {occurrences} exact matches for old_string. Provide more surrounding context or set replace_all=true."))),
            attachments: vec![],
        });
    }

    let replaced = if replace_all { occurrences } else { 1 };
    let new_content = if replace_all {
        content.replace(old_string, new_string)
    } else {
        content.replacen(old_string, new_string, 1)
    };
    tokio::fs::write(&resolved, new_content).await
        .map_err(|e| ToolRuntimeError::Runtime(e.to_string()))?;
    Ok(ToolOutcome {
        output: Ok(json!({
            "path": resolved.to_string_lossy(),
            "replaced": replaced,
            "old_lines": old_string.lines().count(),
            "new_lines": new_string.lines().count(),
        })),
        attachments: vec![],
    })
}

// ── external-tool detection ─────────────────────────────────────────────────

/// Cap on grep matches assembled from a streamed `rg --json` run. Hitting it
/// stops the read and kills rg (rather than draining the whole match set),
/// mirroring how the glob tools cap a walk at [`MAX_FS_GLOB_RESULTS`].
const MAX_GREP_MATCHES: usize = 5_000;

/// Probe PATH for an external search tool, trying each candidate name in order
/// and returning the first that answers `--version`. The probe spawns a child
/// once; the result is cached for the process lifetime, so every later call is
/// a lock-free read. `None` means the built-in implementation is used instead
/// — this library never downloads a binary.
fn probe_tool(names: &[&str]) -> Option<String> {
    for name in names {
        let ok = std::process::Command::new(name)
            .arg("--version")
            .stdin(std::process::Stdio::null())
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::null())
            .status()
            .map(|s| s.success())
            .unwrap_or(false);
        if ok {
            return Some((*name).to_string());
        }
    }
    None
}

/// Path/command name for ripgrep, or `None` when it is not installed. Used by
/// the grep tool; glob uses the `ignore` crate directly and needs no binary.
fn ripgrep_bin() -> Option<&'static str> {
    static RG: OnceLock<Option<String>> = OnceLock::new();
    RG.get_or_init(|| probe_tool(&["rg"])).as_deref()
}

// ── glob ──────────────────────────────────────────────────────────────────────

async fn glob_invoke(inv: ToolInvocation, rt: &LocalToolRuntime) -> Result<ToolOutcome, ToolRuntimeError> {
    let pattern = req_str(&inv, "pattern")?.to_string();
    let base = match inv.input.get("path").and_then(Value::as_str).filter(|s| !s.is_empty()) {
        Some(p) => rt.resolve(p),
        None    => rt.cwd.clone(),
    };

    // Glob is served entirely by the `ignore` crate — the same walker +
    // gitignore engine ripgrep/fd are built on. Using it directly (rather than
    // shelling out to fd) gives one deterministic, gitignore-aware behaviour
    // with no subprocess, no PATH dependency, and no fd-vs-fallback divergence.
    let (matches, truncated) = glob_with_ignore(&pattern, &base);

    Ok(ToolOutcome {
        output: Ok(json!({
            "pattern": pattern,
            "count": matches.len(),
            "matches": matches,
            "truncated": truncated,
        })),
        attachments: vec![],
    })
}

/// Glob implementation, built on the `ignore` crate — the same walker +
/// gitignore engine that ripgrep and fd use internally. A recursive,
/// .gitignore/.ignore-aware walk in which a slash-less pattern like `*.rs`
/// matches by file name at any depth, `**/*.rs` includes top-level files, and a
/// `/`-bearing pattern (`src/*.rs`) is anchored to the search root (matching
/// `rg -g` / git semantics). Returns files only, relative to `base`, sorted,
/// capped at [`MAX_FS_GLOB_RESULTS`]. Honours gitignore in and out of a repo;
/// hidden paths are searched only for dot patterns.
fn glob_with_ignore(pattern: &str, base: &Path) -> (Vec<String>, bool) {
    use ignore::overrides::OverrideBuilder;
    use ignore::WalkBuilder;

    // The pattern becomes a whitelist override. Override globs use the same
    // gitignore matching semantics fd applies, which is what keeps the two
    // backends in lockstep.
    let mut ob = OverrideBuilder::new(base);
    if ob.add(pattern).is_err() {
        return (Vec::new(), false);
    }
    let overrides = match ob.build() {
        Ok(o) => o,
        Err(_) => return (Vec::new(), false),
    };

    let mut wb = WalkBuilder::new(base);
    wb.overrides(overrides)
        // fd --no-require-git: honour .gitignore/.ignore in and out of a repo.
        .require_git(false)
        // hidden(true) skips hidden entries; only search them for dot patterns.
        .hidden(!pattern.starts_with('.'));

    let mut matches: Vec<String> = Vec::new();
    let mut truncated = false;
    for result in wb.build() {
        let Ok(entry) = result else { continue };
        // Files only (the tool lists files; fd uses --type f).
        if entry.file_type().is_none_or(|t| t.is_dir()) {
            continue;
        }
        let Ok(rel) = entry.path().strip_prefix(base) else { continue };
        if matches.len() >= MAX_FS_GLOB_RESULTS {
            truncated = true;
            break;
        }
        matches.push(rel.to_string_lossy().replace('\\', "/"));
    }
    matches.sort();
    (matches, truncated)
}

// ── grep ──────────────────────────────────────────────────────────────────────

async fn grep_invoke(inv: ToolInvocation, rt: &LocalToolRuntime) -> Result<ToolOutcome, ToolRuntimeError> {
    let pattern = req_str(&inv, "pattern")?.to_string();
    let ci = inv.input.get("case_insensitive").and_then(Value::as_bool).unwrap_or(false);
    let search = match inv.input.get("path").and_then(Value::as_str).filter(|s| !s.is_empty()) {
        Some(p) => rt.resolve(p),
        None    => rt.cwd.clone(),
    };

    // Prefer ripgrep: gitignore-aware, faster, and streamed via `--json` so we
    // can stop (and kill the child) the moment we have enough matches. Falls
    // back to system grep when rg is absent or fails to spawn.
    if let Some(rg) = ripgrep_bin() {
        if let Some(outcome) = grep_with_rg(rg, &pattern, ci, &search, &rt.cwd, &inv.id).await? {
            return Ok(outcome);
        }
    }
    grep_with_system(&pattern, ci, &search, &rt.cwd, &inv.id).await
}

/// Stream `rg --json` and rebuild the same `path:line:text` lines the
/// system-grep fallback emits. Returns `Ok(None)` when rg could not be spawned
/// (caller falls back); `Ok(Some(outcome))` otherwise.
async fn grep_with_rg(
    rg: &str,
    pattern: &str,
    ci: bool,
    search: &Path,
    cwd: &Path,
    id: &str,
) -> Result<Option<ToolOutcome>, ToolRuntimeError> {
    let mut cmd = Command::new(rg);
    cmd.arg("--json");
    if ci {
        cmd.arg("-i");
    }
    cmd.arg("-e").arg(pattern).arg("--").arg(search);
    cmd.current_dir(cwd)
        .kill_on_drop(true)
        .stdin(std::process::Stdio::null())
        .stdout(std::process::Stdio::piped())
        .stderr(std::process::Stdio::null());

    let mut child = match cmd.spawn() {
        Ok(c) => c,
        Err(_) => return Ok(None), // rg vanished between probe and spawn → fall back
    };
    let raw_stdout = child.stdout.take().expect("stdout piped");

    let mut out = String::new();
    let mut count = 0usize;
    let mut truncated = false;
    let mut lines = tokio::io::BufReader::new(raw_stdout).lines();
    let read = async {
        while let Some(line) = lines.next_line().await.ok().flatten() {
            let Ok(ev) = serde_json::from_str::<Value>(&line) else { continue };
            if ev.get("type").and_then(Value::as_str) != Some("match") {
                continue;
            }
            let data = &ev["data"];
            // Non-UTF8 matches arrive as {"bytes": …} instead of {"text": …}; skip them.
            let (Some(path), Some(line_no), Some(text)) = (
                data["path"]["text"].as_str(),
                data["line_number"].as_u64(),
                data["lines"]["text"].as_str(),
            ) else {
                continue;
            };
            out.push_str(path);
            out.push(':');
            out.push_str(&line_no.to_string());
            out.push(':');
            out.push_str(text.trim_end_matches('\n'));
            out.push('\n');
            count += 1;
            if count >= MAX_GREP_MATCHES || out.len() > MAX_OUTPUT_BYTES {
                truncated = true;
                break;
            }
        }
    };
    // Same 30s ceiling as the system-grep path.
    let timed_out = tokio::time::timeout(Duration::from_secs(30), read).await.is_err();
    // Stop rg regardless of why we finished (limit hit / EOF / timeout).
    let _ = child.start_kill();
    let _ = tokio::time::timeout(CHILD_WAIT_AFTER_KILL, child.wait()).await;

    if timed_out {
        return Ok(Some(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::Timeout, "grep timed out after 30s")),
            attachments: vec![],
        }));
    }
    Ok(Some(ToolOutcome {
        output: Ok(json!({
            "pattern": pattern,
            "matches": bound_output(out, id, "matches"),
            "truncated": truncated,
        })),
        attachments: vec![],
    }))
}

/// Fallback grep: shell out to system `grep -rnE` with a fixed set of pruned
/// directories. Used when ripgrep is not installed.
async fn grep_with_system(
    pattern: &str,
    ci: bool,
    search: &Path,
    cwd: &Path,
    id: &str,
) -> Result<ToolOutcome, ToolRuntimeError> {
    let mut cmd = Command::new("grep");
    cmd.args(["-rn", "-E"]);
    if ci { cmd.arg("-i"); }
    cmd.args([
        "--exclude-dir=node_modules",
        "--exclude-dir=target",
        "--exclude-dir=.git",
        "--exclude-dir=dist",
        "--exclude-dir=build",
        "--exclude-dir=__pycache__",
        "--exclude-dir=.venv",
        "--exclude-dir=vendor",
        "--exclude-dir=.next",
    ]);
    cmd.arg("-e").arg(pattern).arg("--").arg(search);
    cmd.current_dir(cwd);

    match tokio::time::timeout(Duration::from_secs(30), cmd.output()).await {
        Err(_) => Ok(ToolOutcome {
            output: Err(ToolFailure::new(ToolFailureKind::Timeout, "grep timed out after 30s")),
            attachments: vec![],
        }),
        Ok(Err(e)) => Err(ToolRuntimeError::Runtime(format!("grep spawn failed: {e}"))),
        Ok(Ok(out)) => {
            let code = out.status.code().unwrap_or(-1);
            if code >= 2 {
                let stderr = String::from_utf8_lossy(&out.stderr).into_owned();
                return Ok(ToolOutcome {
                    output: Err(ToolFailure::new(ToolFailureKind::Runtime,
                        truncate(format!("grep error: {stderr}")))),
                    attachments: vec![],
                });
            }
            let stdout = String::from_utf8_lossy(&out.stdout).into_owned();
            Ok(ToolOutcome {
                output: Ok(json!({
                    "pattern": pattern,
                    "matches": bound_output(stdout, id, "matches"),
                    "truncated": false,
                })),
                attachments: vec![],
            })
        }
    }
}

// ── output cap ────────────────────────────────────────────────────────────────

/// Write `content` to `/tmp/harness_out_{id}_{suffix}.txt` and return an
/// error-aware head+tail preview with the path. Used for bash stdout/stderr
/// and grep matches. Spills only when over budget (see `bounded_preview`).
fn bound_output(content: String, id: &str, suffix: &str) -> String {
    let path = format!("/tmp/harness_out_{id}_{suffix}.txt");
    match crate::tools::bounded_preview(&content, &path) {
        None => content,
        Some(preview) => {
            let _ = std::fs::write(&path, &content);
            preview
        }
    }
}

/// Simple safety truncation used only by the read tool as a backstop for
/// pages that exceed the output budget after pagination.
fn truncate(s: String) -> String {
    crate::tools::clip_head(s)
}

// ── helpers ───────────────────────────────────────────────────────────────────

fn req_str<'a>(inv: &'a ToolInvocation, key: &str) -> Result<&'a str, ToolRuntimeError> {
    inv.input
        .get(key)
        .and_then(Value::as_str)
        .filter(|s| !s.is_empty())
        .ok_or_else(|| ToolRuntimeError::InvalidInput {
            tool: inv.name.clone(),
            message: format!("missing field `{key}`"),
        })
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::tools::approval::YoloApproval;

    fn runtime() -> LocalToolRuntime {
        LocalToolRuntime::new(LocalToolConfig {
            cwd: Some(std::env::temp_dir()),
            approval: Arc::new(YoloApproval),
            emit: Arc::new(|_| {}),
        })
    }

    #[cfg(unix)]
    async fn processes_matching_marker(marker: &str) -> Vec<(libc::pid_t, libc::pid_t, String)> {
        let output = Command::new("ps")
            .args(["-axo", "pid=,pgid=,command="])
            .output()
            .await
            .expect("ps should run");
        String::from_utf8_lossy(&output.stdout)
            .lines()
            .filter(|line| line.contains(marker))
            .filter_map(|line| {
                let mut parts = line.split_whitespace();
                let pid = parts.next()?.parse().ok()?;
                let pgid = parts.next()?.parse().ok()?;
                let command = parts.collect::<Vec<_>>().join(" ");
                Some((pid, pgid, command))
            })
            .collect()
    }

    #[cfg(unix)]
    async fn cleanup_marker_processes(marker: &str) {
        for (_, pgid, _) in processes_matching_marker(marker).await {
            unsafe {
                libc::kill(-pgid, libc::SIGKILL);
            }
        }
    }

    #[cfg(unix)]
    fn shell_quote(value: &str) -> String {
        format!("'{}'", value.replace('\'', "'\\''"))
    }

    #[cfg(unix)]
    async fn wait_for_marker_process(marker: &str) -> bool {
        let deadline = tokio::time::Instant::now() + Duration::from_secs(2);
        while tokio::time::Instant::now() < deadline {
            if !processes_matching_marker(marker).await.is_empty() {
                return true;
            }
            tokio::time::sleep(Duration::from_millis(50)).await;
        }
        false
    }

    #[tokio::test]
    async fn bash_non_zero_exit_returns_structured_result() {
        let out = runtime()
            .invoke(ToolInvocation {
                id: "tc_nonzero".into(),
                name: "bash".into(),
                input: json!({"command": "printf nope >&2; exit 7"}),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        assert_eq!(out["exit_code"], 7);
        assert_eq!(out["success"], false);
        assert_eq!(out["stderr"], "nope\n");
    }

    #[tokio::test]
    async fn bash_timeout_returns_structured_result() {
        let out = runtime()
            .invoke(ToolInvocation {
                id: "tc_timeout".into(),
                name: "bash".into(),
                input: json!({
                    "command": "sleep 2",
                    "soft_timeout_ms": 1000,
                    "timeout_ms": 5000
                }),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        assert_eq!(out["success"], false);
        assert_eq!(out["timed_out"], true);
        assert_eq!(out["timeout_kind"], "soft");
    }

    #[cfg(unix)]
    #[tokio::test]
    async fn bash_timeout_kills_process_group_children() {
        let marker = format!("harness-timeout-pgid-{}", epoch_ms());
        cleanup_marker_processes(&marker).await;

        let command = format!("sh -c 'while :; do sleep 5; done' {marker} & wait");
        let out = runtime()
            .invoke(ToolInvocation {
                id: "tc_timeout_pgid".into(),
                name: "bash".into(),
                input: json!({
                    "command": command,
                    "soft_timeout_ms": 200,
                    "timeout_ms": 5000
                }),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        assert_eq!(out["success"], false);
        assert_eq!(out["timed_out"], true);

        tokio::time::sleep(Duration::from_millis(500)).await;
        let leftovers = processes_matching_marker(&marker).await;
        cleanup_marker_processes(&marker).await;
        assert!(
            leftovers.is_empty(),
            "timeout left child processes running: {leftovers:?}"
        );
    }

    #[cfg(unix)]
    #[tokio::test]
    async fn bash_cancel_sends_sigterm_then_kills_process_group_children() {
        let marker = format!("harness-cancel-pgid-{}", epoch_ms());
        let cleanup_path = std::env::temp_dir().join(format!("{marker}.cleanup"));
        let _ = tokio::fs::remove_file(&cleanup_path).await;
        cleanup_marker_processes(&marker).await;

        let command = format!(
            r#"trap "printf cleanup > {}; exit 0" TERM; sh -c 'trap "" TERM; while :; do sleep 5; done' {} & wait"#,
            shell_quote(&cleanup_path.to_string_lossy()),
            shell_quote(&marker),
        );
        let cancel = CancellationToken::new();
        let cancel_for_task = cancel.clone();
        let handle = tokio::spawn(async move {
            runtime()
                .invoke_cancellable(
                    ToolInvocation {
                        id: "tc_cancel_pgid".into(),
                        name: "bash".into(),
                        input: json!({
                            "command": command,
                            "soft_timeout_ms": 5000,
                            "timeout_ms": 10000
                        }),
                        raw_emitted_args: None,
                    },
                    Some(&cancel_for_task),
                )
                .await
        });

        assert!(
            wait_for_marker_process(&marker).await,
            "test command did not start"
        );

        cancel.cancel();
        let outcome = tokio::time::timeout(Duration::from_secs(5), handle)
            .await
            .expect("cancelled bash invocation should return promptly")
            .expect("join should succeed")
            .expect("runtime should return a ToolOutcome");
        let failure = outcome.output.expect_err("cancel should be surfaced as failure");
        assert_eq!(failure.kind, ToolFailureKind::Runtime);
        assert_eq!(failure.message, "cancelled");

        tokio::time::sleep(Duration::from_millis(500)).await;
        assert!(
            cleanup_path.exists(),
            "parent shell did not get SIGTERM cleanup window"
        );
        let leftovers = processes_matching_marker(&marker).await;
        cleanup_marker_processes(&marker).await;
        let _ = tokio::fs::remove_file(&cleanup_path).await;
        assert!(
            leftovers.is_empty(),
            "cancel left child processes running: {leftovers:?}"
        );
    }

    #[tokio::test]
    async fn bash_tool_supports_bash_syntax_when_bash_exists() {
        if !Path::new("/bin/bash").exists() {
            return;
        }
        let out = runtime()
            .invoke(ToolInvocation {
                id: "tc_bash_syntax".into(),
                name: "bash".into(),
                input: json!({"command": "diff <(printf a) <(printf a)"}),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        assert_eq!(out["success"], true);
        assert_eq!(out["exit_code"], 0);
        assert_eq!(out["shell"], "/bin/bash");
    }

    // ── grep / glob: rg/fd with fallback ────────────────────────────────────

    /// A runtime rooted at `dir` (grep/glob resolve relative paths and default
    /// their search root to it).
    fn runtime_in(dir: &Path) -> LocalToolRuntime {
        LocalToolRuntime::new(LocalToolConfig {
            cwd: Some(dir.to_path_buf()),
            approval: Arc::new(YoloApproval),
            emit: Arc::new(|_| {}),
        })
    }

    /// Fresh, empty temp dir unique to this test name + process.
    fn scratch(tag: &str) -> PathBuf {
        let dir = std::env::temp_dir().join(format!("harness_{tag}_{}", std::process::id()));
        let _ = std::fs::remove_dir_all(&dir);
        std::fs::create_dir_all(&dir).unwrap();
        dir
    }

    #[tokio::test]
    async fn grep_returns_path_line_text_and_truncated_flag() {
        let dir = scratch("grep_shape");
        std::fs::create_dir_all(dir.join("src")).unwrap();
        std::fs::write(dir.join("src/a.rs"), "let needle = 1;\nother\n").unwrap();
        std::fs::write(dir.join("b.txt"), "no match here\n").unwrap();

        let out = runtime_in(&dir)
            .invoke(ToolInvocation {
                id: "tc_grep_shape".into(),
                name: "grep".into(),
                input: json!({ "pattern": "needle" }),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        let _ = std::fs::remove_dir_all(&dir);

        // Shape is identical whether ripgrep or system grep served the request.
        assert!(out.get("truncated").is_some(), "missing truncated flag: {out}");
        let matches = out["matches"].as_str().unwrap();
        // One `path:line:text` line pointing at src/a.rs line 1.
        assert!(matches.contains("a.rs:1:"), "unexpected matches: {matches:?}");
        assert!(matches.contains("needle"), "unexpected matches: {matches:?}");
        assert!(!matches.contains("no match here"), "unexpected matches: {matches:?}");
    }

    #[tokio::test]
    async fn grep_with_rg_honours_ignore_file() {
        // `.ignore` is honoured by ripgrep unconditionally; system grep is not,
        // so this asserts the rg path specifically.
        let Some(rg) = ripgrep_bin() else { return };
        let dir = scratch("grep_ignore");
        std::fs::create_dir_all(dir.join("skip")).unwrap();
        std::fs::write(dir.join("keep.txt"), "needle\n").unwrap();
        std::fs::write(dir.join("skip/hit.txt"), "needle\n").unwrap();
        std::fs::write(dir.join(".ignore"), "skip/\n").unwrap();

        let outcome = grep_with_rg(rg, "needle", false, &dir, &dir, "tc_grep_ignore")
            .await
            .unwrap()
            .expect("rg should have produced an outcome");
        let out = outcome.output.unwrap();
        let _ = std::fs::remove_dir_all(&dir);

        let matches = out["matches"].as_str().unwrap();
        assert!(matches.contains("keep.txt"), "{matches:?}");
        assert!(!matches.contains("hit.txt"), "ignored dir leaked: {matches:?}");
    }

    #[tokio::test]
    async fn glob_returns_relative_paths_and_truncated_flag() {
        let dir = scratch("glob_shape");
        std::fs::create_dir_all(dir.join("src/sub")).unwrap();
        std::fs::write(dir.join("top.rs"), "").unwrap();
        std::fs::write(dir.join("src/lib.rs"), "").unwrap();
        std::fs::write(dir.join("src/sub/deep.rs"), "").unwrap();

        let out = runtime_in(&dir)
            .invoke(ToolInvocation {
                id: "tc_glob_shape".into(),
                name: "glob".into(),
                input: json!({ "pattern": "**/*.rs" }),
                raw_emitted_args: None,
            })
            .await
            .unwrap()
            .output
            .unwrap();
        let _ = std::fs::remove_dir_all(&dir);

        assert!(out.get("truncated").is_some(), "missing truncated flag: {out}");
        let matches: Vec<String> = out["matches"]
            .as_array()
            .unwrap()
            .iter()
            .map(|v| v.as_str().unwrap().to_string())
            .collect();
        // `**/*.rs` is recursive AND includes top-level files under both backends.
        assert!(matches.iter().any(|m| m == "top.rs"), "{matches:?}");
        assert!(matches.iter().any(|m| m == "src/lib.rs"), "{matches:?}");
        assert!(matches.iter().any(|m| m == "src/sub/deep.rs"), "{matches:?}");
        // Paths are relative to the search root, never absolute.
        assert!(matches.iter().all(|m| !m.starts_with('/')), "{matches:?}");
    }

    #[test]
    fn glob_matches_bare_pattern_at_any_depth() {
        // A slash-less pattern recurses (matches by file name at any depth) —
        // unlike the old hand-written walk, which only matched the top level.
        let dir = scratch("glob_recurse");
        std::fs::create_dir_all(dir.join("src/sub")).unwrap();
        std::fs::write(dir.join("top.rs"), "").unwrap();
        std::fs::write(dir.join("src/a.rs"), "").unwrap();
        std::fs::write(dir.join("src/sub/deep.rs"), "").unwrap();

        let (matches, truncated) = glob_with_ignore("*.rs", &dir);
        let _ = std::fs::remove_dir_all(&dir);

        assert!(!truncated);
        assert_eq!(
            matches,
            vec!["src/a.rs", "src/sub/deep.rs", "top.rs"],
            "bare `*.rs` should match .rs files at every depth"
        );
    }

    #[test]
    fn glob_slash_pattern_is_anchored_to_root() {
        // A `/`-bearing pattern anchors to the search root (rg -g / git
        // semantics): `src/*.rs` matches src/a.rs but not sub/src or top-level.
        let dir = scratch("glob_anchor");
        std::fs::create_dir_all(dir.join("src/sub")).unwrap();
        std::fs::write(dir.join("top.rs"), "").unwrap();
        std::fs::write(dir.join("src/a.rs"), "").unwrap();
        std::fs::write(dir.join("src/sub/deep.rs"), "").unwrap();

        let (matches, _) = glob_with_ignore("src/*.rs", &dir);
        let _ = std::fs::remove_dir_all(&dir);

        assert_eq!(matches, vec!["src/a.rs"], "`src/*.rs` should anchor to the root");
    }

    #[test]
    fn glob_honours_ignore_file() {
        let dir = scratch("glob_ignore");
        std::fs::create_dir_all(dir.join("skip")).unwrap();
        std::fs::write(dir.join("keep.rs"), "").unwrap();
        std::fs::write(dir.join("skip/hidden.rs"), "").unwrap();
        std::fs::write(dir.join(".ignore"), "skip/\n").unwrap();

        let (matches, _) = glob_with_ignore("**/*.rs", &dir);
        let _ = std::fs::remove_dir_all(&dir);

        assert!(matches.iter().any(|m| m == "keep.rs"), "{matches:?}");
        assert!(!matches.iter().any(|m| m.contains("hidden.rs")), "ignored dir leaked: {matches:?}");
    }

    #[tokio::test]
    async fn grep_system_fallback_produces_expected_shape() {
        // Directly exercise the fallback so it is covered even where rg exists.
        let dir = scratch("grep_fallback");
        std::fs::write(dir.join("f.rs"), "let needle = 1;\n").unwrap();

        let out = grep_with_system("needle", false, &dir, &dir, "tc_grep_fallback")
            .await
            .unwrap()
            .output
            .unwrap();
        let _ = std::fs::remove_dir_all(&dir);

        assert_eq!(out["truncated"], false);
        assert!(out["matches"].as_str().unwrap().contains("needle"), "{out}");
    }
}