palette.rs

  1use std::marker::PhantomData;
  2
  3use crate::prelude::*;
  4use crate::theme::theme;
  5use crate::{h_stack, v_stack, Keybinding, Label, LabelColor};
  6
  7#[derive(Element)]
  8pub struct Palette<S: 'static + Send + Sync + Clone> {
  9    state_type: PhantomData<S>,
 10    scroll_state: ScrollState,
 11    input_placeholder: &'static str,
 12    empty_string: &'static str,
 13    items: Vec<PaletteItem<S>>,
 14    default_order: OrderMethod,
 15}
 16
 17impl<S: 'static + Send + Sync + Clone> Palette<S> {
 18    pub fn new(scroll_state: ScrollState) -> Self {
 19        Self {
 20            state_type: PhantomData,
 21            scroll_state,
 22            input_placeholder: "Find something...",
 23            empty_string: "No items found.",
 24            items: vec![],
 25            default_order: OrderMethod::default(),
 26        }
 27    }
 28
 29    pub fn items(mut self, items: Vec<PaletteItem<S>>) -> Self {
 30        self.items = items;
 31        self
 32    }
 33
 34    pub fn placeholder(mut self, input_placeholder: &'static str) -> Self {
 35        self.input_placeholder = input_placeholder;
 36        self
 37    }
 38
 39    pub fn empty_string(mut self, empty_string: &'static str) -> Self {
 40        self.empty_string = empty_string;
 41        self
 42    }
 43
 44    // TODO: Hook up sort order
 45    pub fn default_order(mut self, default_order: OrderMethod) -> Self {
 46        self.default_order = default_order;
 47        self
 48    }
 49
 50    fn render(&mut self, _view: &mut S, cx: &mut ViewContext<S>) -> impl Element<ViewState = S> {
 51        let theme = theme(cx);
 52
 53        v_stack()
 54            .w_96()
 55            .rounded_lg()
 56            .bg(theme.lowest.base.default.background)
 57            .border()
 58            .border_color(theme.lowest.base.default.border)
 59            .child(
 60                v_stack()
 61                    .gap_px()
 62                    .child(v_stack().py_0p5().px_1().child(
 63                        div().px_2().py_0p5().child(
 64                            Label::new(self.input_placeholder).color(LabelColor::Placeholder),
 65                        ),
 66                    ))
 67                    .child(div().h_px().w_full().bg(theme.lowest.base.default.border))
 68                    .child(
 69                        v_stack()
 70                            .py_0p5()
 71                            .px_1()
 72                            .grow()
 73                            .max_h_96()
 74                            .overflow_y_scroll(self.scroll_state.clone())
 75                            .children(
 76                                vec![if self.items.is_empty() {
 77                                    Some(h_stack().justify_between().px_2().py_1().child(
 78                                        Label::new(self.empty_string).color(LabelColor::Muted),
 79                                    ))
 80                                } else {
 81                                    None
 82                                }]
 83                                .into_iter()
 84                                .flatten(),
 85                            )
 86                            .children(self.items.iter().map(|item| {
 87                                h_stack()
 88                                    .justify_between()
 89                                    .px_2()
 90                                    .py_0p5()
 91                                    .rounded_lg()
 92                                    .hover(|style| style.bg(theme.lowest.base.hovered.background))
 93                                    // .active()
 94                                    // .fill(theme.lowest.base.pressed.background)
 95                                    .child(item.clone())
 96                            })),
 97                    ),
 98            )
 99    }
100}
101
102#[derive(Element, Clone)]
103pub struct PaletteItem<S: 'static + Send + Sync + Clone> {
104    pub label: &'static str,
105    pub sublabel: Option<&'static str>,
106    pub keybinding: Option<Keybinding<S>>,
107}
108
109impl<S: 'static + Send + Sync + Clone> PaletteItem<S> {
110    pub fn new(label: &'static str) -> Self {
111        Self {
112            label,
113            sublabel: None,
114            keybinding: None,
115        }
116    }
117
118    pub fn label(mut self, label: &'static str) -> Self {
119        self.label = label;
120        self
121    }
122
123    pub fn sublabel<L: Into<Option<&'static str>>>(mut self, sublabel: L) -> Self {
124        self.sublabel = sublabel.into();
125        self
126    }
127
128    pub fn keybinding<K>(mut self, keybinding: K) -> Self
129    where
130        K: Into<Option<Keybinding<S>>>,
131    {
132        self.keybinding = keybinding.into();
133        self
134    }
135
136    fn render(&mut self, _view: &mut S, cx: &mut ViewContext<S>) -> impl Element<ViewState = S> {
137        let theme = theme(cx);
138
139        div()
140            .flex()
141            .flex_row()
142            .grow()
143            .justify_between()
144            .child(
145                v_stack()
146                    .child(Label::new(self.label))
147                    .children(self.sublabel.map(|sublabel| Label::new(sublabel))),
148            )
149            .children(self.keybinding.clone())
150    }
151}
152
153#[cfg(feature = "stories")]
154pub use stories::*;
155
156#[cfg(feature = "stories")]
157mod stories {
158    use crate::{ModifierKeys, Story};
159
160    use super::*;
161
162    #[derive(Element)]
163    pub struct PaletteStory<S: 'static + Send + Sync + Clone> {
164        state_type: PhantomData<S>,
165    }
166
167    impl<S: 'static + Send + Sync + Clone> PaletteStory<S> {
168        pub fn new() -> Self {
169            Self {
170                state_type: PhantomData,
171            }
172        }
173
174        fn render(
175            &mut self,
176            _view: &mut S,
177            cx: &mut ViewContext<S>,
178        ) -> impl Element<ViewState = S> {
179            Story::container(cx)
180                .child(Story::title_for::<_, Palette<S>>(cx))
181                .child(Story::label(cx, "Default"))
182                .child(Palette::new(ScrollState::default()))
183                .child(Story::label(cx, "With Items"))
184                .child(
185                    Palette::new(ScrollState::default())
186                        .placeholder("Execute a command...")
187                        .items(vec![
188                            PaletteItem::new("theme selector: toggle").keybinding(
189                                Keybinding::new_chord(
190                                    ("k".to_string(), ModifierKeys::new().command(true)),
191                                    ("t".to_string(), ModifierKeys::new().command(true)),
192                                ),
193                            ),
194                            PaletteItem::new("assistant: inline assist").keybinding(
195                                Keybinding::new(
196                                    "enter".to_string(),
197                                    ModifierKeys::new().command(true),
198                                ),
199                            ),
200                            PaletteItem::new("assistant: quote selection").keybinding(
201                                Keybinding::new(">".to_string(), ModifierKeys::new().command(true)),
202                            ),
203                            PaletteItem::new("assistant: toggle focus").keybinding(
204                                Keybinding::new("?".to_string(), ModifierKeys::new().command(true)),
205                            ),
206                            PaletteItem::new("auto update: check"),
207                            PaletteItem::new("auto update: view release notes"),
208                            PaletteItem::new("branches: open recent").keybinding(Keybinding::new(
209                                "b".to_string(),
210                                ModifierKeys::new().command(true).alt(true),
211                            )),
212                            PaletteItem::new("chat panel: toggle focus"),
213                            PaletteItem::new("cli: install"),
214                            PaletteItem::new("client: sign in"),
215                            PaletteItem::new("client: sign out"),
216                            PaletteItem::new("editor: cancel").keybinding(Keybinding::new(
217                                "escape".to_string(),
218                                ModifierKeys::new(),
219                            )),
220                        ]),
221                )
222        }
223    }
224}