oak-gpui/zed/src/file_finder.rs

use crate::{
    editor::{buffer_view, BufferView},
    settings::Settings,
    util, watch,
    workspace::{Workspace, WorkspaceView},
    worktree::{match_paths, PathMatch, Worktree},
};
use gpui::{
    color::{ColorF, ColorU},
    elements::*,
    fonts::{Properties, Weight},
    geometry::vector::vec2f,
    keymap::{self, Binding},
    App, AppContext, Axis, Border, Entity, ModelHandle, View, ViewContext, ViewHandle,
    WeakViewHandle,
};
use std::cmp;

pub struct FileFinder {
    handle: WeakViewHandle<Self>,
    settings: watch::Receiver<Settings>,
    workspace: ModelHandle<Workspace>,
    query_buffer: ViewHandle<BufferView>,
    search_count: usize,
    latest_search_id: usize,
    matches: Vec<PathMatch>,
    selected: usize,
    list_state: UniformListState,
}

pub fn init(app: &mut App) {
    app.add_action("file_finder:toggle", FileFinder::toggle);
    app.add_action("file_finder:confirm", FileFinder::confirm);
    app.add_action("file_finder:select", FileFinder::select);
    app.add_action("buffer:move_up", FileFinder::select_prev);
    app.add_action("buffer:move_down", FileFinder::select_next);
    app.add_action("uniform_list:scroll", FileFinder::scroll);

    app.add_bindings(vec![
        Binding::new("cmd-p", "file_finder:toggle", None),
        Binding::new("escape", "file_finder:toggle", Some("FileFinder")),
        Binding::new("enter", "file_finder:confirm", Some("FileFinder")),
    ]);
}

pub enum Event {
    Selected(usize, usize),
    Dismissed,
}

impl Entity for FileFinder {
    type Event = Event;
}

impl View for FileFinder {
    fn ui_name() -> &'static str {
        "FileFinder"
    }

    fn render(&self, _: &AppContext) -> Box<dyn Element> {
        Align::new(
            ConstrainedBox::new(
                Container::new(
                    Flex::new(Axis::Vertical)
                        .with_child(ChildView::new(self.query_buffer.id()).boxed())
                        .with_child(Expanded::new(1.0, self.render_matches()).boxed())
                        .boxed(),
                )
                .with_margin_top(12.0)
                .with_uniform_padding(6.0)
                .with_corner_radius(6.0)
                .with_background_color(ColorU::new(0xff, 0xf2, 0xf2, 0xff))
                // .with_background_color(ColorU::new(0xf2, 0xf2, 0xf2, 0xff))
                .with_shadow(
                    vec2f(0.0, 4.0),
                    12.0,
                    ColorF::new(0.0, 0.0, 0.0, 0.25).to_u8(),
                )
                .boxed(),
            )
            .with_max_width(600.0)
            .with_max_height(400.0)
            .boxed(),
        )
        .top_center()
        .boxed()
    }

    fn on_focus(&mut self, ctx: &mut ViewContext<Self>) {
        ctx.focus(&self.query_buffer);
    }

    fn keymap_context(&self, _: &AppContext) -> keymap::Context {
        let mut ctx = Self::default_keymap_context();
        ctx.set.insert("menu".into());
        ctx
    }
}

impl FileFinder {
    fn render_matches(&self) -> Box<dyn Element> {
        if self.matches.is_empty() {
            let settings = smol::block_on(self.settings.read());
            return Container::new(
                Label::new(
                    "No matches".into(),
                    settings.ui_font_family,
                    settings.ui_font_size,
                )
                .boxed(),
            )
            .with_margin_top(6.0)
            .boxed();
        }

        let handle = self.handle.clone();
        let list = UniformList::new(
            self.list_state.clone(),
            self.matches.len(),
            move |mut range, items, app| {
                let finder = handle.upgrade(app).unwrap();
                let finder = finder.as_ref(app);
                let start = range.start;
                range.end = cmp::min(range.end, finder.matches.len());
                items.extend(finder.matches[range].iter().enumerate().filter_map(
                    move |(i, path_match)| finder.render_match(path_match, start + i, app),
                ));
            },
        );

        Container::new(list.boxed())
            .with_background_color(ColorU::new(0xf7, 0xf7, 0xf7, 0xff))
            .with_border(Border::all(1.0, ColorU::new(0xdb, 0xdb, 0xdc, 0xff)))
            .with_margin_top(6.0)
            .boxed()
    }

