Create rust bins

create-worktree/Cargo.toml

@@ -0,0 +1,14 @@
+[package]
+name = "create-worktree"
+version = "0.1.0"
+edition = "2024"
+authors = ["Noah Knegt <noah.knegt@priva.com>"]
+description = "A tool to create git worktrees with convenient branch management"
+
+[dependencies]
+clap = { version = "4.5", features = ["derive"] }
+anyhow = "1.0"
+colored = "3.0"
+indicatif = "0.17"
+dirs = "6.0"
+thiserror = { version = "2.0" }

create-worktree/src/main.rs

@@ -0,0 +1,231 @@
+use std::env;
+use std::path::Path;
+use std::process::{Command, Stdio};
+
+use anyhow::{Context, Result};
+use clap::{ArgAction, Parser};
+use colored::*;
+use indicatif::{ProgressBar, ProgressStyle};
+
+/// Tool to create git worktrees with convenient branch management
+#[derive(Parser, Debug)]
+#[command(author, version, about)]
+struct Args {
+    /// Path for the new worktree
+    worktree: String,
+
+    /// Branch to create
+    #[arg(short, long)]
+    branch: Option<String>,
+
+    /// Base branch to use for the new worktree
+    #[arg(short = 'B', long, default_value = "origin/main")]
+    base: String,
+
+    /// Prefix to apply to the branch name
+    #[arg(short, long, default_value = "")]
+    prefix: String,
+
+    /// Do not create an upstream branch
+    #[arg(short = 'N', long, action = ArgAction::SetFalse)]
+    no_create_upstream: bool,
+
+    /// Enable verbose output
+    #[arg(short, long)]
+    verbose: bool,
+
+    /// Disable colored output
+    #[arg(long)]
+    no_color: bool,
+}
+
+/// Git operations error type
+#[derive(Debug, thiserror::Error)]
+enum GitError {
+    #[error("Command failed with exit code: {0}")]
+    Failed(i32),
+
+    #[error("Command failed without exit code")]
+    FailedNoCode,
+
+    #[error("Failed to execute command: {0}")]
+    ExecutionError(#[from] std::io::Error),
+}
+
+/// Runs a git command with a progress spinner
+fn run_command(command: &mut Command, message: &str) -> Result<()> {
+    let spinner = ProgressBar::new_spinner();
+    spinner.set_style(
+        ProgressStyle::default_spinner()
+            .tick_chars("⣾⣽⣻⢿⡿⣟⣯⣷")
+            .template("{spinner:.green} {msg}")
+            .expect("Invalid template format"),
+    );
+    spinner.set_message(message.to_string());
+
+    // Configure the command to not show output
+    command.stdout(Stdio::null()).stderr(Stdio::null());
+
+    // Execute the command and wait for it to complete
+    spinner.enable_steady_tick(std::time::Duration::from_millis(100));
+
+    let status = command.status().context("Failed to execute command")?;
+    spinner.finish_and_clear();
+
+    if status.success() {
+        println!("{message} {}", "Done.".green());
+        Ok(())
+    } else {
+        println!("{message} {}", "FAILED.".red());
+
+        let code = status.code();
+        match code {
+            Some(code) => Err(GitError::Failed(code).into()),
+            None => Err(GitError::FailedNoCode.into()),
+        }
+    }
+}
+
+/// Git command wrapper
+struct Git;
+
+impl Git {
+    /// Check if a branch exists locally
+    fn branch_exists_locally(branch: &str) -> Result<bool> {
+        let output = Command::new("git")
+            .args([
+                "branch",
+                "--list",
+                branch
+            ])
+            .output()
+            .context("Failed to check branch existence")?;
+
+        Ok(!output.stdout.is_empty())
+    }
+
+    /// Check if a branch exists on the remote
+    fn branch_exists_on_remote(branch: &str) -> Result<bool> {
+        let output = Command::new("git")
+            .args([
+                "ls-remote",
+                "--heads",
+                "origin",
+                branch
+            ])
+            .output()
+            .context("Failed to check remote branch existence")?;
+
+        Ok(!output.stdout.is_empty())
+    }
+
+    /// Create a new worktree with an existing branch
+    fn create_worktree_existing_branch(worktree_path: &str, branch: &str) -> Result<()> {
+        let mut cmd = Command::new("git");
+        cmd.args([
+            "worktree",
+            "add",
+            worktree_path,
+            branch
+        ]);
+        run_command(&mut cmd, &format!("Generating new worktree from existing branch: {branch}"))
+    }
+
+    /// Create a new worktree with a new branch
+    fn create_worktree_new_branch(worktree_path: &str, branch: &str, base: &str) -> Result<()> {
+        let mut cmd = Command::new("git");
+        cmd.args([
+            "worktree",
+            "add",
+            "-b", branch,
+            worktree_path,
+            base
+        ]);
+        run_command(&mut cmd, &format!("Generating new worktree: {worktree_path}"))
+    }
+
+    /// Create and push a new remote branch
+    fn create_remote_branch(branch: &str) -> Result<()> {
+        let mut cmd = Command::new("git");
+        cmd.args([
+            "push",
+            "-u", "origin",
+            branch
+        ]);
+        run_command(&mut cmd, &format!("Creating remote branch {branch}..."))
+    }
+
+    /// Set the upstream branch
+    fn set_upstream_branch(branch: &str) -> Result<()> {
+        let mut cmd = Command::new("git");
+        cmd.args([
+            "branch",
+            "--set-upstream-to", &format!("origin/{branch}")
+        ]);
+        run_command(&mut cmd, &format!("Setting upstream branch to 'origin/{branch}'"))
+    }
+}
+
+/// Update or create the remote tracking branch
+fn update_remote(branch: &str, create_upstream: bool) -> Result<()> {
+    // Do nothing if create_upstream is disabled
+    if !create_upstream {
+        return Ok(());
+    }
+
+    // Check if branch exists on remote
+    let has_remote = Git::branch_exists_on_remote(branch)?;
+
+    if !has_remote {
+        // Create remote branch
+        println!("{}", format!("Branch '{branch}' does not exist on remote. Creating.").dimmed());
+        Git::set_upstream_branch(branch)?;
+        // Git::create_remote_branch(branch)?;
+    } else {
+        println!("{}", format!("Branch '{branch}' exists. Setting upstream.").dimmed());
+        Git::set_upstream_branch(branch)?;
+    }
+
+    Ok(())
+}
+
+fn main() -> Result<()> {
+    // Parse arguments
+    let args = Args::parse();
+
+    // Enable or disable colored output
+    colored::control::set_override(!args.no_color);
+
+    // Print verbose information if enabled
+    if args.verbose {
+        println!("{}", "Verbose mode enabled".dimmed());
+        println!("{}", format!("Base branch: {}", args.base).dimmed());
+        println!("{}", format!("Create upstream: {}", args.no_create_upstream).dimmed());
+    }
+
+    // Determine branch name if not specified
+    let branch = args.branch.unwrap_or_else(|| format!("{}{}", args.prefix, args.worktree));
+
+    // Normalize paths
+    let worktree_path = Path::new(&args.worktree).to_string_lossy();
+
+    // Check if branch already exists
+    let branch_exists = Git::branch_exists_locally(&branch)?;
+
+    // Create the worktree
+    if branch_exists {
+        Git::create_worktree_existing_branch(&args.worktree, &branch)?;
+    } else {
+        Git::create_worktree_new_branch(&args.worktree, &branch, &args.base)?;
+    }
+
+    // Change to worktree directory
+    println!("{}", format!("Moving into worktree: {worktree_path}").dimmed());
+    env::set_current_dir(&args.worktree).context("Failed to change directory")?;
+
+    // Update remote
+    update_remote(&branch, args.no_create_upstream)?;
+
+    println!("{}", "Success.".green());
+    Ok(())
+}