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
102
103
104
105
106

use anyhow::Result;
use std::fs;
use std::path::Path;
use std::process::Command;

use crate::utils::project_paths::ProjectPaths;

/// Embedded git hook scripts
pub const POST_COMMIT: &str = include_str!("scripts/post-commit");

fn set_git_hooks_path(repo_path: &Path, hooks_path: &Path) -> std::io::Result<()> {
    let status = Command::new("git")
        .arg("config")
        .arg("core.hooksPath")
        .arg(hooks_path)
        .current_dir(repo_path)
        .status()?; // run in the repo directory

    if !status.success() {
        Err(std::io::Error::new(
            std::io::ErrorKind::Other,
            "Failed to set core.hooksPath",
        ))
    } else {
        Ok(())
    }
}

/// Install git hooks into .git/hooks/
pub fn install_hooks() -> Result<()> {
    let paths = ProjectPaths::get_paths()?;
    let hooks_dir = paths.hooks_dir;

    fs::create_dir_all(&hooks_dir)?;

    install_hook(&hooks_dir, "post-commit", POST_COMMIT)?;

    set_git_hooks_path(&paths.root.join(".git"), &hooks_dir)?;

    Ok(())
}

/// Install a single hook, backing up if exists
fn install_hook(hooks_dir: &Path, name: &str, content: &str) -> Result<()> {
    let hook_path = hooks_dir.join(name);

    // Backup existing hook if present
    if hook_path.exists() {
        let backup_path = hooks_dir.join(format!("{}.tally-backup", name));
        fs::copy(&hook_path, backup_path)?;
    }

    // Write new hook
    fs::write(&hook_path, content)?;

    // Make executable (Unix only)
    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        let mut perms = fs::metadata(&hook_path)?.permissions();
        perms.set_mode(0o755);
        fs::set_permissions(&hook_path, perms)?;
    }

    Ok(())
}

/// Uninstall git hooks
pub fn uninstall_hooks() -> Result<()> {
    let hooks_dir = ProjectPaths::get_paths()?.hooks_dir;

    uninstall_hook(&hooks_dir, "post-commit")?;

    Ok(())
}

fn uninstall_hook(hooks_dir: &Path, name: &str) -> Result<()> {
    let hook_path = hooks_dir.join(name);

    fs::remove_file(&hook_path)?;

    Ok(())
}


/// Update hooks (rewrite with latest version)
pub fn update_hooks() -> Result<()> {
    let hooks_dir = ProjectPaths::get_paths()?.hooks_dir;

    fs::write(hooks_dir.join("post-commit"), POST_COMMIT)?;

    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        for name in &[
            "post-commit",
        ] {
            let path = hooks_dir.join(name);
            let mut perms = fs::metadata(&path)?.permissions();
            perms.set_mode(0o755);
            fs::set_permissions(&path, perms)?;
        }
    }

    Ok(())
}