use crate::model::NodeId;
use crate::ops::{SwapKind, SwapVariant};
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum MenuItemKind {
Delete,
Swap,
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct MenuItem {
pub label: String,
pub kind: MenuItemKind,
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct MenuView {
pub items: Vec<MenuItem>,
pub selected: usize,
pub query: String,
}
pub(crate) enum RowEffect<'a> {
Delete,
Swap(&'a SwapKind),
}
pub(crate) struct Menu {
pub(crate) anchor: NodeId,
variants: Vec<SwapVariant>,
pub(crate) query: String,
pub(crate) selected: usize,
}
impl Menu {
pub(crate) fn for_node(anchor: NodeId, variants: Vec<SwapVariant>) -> Self {
Self { anchor, variants, query: String::new(), selected: 0 }
}
fn visible_variants(&self) -> impl Iterator<Item = &SwapVariant> {
let q = self.query.to_lowercase();
self.variants.iter().filter(move |v| q.is_empty() || v.label.to_lowercase().contains(&q))
}
pub(crate) fn visible(&self) -> Vec<RowEffect<'_>> {
std::iter::once(RowEffect::Delete).chain(self.visible_variants().map(|v| RowEffect::Swap(&v.kind))).collect()
}
pub(crate) fn view(&self) -> MenuView {
let delete = MenuItem { label: "Delete".to_string(), kind: MenuItemKind::Delete };
let swaps = self.visible_variants().map(|v| MenuItem { label: v.label.clone(), kind: MenuItemKind::Swap });
MenuView { items: std::iter::once(delete).chain(swaps).collect(), selected: self.selected, query: self.query.clone() }
}
}