cursive/src/views/select_view.rs

933 lines
29 KiB
Rust
Raw Normal View History

use crate::align::{Align, HAlign, VAlign};
use crate::direction::Direction;
2019-03-01 00:04:14 +00:00
use crate::event::{
Callback, Event, EventResult, Key, MouseButton, MouseEvent,
};
use crate::menu::MenuTree;
use crate::rect::Rect;
use crate::theme::ColorStyle;
use crate::utils::markup::StyledString;
use crate::vec::Vec2;
use crate::view::{Position, View};
use crate::views::MenuPopup;
use crate::Cursive;
use crate::Printer;
use crate::With;
2019-03-01 00:04:14 +00:00
use std::borrow::Borrow;
use std::cell::Cell;
use std::cmp::{min, Ordering};
2019-03-01 00:04:14 +00:00
use std::rc::Rc;
2016-07-04 23:04:32 +00:00
/// View to select an item among a list.
///
/// It contains a list of values of type T, with associated labels.
2016-07-21 05:08:06 +00:00
///
/// # Examples
///
/// ```rust
/// # use cursive::Cursive;
/// # use cursive::views::{SelectView, Dialog, TextView};
2016-07-21 05:08:06 +00:00
/// # use cursive::align::HAlign;
/// # fn main() {
/// let mut time_select = SelectView::new().h_align(HAlign::Center);
/// time_select.add_item("Short", 1);
/// time_select.add_item("Medium", 5);
/// time_select.add_item("Long", 10);
///
/// time_select.set_on_submit(|s, time| {
2016-07-21 05:08:06 +00:00
/// s.pop_layer();
/// let text = format!("You will wait for {} minutes...", time);
/// s.add_layer(Dialog::around(TextView::new(text))
2016-07-21 05:08:06 +00:00
/// .button("Quit", |s| s.quit()));
/// });
///
/// let mut siv = Cursive::dummy();
/// siv.add_layer(Dialog::around(time_select)
/// .title("How long is your wait?"));
2016-07-21 05:08:06 +00:00
/// # }
///
/// ```
2016-03-15 22:37:57 +00:00
pub struct SelectView<T = String> {
2018-09-27 23:01:37 +00:00
// The core of the view: we store a list of items
// `Item` is more or less a `(String, Rc<T>)`.
items: Vec<Item<T>>,
2018-09-27 23:01:37 +00:00
// When disabled, we cannot change selection.
2016-07-17 01:46:18 +00:00
enabled: bool,
2018-09-27 23:01:37 +00:00
// Callbacks may need to manipulate focus, so give it some mutability.
focus: Rc<Cell<usize>>,
2018-09-27 23:01:37 +00:00
// This is a custom callback to include a &T.
2018-09-27 23:01:37 +00:00
// It will be called whenever "Enter" is pressed or when an item is clicked.
2019-02-28 23:55:02 +00:00
on_submit: Option<Rc<dyn Fn(&mut Cursive, &T)>>,
2018-09-27 23:01:37 +00:00
// This callback is called when the selection is changed.
2018-09-27 23:01:37 +00:00
// TODO: add the previous selection? Indices?
2019-02-28 23:55:02 +00:00
on_select: Option<Rc<dyn Fn(&mut Cursive, &T)>>,
2018-09-27 23:01:37 +00:00
// If `true`, when a character is pressed, jump to the next item starting
// with this character.
autojump: bool,
align: Align,
2018-09-27 23:01:37 +00:00
// `true` if we show a one-line view, with popup on selection.
popup: bool,
2018-09-27 23:01:37 +00:00
// We need the last offset to place the popup window
// We "cache" it during the draw, so we need interior mutability.
last_offset: Cell<Vec2>,
last_size: Vec2,
}
2017-08-14 23:32:01 +00:00
impl<T: 'static> Default for SelectView<T> {
fn default() -> Self {
Self::new()
}
}
2016-06-25 23:36:22 +00:00
impl<T: 'static> SelectView<T> {
/// Creates a new empty SelectView.
pub fn new() -> Self {
SelectView {
items: Vec::new(),
2016-07-17 01:46:18 +00:00
enabled: true,
focus: Rc::new(Cell::new(0)),
on_select: None,
on_submit: None,
align: Align::top_left(),
popup: false,
2018-09-27 23:01:37 +00:00
autojump: false,
last_offset: Cell::new(Vec2::zero()),
last_size: Vec2::zero(),
}
}
2018-09-27 23:01:37 +00:00
/// Sets the "auto-jump" property for this view.
///
/// If enabled, when a key is pressed, the selection will jump to the next
/// item beginning with the pressed letter.
pub fn set_autojump(&mut self, autojump: bool) {
self.autojump = autojump;
}
/// Sets the "auto-jump" property for this view.
///
/// If enabled, when a key is pressed, the selection will jump to the next
/// item beginning with the pressed letter.
///
/// Chainable variant.
pub fn autojump(self) -> Self {
self.with(|s| s.set_autojump(true))
}
/// Turns `self` into a popup select view.
///
/// Chainable variant.
pub fn popup(self) -> Self {
self.with(|s| s.set_popup(true))
}
/// Turns `self` into a popup select view.
pub fn set_popup(&mut self, popup: bool) {
self.popup = popup;
}
2016-07-17 01:46:18 +00:00
/// Disables this view.
///
/// A disabled view cannot be selected.
pub fn disable(&mut self) {
self.enabled = false;
}
/// Disables this view.
///
/// Chainable variant.
pub fn disabled(self) -> Self {
self.with(Self::disable)
}
/// Re-enables this view.
pub fn enable(&mut self) {
self.enabled = true;
}
/// Enable or disable this view.
pub fn set_enabled(&mut self, enabled: bool) {
self.enabled = enabled;
}
/// Returns `true` if this view is enabled.
pub fn is_enabled(&self) -> bool {
self.enabled
}
/// Sets a callback to be used when an item is selected.
2015-07-28 19:54:32 +00:00
pub fn set_on_select<F>(&mut self, cb: F)
2017-08-23 23:48:54 +00:00
where
F: Fn(&mut Cursive, &T) + 'static,
2015-07-28 19:54:32 +00:00
{
self.on_select = Some(Rc::new(cb));
2015-07-28 19:54:32 +00:00
}
/// Sets a callback to be used when an item is selected.
///
/// Chainable variant.
pub fn on_select<F>(self, cb: F) -> Self
2017-08-23 23:48:54 +00:00
where
F: Fn(&mut Cursive, &T) + 'static,
{
self.with(|s| s.set_on_select(cb))
}
/// Sets a callback to be used when `<Enter>` is pressed.
///
2018-09-27 23:01:37 +00:00
/// Also happens if the user clicks an item.
///
/// The item currently selected will be given to the callback.
///
/// Here, `V` can be `T` itself, or a type that can be borrowed from `T`.
pub fn set_on_submit<F, R, V: ?Sized>(&mut self, cb: F)
2017-08-23 23:48:54 +00:00
where
F: 'static + Fn(&mut Cursive, &V) -> R,
2017-08-23 23:48:54 +00:00
T: Borrow<V>,
{
self.on_submit = Some(Rc::new(move |s, t| {
cb(s, t.borrow());
}));
}
/// Sets a callback to be used when `<Enter>` is pressed.
///
2018-09-27 23:01:37 +00:00
/// Also happens if the user clicks an item.
///
/// The item currently selected will be given to the callback.
///
/// Chainable variant.
pub fn on_submit<F, V: ?Sized>(self, cb: F) -> Self
2017-08-23 23:48:54 +00:00
where
F: Fn(&mut Cursive, &V) + 'static,
T: Borrow<V>,
{
self.with(|s| s.set_on_submit(cb))
}
2015-06-03 22:36:51 +00:00
/// Sets the alignment for this view.
pub fn align(mut self, align: Align) -> Self {
self.align = align;
self
}
2015-06-03 22:36:51 +00:00
/// Sets the vertical alignment for this view.
2016-06-26 16:45:53 +00:00
/// (If the view is given too much space vertically.)
pub fn v_align(mut self, v: VAlign) -> Self {
self.align.v = v;
self
}
2015-06-03 22:36:51 +00:00
/// Sets the horizontal alignment for this view.
pub fn h_align(mut self, h: HAlign) -> Self {
self.align.h = h;
self
}
2016-06-28 05:10:59 +00:00
/// Returns the value of the currently selected item.
///
2018-07-23 03:26:17 +00:00
/// Returns `None` if the list is empty.
pub fn selection(&self) -> Option<Rc<T>> {
let focus = self.focus();
if self.len() <= focus {
None
} else {
Some(Rc::clone(&self.items[focus].value))
}
}
/// Removes all items from this view.
pub fn clear(&mut self) {
self.items.clear();
2017-03-27 20:07:22 +00:00
self.focus.set(0);
}
/// Adds a item to the list, with given label and value.
2018-11-30 21:19:08 +00:00
pub fn add_item<S: Into<StyledString>>(&mut self, label: S, value: T) {
self.items.push(Item::new(label.into(), value));
}
2018-01-26 12:25:19 +00:00
/// Gets an item at given idx or None.
///
/// ```
/// use cursive::Cursive;
/// use cursive::views::{SelectView, TextView};
/// let select = SelectView::new()
/// .item("Short", 1);
/// assert_eq!(select.get_item(0), Some(("Short", &1)));
/// ```
pub fn get_item(&self, i: usize) -> Option<(&str, &T)> {
2018-11-30 21:19:08 +00:00
self.iter().nth(i)
2018-01-26 12:25:19 +00:00
}
2018-01-26 16:24:44 +00:00
/// Gets a mut item at given idx or None.
2018-11-30 21:19:08 +00:00
pub fn get_item_mut(
2019-07-30 23:08:05 +00:00
&mut self,
i: usize,
2018-11-30 21:19:08 +00:00
) -> Option<(&mut StyledString, &mut T)> {
2018-01-26 16:24:44 +00:00
if i >= self.items.len() {
None
} else {
let item = &mut self.items[i];
if let Some(t) = Rc::get_mut(&mut item.value) {
let label = &mut item.label;
Some((label, t))
} else {
None
}
}
}
2018-09-27 23:01:37 +00:00
/// Iterate on the items in this view.
///
/// Returns an iterator with each item and their labels.
pub fn iter(&self) -> impl Iterator<Item = (&str, &T)> {
self.items
.iter()
2018-11-30 21:19:08 +00:00
.map(|item| (item.label.source(), &*item.value))
2018-09-27 23:01:37 +00:00
}
/// Removes an item from the list.
///
/// Returns a callback in response to the selection change.
///
/// You should run this callback with a `&mut Cursive`.
pub fn remove_item(&mut self, id: usize) -> Callback {
self.items.remove(id);
let focus = self.focus();
if focus >= id && focus > 0 {
self.focus.set(focus - 1);
}
2018-05-18 00:37:39 +00:00
self.make_select_cb().unwrap_or_else(Callback::dummy)
}
2018-03-25 07:30:48 +00:00
/// Inserts an item at position `index`, shifting all elements after it to
/// the right.
pub fn insert_item<S>(&mut self, index: usize, label: S, value: T)
where
2018-11-30 21:19:08 +00:00
S: Into<StyledString>,
2018-03-25 07:30:48 +00:00
{
self.items.insert(index, Item::new(label.into(), value));
}
/// Chainable variant of add_item
pub fn item<S: Into<String>>(self, label: S, value: T) -> Self {
self.with(|s| s.add_item(label, value))
}
/// Adds all items from from an iterator.
pub fn add_all<S, I>(&mut self, iter: I)
2017-08-23 23:48:54 +00:00
where
2018-11-30 21:19:08 +00:00
S: Into<StyledString>,
2017-08-23 23:48:54 +00:00
I: IntoIterator<Item = (S, T)>,
{
for (s, t) in iter {
self.add_item(s, t);
}
}
/// Adds all items from from an iterator.
///
/// Chainable variant.
pub fn with_all<S, I>(self, iter: I) -> Self
2017-08-23 23:48:54 +00:00
where
S: Into<String>,
I: IntoIterator<Item = (S, T)>,
{
self.with(|s| s.add_all(iter))
}
2016-07-17 01:46:18 +00:00
2019-02-28 23:55:02 +00:00
fn draw_item(&self, printer: &Printer<'_, '_>, i: usize) {
2016-07-17 01:46:18 +00:00
let l = self.items[i].label.width();
let x = self.align.h.get_offset(l, printer.size.x);
printer.print_hline((0, 0), x, " ");
2018-11-30 21:19:08 +00:00
printer.print_styled((x, 0), (&self.items[i].label).into());
2016-07-17 08:20:41 +00:00
if l < printer.size.x {
2017-08-14 23:32:01 +00:00
assert!((l + x) <= printer.size.x);
printer.print_hline((x + l, 0), printer.size.x - (l + x), " ");
2016-07-17 08:20:41 +00:00
}
2016-07-17 01:46:18 +00:00
}
/// Returns the id of the item currently selected.
///
/// Returns `None` if the list is empty.
pub fn selected_id(&self) -> Option<usize> {
if self.items.is_empty() {
None
} else {
Some(self.focus())
}
}
/// Returns the number of items in this list.
pub fn len(&self) -> usize {
self.items.len()
}
/// Returns `true` if this list has no item.
pub fn is_empty(&self) -> bool {
self.items.is_empty()
}
fn focus(&self) -> usize {
self.focus.get()
}
2019-04-12 00:08:55 +00:00
/// Sort the current items lexicographically by their label.
/// Note that this does not change the current focus index, which means that the current
/// selection will likely be changed by the sorting.
/// This sort is stable: items with identical label will not be reordered.
pub fn sort_by_label(&mut self) {
2019-04-12 00:08:55 +00:00
self.items
.sort_by(|a, b| a.label.source().cmp(b.label.source()));
}
2019-04-12 00:08:55 +00:00
/// Sort the current items with the given comparator function.
/// Note that this does not change the current focus index, which means that the current
/// selection will likely be changed by the sorting.
/// The given comparator function must define a total order for the items.
/// If the comparator function does not define a total order, then the order after the sort is
/// unspecified.
2019-04-12 00:08:55 +00:00
/// This sort is stable: equal items will not be reordered.
pub fn sort_by<F>(&mut self, mut compare: F)
2019-04-12 00:08:55 +00:00
where
F: FnMut(&T, &T) -> Ordering,
{
self.items.sort_by(|a, b| compare(&a.value, &b.value));
}
2019-04-12 00:08:55 +00:00
/// Sort the current items with the given key extraction function.
/// Note that this does not change the current focus index, which means that the current
/// selection will likely be changed by the sorting.
/// This sort is stable: items with equal keys will not be reordered.
pub fn sort_by_key<K, F>(&mut self, mut key_of: F)
2019-04-12 00:08:55 +00:00
where
F: FnMut(&T) -> K,
K: Ord,
{
self.items.sort_by_key(|item| key_of(&item.value));
}
2017-03-27 20:14:03 +00:00
/// Moves the selection to the given position.
///
/// Returns a callback in response to the selection change.
///
/// You should run this callback with a `&mut Cursive`.
pub fn set_selection(&mut self, i: usize) -> Callback {
// TODO: Check if `i >= self.len()` ?
// assert!(i < self.len(), "SelectView: trying to select out-of-bound");
// Or just cap the ID?
2018-04-10 18:53:25 +00:00
let i = if self.is_empty() {
0
} else {
min(i, self.len() - 1)
};
2017-03-27 20:14:03 +00:00
self.focus.set(i);
2018-05-18 00:37:39 +00:00
self.make_select_cb().unwrap_or_else(Callback::dummy)
2017-03-27 20:14:03 +00:00
}
2018-02-12 03:26:44 +00:00
/// Sets the selection to the given position.
///
/// Chainable variant.
///
/// Does not apply `on_select` callbacks.
2018-02-12 03:26:44 +00:00
pub fn selected(self, i: usize) -> Self {
self.with(|s| {
s.set_selection(i);
})
2018-02-12 03:26:44 +00:00
}
2017-08-23 23:48:54 +00:00
/// Moves the selection up by the given number of rows.
///
/// Returns a callback in response to the selection change.
///
/// You should run this callback with a `&mut Cursive`:
///
/// ```rust
/// # use cursive::Cursive;
/// # use cursive::views::SelectView;
/// # fn main() {}
/// fn select_up(siv: &mut Cursive, view: &mut SelectView<()>) {
/// let cb = view.select_up(1);
/// cb(siv);
/// }
/// ```
pub fn select_up(&mut self, n: usize) -> Callback {
2017-08-23 23:48:54 +00:00
self.focus_up(n);
2018-05-18 00:37:39 +00:00
self.make_select_cb().unwrap_or_else(Callback::dummy)
2017-08-23 23:48:54 +00:00
}
/// Moves the selection down by the given number of rows.
///
/// Returns a callback in response to the selection change.
///
/// You should run this callback with a `&mut Cursive`.
pub fn select_down(&mut self, n: usize) -> Callback {
2017-08-23 23:48:54 +00:00
self.focus_down(n);
2018-05-18 00:37:39 +00:00
self.make_select_cb().unwrap_or_else(Callback::dummy)
2017-08-23 23:48:54 +00:00
}
fn focus_up(&mut self, n: usize) {
let focus = self.focus().saturating_sub(n);
self.focus.set(focus);
}
fn focus_down(&mut self, n: usize) {
2018-05-18 00:37:39 +00:00
let focus = min(self.focus() + n, self.items.len().saturating_sub(1));
self.focus.set(focus);
}
fn submit(&mut self) -> EventResult {
let cb = self.on_submit.clone().unwrap();
// We return a Callback Rc<|s| cb(s, &*v)>
EventResult::Consumed(
self.selection()
.map(|v| Callback::from_fn(move |s| cb(s, &v))),
)
}
2018-09-27 23:01:37 +00:00
fn on_char_event(&mut self, c: char) -> EventResult {
let i = {
// * Starting from the current focus, find the first item that
// match the char.
// * Cycle back to the beginning of the list when we reach the end.
// * This is achieved by chaining twice the iterator.
let iter = self.iter().chain(self.iter());
// We'll do a lowercase check.
let lower_c: Vec<char> = c.to_lowercase().collect();
let lower_c: &[char] = &lower_c;
if let Some((i, _)) = iter.enumerate().skip(self.focus() + 1).find(
|&(_, (label, _))| label.to_lowercase().starts_with(lower_c),
) {
i % self.len()
} else {
return EventResult::Ignored;
}
};
self.focus.set(i);
// Apply modulo in case we have a hit from the chained iterator
let cb = self.set_selection(i);
EventResult::Consumed(Some(cb))
}
fn on_event_regular(&mut self, event: Event) -> EventResult {
match event {
Event::Key(Key::Up) if self.focus() > 0 => self.focus_up(1),
Event::Key(Key::Down) if self.focus() + 1 < self.items.len() => {
self.focus_down(1)
}
Event::Key(Key::PageUp) => self.focus_up(10),
Event::Key(Key::PageDown) => self.focus_down(10),
Event::Key(Key::Home) => self.focus.set(0),
2018-05-18 00:37:39 +00:00
Event::Key(Key::End) => {
self.focus.set(self.items.len().saturating_sub(1))
}
2017-10-15 04:18:50 +00:00
Event::Mouse {
event: MouseEvent::Press(_),
position,
offset,
2018-11-09 22:08:43 +00:00
} if position
.checked_sub(offset)
.map(|position| {
position < self.last_size && position.y < self.len()
})
.unwrap_or(false) =>
2017-10-15 04:18:50 +00:00
{
self.focus.set(position.y - offset.y)
}
Event::Mouse {
event: MouseEvent::Release(MouseButton::Left),
position,
offset,
2018-11-09 22:08:43 +00:00
} if self.on_submit.is_some()
&& position
2018-08-22 20:33:29 +00:00
.checked_sub(offset)
.map(|position| {
position < self.last_size && position.y == self.focus()
})
.unwrap_or(false) =>
{
return self.submit();
}
Event::Key(Key::Enter) if self.on_submit.is_some() => {
return self.submit();
}
2018-09-27 23:01:37 +00:00
Event::Char(c) if self.autojump => return self.on_char_event(c),
_ => return EventResult::Ignored,
}
EventResult::Consumed(self.make_select_cb())
}
/// Returns a callback from selection change.
fn make_select_cb(&self) -> Option<Callback> {
self.on_select.clone().and_then(|cb| {
self.selection()
.map(|v| Callback::from_fn(move |s| cb(s, &v)))
})
}
fn open_popup(&mut self) -> EventResult {
// Build a shallow menu tree to mimick the items array.
// TODO: cache it?
let mut tree = MenuTree::new();
for (i, item) in self.items.iter().enumerate() {
2017-10-13 18:22:02 +00:00
let focus = Rc::clone(&self.focus);
let on_submit = self.on_submit.as_ref().cloned();
2017-10-13 18:22:02 +00:00
let value = Rc::clone(&item.value);
2018-11-30 21:19:08 +00:00
tree.add_leaf(item.label.source(), move |s| {
// TODO: What if an item was removed in the meantime?
focus.set(i);
if let Some(ref on_submit) = on_submit {
on_submit(s, &value);
}
});
}
// Let's keep the tree around,
// the callback will want to use it.
let tree = Rc::new(tree);
let focus = self.focus();
// This is the offset for the label text.
// We'll want to show the popup so that the text matches.
// It'll be soo cool.
2018-11-30 21:19:08 +00:00
let item_length = self.items[focus].label.width();
let text_offset = (self.last_size.x.saturating_sub(item_length)) / 2;
// The total offset for the window is:
// * the last absolute offset at which we drew this view
// * shifted to the right of the text offset
// * shifted to the top of the focus (so the line matches)
// * shifted top-left of the border+padding of the popup
let offset = self.last_offset.get();
let offset = offset + (text_offset, 0);
let offset = offset.saturating_sub((0, focus));
let offset = offset.saturating_sub((2, 1));
2017-10-12 23:42:17 +00:00
// And now, we can return the callback that will create the popup.
EventResult::with_cb(move |s| {
// The callback will want to work with a fresh Rc
2017-10-13 18:22:02 +00:00
let tree = Rc::clone(&tree);
// We'll relativise the absolute position,
// So that we are locked to the parent view.
// A nice effect is that window resizes will keep both
// layers together.
let current_offset = s.screen().offset();
let offset = offset.signed() - current_offset;
// And finally, put the view in view!
s.screen_mut().add_layer_at(
Position::parent(offset),
MenuPopup::new(tree).focus(focus),
);
})
}
// A popup view only does one thing: open the popup on Enter.
fn on_event_popup(&mut self, event: Event) -> EventResult {
match event {
// TODO: add Left/Right support for quick-switch?
Event::Key(Key::Enter) => self.open_popup(),
Event::Mouse {
2017-10-12 23:42:17 +00:00
event: MouseEvent::Release(MouseButton::Left),
position,
offset,
2018-11-09 22:08:43 +00:00
} if position.fits_in_rect(offset, self.last_size) => {
self.open_popup()
}
_ => EventResult::Ignored,
}
}
}
impl SelectView<String> {
2016-06-28 05:10:59 +00:00
/// Convenient method to use the label as value.
pub fn add_item_str<S: Into<String>>(&mut self, label: S) {
let label = label.into();
self.add_item(label.clone(), label);
2015-05-31 23:58:55 +00:00
}
/// Chainable variant of add_item_str
pub fn item_str<S: Into<String>>(self, label: S) -> Self {
self.with(|s| s.add_item_str(label))
}
2018-03-25 07:30:48 +00:00
/// Convenient method to use the label as value.
2018-05-18 00:37:39 +00:00
pub fn insert_item_str<S>(&mut self, index: usize, label: S)
where
S: Into<String>,
{
2018-03-25 07:30:48 +00:00
let label = label.into();
self.insert_item(index, label.clone(), label);
}
/// Adds all strings from an iterator.
///
/// # Examples
///
/// ```
/// # use cursive::views::SelectView;
/// let mut select_view = SelectView::new();
/// select_view.add_all_str(vec!["a", "b", "c"]);
/// ```
pub fn add_all_str<S, I>(&mut self, iter: I)
2017-08-23 23:48:54 +00:00
where
S: Into<String>,
I: IntoIterator<Item = S>,
{
for s in iter {
self.add_item_str(s);
}
}
/// Adds all strings from an iterator.
///
/// Chainable variant.
pub fn with_all_str<S, I>(self, iter: I) -> Self
2017-08-23 23:48:54 +00:00
where
S: Into<String>,
I: IntoIterator<Item = S>,
{
self.with(|s| s.add_all_str(iter))
}
}
2019-04-12 00:08:55 +00:00
impl<T: 'static> SelectView<T>
where
T: Ord,
{
/// Sort the current items by their natural ordering.
/// Note that this does not change the current focus index, which means that the current
/// selection will likely be changed by the sorting.
/// This sort is stable: items that are equal will not be reordered.
pub fn sort(&mut self) {
self.items.sort_by(|a, b| a.value.cmp(&b.value));
}
}
2016-06-25 23:36:22 +00:00
impl<T: 'static> View for SelectView<T> {
2019-02-28 23:55:02 +00:00
fn draw(&self, printer: &Printer<'_, '_>) {
self.last_offset.set(printer.offset);
if self.popup {
// Popup-select only draw the active element.
// We'll draw the full list in a popup if needed.
2018-11-09 22:08:43 +00:00
let style = if !(self.enabled && printer.enabled) {
ColorStyle::secondary()
2019-03-04 18:31:36 +00:00
} else if printer.focused {
ColorStyle::highlight()
2019-03-04 18:31:36 +00:00
} else {
ColorStyle::primary()
};
2017-08-14 23:32:01 +00:00
let x = match printer.size.x.checked_sub(1) {
Some(x) => x,
None => return,
};
printer.with_color(style, |printer| {
// Prepare the entire background
2017-08-14 23:32:01 +00:00
printer.print_hline((1, 0), x, " ");
// Draw the borders
printer.print((0, 0), "<");
2017-08-14 23:32:01 +00:00
printer.print((x, 0), ">");
let label = &self.items[self.focus()].label;
// And center the text?
2018-11-30 21:19:08 +00:00
let offset = HAlign::Center.get_offset(label.width(), x + 1);
2018-11-30 21:19:08 +00:00
printer.print_styled((offset, 0), label.into());
});
} else {
// Non-popup mode: we always print the entire list.
let h = self.items.len();
let offset = self.align.v.get_offset(h, printer.size.y);
let printer = &printer.offset((0, offset));
for i in 0..self.len() {
printer.offset((0, i)).with_selection(
i == self.focus(),
|printer| {
2018-11-09 22:08:43 +00:00
if i != self.focus()
&& !(self.enabled && printer.enabled)
{
printer.with_color(
ColorStyle::secondary(),
|printer| self.draw_item(printer, i),
);
} else {
self.draw_item(printer, i);
}
},
);
}
}
}
fn required_size(&mut self, _: Vec2) -> Vec2 {
// Items here are not compressible.
// So no matter what the horizontal requirements are,
// we'll still return our longest item.
2018-06-11 06:29:10 +00:00
let w = self
.items
2016-07-10 02:05:51 +00:00
.iter()
.map(|item| item.label.width())
.max()
.unwrap_or(1);
if self.popup {
Vec2::new(w + 2, 1)
} else {
let h = self.items.len();
Vec2::new(w, h)
}
}
fn on_event(&mut self, event: Event) -> EventResult {
if self.popup {
self.on_event_popup(event)
} else {
self.on_event_regular(event)
}
}
fn take_focus(&mut self, _: Direction) -> bool {
2016-07-17 01:46:18 +00:00
self.enabled && !self.items.is_empty()
}
fn layout(&mut self, size: Vec2) {
self.last_size = size;
}
fn important_area(&self, size: Vec2) -> Rect {
self.selected_id()
.map(|i| Rect::from_size((0, i), (size.x, 1)))
2018-05-18 00:35:57 +00:00
.unwrap_or_else(|| Rect::from((0, 0)))
}
}
2018-09-27 23:01:37 +00:00
// We wrap each value in a `Rc` and add a label
struct Item<T> {
2018-11-30 21:19:08 +00:00
label: StyledString,
value: Rc<T>,
}
impl<T> Item<T> {
2018-11-30 21:19:08 +00:00
fn new(label: StyledString, value: T) -> Self {
let value = Rc::new(value);
Item { label, value }
}
}
#[cfg(test)]
mod tests {
use super::*;
2019-04-12 00:08:55 +00:00
#[test]
fn select_view_sorting() {
// We add items in no particular order, from going by their label.
let mut view = SelectView::new();
view.add_item_str("Y");
view.add_item_str("Z");
view.add_item_str("X");
// Then sorting the list...
view.sort_by_label();
// ... should observe the items in sorted order.
// And focus is NOT changed by the sorting, so the first item is "X".
assert_eq!(view.selection(), Some(Rc::new(String::from("X"))));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(String::from("Y"))));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(String::from("Z"))));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(String::from("Z"))));
}
2019-04-12 00:08:55 +00:00
#[test]
fn select_view_sorting_with_comparator() {
// We add items in no particular order, from going by their value.
let mut view = SelectView::new();
view.add_item("Y", 2);
view.add_item("Z", 1);
view.add_item("X", 3);
// Then sorting the list...
view.sort_by(|a, b| a.cmp(b));
// ... should observe the items in sorted order.
// And focus is NOT changed by the sorting, so the first item is "X".
assert_eq!(view.selection(), Some(Rc::new(1)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(2)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(3)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(3)));
}
2019-04-12 00:08:55 +00:00
#[test]
fn select_view_sorting_by_key() {
// We add items in no particular order, from going by their key value.
#[derive(Eq, PartialEq, Debug)]
struct MyStruct {
2019-04-12 00:08:55 +00:00
key: i32,
}
2019-04-12 00:08:55 +00:00
let mut view = SelectView::new();
2019-04-12 00:08:55 +00:00
view.add_item("Y", MyStruct { key: 2 });
view.add_item("Z", MyStruct { key: 1 });
view.add_item("X", MyStruct { key: 3 });
// Then sorting the list...
view.sort_by_key(|s| s.key);
// ... should observe the items in sorted order.
// And focus is NOT changed by the sorting, so the first item is "X".
assert_eq!(view.selection(), Some(Rc::new(MyStruct { key: 1 })));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(MyStruct { key: 2 })));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(MyStruct { key: 3 })));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(MyStruct { key: 3 })));
}
#[test]
fn select_view_sorting_orderable_items() {
// We add items in no particular order, from going by their value.
let mut view = SelectView::new();
view.add_item("Y", 2);
view.add_item("Z", 1);
view.add_item("X", 3);
// Then sorting the list...
view.sort();
// ... should observe the items in sorted order.
// And focus is NOT changed by the sorting, so the first item is "X".
assert_eq!(view.selection(), Some(Rc::new(1)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(2)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(3)));
view.on_event(Event::Key(Key::Down));
assert_eq!(view.selection(), Some(Rc::new(3)));
}
}