172 lines · 6.5 KB · Rust
RawDownload
use super::edit_file_tool::{    SensitiveSettingsKind, is_sensitive_settings_path, sensitive_settings_kind,};use crate::{AgentTool, ToolCallEventStream, ToolPermissionDecision, decide_permission_for_paths};use agent_client_protocol::ToolKind;use agent_settings::AgentSettings;use anyhow::{Context as _, Result, anyhow};use futures::FutureExt as _;use gpui::{App, Entity, SharedString, Task};use project::Project;use schemars::JsonSchema;use serde::{Deserialize, Serialize};use settings::Settings;use std::{path::Path, sync::Arc};use util::markdown::MarkdownInlineCode; /// Moves or rename a file or directory in the project, and returns confirmation that the move succeeded.////// If the source and destination directories are the same, but the filename is different, this performs a rename. Otherwise, it performs a move.////// This tool should be used when it's desirable to move or rename a file or directory without changing its contents at all.#[derive(Debug, Serialize, Deserialize, JsonSchema)]pub struct MovePathToolInput {    /// The source path of the file or directory to move/rename.    ///    /// <example>    /// If the project has the following files:    ///    /// - directory1/a/something.txt    /// - directory2/a/things.txt    /// - directory3/a/other.txt    ///    /// You can move the first file by providing a source_path of "directory1/a/something.txt"    /// </example>    pub source_path: String,     /// The destination path where the file or directory should be moved/renamed to.    /// If the paths are the same except for the filename, then this will be a rename.    ///    /// <example>    /// To move "directory1/a/something.txt" to "directory2/b/renamed.txt",    /// provide a destination_path of "directory2/b/renamed.txt"    /// </example>    pub destination_path: String,} pub struct MovePathTool {    project: Entity<Project>,} impl MovePathTool {    pub fn new(project: Entity<Project>) -> Self {        Self { project }    }} impl AgentTool for MovePathTool {    type Input = MovePathToolInput;    type Output = String;     const NAME: &'static str = "move_path";     fn kind() -> ToolKind {        ToolKind::Move    }     fn initial_title(        &self,        input: Result<Self::Input, serde_json::Value>,        _cx: &mut App,    ) -> SharedString {        if let Ok(input) = input {            let src = MarkdownInlineCode(&input.source_path);            let dest = MarkdownInlineCode(&input.destination_path);            let src_path = Path::new(&input.source_path);            let dest_path = Path::new(&input.destination_path);             match dest_path                .file_name()                .and_then(|os_str| os_str.to_os_string().into_string().ok())            {                Some(filename) if src_path.parent() == dest_path.parent() => {                    let filename = MarkdownInlineCode(&filename);                    format!("Rename {src} to {filename}").into()                }                _ => format!("Move {src} to {dest}").into(),            }        } else {            "Move path".into()        }    }     fn run(        self: Arc<Self>,        input: Self::Input,        event_stream: ToolCallEventStream,        cx: &mut App,    ) -> Task<Result<Self::Output>> {        let settings = AgentSettings::get_global(cx);         let paths = vec![input.source_path.clone(), input.destination_path.clone()];        let decision = decide_permission_for_paths(Self::NAME, &paths, settings);        if let ToolPermissionDecision::Deny(reason) = decision {            return Task::ready(Err(anyhow!("{}", reason)));        }         let needs_confirmation = matches!(decision, ToolPermissionDecision::Confirm)            || (matches!(decision, ToolPermissionDecision::Allow)                && (is_sensitive_settings_path(Path::new(&input.source_path))                    || is_sensitive_settings_path(Path::new(&input.destination_path))));         let authorize = if needs_confirmation {            let src = MarkdownInlineCode(&input.source_path);            let dest = MarkdownInlineCode(&input.destination_path);            let context = crate::ToolPermissionContext {                tool_name: Self::NAME.to_string(),                input_values: vec![input.source_path.clone(), input.destination_path.clone()],            };            let title = format!("Move {src} to {dest}");            let settings_kind = sensitive_settings_kind(Path::new(&input.source_path))                .or_else(|| sensitive_settings_kind(Path::new(&input.destination_path)));            let title = match settings_kind {                Some(SensitiveSettingsKind::Local) => format!("{title} (local settings)"),                Some(SensitiveSettingsKind::Global) => format!("{title} (settings)"),                None => title,            };            Some(event_stream.authorize(title, context, cx))        } else {            None        };         let project = self.project.clone();        cx.spawn(async move |cx| {            if let Some(authorize) = authorize {                authorize.await?;            }             let rename_task = project.update(cx, |project, cx| {                match project                    .find_project_path(&input.source_path, cx)                    .and_then(|project_path| project.entry_for_path(&project_path, cx))                {                    Some(entity) => match project.find_project_path(&input.destination_path, cx) {                        Some(project_path) => Ok(project.rename_entry(entity.id, project_path, cx)),                        None => Err(anyhow!(                            "Destination path {} was outside the project.",                            input.destination_path                        )),                    },                    None => Err(anyhow!(                        "Source path {} was not found in the project.",                        input.source_path                    )),                }            })?;             let result = futures::select! {                result = rename_task.fuse() => result,                _ = event_stream.cancelled_by_user().fuse() => {                    anyhow::bail!("Move cancelled by user");                }            };            result.with_context(|| {                format!("Moving {} to {}", input.source_path, input.destination_path)            })?;            Ok(format!(                "Moved {} to {}",                input.source_path, input.destination_path            ))        })    }}