    fn render_match(
        &self,
        path_match: &PathMatch,
        index: usize,
        app: &AppContext,
    ) -> Option<Box<dyn Element>> {
        let tree_id = path_match.tree_id;
        let entry_id = path_match.entry_id;

        self.worktree(tree_id, app).map(|tree| {
            let path = tree.entry_path(entry_id).unwrap();
            let file_name = path
                .file_name()
                .unwrap_or_default()
                .to_string_lossy()
                .to_string();

            let mut path = path.to_string_lossy().to_string();
            if path_match.skipped_prefix_len > 0 {
                let mut i = 0;
                path.retain(|_| util::post_inc(&mut i) >= path_match.skipped_prefix_len)
            }

            let path_positions = path_match.positions.clone();
            let file_name_start = path.chars().count() - file_name.chars().count();
            let mut file_name_positions = Vec::new();
            file_name_positions.extend(path_positions.iter().filter_map(|pos| {
                if pos >= &file_name_start {
                    Some(pos - file_name_start)
                } else {
                    None
                }
            }));

            let settings = smol::block_on(self.settings.read());
            let highlight_color = ColorU::new(0x30, 0x4e, 0xe2, 0xff);
            let bold = *Properties::new().weight(Weight::BOLD);

            let mut container = Container::new(
                Flex::row()
                    .with_child(
                        Container::new(
                            LineBox::new(
                                settings.ui_font_family,
                                settings.ui_font_size,
                                Svg::new("icons/file-16.svg".into()).boxed(),
                            )
                            .boxed(),
                        )
                        .with_padding_right(6.0)
                        .boxed(),
                    )
                    .with_child(
                        Expanded::new(
                            1.0,
                            Flex::column()
                                .with_child(
                                    Label::new(
                                        file_name,
                                        settings.ui_font_family,
                                        settings.ui_font_size,
                                    )
                                    .with_highlights(highlight_color, bold, file_name_positions)
                                    .boxed(),
                                )
                                .with_child(
                                    Label::new(
                                        path.into(),
                                        settings.ui_font_family,
                                        settings.ui_font_size,
                                    )
                                    .with_highlights(highlight_color, bold, path_positions)
                                    .boxed(),
                                )
                                .boxed(),
                        )
                        .boxed(),
                    )
                    .boxed(),
            )
            .with_uniform_padding(6.0);

            if index == self.selected || index < self.matches.len() - 1 {
                container =
                    container.with_border(Border::bottom(1.0, ColorU::new(0xdb, 0xdb, 0xdc, 0xff)));
            }

            if index == self.selected {
                container = container.with_background_color(ColorU::new(0xdb, 0xdb, 0xdc, 0xff));
            }

            EventHandler::new(container.boxed())
                .on_mouse_down(move |ctx, _| {
                    ctx.dispatch_action("file_finder:select", (tree_id, entry_id));
                    true
                })
                .boxed()
        })
    }

    fn toggle(workspace_view: &mut WorkspaceView, _: &(), ctx: &mut ViewContext<WorkspaceView>) {
        workspace_view.toggle_modal(ctx, |ctx, workspace_view| {
            let handle = ctx.add_view(|ctx| {
                Self::new(
                    workspace_view.settings.clone(),
                    workspace_view.workspace.clone(),
                    ctx,
                )
            });
            ctx.subscribe_to_view(&handle, Self::on_event);
            handle
        });
    }

    fn on_event(
        workspace_view: &mut WorkspaceView,
        _: ViewHandle<FileFinder>,
        event: &Event,
        ctx: &mut ViewContext<WorkspaceView>,
    ) {
        match event {
            Event::Selected(tree_id, entry_id) => {
                workspace_view.open_entry((*tree_id, *entry_id), ctx);
                workspace_view.dismiss_modal(ctx);
            }
            Event::Dismissed => {
                workspace_view.dismiss_modal(ctx);
            }
        }
    }

    pub fn new(
        settings: watch::Receiver<Settings>,
        workspace: ModelHandle<Workspace>,
        ctx: &mut ViewContext<Self>,
    ) -> Self {
        ctx.observe(&workspace, Self::workspace_updated);

        let query_buffer = ctx.add_view(|ctx| BufferView::single_line(settings.clone(), ctx));
        ctx.subscribe_to_view(&query_buffer, Self::on_query_buffer_event);

        settings.notify_view_on_change(ctx);

        Self {
            handle: ctx.handle(),
            settings,
            workspace,
            query_buffer,
            search_count: 0,
            latest_search_id: 0,
            matches: Vec::new(),
            selected: 0,
            list_state: UniformListState::new(),
        }
    }

    fn workspace_updated(&mut self, _: ModelHandle<Workspace>, ctx: &mut ViewContext<Self>) {
        self.spawn_search(self.query_buffer.as_ref(ctx).text(ctx.app()), ctx);
    }

    fn on_query_buffer_event(
        &mut self,
        _: ViewHandle<BufferView>,
        event: &buffer_view::Event,
        ctx: &mut ViewContext<Self>,
    ) {
        use buffer_view::Event::*;
        match event {
            Edited => {
                let query = self.query_buffer.as_ref(ctx).text(ctx.app());
                if query.is_empty() {
                    self.latest_search_id = util::post_inc(&mut self.search_count);
                    self.matches.clear();
                    ctx.notify();
                } else {
                    self.spawn_search(query, ctx);
                }
            }
            Blurred => ctx.emit(Event::Dismissed),
            Activate => {}
        }
    }

    fn select_prev(&mut self, _: &(), ctx: &mut ViewContext<Self>) {
        if self.selected > 0 {
            self.selected -= 1;
        }
        self.list_state.scroll_to(self.selected);
        ctx.notify();
    }

