tally-todo 0.5.2

Make TODO management a little more automatic
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
100
101

use anyhow::{Result, anyhow};
use std::{env, fs};
use std::path::PathBuf;
use dirs;

pub fn find_project_root() -> Result<PathBuf> {
    let mut current = env::current_dir()?;

    loop {
        let tally_dir = current.join(".tally");
        if tally_dir.exists() && tally_dir.is_dir() {
            return Ok(current);
        }

        // Try to move to parent directory
        match current.parent() {
            Some(parent) => current = parent.to_path_buf(),
            None => {
                return Err(anyhow!(
                    "No .tally/ directory found. use 'tally init' to create a new one"
                ));
            }
        }
    }
}

pub struct ProjectPaths {
    pub todo_file: PathBuf,
    pub history_file: PathBuf,
    pub config_file: PathBuf,
    pub hooks_dir: PathBuf,
    pub ignore_file: PathBuf,
    pub tally_dir: PathBuf,
    pub root: PathBuf,
}

impl ProjectPaths {
    /// Get paths for the current project
    pub fn get_paths() -> Result<Self> {
        let root = find_project_root()?;
        let tally_dir = root.join(".tally");

        let config_file: PathBuf;
        let conf = tally_dir.join("config.toml");


        if conf.exists() {
            config_file = conf;
        } else {
            let config_dir = dirs::config_dir().unwrap().join("tally");
            config_file = config_dir.join("config.toml");
        }

        Ok(Self {
            todo_file: root.join("TODO.md"),
            history_file: tally_dir.join("history.json"),
            config_file: config_file,
            ignore_file: tally_dir.join("ignore"),
            hooks_dir: tally_dir.join("hooks"),
            tally_dir,
            root,
        })
    }

    /// Initialize a new project
    pub fn init_here() -> Result<Self> {
        let root = env::current_dir()?;
        let tally_dir = root.join(".tally");
        let hooks_dir = tally_dir.join("hooks");

        if tally_dir.exists() {
            return Err(anyhow!(
                "Project already initialized at {}",
                tally_dir.display()
            ));
        }

        let config_file: PathBuf;
        let conf = tally_dir.join("config.toml");

        if conf.exists() {
            config_file = conf;
        } else {
            let config_dir = dirs::config_dir().unwrap().join("tally");
            config_file = config_dir.join("config.toml");
        }

        std::fs::create_dir_all(&tally_dir)?;
        std::fs::create_dir_all(&hooks_dir)?;

        Ok(Self {
            todo_file: root.join("TODO.md"),
            history_file: tally_dir.join("history.json"),
            config_file: config_file,
            ignore_file: tally_dir.join("ignore"),
            hooks_dir: tally_dir.join("hooks"),
            tally_dir,
            root,
        })
    }
}