tracc/src/timesheet.rs

use super::listview::ListView;
use itertools::Itertools;
use serde::{Deserialize, Serialize};
use serde_json::from_reader;
use std::{collections, default, fmt, fs, io};
use time::{Duration, OffsetDateTime, Time};

pub struct TimeSheet {
    pub times: Vec<TimePoint>,
    pub selected: usize,
    pub register: Option<TimePoint>,
}

const MAIN_PAUSE_TEXT: &str = "pause";
const PAUSE_TEXTS: [&str; 4] = [MAIN_PAUSE_TEXT, "lunch", "mittag", "break"];
const END_TEXT: &str = "end";

lazy_static! {
    static ref OVERRIDE_REGEX: regex::Regex = regex::Regex::new("\\[(.*)\\]").unwrap();
}

#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
pub struct TimePoint {
    text: String,
    time: Time,
}

impl TimePoint {
    pub fn new(text: &str) -> Self {
        Self {
            text: String::from(text),
            time: now(),
        }
    }
}

fn now() -> Time {
    let raw_time = OffsetDateTime::now_local().time();
    Time::try_from_hms(raw_time.hour(), raw_time.minute(), 0).unwrap()
}

impl fmt::Display for TimePoint {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "[{}] {}", self.time.format("%H:%M"), self.text)
    }
}

impl default::Default for TimePoint {
    fn default() -> Self {
        TimePoint::new("")
    }
}

fn read_times(path: &str) -> Option<Vec<TimePoint>> {
    fs::File::open(path)
        .ok()
        .map(io::BufReader::new)
        .and_then(|r| from_reader(r).ok())
}

/**
 * If a time text contains "[something]",
 * only use the message inside the brackets.
 */
fn effective_text(s: String) -> String {
    let text = OVERRIDE_REGEX
        .captures(&s)
        // index 0 is the entire string
        .and_then(|caps| caps.get(1))
        .map(|m| m.as_str())
        .unwrap_or(&s);
    if PAUSE_TEXTS.contains(&text) {
        MAIN_PAUSE_TEXT
    } else {
        text
    }.to_string()
}

impl TimeSheet {
    pub fn open_or_create(path: &str) -> Self {
        let times = read_times(path).unwrap_or_else(|| vec![TimePoint::new("start")]);
        let selected = times.len().saturating_sub(1);
        Self {
            times,
            selected,
            register: None,
        }
    }

    pub fn printable(&self) -> Vec<String> {
        self.times.iter().map(TimePoint::to_string).collect()
    }

    /**
     * Adjust the current time by `minutes` and round the result to a multiple of `minutes`.
     * This is so I can adjust in steps of 5 but still get nice, even numbers in the output.
     */
    pub fn shift_current(&mut self, minutes: i64) {
        let time = &mut self.times[self.selected].time;
        *time += Duration::minutes(minutes);
        *time -= Duration::minutes(time.minute() as i64 % 5);
        let timepoint = self.times[self.selected].clone();
        self.times.sort_by_key(|tp| tp.time);
        self.selected = self.times.iter().position(|tp| tp == &timepoint).unwrap();
    }

    fn current(&self) -> &TimePoint {
        &self.times[self.selected]
    }

    fn grouped_times(&self) -> collections::BTreeMap<String, Duration> {
        let last_time = self.times.last();
        self.times
            .iter()
            .chain(TimeSheet::maybe_end_time(last_time).iter())
            .tuple_windows()
            .map(|(prev, next)| (prev.text.clone(), next.time - prev.time))
            // Fold into a map to group by description.
            // I use a BTreeMap because I need a stable output order for the iterator
            // (otherwise the summary list will jump around on every input).
            .fold(collections::BTreeMap::new(), |mut map, (text, duration)| {
                *map.entry(effective_text(text))
                    .or_insert_with(Duration::zero) += duration;
                map
            })
    }

    fn maybe_end_time(last_time: Option<&TimePoint>) -> Option<TimePoint> {
        match last_time {
            Some(tp) if PAUSE_TEXTS.contains(&&tp.text[..]) => None,
            Some(tp) if tp.text == END_TEXT => None,
            Some(tp) if tp.time > now() => None,
            _ => Some(TimePoint::new(END_TEXT)),
        }
    }

    pub fn time_by_tasks(&self) -> String {
        self.grouped_times()
            .into_iter()
            .filter(|(text, _)| text != MAIN_PAUSE_TEXT)
            .map(|(text, duration)| format!("{}: {}", text, format_duration(&duration)))
            .join("\n")
    }

    pub fn sum_as_str(&self) -> String {
        let total = self.grouped_times()
            .into_iter()
            .filter(|(text, _)| text != MAIN_PAUSE_TEXT)
            .fold(Duration::zero(), |total, (_, d)| total + d);
        format_duration(&total)
    }

    pub fn pause_time(&self) -> String {
        let times = self.grouped_times();
        let duration = times
            .get(MAIN_PAUSE_TEXT)
            .map(Duration::clone)
            .unwrap_or_else(Duration::zero);
        format!("{}: {}", MAIN_PAUSE_TEXT, format_duration(&duration))
    }
}

fn format_duration(d: &Duration) -> String {
    format!("{}:{:02}", d.whole_hours(), d.whole_minutes() % 60)
}

impl ListView<TimePoint> for TimeSheet {
    fn selection_pointer(&mut self) -> &mut usize {
        &mut self.selected
    }

    fn list(&mut self) -> &mut Vec<TimePoint> {
        &mut self.times
    }

    fn register(&mut self) -> &mut Option<TimePoint> {
        &mut self.register
    }

    fn normal_mode(&mut self) {
        if self.current().text.is_empty() {
            self.remove_current();
            self.selected = self.selected.saturating_sub(1);
        }
        self.times.sort_by_key(|t| t.time);
    }

    fn append_to_current(&mut self, chr: char) {
        self.times[self.selected].text.push(chr);
    }

    fn backspace(&mut self) {
        self.times[self.selected].text.pop();
    }
}