feat: add branch manipulation features

Author: terakoya76

README.md

@@ -44,6 +44,22 @@ $ git branch
   main
 $ popd
 
+# List branch directories under the current repository
+$ git-replicator branch
+base
+foo
+
+# Delete a branch directory under the current repository
+$ git-replicator delete foo
+Deleted branch directory: foo
+
+$ git-replicator switch foo
+Enumerating objects: 5, done.
+Counting objects: 100% (5/5), done.
+Compressing objects: 100% (4/4), done.
+Total 5 (delta 0), reused 0 (delta 0), pack-reused 0 (from 0)
+cloned branch: foo to dir: /home/$USER/git-replicator/github.com/terakoya76/git-replicator-test/foo
+
 # From now on, you can let the agent use this directory as it pleases.
 # e.g. 
 # $ cursor ./foo/
@@ -53,6 +69,8 @@ $ popd
 - Clone a git repository into a structured local directory (`get <url>`)
 - List all managed repositories (`list`)
 - Clone current repo into a new branch directory (`switch <branch>`, like `git switch`)
+- List branch directories under the current repository (`branch`)
+- Delete a branch directory under the current repository (`delete <branch>`)
 
 ## Development
 

cmd/branch.go

@@ -0,0 +1,42 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/terakoya76/git-replicator/internal/handlers"
+	"github.com/terakoya76/git-replicator/internal/utils"
+)
+
+var branchCmd = &cobra.Command{
+	Use:   "branch",
+	Short: "List branch directories under the current repository (like git switch)",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		cwd, err := os.Getwd()
+		if err != nil {
+			return fmt.Errorf("failed to get current directory: %w", err)
+		}
+		rootDir, err := utils.GetGitReplicatorRoot()
+		if err != nil {
+			return fmt.Errorf("failed to get git-replicator root: %w", err)
+		}
+		repoDir, err := utils.FindRepoDir(cwd, rootDir)
+		if err != nil {
+			return err
+		}
+		branches, err := handlers.ListBranchDirs(context.Background(), repoDir)
+		if err != nil {
+			return err
+		}
+		for _, b := range branches {
+			fmt.Println(b)
+		}
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(branchCmd)
+}

cmd/delete.go

@@ -0,0 +1,41 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/terakoya76/git-replicator/internal/handlers"
+	"github.com/terakoya76/git-replicator/internal/utils"
+)
+
+var deleteCmd = &cobra.Command{
+	Use:   "delete <branch>",
+	Short: "Delete a branch directory under the current repository",
+	Args:  cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		branch := args[0]
+		cwd, err := os.Getwd()
+		if err != nil {
+			return fmt.Errorf("failed to get current directory: %w", err)
+		}
+		rootDir, err := utils.GetGitReplicatorRoot()
+		if err != nil {
+			return fmt.Errorf("failed to get git-replicator root: %w", err)
+		}
+		repoDir, err := utils.FindRepoDir(cwd, rootDir)
+		if err != nil {
+			return err
+		}
+		if err := handlers.DeleteBranchDir(context.Background(), repoDir, branch); err != nil {
+			return err
+		}
+		fmt.Printf("Deleted branch directory: %s\n", branch)
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(deleteCmd)
+}

cmd/get.go

