1use gpui::{Action, AnyView};
  2
  3use crate::prelude::*;
  4use crate::{ButtonCommon, ButtonLike, ButtonSize, ButtonStyle, Icon, IconSize};
  5
  6use super::button_icon::ButtonIcon;
  7
  8#[derive(IntoElement)]
  9pub struct IconButton {
 10    base: ButtonLike,
 11    icon: Icon,
 12    icon_size: IconSize,
 13    icon_color: Color,
 14    selected_icon: Option<Icon>,
 15}
 16
 17impl IconButton {
 18    pub fn new(id: impl Into<ElementId>, icon: Icon) -> Self {
 19        Self {
 20            base: ButtonLike::new(id),
 21            icon,
 22            icon_size: IconSize::default(),
 23            icon_color: Color::Default,
 24            selected_icon: None,
 25        }
 26    }
 27
 28    pub fn icon_size(mut self, icon_size: IconSize) -> Self {
 29        self.icon_size = icon_size;
 30        self
 31    }
 32
 33    pub fn icon_color(mut self, icon_color: Color) -> Self {
 34        self.icon_color = icon_color;
 35        self
 36    }
 37
 38    pub fn selected_icon(mut self, icon: impl Into<Option<Icon>>) -> Self {
 39        self.selected_icon = icon.into();
 40        self
 41    }
 42
 43    pub fn action(self, action: Box<dyn Action>) -> Self {
 44        self.on_click(move |_event, cx| cx.dispatch_action(action.boxed_clone()))
 45    }
 46}
 47
 48impl Disableable for IconButton {
 49    fn disabled(mut self, disabled: bool) -> Self {
 50        self.base = self.base.disabled(disabled);
 51        self
 52    }
 53}
 54
 55impl Selectable for IconButton {
 56    fn selected(mut self, selected: bool) -> Self {
 57        self.base = self.base.selected(selected);
 58        self
 59    }
 60}
 61
 62impl Clickable for IconButton {
 63    fn on_click(
 64        mut self,
 65        handler: impl Fn(&gpui::ClickEvent, &mut WindowContext) + 'static,
 66    ) -> Self {
 67        self.base = self.base.on_click(handler);
 68        self
 69    }
 70}
 71
 72impl ButtonCommon for IconButton {
 73    fn id(&self) -> &ElementId {
 74        self.base.id()
 75    }
 76
 77    fn style(mut self, style: ButtonStyle) -> Self {
 78        self.base = self.base.style(style);
 79        self
 80    }
 81
 82    fn size(mut self, size: ButtonSize) -> Self {
 83        self.base = self.base.size(size);
 84        self
 85    }
 86
 87    fn tooltip(mut self, tooltip: impl Fn(&mut WindowContext) -> AnyView + 'static) -> Self {
 88        self.base = self.base.tooltip(tooltip);
 89        self
 90    }
 91}
 92
 93impl RenderOnce for IconButton {
 94    type Rendered = ButtonLike;
 95
 96    fn render(self, _cx: &mut WindowContext) -> Self::Rendered {
 97        let is_disabled = self.base.disabled;
 98        let is_selected = self.base.selected;
 99
100        self.base.child(
101            ButtonIcon::new(self.icon)
102                .disabled(is_disabled)
103                .selected(is_selected)
104                .selected_icon(self.selected_icon)
105                .size(self.icon_size)
106                .color(self.icon_color),
107        )
108    }
109}