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
86
87
88
89
90
91
92
93
94
95
96
97
98
99

//! Rollover is an operation that copies incomplete
//! tasks from the latest devlog entry file to a new devlog entry file.

use crate::config::Config;
use crate::error::Error;
use crate::file::LogFile;
use crate::hook::{execute_hook, HookType};
use crate::path::LogPath;
use crate::task::{Task, TaskStatus};
use std::fs::OpenOptions;
use std::io::Write;
use std::path::Path;

/// Copies incomplete tasks from the latest devlog entry file
/// to a new devlog entry file with the next sequence number.
/// If available, the before-rollover and after-rollover hooks are invoked.
pub fn rollover<W: Write>(
    w: &mut W,
    config: &Config,
    p: &LogPath,
) -> Result<(LogPath, usize), Error> {
    let path = p.path();
    let next = p.next()?;
    let next_path = next.path();

    execute_hook(w, config, &HookType::BeforeRollover, &[path.as_os_str()])?;
    let tasks = load_carryover_tasks(path)?;
    create_new_logfile(&next_path, &tasks)?;
    execute_hook(
        w,
        config,
        &HookType::AfterRollover,
        &[path.as_os_str(), next_path.as_os_str()],
    )?;

    Ok((next, tasks.len()))
}

fn load_carryover_tasks(path: &Path) -> Result<Vec<Task>, Error> {
    let prev = LogFile::load(path)?;
    let mut tasks = Vec::new();
    prev.tasks().iter().for_each(|t| {
        if let TaskStatus::ToDo | TaskStatus::Started | TaskStatus::Blocked = t.status() {
            tasks.push(t.clone());
        }
    });
    Ok(tasks)
}

fn create_new_logfile(next_path: &Path, tasks: &[Task]) -> Result<(), Error> {
    let mut f = OpenOptions::new()
        .write(true)
        .create_new(true)
        .open(next_path)?;

    for t in tasks {
        write!(f, "{}\n", t)?;
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::repository::LogRepository;
    use tempfile::tempdir;

    #[test]
    fn test_rollover() {
        let mut out = Vec::new();
        let dir = tempdir().unwrap();
        let repo = LogRepository::new(dir.path());
        let config = Config::new(dir.path(), "");

        // Initialize the repository, which creates a single
        // logfile with three example tasks.
        repo.init().unwrap();
        let first_logpath = repo.latest().unwrap().unwrap();

        // Rollover, then check that only todo/started/blocked tasks
        // were imported into the new logfile
        let (new_logpath, num_imported) = rollover(&mut out, &config, &first_logpath).unwrap();
        assert_eq!(num_imported, 3);

        // Check tasks in the new logfile
        let logfile = LogFile::load(new_logpath.path()).unwrap();
        let task_statuses: Vec<TaskStatus> = logfile.tasks().iter().map(|t| t.status()).collect();
        assert_eq!(
            task_statuses,
            vec![TaskStatus::ToDo, TaskStatus::Started, TaskStatus::Blocked]
        );

        // Repo should contain two logfiles
        let mut paths = repo.list().unwrap();
        paths.sort();
        assert_eq!(paths, vec![first_logpath, new_logpath]);
    }
}