@@ -18,7 +18,7 @@ var getCmd = &cobra.Command{
 		url := args[0]
 		rootDir, err := utils.GetGitReplicatorRoot()
 		if err != nil {
-			return fmt.Errorf("failed to get git-replicator root: %w", err)
+			return fmt.Errorf("failed to get git-replicator root ($HOME/git-replicator): %w", err)
 		}
 		if err := handlers.Get(ctx, url, rootDir); err != nil {
 			return fmt.Errorf("failed to clone repository: %w", err)

internal/config/config.go

@@ -8,8 +8,6 @@ import (
 
 type Config struct {
 	// Add your configuration fields here
-	SourceRepo string `mapstructure:"source_repo"`
-	TargetRepo string `mapstructure:"target_repo"`
 }
 
 func Load() (*Config, error) {

internal/handlers/branch.go

@@ -0,0 +1,22 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+	"os"
+)
+
+// ListBranchDirs returns a list of branch directory names under the given repoDir (including 'base').
+func ListBranchDirs(ctx context.Context, repoDir string) ([]string, error) {
+	entries, err := os.ReadDir(repoDir)
+	if err != nil {
+		return nil, fmt.Errorf("failed to read repo directory: %w", err)
+	}
+	var branches []string
+	for _, entry := range entries {
+		if entry.IsDir() {
+			branches = append(branches, entry.Name())
+		}
+	}
+	return branches, nil
+}

internal/handlers/branch_test.go

@@ -0,0 +1,51 @@
+package handlers_test
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/terakoya76/git-replicator/internal/handlers"
+)
+
+func TestListBranchDirs(t *testing.T) {
+	tmpDir := t.TempDir()
+	repoDir := filepath.Join(tmpDir, "repo")
+	if err := os.MkdirAll(repoDir, 0o755); err != nil {
+		t.Fatalf("failed to create repo dir: %v", err)
+	}
+	// Create base and branch directories
+	if err := os.Mkdir(filepath.Join(repoDir, "base"), 0o755); err != nil {
+		t.Fatalf("failed to create base dir: %v", err)
+	}
+	if err := os.Mkdir(filepath.Join(repoDir, "feature-x"), 0o755); err != nil {
+		t.Fatalf("failed to create feature-x dir: %v", err)
+	}
+	if err := os.Mkdir(filepath.Join(repoDir, "bugfix-y"), 0o755); err != nil {
+		t.Fatalf("failed to create bugfix-y dir: %v", err)
+	}
+
+	t.Run("list branch dirs (including base)", func(t *testing.T) {
+		branches, err := handlers.ListBranchDirs(context.Background(), repoDir)
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+		want := map[string]bool{"feature-x": true, "bugfix-y": true, "base": true}
+		if len(branches) != 3 {
+			t.Errorf("expected 3 branches, got %d", len(branches))
+		}
+		for _, b := range branches {
+			if !want[b] {
+				t.Errorf("unexpected branch: %s", b)
+			}
+		}
+	})
+
+	t.Run("repo dir does not exist", func(t *testing.T) {
+		_, err := handlers.ListBranchDirs(context.Background(), filepath.Join(tmpDir, "not-exist"))
+		if err == nil {
+			t.Errorf("expected error for non-existent dir, got nil")
+		}
+	})
+}

internal/handlers/delete.go

@@ -0,0 +1,18 @@
+package handlers
+
+import (
+	"context"
+	"fmt"
+	"path/filepath"
+
+	"github.com/terakoya76/git-replicator/internal/utils"
+)
+
+// DeleteBranchDir deletes the branch directory under the given repo for a branch name.
+func DeleteBranchDir(ctx context.Context, repoDir, branchName string) error {
+	branchDir := filepath.Join(repoDir, branchName)
+	if err := utils.RemoveDir(branchDir); err != nil {
+		return fmt.Errorf("failed to delete branch directory %s: %w", branchDir, err)
+	}
+	return nil
+}

internal/handlers/delete_test.go

@@ -0,0 +1,86 @@
+package handlers_test
+
+import (
+	"context"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/terakoya76/git-replicator/internal/handlers"
+)
+
+func TestDeleteBranchDir(t *testing.T) {
+	tmpDir := t.TempDir()
+	repoDir := filepath.Join(tmpDir, "repo")
+	branchName := "feature-x"
+	branchDir := filepath.Join(repoDir, branchName)
+
+	tests := []struct {
+		name       string
+		branch     string
+		prepare    func()
+		wantErr    bool
+		checkAfter func(branchDir string) error
+	}{
+		{
+			name:   "delete existing branch dir",
+			branch: branchName,
+			prepare: func() {
+				if err := os.MkdirAll(branchDir, 0o755); err != nil {
+					t.Fatalf("failed to create branch dir: %v", err)
+				}
+				filePath := filepath.Join(branchDir, "dummy.txt")
+				if err := os.WriteFile(filePath, []byte("dummy"), 0o644); err != nil {
+					t.Fatalf("failed to create file: %v", err)
+				}
+			},
+			wantErr: false,
+			checkAfter: func(branchDir string) error {
+				if _, err := os.Stat(branchDir); !os.IsNotExist(err) {
+					return err
+				}
+				return nil
+			},
+		},
+		{
+			name:   "delete already deleted branch dir",
+			branch: branchName,
+			prepare: func() {
+				_ = os.RemoveAll(branchDir)
+			},
+			wantErr: false,
+			checkAfter: func(branchDir string) error {
+				if _, err := os.Stat(branchDir); !os.IsNotExist(err) {
+					return err
+				}
+				return nil
+			},
+		},
+		{
+			name:    "delete non-existent branch dir",
+			branch:  "nonexistent",
+			prepare: func() {},
+			wantErr: false,
+			checkAfter: func(branchDir string) error {
+				if _, err := os.Stat(branchDir); !os.IsNotExist(err) {
+					return err
+				}
+				return nil
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			tt.prepare()
+			branchDirPath := filepath.Join(repoDir, tt.branch)
+			err := handlers.DeleteBranchDir(context.Background(), repoDir, tt.branch)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("DeleteBranchDir() error = %v, wantErr %v", err, tt.wantErr)
+			}
+			if err := tt.checkAfter(branchDirPath); err != nil {
+				t.Errorf("post-check failed: %v", err)
+			}
+		})
+	}
+}

internal/utils/filesystem.go

@@ -26,8 +26,16 @@ func FindRepoDir(cwd, gitReplicatorRoot string) (string, error) {
 			return dir, nil
 		}
 		if dir == gitReplicatorRoot || dir == "/" || dir == "." {
-			return "", fmt.Errorf("could not find repo directory under git-replicator root")
+			return "", fmt.Errorf("could not find repo directory, so move to the repo directory ($HOME/git-replicator/<host>/<owner>/<repo>)")
 		}
 		dir = parent
 	}
 }
+
+// RemoveDir deletes the specified directory and all its contents.
+func RemoveDir(dir string) error {
+	if err := os.RemoveAll(dir); err != nil {
+		return fmt.Errorf("failed to remove directory %s: %w", dir, err)
+	}
+	return nil
+}