git-ea/cmd/cleanup.go

76 lines
1.5 KiB
Go
Raw Normal View History

package cmd
import (
"context"
"errors"
"flag"
"os"
"github.com/AlecAivazis/survey/v2"
"github.com/peterbourgon/ff/v3/ffcli"
)
func (h *Handler) Cleanup() *ffcli.Command {
fs := flag.NewFlagSet("cleanup", flag.ContinueOnError)
return &ffcli.Command{
Name: "cleanup",
FlagSet: fs,
ShortUsage: "cleanup [branches...]",
ShortHelp: "cleanup removes named branches, or interactive if no arguments",
Exec: func(ctx context.Context, args []string) error {
if err := h.checkInit(); err != nil {
return err
}
if len(args) > 0 {
for _, arg := range args {
if err := removeWorktree(h, ctx, arg); err != nil {
return err
}
}
return nil
}
dirs, err := os.ReadDir(h.Config.Workspace())
if err != nil {
return err
}
opts := make([]string, 0, len(dirs))
for _, dir := range dirs {
if !dir.IsDir() {
continue
}
opts = append(opts, dir.Name())
}
if len(opts) == 0 {
return errors.New("no worktrees currently exist")
}
var remove []string
if err := survey.AskOne(&survey.MultiSelect{
Message: "Worktrees to remove",
Options: opts,
}, &remove); err != nil {
return err
}
for _, rm := range remove {
if err := removeWorktree(h, ctx, rm); err != nil {
return err
}
}
return nil
},
}
}
func removeWorktree(h *Handler, ctx context.Context, name string) error {
if err := h.run(ctx, "git", "worktree", "remove", name); err != nil {
return nil
}
return h.run(ctx, "git", "branch", "-D", name)
}