feat: add icon, tooltip, and outline variant to badge

crates/core/src/badge.rs

@@ -1,11 +1,19 @@
+use std::rc::Rc;
+
+use gpui::prelude::FluentBuilder as _;
 use gpui::{
-    App, Hsla, IntoElement, ParentElement as _, RenderOnce, SharedString, StyleRefinement,
-    Styled as _, Window,
+    AnyView, App, ElementId, Hsla, InteractiveElement as _, IntoElement, ParentElement as _,
+    RenderOnce, SharedString, StatefulInteractiveElement as _, StyleRefinement, Styled as _,
+    Window, transparent_black,
 };
+use gpui_native_icons::Icon;
 use gpui_native_theme::{ActiveTheme as _, Size, Theme};
 
 use crate::common::{Sizable, StyledExt, h_flex};
 
+/// The tooltip view builder gpui's `.tooltip(...)` expects.
+type TooltipBuilder = Rc<dyn Fn(&mut Window, &mut App) -> AnyView>;
+
 /// How loud a badge is.
 #[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
 pub enum BadgeVariant {
@@ -20,18 +28,42 @@ pub enum BadgeVariant {
     Warning,
     /// The destructive ramp.
     Danger,
+    /// No fill, just a border; the quietest badge.
+    Outline,
 }
 
 impl BadgeVariant {
-    fn tokens(self, theme: &Theme) -> (Hsla, Hsla) {
+    fn tokens(self, theme: &Theme) -> (Hsla, Hsla, Hsla) {
         match self {
-            Self::Neutral => (theme.colors.muted, theme.colors.muted_foreground),
-            Self::Accent => (theme.ramps.primary.base, theme.colors.primary_foreground),
-            Self::Success => (theme.ramps.success.base, theme.colors.success_foreground),
-            Self::Warning => (theme.ramps.warning.base, theme.colors.warning_foreground),
+            Self::Neutral => (
+                theme.colors.muted,
+                theme.colors.muted_foreground,
+                transparent_black(),
+            ),
+            Self::Accent => (
+                theme.ramps.primary.base,
+                theme.colors.primary_foreground,
+                transparent_black(),
+            ),
+            Self::Success => (
+                theme.ramps.success.base,
+                theme.colors.success_foreground,
+                transparent_black(),
+            ),
+            Self::Warning => (
+                theme.ramps.warning.base,
+                theme.colors.warning_foreground,
+                transparent_black(),
+            ),
             Self::Danger => (
                 theme.ramps.destructive.base,
                 theme.colors.destructive_foreground,
+                transparent_black(),
+            ),
+            Self::Outline => (
+                transparent_black(),
+                theme.colors.foreground,
+                theme.colors.border,
             ),
         }
     }
@@ -40,22 +72,29 @@ impl BadgeVariant {
 /// A small pill for a count or a state.
 #[derive(IntoElement)]
 pub struct Badge {
+    id: ElementId,
     label: SharedString,
+    icon: Option<Icon>,
     variant: BadgeVariant,
     size: Size,
+    tooltip: Option<TooltipBuilder>,
     style: StyleRefinement,
 }
 
 impl Badge {
     /// Creates a badge showing `label` — a count or state word the caller formats.
-    pub fn new<L>(label: L) -> Self
+    pub fn new<I, L>(id: I, label: L) -> Self
     where
+        I: Into<ElementId>,
         L: Into<SharedString>,
     {
         Self {
+            id: id.into(),
             label: label.into(),
+            icon: None,
             variant: BadgeVariant::default(),
             size: Size::default(),
+            tooltip: None,
             style: StyleRefinement::default(),
         }
     }
@@ -65,6 +104,24 @@ impl Badge {
         self.variant = variant;
         self
     }
+
+    /// Shows an icon before the label.
+    pub fn icon<I>(mut self, icon: I) -> Self
+    where
+        I: Into<Icon>,
+    {
+        self.icon = Some(icon.into());
+        self
+    }
+
+    /// Shows a tooltip view next to the badge after it is hovered.
+    pub fn tooltip<F>(mut self, tooltip: F) -> Self
+    where
+        F: Fn(&mut Window, &mut App) -> AnyView + 'static,
+    {
+        self.tooltip = Some(Rc::new(tooltip));
+        self
+    }
 }
 
 impl Sizable for Badge {
@@ -84,18 +141,28 @@ impl RenderOnce for Badge {
         let theme = cx.theme();
         let metrics = theme.sizes.resolve(self.size);
         let label_size = metrics.caption_size;
-        let (background, foreground) = self.variant.tokens(theme);
+        let (background, foreground, border) = self.variant.tokens(theme);
 
         h_flex()
+            .id(self.id.clone())
             .flex_none()
             .justify_center()
+            .gap_1()
             .rounded(theme.radius.full)
+            .border_1()
+            .border_color(border)
             .bg(background)
             .text_color(foreground)
             .text_size(label_size)
             .px(metrics.dense_padding_x)
             .py_0p5()
+            .when_some(self.icon, |badge, icon| {
+                badge.child(icon.size(metrics.dense_icon_size).color(foreground))
+            })
             .child(self.label)
             .refine_style(&self.style)
+            .when_some(self.tooltip, |badge, tooltip| {
+                badge.tooltip(move |window, cx| tooltip(window, cx))
+            })
     }
 }