use crate::context::{AppContext, MatchContext};
use crate::error::AppResult;
use super::cursor_move;
#[derive(Clone, Copy, Debug)]
pub struct SelectOption {
pub toggle: bool,
pub all: bool,
pub reverse: bool,
}
impl std::default::Default for SelectOption {
fn default() -> Self {
Self {
toggle: true,
all: false,
reverse: false,
}
}
}
impl std::fmt::Display for SelectOption {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
write!(
f,
"--toggle={} --all={} --deselect={}",
self.toggle, self.all, self.reverse
)
}
}
pub fn select_files(
context: &mut AppContext,
pattern: &MatchContext,
options: &SelectOption,
) -> AppResult {
if pattern.is_none() {
select_without_pattern(context, options)
} else {
select_with_pattern(context, pattern, options)
}
}
fn select_without_pattern(context: &mut AppContext, options: &SelectOption) -> AppResult {
if options.all {
if let Some(curr_list) = context.tab_context_mut().curr_tab_mut().curr_list_mut() {
curr_list.iter_mut().for_each(|e| {
if options.reverse {
e.set_permanent_selected(false);
} else if options.toggle {
e.set_permanent_selected(!e.is_selected());
} else {
e.set_permanent_selected(true);
}
});
}
} else if let Some(entry) = context
.tab_context_mut()
.curr_tab_mut()
.curr_list_mut()
.and_then(|s| s.curr_entry_mut())
{
if options.reverse {
entry.set_permanent_selected(false);
} else if options.toggle {
entry.set_permanent_selected(!entry.is_selected());
} else {
entry.set_permanent_selected(true);
}
cursor_move::down(context, 1)?;
}
Ok(())
}
fn select_with_pattern(
context: &mut AppContext,
pattern: &MatchContext,
options: &SelectOption,
) -> AppResult {
if let Some(curr_list) = context.tab_context_mut().curr_tab_mut().curr_list_mut() {
let mut found = 0;
curr_list
.iter_mut()
.filter(|e| pattern.is_match(e.file_name()))
.for_each(|e| {
found += 1;
if options.reverse {
e.set_permanent_selected(false);
} else if options.toggle {
e.set_permanent_selected(!e.is_selected());
} else {
e.set_permanent_selected(true);
}
});
context
.message_queue_mut()
.push_info(format!("{} files selected", found));
}
Ok(())
}