1use anyhow::{anyhow, Result};
 2use gpui::{App, AppContext, Entity, SharedString, Task};
 3use project::Project;
 4use schemars::JsonSchema;
 5use serde::Deserialize;
 6use std::{path::PathBuf, sync::Arc};
 7use util::paths::PathMatcher;
 8use worktree::Snapshot as WorktreeSnapshot;
 9
10use crate::{
11    templates::{GlobTemplate, Template, Templates},
12    thread::AgentTool,
13};
14
15// Description is dynamic, see `fn description` below
16#[derive(Deserialize, JsonSchema)]
17struct GlobInput {
18    /// A POSIX glob pattern
19    glob: SharedString,
20}
21
22struct GlobTool {
23    project: Entity<Project>,
24    templates: Arc<Templates>,
25}
26
27impl AgentTool for GlobTool {
28    type Input = GlobInput;
29
30    fn name(&self) -> SharedString {
31        "glob".into()
32    }
33
34    fn description(&self, cx: &mut App) -> SharedString {
35        let project_roots = self
36            .project
37            .read(cx)
38            .worktrees(cx)
39            .map(|worktree| worktree.read(cx).root_name().into())
40            .collect::<Vec<String>>()
41            .join("\n");
42
43        GlobTemplate { project_roots }
44            .render(&self.templates)
45            .expect("template failed to render")
46            .into()
47    }
48
49    fn run(self: Arc<Self>, input: Self::Input, cx: &mut App) -> Task<Result<String>> {
50        let path_matcher = match PathMatcher::new([&input.glob]) {
51            Ok(matcher) => matcher,
52            Err(error) => return Task::ready(Err(anyhow!(error))),
53        };
54
55        let snapshots: Vec<WorktreeSnapshot> = self
56            .project
57            .read(cx)
58            .worktrees(cx)
59            .map(|worktree| worktree.read(cx).snapshot())
60            .collect();
61
62        cx.background_spawn(async move {
63            let paths = snapshots.iter().flat_map(|snapshot| {
64                let root_name = PathBuf::from(snapshot.root_name());
65                snapshot
66                    .entries(false, 0)
67                    .map(move |entry| root_name.join(&entry.path))
68                    .filter(|path| path_matcher.is_match(&path))
69            });
70            let output = paths
71                .map(|path| format!("{}\n", path.display()))
72                .collect::<String>();
73            Ok(output)
74        })
75    }
76}