898 lines · 28.8 KB · Rust
RawDownload
#[cfg(test)]mod tab_switcher_tests; use collections::{HashMap, HashSet};use editor::items::{    entry_diagnostic_aware_icon_decoration_and_color, entry_git_aware_label_color,};use fuzzy_nucleo::StringMatchCandidate;use gpui::{    Action, AnyElement, App, Context, DismissEvent, Entity, EntityId, EventEmitter, FocusHandle,    Focusable, Modifiers, ModifiersChangedEvent, MouseButton, MouseUpEvent, ParentElement, Point,    Render, Styled, Task, TaskExt, WeakEntity, Window, actions, rems,};use picker::{Picker, PickerDelegate};use project::Project;use schemars::JsonSchema;use serde::Deserialize;use settings::Settings;use std::{cmp::Reverse, sync::Arc};use ui::{    DecoratedIcon, IconDecoration, IconDecorationKind, ListItem, ListItemSpacing, Tooltip,    prelude::*,};use util::ResultExt;use workspace::{    Event as WorkspaceEvent, ModalView, Pane, SaveIntent, Workspace,    item::{ItemHandle, ItemSettings, ShowDiagnostics, TabContentParams},    pane::{render_item_indicator, tab_details},}; const PANEL_WIDTH_REMS: f32 = 28.; /// Toggles the tab switcher interface.#[derive(PartialEq, Clone, Deserialize, JsonSchema, Default, Action)]#[action(namespace = tab_switcher)]#[serde(deny_unknown_fields)]pub struct Toggle {    #[serde(default)]    pub select_last: bool,}actions!(    tab_switcher,    [        /// Closes the selected item in the tab switcher.        CloseSelectedItem,        /// Toggles between showing all tabs or just the current pane's tabs.        ToggleAll,        /// Toggles the tab switcher showing all tabs across all panes, deduplicated by path.        /// Opens selected items in the active pane.        OpenInActivePane,    ]); pub struct TabSwitcher {    picker: Entity<Picker<TabSwitcherDelegate>>,    init_modifiers: Option<Modifiers>,} impl ModalView for TabSwitcher {} pub fn init(cx: &mut App) {    cx.observe_new(TabSwitcher::register).detach();} impl TabSwitcher {    fn register(        workspace: &mut Workspace,        _window: Option<&mut Window>,        _: &mut Context<Workspace>,    ) {        workspace.register_action(|workspace, action: &Toggle, window, cx| {            let Some(tab_switcher) = workspace.active_modal::<Self>(cx) else {                Self::open(workspace, action.select_last, false, false, window, cx);                return;            };             tab_switcher.update(cx, |tab_switcher, cx| {                tab_switcher                    .picker                    .update(cx, |picker, cx| picker.cycle_selection(window, cx))            });        });        workspace.register_action(|workspace, _action: &ToggleAll, window, cx| {            let Some(tab_switcher) = workspace.active_modal::<Self>(cx) else {                Self::open(workspace, false, true, false, window, cx);                return;            };             tab_switcher.update(cx, |tab_switcher, cx| {                tab_switcher                    .picker                    .update(cx, |picker, cx| picker.cycle_selection(window, cx))            });        });        workspace.register_action(|workspace, _action: &OpenInActivePane, window, cx| {            let Some(tab_switcher) = workspace.active_modal::<Self>(cx) else {                Self::open(workspace, false, true, true, window, cx);                return;            };             tab_switcher.update(cx, |tab_switcher, cx| {                tab_switcher                    .picker                    .update(cx, |picker, cx| picker.cycle_selection(window, cx))            });        });    }     fn open(        workspace: &mut Workspace,        select_last: bool,        is_global: bool,        open_in_active_pane: bool,        window: &mut Window,        cx: &mut Context<Workspace>,    ) {        let mut weak_pane = workspace.active_pane().downgrade();        for dock in [            workspace.left_dock(),            workspace.bottom_dock(),            workspace.right_dock(),        ] {            dock.update(cx, |this, cx| {                let Some(panel) = this                    .active_panel()                    .filter(|panel| panel.panel_focus_handle(cx).contains_focused(window, cx))                else {                    return;                };                if let Some(pane) = panel.pane(cx) {                    weak_pane = pane.downgrade();                }            })        }         let weak_workspace = workspace.weak_handle();         let project = workspace.project().clone();        let original_items: Vec<_> = workspace            .panes()            .iter()            .map(|p| (p.clone(), p.read(cx).active_item_index()))            .collect();        workspace.toggle_modal(window, cx, |window, cx| {            let delegate = TabSwitcherDelegate::new(                project,                select_last,                cx.entity().downgrade(),                weak_pane,                weak_workspace,                is_global,                open_in_active_pane,                window,                cx,                original_items,            );            TabSwitcher::new(delegate, window, is_global, cx)        });    }     fn new(        delegate: TabSwitcherDelegate,        window: &mut Window,        is_global: bool,        cx: &mut Context<Self>,    ) -> Self {        let init_modifiers = if is_global {            None        } else {            window.modifiers().modified().then_some(window.modifiers())        };        Self {            picker: cx.new(|cx| {                if is_global {                    Picker::list(delegate, window, cx)                } else {                    Picker::nonsearchable_list(delegate, window, cx)                }                .initial_width(rems(PANEL_WIDTH_REMS))            }),            init_modifiers,        }    }     fn handle_modifiers_changed(        &mut self,        event: &ModifiersChangedEvent,        window: &mut Window,        cx: &mut Context<Self>,    ) {        let Some(init_modifiers) = self.init_modifiers else {            return;        };        if !event.modified() || !init_modifiers.is_subset_of(event) {            self.init_modifiers = None;            if self.picker.read(cx).delegate.matches.is_empty() {                cx.emit(DismissEvent)            } else {                window.dispatch_action(menu::Confirm.boxed_clone(), cx);            }        }    }     fn handle_close_selected_item(        &mut self,        _: &CloseSelectedItem,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.picker.update(cx, |picker, cx| {            picker                .delegate                .close_item_at(picker.delegate.selected_index(), window, cx)        });    }} impl EventEmitter<DismissEvent> for TabSwitcher {} impl Focusable for TabSwitcher {    fn focus_handle(&self, cx: &App) -> FocusHandle {        self.picker.focus_handle(cx)    }} impl Render for TabSwitcher {    fn render(&mut self, _: &mut Window, cx: &mut Context<Self>) -> impl IntoElement {        v_flex()            .key_context("TabSwitcher")            .w(rems(PANEL_WIDTH_REMS))            .on_modifiers_changed(cx.listener(Self::handle_modifiers_changed))            .on_action(cx.listener(Self::handle_close_selected_item))            .child(self.picker.clone())    }} #[derive(Clone)]struct TabMatch {    pane: WeakEntity<Pane>,    item_index: usize,    item: Box<dyn ItemHandle>,    detail: usize,    preview: bool,} pub struct TabSwitcherDelegate {    select_last: bool,    tab_switcher: WeakEntity<TabSwitcher>,    selected_index: usize,    pane: WeakEntity<Pane>,    workspace: WeakEntity<Workspace>,    project: Entity<Project>,    matches: Vec<TabMatch>,    original_items: Vec<(Entity<Pane>, usize)>,    is_all_panes: bool,    open_in_active_pane: bool,    restored_items: bool,} impl TabMatch {    fn icon(        &self,        project: &Entity<Project>,        selected: bool,        window: &Window,        cx: &App,    ) -> Option<DecoratedIcon> {        let icon = self.item.tab_icon(window, cx)?;        let item_settings = ItemSettings::get_global(cx);        let show_diagnostics = item_settings.show_diagnostics;        let git_status_color = item_settings            .git_status            .then(|| {                let path = self.item.project_path(cx)?;                let project = project.read(cx);                let entry = project.entry_for_path(&path, cx)?;                let git_status = project                    .git_store()                    .read(cx)                    .display_status_for_project_path(&path, cx)                    .map(|status| status.summary())                    .unwrap_or_default();                Some(entry_git_aware_label_color(                    git_status,                    entry.is_ignored,                    selected,                ))            })            .flatten();        let colored_icon = icon.color(git_status_color.unwrap_or_default());         let most_severe_diagnostic_level = if show_diagnostics == ShowDiagnostics::Off {            None        } else {            let buffer_store = project.read(cx).buffer_store().read(cx);            let buffer = self                .item                .project_path(cx)                .and_then(|path| buffer_store.get_by_path(&path))                .map(|buffer| buffer.read(cx));            buffer.and_then(|buffer| {                buffer                    .buffer_diagnostics(None)                    .iter()                    .map(|diagnostic_entry| diagnostic_entry.diagnostic.severity)                    .min()            })        };         let decorations =            entry_diagnostic_aware_icon_decoration_and_color(most_severe_diagnostic_level)                .filter(|(d, _)| {                    *d != IconDecorationKind::Triangle                        || show_diagnostics != ShowDiagnostics::Errors                })                .map(|(icon, color)| {                    let knockout_item_color = if selected {                        cx.theme().colors().element_selected                    } else {                        cx.theme().colors().element_background                    };                    IconDecoration::new(icon, knockout_item_color, cx)                        .color(color.color(cx))                        .position(Point {                            x: px(-2.),                            y: px(-2.),                        })                });        Some(DecoratedIcon::new(colored_icon, decorations))    }} impl TabSwitcherDelegate {    #[allow(clippy::complexity)]    fn new(        project: Entity<Project>,        select_last: bool,        tab_switcher: WeakEntity<TabSwitcher>,        pane: WeakEntity<Pane>,        workspace: WeakEntity<Workspace>,        is_all_panes: bool,        open_in_active_pane: bool,        window: &mut Window,        cx: &mut Context<TabSwitcher>,        original_items: Vec<(Entity<Pane>, usize)>,    ) -> Self {        Self::subscribe_to_updates(&workspace, window, cx);        Self {            select_last,            tab_switcher,            selected_index: 0,            pane,            workspace,            project,            matches: Vec::new(),            is_all_panes,            open_in_active_pane,            original_items,            restored_items: false,        }    }     fn subscribe_to_updates(        workspace: &WeakEntity<Workspace>,        window: &mut Window,        cx: &mut Context<TabSwitcher>,    ) {        let Some(workspace) = workspace.upgrade() else {            return;        };        cx.subscribe_in(&workspace, window, |tab_switcher, _, event, window, cx| {            match event {                WorkspaceEvent::ItemAdded { .. } | WorkspaceEvent::PaneRemoved => {                    tab_switcher.picker.update(cx, |picker, cx| {                        let query = picker.query(cx);                        picker.delegate.update_matches(query, window, cx);                        cx.notify();                    })                }                WorkspaceEvent::ItemRemoved { .. } => {                    tab_switcher.picker.update(cx, |picker, cx| {                        let query = picker.query(cx);                        picker.delegate.update_matches(query, window, cx);                         // When the Tab Switcher is being used and an item is                        // removed, there's a chance that the new selected index                        // will not match the actual tab that is now being displayed                        // by the pane, as such, the selected index needs to be                        // updated to match the pane's state.                        picker.delegate.sync_selected_index(cx);                        cx.notify();                    })                }                _ => {}            };        })        .detach();    }     fn update_all_pane_matches(        &mut self,        query: String,        window: &mut Window,        cx: &mut Context<Picker<Self>>,    ) {        let Some(workspace) = self.workspace.upgrade() else {            return;        };        let mut all_items = Vec::new();        let mut item_index = 0;        for pane_handle in workspace.read(cx).panes() {            let pane = pane_handle.read(cx);            let items: Vec<Box<dyn ItemHandle>> =                pane.items().map(|item| item.boxed_clone()).collect();            for ((_detail, item), detail) in items                .iter()                .enumerate()                .zip(tab_details(&items, window, cx))            {                all_items.push(TabMatch {                    pane: pane_handle.downgrade(),                    item_index,                    item: item.clone(),                    detail,                    preview: pane.is_active_preview_item(item.item_id()),                });                item_index += 1;            }        }         let mut matches = if query.is_empty() {            let history = workspace.read(cx).recently_activated_items(cx);            all_items                .sort_by_key(|tab| (Reverse(history.get(&tab.item.item_id())), tab.item_index));            all_items        } else {            let candidates = all_items                .iter()                .enumerate()                .flat_map(|(ix, tab_match)| {                    Some(StringMatchCandidate::new(                        ix,                        &tab_match.item.tab_content_text(0, cx),                    ))                })                .collect::<Vec<_>>();            fuzzy_nucleo::match_strings(                &candidates,                &query,                fuzzy_nucleo::Case::Smart,                fuzzy_nucleo::LengthPenalty::On,                10000,            )            .into_iter()            .map(|m| all_items[m.candidate_id].clone())            .collect()        };         if self.open_in_active_pane {            let mut seen_paths: HashSet<project::ProjectPath> = HashSet::default();            matches.retain(|tab| {                if let Some(path) = tab.item.project_path(cx) {                    seen_paths.insert(path)                } else {                    true                }            });        }         let selected_item_id = self.selected_item_id();        self.matches = matches;        self.selected_index = if query.is_empty() {            self.compute_selected_index(selected_item_id, window, cx)        } else {            0        }    }     fn update_matches(        &mut self,        query: String,        window: &mut Window,        cx: &mut Context<Picker<Self>>,    ) {        if self.is_all_panes {            // needed because we need to borrow the workspace, but that may be borrowed when the picker            // calls update_matches.            let this = cx.entity();            window.defer(cx, move |window, cx| {                this.update(cx, |this, cx| {                    this.delegate.update_all_pane_matches(query, window, cx);                })            });            return;        }        let selected_item_id = self.selected_item_id();        self.matches.clear();        let Some(pane) = self.pane.upgrade() else {            return;        };

Lines 1–500 of 898.