diff options
Diffstat (limited to 'bin/domain/imag-timetrack/src/track.rs')
-rw-r--r-- | bin/domain/imag-timetrack/src/track.rs | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/bin/domain/imag-timetrack/src/track.rs b/bin/domain/imag-timetrack/src/track.rs new file mode 100644 index 00000000..a723b909 --- /dev/null +++ b/bin/domain/imag-timetrack/src/track.rs @@ -0,0 +1,75 @@ +// +// imag - the personal information management suite for the commandline +// Copyright (C) 2015, 2016 Matthias Beyer <mail@beyermatthias.de> and contributors +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; version +// 2.1 of the License. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +// + +use std::str::FromStr; + +use clap::ArgMatches; +use chrono::naive::NaiveDateTime; + +use libimagrt::runtime::Runtime; +use libimagerror::trace::trace_error; +use libimagentrytimetrack::tag::TimeTrackingTag; +use libimagentrytimetrack::timetrackingstore::TimeTrackStore; +use libimagerror::trace::MapErrTrace; + +pub fn track(rt: &Runtime) -> i32 { + let (_, cmd) = rt.cli().subcommand(); + let cmd = cmd.unwrap(); // checked in main() + + // Gets the appropriate time from the commandline or None on error (errors already logged, so + // callee can directly return in case of error + fn get_time(cmd: &ArgMatches, clap_name: &str, errname: &str) -> Option<NaiveDateTime> { + let val = cmd + .value_of(clap_name) + .map(::chrono::naive::NaiveDateTime::from_str) + .unwrap(); // clap has our back + + match val { + Ok(ndt) => Some(ndt), + Err(e) => { + trace_error(&e); + error!("Cannot continue, not having {} time", errname); + None + }, + } + } + + let start = match get_time(&cmd, "start-time", "start") { + Some(t) => t, + None => return 1, + }; + + let stop = match get_time(&cmd, "stop-time", "stop") { + Some(t) => t, + None => return 1, + }; + + cmd.values_of("tags") + .unwrap() // enforced by clap + .map(String::from) + .map(TimeTrackingTag::from) + .fold(0, |acc, ttt| { + rt.store() + .create_timetracking(&start, &stop, &ttt) + .map_err_trace() + .map(|_| acc) + .unwrap_or(1) + }) +} + |