2017-07-20 21:21:57 +00:00
|
|
|
//
|
|
|
|
// imag - the personal information management suite for the commandline
|
2018-02-07 01:48:53 +00:00
|
|
|
// Copyright (C) 2015-2018 Matthias Beyer <mail@beyermatthias.de> and contributors
|
2017-07-20 21:21:57 +00:00
|
|
|
//
|
|
|
|
// 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
|
|
|
|
//
|
|
|
|
|
2018-01-13 13:31:24 +00:00
|
|
|
use chrono::NaiveDateTime;
|
2017-07-20 21:21:57 +00:00
|
|
|
use filters::filter::Filter;
|
2018-01-13 02:19:41 +00:00
|
|
|
use prettytable::Table;
|
|
|
|
use prettytable::row::Row;
|
|
|
|
use prettytable::cell::Cell;
|
2018-02-11 20:25:33 +00:00
|
|
|
use kairos::parser::Parsed;
|
|
|
|
use kairos::parser::parse as kairos_parse;
|
|
|
|
use clap::ArgMatches;
|
2017-07-20 21:21:57 +00:00
|
|
|
|
|
|
|
use libimagerror::trace::trace_error;
|
|
|
|
use libimagerror::trace::MapErrTrace;
|
|
|
|
use libimagerror::iter::TraceIterator;
|
|
|
|
use libimagstore::store::FileLockEntry;
|
2018-02-11 18:58:53 +00:00
|
|
|
use libimagtimetrack::error::TimeTrackError;
|
2017-08-27 12:56:37 +00:00
|
|
|
use libimagtimetrack::timetrackingstore::TimeTrackStore;
|
|
|
|
use libimagtimetrack::timetracking::TimeTracking;
|
2018-01-13 02:19:41 +00:00
|
|
|
use libimagtimetrack::error::Result;
|
2017-07-20 21:21:57 +00:00
|
|
|
|
|
|
|
use libimagrt::runtime::Runtime;
|
|
|
|
|
|
|
|
pub fn list(rt: &Runtime) -> i32 {
|
|
|
|
let (_, cmd) = rt.cli().subcommand();
|
|
|
|
let cmd = cmd.unwrap(); // checked in main()
|
|
|
|
|
2018-02-11 20:25:33 +00:00
|
|
|
let gettime = |cmd: &ArgMatches, name| {
|
|
|
|
match cmd.value_of(name).map(kairos_parse) {
|
|
|
|
Some(Ok(Parsed::TimeType(tt))) => match tt.calculate() {
|
|
|
|
Ok(tt) => {
|
|
|
|
let dt = tt.get_moment().unwrap_or_else(|| {
|
|
|
|
error!("Failed to get date from '{}'", cmd.value_of(name).unwrap());
|
|
|
|
::std::process::exit(1)
|
|
|
|
});
|
|
|
|
|
|
|
|
Some(dt.clone())
|
|
|
|
},
|
|
|
|
Err(e) => {
|
|
|
|
error!("Failed to calculate date from '{}': {:?}",
|
|
|
|
cmd.value_of(name).unwrap(), e);
|
|
|
|
::std::process::exit(1)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
Some(Ok(Parsed::Iterator(_))) => {
|
|
|
|
error!("Expected single point in time, got '{}', which yields a list of dates", cmd.value_of(name).unwrap());
|
|
|
|
::std::process::exit(1)
|
|
|
|
},
|
|
|
|
Some(Err(e)) => {
|
|
|
|
trace_error(&e);
|
|
|
|
::std::process::exit(1)
|
|
|
|
}
|
|
|
|
None => None,
|
2017-07-20 21:21:57 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2018-02-11 20:25:33 +00:00
|
|
|
let start = gettime(&cmd, "start-time");
|
|
|
|
let end = gettime(&cmd, "end-time");
|
|
|
|
|
2017-07-20 21:21:57 +00:00
|
|
|
let list_not_ended = cmd.is_present("list-not-ended");
|
|
|
|
|
2018-01-13 13:31:24 +00:00
|
|
|
list_impl(rt, start, end, list_not_ended)
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn list_impl(rt: &Runtime,
|
|
|
|
start: Option<NaiveDateTime>,
|
|
|
|
end: Option<NaiveDateTime>,
|
|
|
|
list_not_ended: bool)
|
|
|
|
-> i32
|
|
|
|
{
|
|
|
|
|
2017-07-20 21:21:57 +00:00
|
|
|
let start_time_filter = |timetracking: &FileLockEntry| {
|
|
|
|
start.map(|s| match timetracking.get_start_datetime() {
|
|
|
|
Ok(Some(dt)) => dt >= s,
|
|
|
|
Ok(None) => {
|
|
|
|
warn!("Funny things are happening: Timetracking has no start time");
|
|
|
|
false
|
|
|
|
}
|
|
|
|
Err(e) => {
|
|
|
|
trace_error(&e);
|
|
|
|
false
|
|
|
|
}
|
|
|
|
})
|
|
|
|
.unwrap_or(true)
|
|
|
|
};
|
|
|
|
|
|
|
|
let end_time_filter = |timetracking: &FileLockEntry| {
|
2017-08-30 21:19:47 +00:00
|
|
|
end.map(|s| match timetracking.get_end_datetime() {
|
2017-07-20 21:21:57 +00:00
|
|
|
Ok(Some(dt)) => dt <= s,
|
|
|
|
Ok(None) => list_not_ended,
|
|
|
|
Err(e) => {
|
|
|
|
trace_error(&e);
|
|
|
|
false
|
|
|
|
}
|
|
|
|
})
|
|
|
|
.unwrap_or(true)
|
|
|
|
};
|
|
|
|
|
|
|
|
let filter = start_time_filter.and(end_time_filter);
|
|
|
|
|
2018-01-13 02:19:41 +00:00
|
|
|
let mut table = Table::new();
|
|
|
|
table.set_titles(Row::new(["Tag", "Start", "End"].into_iter().map(|s| Cell::new(s)).collect()));
|
|
|
|
|
2017-07-20 21:21:57 +00:00
|
|
|
rt.store()
|
|
|
|
.get_timetrackings()
|
2018-02-11 18:58:53 +00:00
|
|
|
.map_err_trace_exit_unwrap(1)
|
|
|
|
.trace_unwrap()
|
|
|
|
.filter(|e| filter.filter(e))
|
|
|
|
.fold(Ok(table), |acc: Result<_>, e| {
|
|
|
|
acc.and_then(|mut tab: Table| {
|
|
|
|
debug!("Processing {:?}", e.get_location());
|
|
|
|
|
|
|
|
let tag = e.get_timetrack_tag()?;
|
|
|
|
debug!(" -> tag = {:?}", tag);
|
|
|
|
|
|
|
|
let start = e.get_start_datetime()?;
|
|
|
|
debug!(" -> start = {:?}", start);
|
|
|
|
|
|
|
|
let end = e.get_end_datetime()?;
|
|
|
|
debug!(" -> end = {:?}", end);
|
|
|
|
|
|
|
|
let v = match (start, end) {
|
|
|
|
(None, _) => vec![String::from(tag.as_str()), String::from(""), String::from("")],
|
|
|
|
(Some(s), None) => {
|
|
|
|
vec![
|
|
|
|
String::from(tag.as_str()),
|
|
|
|
format!("{}", s),
|
|
|
|
String::from(""),
|
|
|
|
]
|
|
|
|
},
|
|
|
|
(Some(s), Some(e)) => {
|
|
|
|
vec![
|
|
|
|
String::from(tag.as_str()),
|
|
|
|
format!("{}", s),
|
|
|
|
format!("{}", e),
|
|
|
|
]
|
|
|
|
},
|
|
|
|
};
|
|
|
|
|
|
|
|
let cells : Vec<Cell> = v
|
|
|
|
.into_iter()
|
|
|
|
.map(|s| Cell::new(&s))
|
|
|
|
.collect();
|
|
|
|
tab.add_row(Row::new(cells));
|
|
|
|
|
|
|
|
Ok(tab)
|
|
|
|
})
|
2017-07-20 21:21:57 +00:00
|
|
|
})
|
2018-02-11 18:58:53 +00:00
|
|
|
.map_err_trace_exit_unwrap(1)
|
2018-02-26 21:57:17 +00:00
|
|
|
.print(&mut rt.stdout())
|
2018-02-11 18:58:53 +00:00
|
|
|
.map_err(|_| TimeTrackError::from(String::from("Failed printing table")))
|
2017-07-20 21:21:57 +00:00
|
|
|
.map(|_| 0)
|
|
|
|
.map_err_trace()
|
|
|
|
.unwrap_or(1)
|
|
|
|
}
|
|
|
|
|