From 3884d7319c7615d12b6c2250f612c5d12f24219b Mon Sep 17 00:00:00 2001
From: Matthias Beyer <mail@beyermatthias.de>
Date: Fri, 17 May 2019 22:28:42 +0200
Subject: [PATCH] Add duration listing in imag-timetrack

This patch adds the -d / --show-duration feature to the "list"
subcommand, which makes imag-timetrack list the duration of the
timetracking in the output.

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>
---
 bin/domain/imag-timetrack/src/list.rs | 48 ++++++++++++++++++++++-----
 bin/domain/imag-timetrack/src/main.rs |  2 +-
 bin/domain/imag-timetrack/src/ui.rs   |  8 +++++
 3 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/bin/domain/imag-timetrack/src/list.rs b/bin/domain/imag-timetrack/src/list.rs
index 8df8f7e5..55002e23 100644
--- a/bin/domain/imag-timetrack/src/list.rs
+++ b/bin/domain/imag-timetrack/src/list.rs
@@ -78,14 +78,16 @@ pub fn list(rt: &Runtime) -> i32 {
     let end   = gettime(&cmd, "end-time");
 
     let list_not_ended = cmd.is_present("list-not-ended");
+    let show_duration  = cmd.is_present("show-duration");
 
-    list_impl(rt, start, end, list_not_ended)
+    list_impl(rt, start, end, list_not_ended, show_duration)
 }
 
 pub fn list_impl(rt: &Runtime,
                  start: Option<NaiveDateTime>,
                  end: Option<NaiveDateTime>,
-                 list_not_ended: bool)
+                 list_not_ended: bool,
+                 show_duration: bool)
     -> i32
 {
 
@@ -119,7 +121,12 @@ pub fn list_impl(rt: &Runtime,
     let filter = start_time_filter.and(end_time_filter);
 
     let mut table = Table::new();
-    table.set_titles(Row::new(["Tag", "Start", "End"].into_iter().map(|s| Cell::new(s)).collect()));
+    let title_row = if !show_duration {
+        Row::new(["Tag", "Start", "End"].iter().map(|s| Cell::new(s)).collect())
+    } else {
+        Row::new(["Tag", "Start", "End", "Duration"].iter().map(|s| Cell::new(s)).collect())
+    };
+    table.set_titles(title_row);
 
     let mut table_empty = true;
 
@@ -142,20 +149,45 @@ pub fn list_impl(rt: &Runtime,
                 debug!(" -> end = {:?}", end);
 
                 let v = match (start, end) {
-                    (None, _)          => vec![String::from(tag.as_str()), String::from(""), String::from("")],
+                    (None, _)          => {
+                        let mut v = vec![String::from(tag.as_str()), String::from(""), String::from("")];
+                        if show_duration {
+                            v.push(String::from(""));
+                        }
+                        v
+                    },
                     (Some(s), None)    => {
-                        vec![
+                        let mut v = vec![
                             String::from(tag.as_str()),
                             format!("{}", s),
                             String::from(""),
-                        ]
+                        ];
+
+                        if show_duration {
+                            v.push(String::from(""));
+                        }
+
+                        v
                     },
                     (Some(s), Some(e)) => {
-                        vec![
+                        let mut v = vec![
                             String::from(tag.as_str()),
                             format!("{}", s),
                             format!("{}", e),
-                        ]
+                        ];
+
+                        if show_duration {
+                            let duration = e - s;
+                            let dur = format!("{days} Days, {hours} Hours, {minutes} Minutes, {seconds} Seconds",
+                                    days    = duration.num_days(),
+                                    hours   = duration.num_hours(),
+                                    minutes = duration.num_minutes(),
+                                    seconds = duration.num_seconds());
+
+                            v.push(dur);
+                        }
+
+                        v
                     },
                 };
 
diff --git a/bin/domain/imag-timetrack/src/main.rs b/bin/domain/imag-timetrack/src/main.rs
index 7ae23d9e..af7f1414 100644
--- a/bin/domain/imag-timetrack/src/main.rs
+++ b/bin/domain/imag-timetrack/src/main.rs
@@ -107,7 +107,7 @@ fn main() {
     } else {
         let start = ::chrono::offset::Local::today().naive_local().and_hms(0, 0, 0);
         let end   = ::chrono::offset::Local::today().naive_local().and_hms(23, 59, 59);
-        list_impl(&rt, Some(start), Some(end), false)
+        list_impl(&rt, Some(start), Some(end), false, false)
     };
 
     ::std::process::exit(retval);
diff --git a/bin/domain/imag-timetrack/src/ui.rs b/bin/domain/imag-timetrack/src/ui.rs
index 16ece5f9..a654a819 100644
--- a/bin/domain/imag-timetrack/src/ui.rs
+++ b/bin/domain/imag-timetrack/src/ui.rs
@@ -46,6 +46,14 @@ pub fn build_ui<'a>(app: App<'a, 'a>) -> App<'a, 'a> {
                         .multiple(false)
                         .required(false)
                         .help("List not yet ended timetrackings even if after 'end-time'"))
+
+                   .arg(Arg::with_name("show-duration")
+                        .short("d")
+                        .long("show-duration")
+                        .takes_value(false)
+                        .multiple(false)
+                        .required(false)
+                        .help("Show durations of ended timetrackings"))
                    )
 
        .subcommand(SubCommand::with_name("start")