ZIm/crates/breadcrumbs/src/breadcrumbs.rs

use editor::{Anchor, Editor};
use gpui::{
    elements::*, AppContext, Entity, ModelHandle, RenderContext, Subscription, View, ViewContext,
    ViewHandle,
};
use language::{Buffer, OutlineItem};
use project::Project;
use search::ProjectSearchView;
use theme::SyntaxTheme;
use workspace::{ItemHandle, Settings, ToolbarItemLocation, ToolbarItemView};

pub enum Event {
    UpdateLocation,
}

pub struct Breadcrumbs {
    project: ModelHandle<Project>,
    editor: Option<ViewHandle<Editor>>,
    project_search: Option<ViewHandle<ProjectSearchView>>,
    subscriptions: Vec<Subscription>,
}

impl Breadcrumbs {
    pub fn new(project: ModelHandle<Project>) -> Self {
        Self {
            project,
            editor: Default::default(),
            subscriptions: Default::default(),
            project_search: Default::default(),
        }
    }

    fn active_symbols(
        &self,
        theme: &SyntaxTheme,
        cx: &AppContext,
    ) -> Option<(ModelHandle<Buffer>, Vec<OutlineItem<Anchor>>)> {
        let editor = self.editor.as_ref()?.read(cx);
        let cursor = editor.newest_anchor_selection().head();
        let multibuffer = &editor.buffer().read(cx);
        let (buffer_id, symbols) = multibuffer
            .read(cx)
            .symbols_containing(cursor, Some(theme))?;
        let buffer = multibuffer.buffer(buffer_id)?;
        Some((buffer, symbols))
    }
}

impl Entity for Breadcrumbs {
    type Event = Event;
}

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

    fn render(&mut self, cx: &mut RenderContext<Self>) -> ElementBox {
        let theme = cx.global::<Settings>().theme.clone();
        let (buffer, symbols) =
            if let Some((buffer, symbols)) = self.active_symbols(&theme.editor.syntax, cx) {
                (buffer, symbols)
            } else {
                return Empty::new().boxed();
            };
        let buffer = buffer.read(cx);
        let filename = if let Some(file) = buffer.file() {
            if file.path().file_name().is_none()
                || self.project.read(cx).visible_worktrees(cx).count() > 1
            {
                file.full_path(cx).to_string_lossy().to_string()
            } else {
                file.path().to_string_lossy().to_string()
            }
        } else {
            "untitled".to_string()
        };

        Flex::row()
            .with_child(Label::new(filename, theme.breadcrumbs.text.clone()).boxed())
            .with_children(symbols.into_iter().flat_map(|symbol| {
                [
                    Label::new(" 〉 ".to_string(), theme.breadcrumbs.text.clone()).boxed(),
                    Text::new(symbol.text, theme.breadcrumbs.text.clone())
                        .with_highlights(symbol.highlight_ranges)
                        .boxed(),
                ]
            }))
            .contained()
            .with_style(theme.breadcrumbs.container)
            .aligned()
            .left()
            .boxed()
    }
}

impl ToolbarItemView for Breadcrumbs {
    fn set_active_pane_item(
        &mut self,
        active_pane_item: Option<&dyn ItemHandle>,
        cx: &mut ViewContext<Self>,
    ) -> ToolbarItemLocation {
        cx.notify();
        self.subscriptions.clear();
        self.editor = None;
        self.project_search = None;
        if let Some(item) = active_pane_item {
            if let Some(editor) = item.act_as::<Editor>(cx) {
                self.subscriptions
                    .push(cx.subscribe(&editor, |_, _, event, cx| match event {
                        editor::Event::BufferEdited
                        | editor::Event::TitleChanged
                        | editor::Event::Saved
                        | editor::Event::Reparsed => cx.notify(),
                        editor::Event::SelectionsChanged { local } if *local => cx.notify(),
                        _ => {}
                    }));
                self.editor = Some(editor);
                if let Some(project_search) = item.downcast::<ProjectSearchView>() {
                    self.subscriptions
                        .push(cx.subscribe(&project_search, |_, _, _, cx| {
                            cx.emit(Event::UpdateLocation);
                        }));
                    self.project_search = Some(project_search.clone());

                    if project_search.read(cx).has_matches() {
                        ToolbarItemLocation::Secondary
                    } else {
                        ToolbarItemLocation::Hidden
                    }
                } else {
                    ToolbarItemLocation::PrimaryLeft { flex: None }
                }
            } else {
                ToolbarItemLocation::Hidden
            }
        } else {
            ToolbarItemLocation::Hidden
        }
    }

    fn location_for_event(
        &self,
        _: &Event,
        current_location: ToolbarItemLocation,
        cx: &AppContext,
    ) -> ToolbarItemLocation {
        if let Some(project_search) = self.project_search.as_ref() {
            if project_search.read(cx).has_matches() {
                ToolbarItemLocation::Secondary
            } else {
                ToolbarItemLocation::Hidden
            }
        } else {
            current_location
        }
    }
}