use gpui::{
Div, ElementId, Hsla, InteractiveElement, IntoElement, MouseButton, MouseDownEvent,
ParentElement, RenderOnce, Styled, div, prelude::FluentBuilder,
};
use crate::theme::{ActionVariantKind, ActiveTheme};
pub fn context_menu_trigger(id: impl Into<ElementId>) -> ContextMenuTrigger {
ContextMenuTrigger::new().id(id)
}
type OpenFn = Box<dyn Fn(&MouseDownEvent, &mut gpui::Window, &mut gpui::App)>;
#[derive(IntoElement)]
pub struct ContextMenuTrigger {
element_id: ElementId,
base: Div,
on_open: Option<OpenFn>,
consume: bool,
enabled: bool,
variant: ActionVariantKind,
bg: Option<Hsla>,
hover_bg: Option<Hsla>,
}
impl Default for ContextMenuTrigger {
fn default() -> Self {
Self::new()
}
}
impl ContextMenuTrigger {
pub fn new() -> Self {
Self {
element_id: "ui:context-menu-trigger".into(),
base: div(),
on_open: None,
consume: true,
enabled: true,
variant: ActionVariantKind::Neutral,
bg: None,
hover_bg: None,
}
}
pub fn id(mut self, id: impl Into<ElementId>) -> Self {
self.element_id = id.into();
self
}
pub fn key(self, key: impl Into<ElementId>) -> Self {
self.id(key)
}
pub fn enabled(mut self, enabled: bool) -> Self {
self.enabled = enabled;
self
}
pub fn consume(mut self, consume: bool) -> Self {
self.consume = consume;
self
}
pub fn variant(mut self, variant: ActionVariantKind) -> Self {
self.variant = variant;
self
}
pub fn on_open<F>(mut self, listener: F) -> Self
where
F: 'static + Fn(&MouseDownEvent, &mut gpui::Window, &mut gpui::App),
{
self.on_open = Some(Box::new(listener));
self
}
pub fn bg(mut self, fill: impl Into<Hsla>) -> Self {
self.bg = Some(fill.into());
self
}
pub fn hover_bg(mut self, fill: impl Into<Hsla>) -> Self {
self.hover_bg = Some(fill.into());
self
}
}
impl ParentElement for ContextMenuTrigger {
fn extend(&mut self, elements: impl IntoIterator<Item = gpui::AnyElement>) {
self.base.extend(elements);
}
}
impl Styled for ContextMenuTrigger {
fn style(&mut self) -> &mut gpui::StyleRefinement {
self.base.style()
}
}
impl InteractiveElement for ContextMenuTrigger {
fn interactivity(&mut self) -> &mut gpui::Interactivity {
self.base.interactivity()
}
}
impl RenderOnce for ContextMenuTrigger {
fn render(self, _window: &mut gpui::Window, _cx: &mut gpui::App) -> impl IntoElement {
let enabled = self.enabled;
let consume = self.consume;
let on_open = self.on_open;
let bg = self.bg;
let hover_bg = self.hover_bg;
let variant = self.variant;
let action_variant = _cx.theme().action_variant(variant);
let hover_bg = hover_bg.unwrap_or(action_variant.hover_bg);
let mut resolved_bg = bg.unwrap_or(action_variant.bg);
if !enabled {
resolved_bg = action_variant.disabled_bg;
}
self.base
.block_mouse_except_scroll()
.id(self.element_id.clone())
.when(enabled, |this| this.cursor_context_menu())
.on_mouse_down(MouseButton::Right, move |ev, window, cx| {
if !enabled {
return;
}
if consume {
cx.stop_propagation();
}
if let Some(handler) = &on_open {
handler(ev, window, cx);
}
})
.bg(resolved_bg)
.hover(move |this| this.bg(hover_bg))
}
}