git-workon 0.1.1

A git plugin for managing worktrees
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

use std::path::PathBuf;

use miette::{Result, WrapErr};
use workon::{
    add_worktree, clone, get_default_branch_name, BranchType, CloneOptions, WorktreeDescriptor,
};

use crate::cli::Clone;
use crate::hooks::execute_post_create_hooks;
use crate::output;

use super::Run;

impl Run for Clone {
    fn run(&self) -> Result<Option<WorktreeDescriptor>> {
        let path = self.path.clone().unwrap_or_else(|| {
            PathBuf::from(
                self.url
                    .trim_end_matches('/')
                    .split('/')
                    .next_back()
                    .unwrap_or(".")
                    .trim_end_matches(".git"),
            )
        });

        // Phase 1: clone the repository with transfer progress
        let pb = output::create_spinner();
        pb.set_message("Cloning repository...");
        let pb_progress = pb.clone();
        let repo = clone(
            path,
            &self.url,
            CloneOptions {
                on_transfer_progress: Box::new(move |received, total, _bytes| {
                    if total > 0 {
                        pb_progress
                            .set_message(format!("Cloning... {}/{} objects", received, total));
                    }
                }),
            },
        )
        .wrap_err(format!("Failed to clone repository from {}", self.url))
        .inspect_err(|_| pb.finish_and_clear())?;
        pb.finish_and_clear();

        // Phase 2: create the initial worktree
        let pb = output::create_spinner();
        pb.set_message("Creating worktree...");
        let config = workon::WorkonConfig::new(&repo)?;
        let default_branch = get_default_branch_name(&repo, repo.find_remote("origin").ok())
            .wrap_err("Failed to determine default branch")
            .inspect_err(|_| pb.finish_and_clear())?;
        let worktree = add_worktree(&repo, &default_branch, BranchType::default(), None)
            .wrap_err(format!(
                "Failed to create worktree for default branch '{}'",
                default_branch
            ))
            .inspect_err(|_| pb.finish_and_clear())?;
        pb.finish_and_clear();

        // Execute post-create hooks after successful worktree creation
        if !self.no_hooks {
            if let Err(e) = execute_post_create_hooks(&worktree, None, &config) {
                output::warn(&format!("Post-create hook failed: {}", e));
                // Continue - worktree is still valid
            }
        }

        Ok(Some(worktree))
    }
}