14 Commits
v0.1.0 ... main

Author SHA1 Message Date
5faffde48b Merge pull request 'build: Remove typo' (#3) from fix-release-pipeline into main
Some checks failed
Release packages / build-and-release-artifact (push) Failing after 1m36s
Reviewed-on: #3
2025-07-03 23:57:20 +02:00
96afdc76c2 build: Remove typo
All checks were successful
Code style / cargo-fmt (pull_request) Successful in 1m34s
Code style / cargo-check (pull_request) Successful in 1m33s
Code style / clippy (pull_request) Successful in 1m33s
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:56:57 +02:00
c2df03303b Merge pull request 'feat(setup-repo): Use URI to generate target directory if not provided' (#2) from optional-target-dir into main
Some checks failed
Release packages / build-and-release-artifact (push) Failing after 0s
Reviewed-on: #2
2025-07-03 23:53:23 +02:00
5d745678f3 chore: Fix formatting
All checks were successful
Code style / cargo-fmt (pull_request) Successful in 1m20s
Code style / cargo-check (pull_request) Successful in 1m14s
Code style / clippy (pull_request) Successful in 1m13s
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:43:13 +02:00
cf5b30015a build: Update release pipeline
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:40:41 +02:00
476f849ded build: Update the linting workflow
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:37:27 +02:00
8c22757f71 chore(create-worktree): Update git lib version to 0.2.0
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:16:39 +02:00
0793c8bbd7 chore(setup-repo): Update the git lib version to 0.2.0
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:16:11 +02:00
ddad92dc32 chore(setup-repo): Fix clippy warnings
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:15:49 +02:00
002c85329e feat(git): Remove the failure code from the error type
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 23:15:19 +02:00
b46621f772 chore(setup-repo): Bump version to 0.2.0
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 22:58:37 +02:00
c0fe91f916 feat(setup-repo): Use the last part of the URI as directory if none is provided
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 22:57:41 +02:00
446a3a2f3f Each entry on their own line
Signed-off-by: Noah Knegt <git@noahknegt.com>
2025-07-03 22:34:44 +02:00
Noah Knegt
69f94c64b6 Create libs for common code
Update libs
2025-05-08 16:50:03 +02:00
16 changed files with 408 additions and 355 deletions

View File

@@ -7,18 +7,36 @@ env:
CARGO_TERM_COLOR: always
jobs:
cargo-fmt:
runs-on: ubuntu-latest
steps:
- name: Checkout source code
uses: actions/checkout@v4
# Cache cargo registry
- name: Hash the lock file
uses: https://gitea.com/seepine/hash-files@v1
id: get-hash
with:
patterns: |
**/Cargo.lock
- name: Setup rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
toolchain: stable
components: rustfmt,clippy
cache-shared-key: cargo-${{ steps.get-hash.outputs.hash }}
- name: Rustfmt Check
uses: actions-rust-lang/rustfmt@v1
cargo-check:
runs-on: ubuntu-latest
steps:
- name: Checkout source code
uses: actions/checkout@v4
- name: Setup rust
uses: actions-rs/toolchain@v1
with:
toolchain: stable
components: rustfmt,clippy
# Cache cargo registry
- name: Hash the lock file
uses: https://gitea.com/seepine/hash-files@v1
@@ -26,37 +44,23 @@ jobs:
with:
patterns: |
**/Cargo.lock
- name: Cache cargo registry
uses: actions/cache@v3
- name: Setup rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
path: |
~/.cargo/bin/
~/.cargo/registry/index/
~/.cargo/registry/cache/
~/.cargo/git/db/
target/
key: cargo-${{ steps.get-hash.outputs.hash }}
restore-keys: |
cargo-
# Cache cargo registry - end
toolchain: stable
components: rustfmt,clippy
cache-shared-key: cargo-${{ steps.get-hash.outputs.hash }}
- name: Cargo check
uses: actions-rs/cargo@v1
with:
command: check
run: cargo check --workspace
clippy:
runs-on: ubuntu-latest
steps:
- name: Checkout source code
- name: Checkout source code
uses: actions/checkout@v4
- name: Setup rust
uses: actions-rs/toolchain@v1
with:
toolchain: stable
components: rustfmt,clippy
# Cache cargo registry
- name: Hash the lock file
uses: https://gitea.com/seepine/hash-files@v1
@@ -64,22 +68,13 @@ jobs:
with:
patterns: |
**/Cargo.lock
- name: Cache cargo registry
uses: actions/cache@v3
with:
path: |
~/.cargo/bin/
~/.cargo/registry/index/
~/.cargo/registry/cache/
~/.cargo/git/db/
target/
key: cargo-${{ steps.get-hash.outputs.hash }}
restore-keys: |
cargo-
# Cache cargo registry - end
- name: Run clippy
uses: actions-rs/clippy-check@v1
- name: Setup rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
token: ${{ secrets.GITHUB_TOKEN }}
args: --all-targets --all-features -- -Dwarnings
toolchain: stable
components: rustfmt,clippy
cache-shared-key: cargo-${{ steps.get-hash.outputs.hash }}
- name: Cargo check
run: cargo clippy --all-targets --all-features -- -Dwarnings

View File

@@ -12,57 +12,36 @@ jobs:
build-and-release-artifact:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Setup rust
uses: actions-rs/toolchain@v1
with:
profile: minimal
toolchain: stable
- name: Checkout code
uses: actions/checkout@v4
# Cache cargo registry
- name: Hash the lock file
uses: https://gitea.com/seepine/hash-files@v1
id: get-hash
with:
patterns: |
**/Cargo.lock
- name: Cache cargo registry
uses: actions/cache@v4
with:
path: |
~/.cargo/bin/
~/.cargo/registry/index/
~/.cargo/registry/cache/
~/.cargo/git/db/
target/
key: cargo-${{ steps.get-hash.outputs.hash }}
restore-keys: |
cargo-
# Cache cargo registry - end
- name: Hash the lock file
uses: https://gitea.com/seepine/hash-files@v1
id: get-hash
with:
patterns: |
**/Cargo.lock
- name: Build application
uses: actions-rs/cargo@v1
with:
command: build
args: --release --all-features --workspace --all-targets
- name: Setup rust
uses: actions-rust-lang/setup-rust-toolchain@v1
with:
toolchain: stable
cache-shared-key: cargo-${{ steps.get-hash.outputs.hash }}
matcher: false
- name: Package application
uses: actions-rs/cargo@v1
with:
command: package
args: --all-features --workspace --allow-dirty
- name: Build application
run: cargo build --release --all-features --workspace --all-targets
- name: Add registry to cargo config
run: echo -e '[registries.gitea]\nindex = "sparse+https://git.noahknegt.com/api/packages/${{ github.repository_owner }}/cargo/"\n' > ~/.cargo/config.toml
- name: Package application
run: cargo package --all-features --workspace --allow-dirty
- name: Publish to cargo registry
uses: actions-rs/cargo@v1
with:
command: publish
args: --no-verify --registry gitea --token ${{ secrets.CARGO_REGISTRY_TOKEN }}
env:
CARGO_NET_GIT_FETCH_WITH_CLI: true
CARGO_NET_RETRY: 2
CARGO_NET_TIMEOUT: 120
- name: Add registry to cargo config
run: echo -e '[registries.gitea]\nindex = "sparse+https://git.noahknegt.com/api/packages/${{ github.repository_owner }}/cargo/"\n' > ~/.cargo/config.toml
- name: Publish to cargo registry
run: cargo publish --no-verify --registry gitea --token ${{ secrets.CARGO_REGISTRY_TOKEN }}
env:
CARGO_NET_GIT_FETCH_WITH_CLI: true
CARGO_NET_RETRY: 2
CARGO_NET_TIMEOUT: 120

28
Cargo.lock generated
View File

@@ -131,6 +131,16 @@ dependencies = [
"windows-sys",
]
[[package]]
name = "command-with-spinner"
version = "0.1.0"
dependencies = [
"anyhow",
"colored",
"indicatif",
"thiserror",
]
[[package]]
name = "console"
version = "0.15.11"
@@ -152,6 +162,7 @@ dependencies = [
"clap",
"colored",
"dirs",
"git",
"indicatif",
"thiserror",
]
@@ -194,6 +205,19 @@ dependencies = [
"wasi",
]
[[package]]
name = "git"
version = "0.2.0"
dependencies = [
"anyhow",
"command-with-spinner",
"thiserror",
]
[[package]]
name = "git-ssh-bitwarden"
version = "0.1.0"
[[package]]
name = "heck"
version = "0.5.0"
@@ -306,13 +330,13 @@ dependencies = [
[[package]]
name = "setup-repo"
version = "0.1.0"
version = "0.2.0"
dependencies = [
"anyhow",
"clap",
"colored",
"git",
"indicatif",
"thiserror",
]
[[package]]

View File

@@ -1,7 +1,10 @@
[workspace]
resolver = "3"
members = [
"create-worktree",
members = [
"command-with-spinner",
"create-worktree",
"git",
"git-ssh-bitwarden",
"setup-repo",
]

View File

@@ -0,0 +1,14 @@
[package]
name = "command-with-spinner"
version = "0.1.0"
edition.workspace = true
authors.workspace = true
license-file.workspace = true
repository.workspace = true
[dependencies]
anyhow = "1.0.98"
colored = "3.0.0"
indicatif = "0.17.11"
thiserror = "2.0"

View File

@@ -0,0 +1,12 @@
/// Git operations error type
#[derive(Debug, thiserror::Error)]
pub enum CommandError {
#[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),
}

View File

@@ -0,0 +1,42 @@
use std::process::{Command, Stdio};
use anyhow::{Context, Result};
use colored::*;
use indicatif::{ProgressBar, ProgressStyle};
mod error;
use error::CommandError;
/// Runs a git command with a progress spinner
pub 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(CommandError::Failed(code).into()),
None => Err(CommandError::FailedNoCode.into()),
}
}
}

View File

@@ -9,9 +9,10 @@ license-file.workspace = true
repository.workspace = true
[dependencies]
clap = { version = "4.5", features = ["derive"] }
anyhow = "1.0"
clap = { version = "4.5", features = ["derive"] }
colored = "3.0"
indicatif = "0.17"
dirs = "6.0"
git = { version = "0.2.0", path = "../git" }
indicatif = "0.17"
thiserror = { version = "2.0" }

View File

@@ -1,11 +1,11 @@
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};
use git::Git;
/// Tool to create git worktrees with convenient branch management
#[derive(Parser, Debug)]
@@ -39,133 +39,6 @@ struct Args {
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
@@ -178,11 +51,17 @@ fn update_remote(branch: &str, create_upstream: bool) -> Result<()> {
if !has_remote {
// Create remote branch
println!("{}", format!("Branch '{branch}' does not exist on remote. Creating.").dimmed());
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());
println!(
"{}",
format!("Branch '{branch}' exists. Setting upstream.").dimmed()
);
Git::set_upstream_branch(branch)?;
}
@@ -200,11 +79,16 @@ fn main() -> Result<()> {
if args.verbose {
println!("{}", "Verbose mode enabled".dimmed());
println!("{}", format!("Base branch: {}", args.base).dimmed());
println!("{}", format!("Create upstream: {}", args.no_create_upstream).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));
let branch = args
.branch
.unwrap_or_else(|| format!("{}{}", args.prefix, args.worktree));
// Normalize paths
let worktree_path = Path::new(&args.worktree).to_string_lossy();
@@ -220,7 +104,10 @@ fn main() -> Result<()> {
}
// Change to worktree directory
println!("{}", format!("Moving into worktree: {worktree_path}").dimmed());
println!(
"{}",
format!("Moving into worktree: {worktree_path}").dimmed()
);
env::set_current_dir(&args.worktree).context("Failed to change directory")?;
// Update remote

View File

@@ -0,0 +1,10 @@
[package]
name = "git-ssh-bitwarden"
version = "0.1.0"
edition.workspace = true
authors.workspace = true
license-file.workspace = true
repository.workspace = true
[dependencies]

View File

@@ -0,0 +1,3 @@
fn main() {
println!("Hello, world!");
}

13
git/Cargo.toml Normal file
View File

@@ -0,0 +1,13 @@
[package]
name = "git"
version = "0.2.0"
edition.workspace = true
authors.workspace = true
license-file.workspace = true
repository.workspace = true
[dependencies]
anyhow = "1.0.98"
command-with-spinner = { version = "0.1.0", path = "../command-with-spinner" }
thiserror = "2.0"

9
git/src/error.rs Normal file
View File

@@ -0,0 +1,9 @@
/// Git operations error type
#[derive(Debug, thiserror::Error)]
pub enum GitError {
#[error("Command failed without exit code")]
FailedNoCode,
#[error("Failed to execute command: {0}")]
ExecutionError(#[from] std::io::Error),
}

148
git/src/lib.rs Normal file
View File

@@ -0,0 +1,148 @@
use std::{path::Path, process::Command};
use anyhow::{Context, Result};
use command_with_spinner::run_command;
use error::GitError;
mod error;
/// Git command wrapper
pub struct Git;
impl Git {
/// Clone a repository as a bare clone in a .bare directory
pub fn clone_bare_repo(repo_url: &str, target_dir: &str) -> Result<()> {
// Create the base directory first
std::fs::create_dir_all(target_dir).context("Failed to create target directory")?;
// Create the .bare subdirectory path
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
// Clone the repository as a bare clone into .bare directory
let mut cmd = Command::new("git");
cmd.args(["clone", "--bare", repo_url, &bare_dir_str]);
match run_command(
&mut cmd,
&format!("Cloning repository as bare clone into {bare_dir_str}"),
) {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Set up the .git file to point to the .bare directory
pub fn setup_git_pointer(target_dir: &str) -> Result<()> {
let git_file_path = Path::new(target_dir).join(".git");
std::fs::write(git_file_path, "gitdir: ./.bare")
.context("Failed to create .git file pointing to .bare directory")
}
/// Configure remote.origin.fetch to fetch all references
pub fn configure_remote_fetch(target_dir: &str) -> Result<()> {
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
let mut cmd = Command::new("git");
cmd.args([
"--git-dir",
&bare_dir_str,
"config",
"remote.origin.fetch",
"+refs/heads/*:refs/remotes/origin/*",
]);
match run_command(&mut cmd, "Configuring remote.origin.fetch") {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Fetch all remotes
pub fn fetch_remotes(target_dir: &str) -> Result<()> {
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
let mut cmd = Command::new("git");
cmd.args(["--git-dir", &bare_dir_str, "fetch", "--all"]);
match run_command(&mut cmd, "Fetching all remotes") {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Check if a branch exists locally
pub 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
pub 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
pub fn create_worktree_existing_branch(worktree_path: &str, branch: &str) -> Result<()> {
let mut cmd = Command::new("git");
cmd.args(["worktree", "add", worktree_path, branch]);
match run_command(
&mut cmd,
&format!("Generating new worktree from existing branch: {branch}"),
) {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Create a new worktree with a new branch
pub 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]);
match run_command(
&mut cmd,
&format!("Generating new worktree: {worktree_path}"),
) {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Create and push a new remote branch
pub fn create_remote_branch(branch: &str) -> Result<()> {
let mut cmd = Command::new("git");
cmd.args(["push", "-u", "origin", branch]);
match run_command(&mut cmd, &format!("Creating remote branch {branch}...")) {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
/// Set the upstream branch
pub fn set_upstream_branch(branch: &str) -> Result<()> {
let mut cmd = Command::new("git");
cmd.args(["branch", "--set-upstream-to", &format!("origin/{branch}")]);
match run_command(
&mut cmd,
&format!("Setting upstream branch to 'origin/{branch}'"),
) {
Ok(_) => Ok(()),
Err(_) => Err(GitError::FailedNoCode.into()),
}
}
}

View File

@@ -1,6 +1,6 @@
[package]
name = "setup-repo"
version = "0.1.0"
version = "0.2.0"
description = "Tool to set up Git repositories for worktree development"
edition.workspace = true
@@ -12,5 +12,5 @@ repository.workspace = true
anyhow = "1.0"
clap = { version = "4.5", features = ["derive"] }
colored = "3.0"
git = { version = "0.2.0", path = "../git" }
indicatif = "0.17"
thiserror = "2.0"

View File

@@ -1,10 +1,8 @@
use std::path::Path;
use std::process::{Command, Stdio};
use anyhow::{Context, Result};
use anyhow::Result;
use clap::Parser;
use colored::*;
use indicatif::{ProgressBar, ProgressStyle};
use git::Git;
/// Tool to set up Git repositories for worktree development
#[derive(Parser, Debug)]
@@ -16,7 +14,7 @@ struct Args {
/// Target directory for the repository setup
#[arg(short, long)]
target_dir: String,
target_dir: Option<String>,
/// Enable verbose output
#[arg(short, long)]
@@ -27,113 +25,6 @@ struct Args {
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 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 {
/// Clone a repository as a bare clone in a .bare directory
fn clone_bare_repo(repo_url: &str, target_dir: &str) -> Result<()> {
// Create the base directory first
std::fs::create_dir_all(target_dir).context("Failed to create target directory")?;
// Create the .bare subdirectory path
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
// Clone the repository as a bare clone into .bare directory
let mut cmd = Command::new("git");
cmd.args([
"clone",
"--bare",
repo_url,
&bare_dir_str
]);
run_command(&mut cmd, &format!("Cloning repository as bare clone into {bare_dir_str}"))
}
/// Set up the .git file to point to the .bare directory
fn setup_git_pointer(target_dir: &str) -> Result<()> {
let git_file_path = Path::new(target_dir).join(".git");
std::fs::write(git_file_path, "gitdir: ./.bare")
.context("Failed to create .git file pointing to .bare directory")
}
/// Configure remote.origin.fetch to fetch all references
fn configure_remote_fetch(target_dir: &str) -> Result<()> {
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
let mut cmd = Command::new("git");
cmd.args([
"--git-dir", &bare_dir_str,
"config",
"remote.origin.fetch", "+refs/heads/*:refs/remotes/origin/*"
]);
run_command(&mut cmd, "Configuring remote.origin.fetch")
}
/// Fetch all remotes
fn fetch_remotes(target_dir: &str) -> Result<()> {
let bare_dir = Path::new(target_dir).join(".bare");
let bare_dir_str = bare_dir.to_string_lossy();
let mut cmd = Command::new("git");
cmd.args([
"--git-dir", &bare_dir_str,
"fetch",
"--all"
]);
run_command(&mut cmd, "Fetching all remotes")
}
}
fn main() -> Result<()> {
// Parse arguments
let args = Args::parse();
@@ -141,28 +32,50 @@ fn main() -> Result<()> {
// Enable or disable colored output
colored::control::set_override(!args.no_color);
let target_dir = match args.target_dir {
Some(dir) => dir,
// If a directory is not provided construct one from the last part of the URI.
// This behavior follows git itself
None => {
let start = args.repo_url.rfind("/").unwrap() + 1;
let end = match args.repo_url.rfind(".git") {
Some(index) => index,
None => args.repo_url.len(),
};
args.repo_url[start..end].to_owned()
}
};
// Print verbose information if enabled
if args.verbose {
println!("{}", "Verbose mode enabled".dimmed());
println!("{}", format!("Repository URL: {}", args.repo_url).dimmed());
println!("{}", format!("Target directory: {}", args.target_dir).dimmed());
println!("{}", format!("Target directory: {target_dir}").dimmed());
}
println!("{}", "Setting up repository for worktree development".blue());
println!(
"{}",
"Setting up repository for worktree development".blue()
);
// Clone the repository as a bare clone
Git::clone_bare_repo(&args.repo_url, &args.target_dir)?;
Git::clone_bare_repo(&args.repo_url, &target_dir)?;
// Set up the .git file to point to the .bare directory
Git::setup_git_pointer(&args.target_dir)?;
Git::setup_git_pointer(&target_dir)?;
// Configure the remote.origin.fetch setting
Git::configure_remote_fetch(&args.target_dir)?;
Git::configure_remote_fetch(&target_dir)?;
// Fetch all remotes
Git::fetch_remotes(&args.target_dir)?;
Git::fetch_remotes(&target_dir)?;
println!("{}", "Repository setup complete.".green());
println!("{}", format!("You can now create worktrees in '{}'.", args.target_dir).green());
println!(
"{}",
format!("You can now create worktrees in '{target_dir}'.").green()
);
Ok(())
}