thor-wt 0.2.0

Worktree workflow commands for Thor
Documentation 
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

use thor_core::{find_repo, list_worktrees};
use std::process::Command;

/// Status of a pull operation on one worktree.
pub enum PullStatus {
    Ok,
    Failed(String),
    Skipped,
}

/// Result of pulling one worktree.
pub struct PullResult {
    pub branch: String,
    pub status: PullStatus,
}

/// Pull --rebase in one or all worktrees.
pub async fn pull(all: bool) -> anyhow::Result<Vec<PullResult>> {
    if !all {
        // Pull in current directory only
        let output = Command::new("git")
            .args(["pull", "--rebase"])
            .output()?;

        let branch = Command::new("git")
            .args(["branch", "--show-current"])
            .output()
            .map(|o| String::from_utf8_lossy(&o.stdout).trim().to_string())
            .unwrap_or_else(|_| "unknown".to_string());

        let status = if output.status.success() {
            PullStatus::Ok
        } else {
            PullStatus::Failed(String::from_utf8_lossy(&output.stderr).trim().to_string())
        };

        return Ok(vec![PullResult { branch, status }]);
    }

    // Pull all worktrees
    let repo = find_repo()?;
    let worktrees = list_worktrees(&repo).await?;
    let mut results = Vec::new();

    for wt in &worktrees {
        let branch = wt.display_name();

        // Check if has upstream by trying to resolve @{u}
        let has_upstream = Command::new("git")
            .args(["rev-parse", "--abbrev-ref", "@{u}"])
            .current_dir(&wt.path)
            .output()
            .map(|o| o.status.success())
            .unwrap_or(false);

        if !has_upstream {
            results.push(PullResult { branch, status: PullStatus::Skipped });
            continue;
        }

        let output = Command::new("git")
            .args(["pull", "--rebase"])
            .current_dir(&wt.path)
            .output()?;

        let status = if output.status.success() {
            PullStatus::Ok
        } else {
            PullStatus::Failed(String::from_utf8_lossy(&output.stderr).trim().to_string())
        };

        results.push(PullResult { branch, status });
    }

    Ok(results)
}