use anyhow::Result;
use anyhow::bail;
use crossterm::event::Event;
use crossterm::event::KeyCode;
use crate::utils::terminal::read_terminal_key_press;
use super::Logger;
use super::LoggerRefreshItemKind;
use super::LoggerTextItem;
struct MultiSelectData<'a> {
prompt: &'a str,
item_hanging_indent: u16,
items: Vec<(bool, &'a String)>,
active_index: usize,
}
pub fn show_multi_select(logger: &Logger, context_name: &str, prompt: &str, item_hanging_indent: u16, items: Vec<(bool, &String)>) -> Result<Vec<usize>> {
let mut data = MultiSelectData {
prompt,
items,
item_hanging_indent,
active_index: 0,
};
loop {
let text_items = render_multi_select(&data);
logger.set_refresh_item(LoggerRefreshItemKind::Selection, text_items);
if let Event::Key(key_event) = read_terminal_key_press()? {
match &key_event.code {
KeyCode::Up | KeyCode::Char('k') => {
if data.active_index == 0 {
data.active_index = data.items.len() - 1;
} else {
data.active_index -= 1;
}
}
KeyCode::Down | KeyCode::Char('j') => {
data.active_index = (data.active_index + 1) % data.items.len();
}
KeyCode::Char(' ') => {
let current_item = data.items.get_mut(data.active_index).unwrap();
current_item.0 = !current_item.0;
}
KeyCode::Enter => {
break;
}
KeyCode::Esc | KeyCode::Char('q') => {
logger.remove_refresh_item(LoggerRefreshItemKind::Selection);
bail!("Selection cancelled.");
}
_ => {}
}
} else {
}
}
logger.remove_refresh_item(LoggerRefreshItemKind::Selection);
logger.log_text_items(&render_complete(&data), context_name);
let mut result = Vec::new();
for (i, (is_selected, _)) in data.items.iter().enumerate() {
if *is_selected {
result.push(i);
}
}
Ok(result)
}
fn render_multi_select(data: &MultiSelectData) -> Vec<LoggerTextItem> {
let mut result = vec![LoggerTextItem::Text(data.prompt.to_string())];
for (i, (is_selected, item_text)) in data.items.iter().enumerate() {
let mut text = String::new();
text.push_str(if i == data.active_index { ">" } else { " " });
text.push_str(" [");
text.push_str(if *is_selected { "x" } else { " " });
text.push_str("] ");
text.push_str(item_text);
result.push(LoggerTextItem::HangingText {
text,
indent: 7 + data.item_hanging_indent,
});
}
result
}
fn render_complete(data: &MultiSelectData) -> Vec<LoggerTextItem> {
let mut result = Vec::new();
if data.items.iter().any(|(is_selected, _)| *is_selected) {
result.push(LoggerTextItem::Text(data.prompt.to_string()));
for (is_selected, item_text) in data.items.iter() {
if *is_selected {
result.push(LoggerTextItem::HangingText {
text: format!(" * {}", item_text),
indent: 3 + data.item_hanging_indent,
});
}
}
}
result
}