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 )) }) }}
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
))
})
}
}