1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

extern crate chrono;

pub mod utils;
pub mod session;

use session::{Session, SessionStatus};

use std::fs::File;
use std::io::Read;
use std::path::Path;

const TEMPUS_DIR_NAME: &str = "/tempus/";
const SESSION_NAME: &str = ".session";
const TEMPUS_LOG_NAME: &str = "tempus_log.txt";

fn get_project_dir_path(project: &str) -> String {
    format!("{}/{}/{}", utils::get_home_dir(), &TEMPUS_DIR_NAME, &project)
}

pub fn calc_total_log_time(project: &str) {
    let log_file_path_str = format!("{}/{}", get_project_dir_path(project), &TEMPUS_LOG_NAME);
    let log_file_path = Path::new(&log_file_path_str);

    let mut file = match File::open(&log_file_path) {
        Ok(file) => file,
        Err(e) => {
            eprintln!("error opening {}: {}", log_file_path.display(), e);
            std::process::exit(1);
        }
    };

    let mut contents = String::new();
    if let Err(e) = file.read_to_string(&mut contents) {
        eprintln!("error reading {}: {}", log_file_path.display(), e);
        std::process::exit(1);
    }

    let mut total_length_hours = 0.0;

    for line in contents.split('\n') {
        let times: Vec<&str> = line.split(',').collect();

        if times.len() < 2 {
            break;
        }

        let start = utils::datetime_from_str(times[0]);
        let end = utils::datetime_from_str(times[1]);

        total_length_hours += utils::get_length_hours(&start, &end);
    }

    println!("{:.3}", total_length_hours);
}

pub fn do_session(project: &str) {
    let project_dir_path = get_project_dir_path(project);
    utils::create_dir(&project_dir_path);

    let mut session = Session::new(&project_dir_path, SESSION_NAME);
    match session.status {
        SessionStatus::Started(start_time) => {
            let end_time = session.end();
            session.record(&TEMPUS_LOG_NAME);

            let length_hours = format!("{:.3}", utils::get_length_hours(&start_time, &end_time));

            println!("{} session ended: {} hours.", &project, &length_hours);
        },
        SessionStatus::NotStarted => {
            let start_time = session.start();
            println!("{} session started at {}.", &project, utils::format_datetime(&start_time));
        }
    };
}

pub fn print_session_start(project: &str) {
    let project_dir_path = get_project_dir_path(project);

    let session = Session::new(&project_dir_path, SESSION_NAME);
    match session.status {
        SessionStatus::Started(start_time) => println!("{}", utils::format_datetime(&start_time)),
        SessionStatus::NotStarted => eprintln!("No session started for {}", project),
    }
}