From 29ab99339700c44a1fa761c815ce10c0093663b8 Mon Sep 17 00:00:00 2001
From: "Dustin J. Mitchell" <dustin@mozilla.com>
Date: Sun, 29 Nov 2020 11:23:19 -0500
Subject: [PATCH] add done

---
 cli/src/cmd/done.rs | 51 +++++++++++++++++++++++++++++++++++++++++++++
 cli/src/cmd/list.rs |  4 ++++
 cli/src/cmd/mod.rs  |  2 ++
 3 files changed, 57 insertions(+)
 create mode 100644 cli/src/cmd/done.rs

diff --git a/cli/src/cmd/done.rs b/cli/src/cmd/done.rs
new file mode 100644
index 000000000..713a747a8
--- /dev/null
+++ b/cli/src/cmd/done.rs
@@ -0,0 +1,51 @@
+use clap::{App, ArgMatches, SubCommand as ClapSubCommand};
+use failure::Fallible;
+use taskchampion::Status;
+
+use crate::cmd::{shared, ArgMatchResult, CommandInvocation};
+
+#[derive(Debug)]
+struct Invocation {
+    task: String,
+}
+
+define_subcommand! {
+    fn decorate_app<'a>(&self, app: App<'a, 'a>) -> App<'a, 'a> {
+        app.subcommand(
+            ClapSubCommand::with_name("done")
+                .about("finish the given task (status Completed)")
+                .arg(shared::task_arg()))
+    }
+
+    fn arg_match<'a>(&self, matches: &ArgMatches<'a>) -> ArgMatchResult {
+        match matches.subcommand() {
+            ("done", Some(matches)) => ArgMatchResult::Ok(Box::new(Invocation {
+                task: matches.value_of("task").unwrap().into(),
+            })),
+            _ => ArgMatchResult::None,
+        }
+    }
+}
+
+subcommand_invocation! {
+    fn run(&self, command: &CommandInvocation) -> Fallible<()> {
+        let mut replica = command.get_replica()?;
+        let task = shared::get_task(&mut replica, &self.task)?;
+        let mut task = task.into_mut(&mut replica);
+        task.stop()?;
+        task.set_status(Status::Completed)?;
+        Ok(())
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn parse_command() {
+        with_subcommand_invocation!(vec!["task", "done", "1"], |inv: &Invocation| {
+            assert_eq!(inv.task, "1".to_string());
+        });
+    }
+}
diff --git a/cli/src/cmd/list.rs b/cli/src/cmd/list.rs
index 4dddd5265..7b1b5078d 100644
--- a/cli/src/cmd/list.rs
+++ b/cli/src/cmd/list.rs
@@ -2,6 +2,7 @@ use crate::table;
 use clap::{App, ArgMatches, SubCommand as ClapSubCommand};
 use failure::Fallible;
 use prettytable::{cell, row, Table};
+use taskchampion::Status;
 
 use crate::cmd::{ArgMatchResult, CommandInvocation};
 
@@ -28,6 +29,9 @@ subcommand_invocation! {
         t.set_format(table::format());
         t.set_titles(row![b->"id", b->"act", b->"description"]);
         for (uuid, task) in replica.all_tasks().unwrap() {
+            if task.get_status() != Status::Pending {
+                continue;
+            }
             let mut id = uuid.to_string();
             if let Some(i) = replica.get_working_set_index(&uuid)? {
                 id = i.to_string();
diff --git a/cli/src/cmd/mod.rs b/cli/src/cmd/mod.rs
index d58237816..c7b992b5c 100644
--- a/cli/src/cmd/mod.rs
+++ b/cli/src/cmd/mod.rs
@@ -7,6 +7,7 @@ mod shared;
 
 mod add;
 mod debug;
+mod done;
 mod gc;
 mod info;
 mod list;
@@ -21,6 +22,7 @@ pub(crate) fn subcommands() -> Vec<Box<dyn SubCommand>> {
     vec![
         add::cmd(),
         debug::cmd(),
+        done::cmd(),
         gc::cmd(),
         info::cmd(),
         list::cmd(),