    fn select_next(&mut self, _: &(), ctx: &mut ViewContext<Self>) {
        if self.selected + 1 < self.matches.len() {
            self.selected += 1;
        }
        self.list_state.scroll_to(self.selected);
        ctx.notify();
    }

    fn scroll(&mut self, _: &f32, ctx: &mut ViewContext<Self>) {
        ctx.notify();
    }

    fn confirm(&mut self, _: &(), ctx: &mut ViewContext<Self>) {
        if let Some(m) = self.matches.get(self.selected) {
            ctx.emit(Event::Selected(m.tree_id, m.entry_id));
        }
    }

    fn select(&mut self, entry: &(usize, usize), ctx: &mut ViewContext<Self>) {
        let (tree_id, entry_id) = *entry;
        log::info!("selected item! {} {}", tree_id, entry_id);
        ctx.emit(Event::Selected(tree_id, entry_id));
    }

    fn spawn_search(&mut self, query: String, ctx: &mut ViewContext<Self>) {
        log::info!("spawn search!");

        let worktrees = self.worktrees(ctx.app());
        let search_id = util::post_inc(&mut self.search_count);
        let task = ctx.background_executor().spawn(async move {
            let matches = match_paths(worktrees.as_slice(), &query, false, false, 100);
            (search_id, matches)
        });

        ctx.spawn(task, Self::update_matches).detach();
    }

    fn update_matches(
        &mut self,
        (search_id, matches): (usize, Vec<PathMatch>),
        ctx: &mut ViewContext<Self>,
    ) {
        if search_id >= self.latest_search_id {
            self.latest_search_id = search_id;
            self.matches = matches;
            self.selected = 0;
            self.list_state.scroll_to(0);
            ctx.notify();
        }
    }

    fn worktree<'a>(&'a self, tree_id: usize, app: &'a AppContext) -> Option<&'a Worktree> {
        self.workspace
            .as_ref(app)
            .worktrees()
            .get(&tree_id)
            .map(|worktree| worktree.as_ref(app))
    }

    fn worktrees(&self, app: &AppContext) -> Vec<Worktree> {
        self.workspace
            .as_ref(app)
            .worktrees()
            .iter()
            .map(|worktree| worktree.as_ref(app).clone())
            .collect()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{
        editor, settings,
        workspace::{Workspace, WorkspaceView},
    };
    use anyhow::Result;
    use gpui::App;
    use smol::fs;
    use tempdir::TempDir;

    #[test]
    fn test_matching_paths() -> Result<()> {
        App::test((), |mut app| async move {
            let tmp_dir = TempDir::new("example")?;
            fs::create_dir(tmp_dir.path().join("a")).await?;
            fs::write(tmp_dir.path().join("a/banana"), "banana").await?;
            fs::write(tmp_dir.path().join("a/bandana"), "bandana").await?;
            super::init(&mut app);
            editor::init(&mut app);

            let settings = settings::channel(&app.fonts()).unwrap().1;
            let workspace = app.add_model(|ctx| Workspace::new(vec![tmp_dir.path().into()], ctx));
            let (window_id, workspace_view) =
                app.add_window(|ctx| WorkspaceView::new(workspace.clone(), settings, ctx));
            app.finish_pending_tasks().await; // Open and populate worktree.
            app.dispatch_action(
                window_id,
                vec![workspace_view.id()],
                "file_finder:toggle".into(),
                (),
            );
            let (finder, query_buffer) = workspace_view.read(&app, |view, ctx| {
                let finder = view
                    .modal()
                    .cloned()
                    .unwrap()
                    .downcast::<FileFinder>()
                    .unwrap();
                let query_buffer = finder.as_ref(ctx).query_buffer.clone();
                (finder, query_buffer)
            });

            let chain = vec![finder.id(), query_buffer.id()];
            app.dispatch_action(window_id, chain.clone(), "buffer:insert", "b".to_string());
            app.dispatch_action(window_id, chain.clone(), "buffer:insert", "n".to_string());
            app.dispatch_action(window_id, chain.clone(), "buffer:insert", "a".to_string());
            app.finish_pending_tasks().await; // Complete path search.

            // let view_state = finder.state(&app);
            // assert!(view_state.matches.len() > 1);
            // app.dispatch_action(
            //     window_id,
            //     vec![workspace_view.id(), finder.id()],
            //     "menu:select_next",
            //     (),
            // );
            // app.dispatch_action(
            //     window_id,
            //     vec![workspace_view.id(), finder.id()],
            //     "file_finder:confirm",
            //     (),
            // );
            // app.finish_pending_tasks().await; // Load Buffer and open BufferView.
            // let active_pane = workspace_view.read(&app, |view, _| view.active_pane().clone());
            // assert_eq!(
            //     active_pane.state(&app),
            //     pane::State {
            //         tabs: vec![pane::TabState {
            //             title: "bandana".into(),
            //             active: true,
            //         }]
            //     }
            // );
            Ok(())
        })
    }
}