use crate::action::*;
use crate::item::Item;
use crate::list::{Line, LineKind, List};
use clap::{ArgMatches, Command};
pub fn get_action() -> Action {
let name = String::from("remove");
let mut command = Command::new("remove")
.aliases(["rm"])
.about("Remove a task or tasks");
command = FileType::TodoTxt.add_args(command);
command = Outputter::add_args(command);
command = SearchTerms::add_args(command);
command = ConfirmationStatus::add_args(command);
Action { name, command }
}
pub fn execute(args: &ArgMatches) {
let todo_filename = FileType::TodoTxt.filename(args);
let list = List::from_url(todo_filename.clone())
.expect("Could not read todo list");
let mut outputter = Outputter::from_argmatches(args);
outputter.line_number_digits = list.lines.len().to_string().len();
let search_terms = SearchTerms::from_argmatches(args);
let confirmation = ConfirmationStatus::from_argmatches(args);
let (new_list, count) = remove_items_from_list(
list,
search_terms,
confirmation,
&mut outputter,
);
if count > 0 {
new_list.to_url(todo_filename);
outputter.write_status(format!("Removed {count} tasks!"));
} else {
outputter.write_status(String::from("No actions taken."));
}
}
pub fn remove_items_from_list(
list: List,
search_terms: SearchTerms,
confirmation: ConfirmationStatus,
outputter: &mut Outputter,
) -> (List, usize) {
let mut new_list = List::new();
let mut count = 0;
for line in list.lines {
match line.kind {
LineKind::Item => {
let item = line.item.clone().unwrap();
if search_terms.item_matches(&item)
&& check_if_delete(&item, outputter, confirmation)
{
count += 1;
new_list.lines.push(Line::new_blank());
} else {
new_list.lines.push(line);
}
}
_ => new_list.lines.push(line),
}
}
(new_list, count)
}
pub fn check_if_delete(
item: &Item,
outputter: &mut Outputter,
status: ConfirmationStatus,
) -> bool {
outputter.write_item(item);
status.check(outputter, "Remove?", "Removing", "Keeping")
}
#[cfg(test)]
mod tests {
use super::*;
use tempfile::tempdir;
#[test]
fn test_get_action() {
assert_eq!(String::from("remove"), get_action().name);
}
#[test]
fn test_check_if_delete() {
let dir = tempdir().unwrap();
let buffer_filename = dir
.path()
.join("buffer.txt")
.display()
.to_string();
let mut i = Item::new();
i.set_description(String::from("XYZ"));
let mut o = Outputter::new(9999);
o.colour = false;
o.io = Box::new(fs::File::create(buffer_filename.clone()).unwrap());
let r = check_if_delete(&i, &mut o, ConfirmationStatus::Yes);
assert_eq!(true, r);
let got_output = fs::read_to_string(buffer_filename.clone()).unwrap();
assert_eq!(String::from(" (?) XYZ\nRemoving\n\n"), got_output);
let mut o = Outputter::new(9999);
o.colour = false;
o.io = Box::new(fs::File::create(buffer_filename.clone()).unwrap());
let r = check_if_delete(&i, &mut o, ConfirmationStatus::No);
assert_eq!(false, r);
let got_output = fs::read_to_string(buffer_filename).unwrap();
assert_eq!(String::from(" (?) XYZ\nKeeping\n\n"), got_output);
}
#[test]
fn test_remove_items_from_list() {
let source_list = List {
lines: Vec::from([
Line::from_string(String::from("Foo1"), 0),
Line::from_string(String::from("x Foo2"), 0),
Line::from_string(String::from(""), 0),
Line::from_string(String::from("Bar"), 0),
]),
path: None,
};
let (got, count) = remove_items_from_list(
source_list,
SearchTerms {
terms: Vec::from([String::from("foo")]),
},
ConfirmationStatus::Yes,
&mut Outputter::new(1000),
);
assert_eq!(2, count);
assert_eq!(LineKind::Blank, got.lines[0].kind);
assert_eq!(LineKind::Blank, got.lines[1].kind);
assert_eq!(LineKind::Blank, got.lines[2].kind);
assert_eq!(LineKind::Item, got.lines[3].kind);
}
}