zed/crates/diagnostics2/src/toolbar_controls.rs

use crate::ProjectDiagnosticsEditor;
use gpui::{div, Div, EventEmitter, ParentElement, Render, ViewContext, WeakView};
use ui::{Icon, IconButton, Tooltip};
use workspace::{item::ItemHandle, ToolbarItemEvent, ToolbarItemLocation, ToolbarItemView};

pub struct ToolbarControls {
    editor: Option<WeakView<ProjectDiagnosticsEditor>>,
}

impl Render<Self> for ToolbarControls {
    type Element = Div<Self>;

    fn render(&mut self, cx: &mut ViewContext<Self>) -> Self::Element {
        let include_warnings = self
            .editor
            .as_ref()
            .and_then(|editor| editor.upgrade())
            .map(|editor| editor.read(cx).include_warnings)
            .unwrap_or(false);

        let tooltip = if include_warnings {
            "Exclude Warnings"
        } else {
            "Include Warnings"
        };

        div().child(
            IconButton::new("toggle-warnings", Icon::ExclamationTriangle)
                .tooltip(move |_, cx| Tooltip::text(tooltip, cx))
                .on_click(|this: &mut Self, cx| {
                    if let Some(editor) = this.editor.as_ref().and_then(|editor| editor.upgrade()) {
                        editor.update(cx, |editor, cx| {
                            editor.toggle_warnings(&Default::default(), cx);
                        });
                    }
                }),
        )
    }
}

impl EventEmitter<ToolbarItemEvent> for ToolbarControls {}

impl ToolbarItemView for ToolbarControls {
    fn set_active_pane_item(
        &mut self,
        active_pane_item: Option<&dyn ItemHandle>,
        _: &mut ViewContext<Self>,
    ) -> ToolbarItemLocation {
        if let Some(pane_item) = active_pane_item.as_ref() {
            if let Some(editor) = pane_item.downcast::<ProjectDiagnosticsEditor>() {
                self.editor = Some(editor.downgrade());
                ToolbarItemLocation::PrimaryRight { flex: None }
            } else {
                ToolbarItemLocation::Hidden
            }
        } else {
            ToolbarItemLocation::Hidden
        }
    }
}

impl ToolbarControls {
    pub fn new() -> Self {
        ToolbarControls { editor: None }
    }
}