use crate::{ AgentTool, ToolCallEventStream, ToolPermissionDecision, decide_permission_from_settings,};use action_log::ActionLog;use agent_client_protocol::ToolKind;use agent_settings::AgentSettings;use anyhow::{Context as _, Result, anyhow};use futures::{FutureExt as _, SinkExt, StreamExt, channel::mpsc};use gpui::{App, AppContext, Entity, SharedString, Task};use project::{Project, ProjectPath};use schemars::JsonSchema;use serde::{Deserialize, Serialize};use settings::Settings;use std::sync::Arc;use util::markdown::MarkdownInlineCode; /// Deletes the file or directory (and the directory's contents, recursively) at the specified path in the project, and returns confirmation of the deletion.#[derive(Debug, Serialize, Deserialize, JsonSchema)]pub struct DeletePathToolInput { /// The path of the file or directory to delete. /// /// <example> /// If the project has the following files: /// /// - directory1/a/something.txt /// - directory2/a/things.txt /// - directory3/a/other.txt /// /// You can delete the first file by providing a path of "directory1/a/something.txt" /// </example> pub path: String,} pub struct DeletePathTool { project: Entity<Project>, action_log: Entity<ActionLog>,} impl DeletePathTool { pub fn new(project: Entity<Project>, action_log: Entity<ActionLog>) -> Self { Self { project, action_log, } }} impl AgentTool for DeletePathTool { type Input = DeletePathToolInput; type Output = String; fn name() -> &'static str { "delete_path" } fn kind() -> ToolKind { ToolKind::Delete } fn initial_title( &self, input: Result<Self::Input, serde_json::Value>, _cx: &mut App, ) -> SharedString { if let Ok(input) = input { format!("Delete “`{}`”", input.path).into() } else { "Delete path".into() } } fn run( self: Arc<Self>, input: Self::Input, event_stream: ToolCallEventStream, cx: &mut App, ) -> Task<Result<Self::Output>> { let path = input.path; let settings = AgentSettings::get_global(cx); let decision = decide_permission_from_settings(Self::name(), &path, settings); let authorize = match decision { ToolPermissionDecision::Allow => None, ToolPermissionDecision::Deny(reason) => { return Task::ready(Err(anyhow!("{}", reason))); } ToolPermissionDecision::Confirm => { let context = crate::ToolPermissionContext { tool_name: "delete_path".to_string(), input_value: path.clone(), }; Some(event_stream.authorize( format!("Delete {}", MarkdownInlineCode(&path)), context, cx, )) } }; let Some(project_path) = self.project.read(cx).find_project_path(&path, cx) else { return Task::ready(Err(anyhow!( "Couldn't delete {path} because that path isn't in this project." ))); }; let Some(worktree) = self .project .read(cx) .worktree_for_id(project_path.worktree_id, cx) else { return Task::ready(Err(anyhow!( "Couldn't delete {path} because that path isn't in this project." ))); }; let worktree_snapshot = worktree.read(cx).snapshot(); let (mut paths_tx, mut paths_rx) = mpsc::channel(256); cx.background_spawn({ let project_path = project_path.clone(); async move { for entry in worktree_snapshot.traverse_from_path(true, false, false, &project_path.path) { if !entry.path.starts_with(&project_path.path) { break; } paths_tx .send(ProjectPath { worktree_id: project_path.worktree_id, path: entry.path.clone(), }) .await?; } anyhow::Ok(()) } }) .detach(); let project = self.project.clone(); let action_log = self.action_log.clone(); cx.spawn(async move |cx| { if let Some(authorize) = authorize { authorize.await?; } loop { let path_result = futures::select! { path = paths_rx.next().fuse() => path, _ = event_stream.cancelled_by_user().fuse() => { anyhow::bail!("Delete cancelled by user"); } }; let Some(path) = path_result else { break; }; if let Ok(buffer) = project .update(cx, |project, cx| project.open_buffer(path, cx)) .await { action_log.update(cx, |action_log, cx| { action_log.will_delete_buffer(buffer.clone(), cx) }); } } let deletion_task = project .update(cx, |project, cx| { project.delete_file(project_path, false, cx) }) .with_context(|| { format!("Couldn't delete {path} because that path isn't in this project.") })?; futures::select! { result = deletion_task.fuse() => { result.with_context(|| format!("Deleting {path}"))?; } _ = event_stream.cancelled_by_user().fuse() => { anyhow::bail!("Delete cancelled by user"); } } Ok(format!("Deleted {path}")) }) }}
use crate::{
AgentTool, ToolCallEventStream, ToolPermissionDecision, decide_permission_from_settings,
};
use action_log::ActionLog;
use agent_client_protocol::ToolKind;
use agent_settings::AgentSettings;
use anyhow::{Context as _, Result, anyhow};
use futures::{FutureExt as _, SinkExt, StreamExt, channel::mpsc};
use gpui::{App, AppContext, Entity, SharedString, Task};
use project::{Project, ProjectPath};
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
use settings::Settings;
use std::sync::Arc;
use util::markdown::MarkdownInlineCode;
/// Deletes the file or directory (and the directory's contents, recursively) at the specified path in the project, and returns confirmation of the deletion.
#[derive(Debug, Serialize, Deserialize, JsonSchema)]
pub struct DeletePathToolInput {
/// The path of the file or directory to delete.
///
/// <example>
/// If the project has the following files:
///
/// - directory1/a/something.txt
/// - directory2/a/things.txt
/// - directory3/a/other.txt
///
/// You can delete the first file by providing a path of "directory1/a/something.txt"
/// </example>
pub path: String,
}
pub struct DeletePathTool {
project: Entity<Project>,
action_log: Entity<ActionLog>,
}
impl DeletePathTool {
pub fn new(project: Entity<Project>, action_log: Entity<ActionLog>) -> Self {
Self {
project,
action_log,
}
}
}
impl AgentTool for DeletePathTool {
type Input = DeletePathToolInput;
type Output = String;
fn name() -> &'static str {
"delete_path"
}
fn kind() -> ToolKind {
ToolKind::Delete
}
fn initial_title(
&self,
input: Result<Self::Input, serde_json::Value>,
_cx: &mut App,
) -> SharedString {
if let Ok(input) = input {
format!("Delete “`{}`”", input.path).into()
} else {
"Delete path".into()
}
}
fn run(
self: Arc<Self>,
input: Self::Input,
event_stream: ToolCallEventStream,
cx: &mut App,
) -> Task<Result<Self::Output>> {
let path = input.path;
let settings = AgentSettings::get_global(cx);
let decision = decide_permission_from_settings(Self::name(), &path, settings);
let authorize = match decision {
ToolPermissionDecision::Allow => None,
ToolPermissionDecision::Deny(reason) => {
return Task::ready(Err(anyhow!("{}", reason)));
}
ToolPermissionDecision::Confirm => {
let context = crate::ToolPermissionContext {
tool_name: "delete_path".to_string(),
input_value: path.clone(),
};
Some(event_stream.authorize(
format!("Delete {}", MarkdownInlineCode(&path)),
context,
cx,
))
}
};
let Some(project_path) = self.project.read(cx).find_project_path(&path, cx) else {
return Task::ready(Err(anyhow!(
"Couldn't delete {path} because that path isn't in this project."
)));
};
let Some(worktree) = self
.project
.read(cx)
.worktree_for_id(project_path.worktree_id, cx)
else {
return Task::ready(Err(anyhow!(
"Couldn't delete {path} because that path isn't in this project."
)));
};
let worktree_snapshot = worktree.read(cx).snapshot();
let (mut paths_tx, mut paths_rx) = mpsc::channel(256);
cx.background_spawn({
let project_path = project_path.clone();
async move {
for entry in
worktree_snapshot.traverse_from_path(true, false, false, &project_path.path)
{
if !entry.path.starts_with(&project_path.path) {
break;
}
paths_tx
.send(ProjectPath {
worktree_id: project_path.worktree_id,
path: entry.path.clone(),
})
.await?;
}
anyhow::Ok(())
}
})
.detach();
let project = self.project.clone();
let action_log = self.action_log.clone();
cx.spawn(async move |cx| {
if let Some(authorize) = authorize {
authorize.await?;
}
loop {
let path_result = futures::select! {
path = paths_rx.next().fuse() => path,
_ = event_stream.cancelled_by_user().fuse() => {
anyhow::bail!("Delete cancelled by user");
}
};
let Some(path) = path_result else {
break;
};
if let Ok(buffer) = project
.update(cx, |project, cx| project.open_buffer(path, cx))
.await
{
action_log.update(cx, |action_log, cx| {
action_log.will_delete_buffer(buffer.clone(), cx)
});
}
}
let deletion_task = project
.update(cx, |project, cx| {
project.delete_file(project_path, false, cx)
})
.with_context(|| {
format!("Couldn't delete {path} because that path isn't in this project.")
})?;
futures::select! {
result = deletion_task.fuse() => {
result.with_context(|| format!("Deleting {path}"))?;
}
_ = event_stream.cancelled_by_user().fuse() => {
anyhow::bail!("Delete cancelled by user");
}
}
Ok(format!("Deleted {path}"))
})
}
}