-
Notifications
You must be signed in to change notification settings - Fork 18
/
up.go
96 lines (84 loc) · 2.14 KB
/
up.go
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
package main
import (
"context"
"fmt"
"github.com/charmbracelet/log"
"go.abhg.dev/gs/internal/git"
"go.abhg.dev/gs/internal/spice"
"go.abhg.dev/gs/internal/spice/state"
"go.abhg.dev/gs/internal/text"
"go.abhg.dev/gs/internal/ui"
"go.abhg.dev/gs/internal/ui/widget"
)
type upCmd struct {
checkoutOptions
N int `arg:"" optional:"" help:"Number of branches to move up." default:"1"`
}
func (*upCmd) Help() string {
return text.Dedent(`
Checks out the branch above the current one.
If there are multiple branches with the current branch as base,
a prompt will allow picking between them.
Use the -n flag to print the branch without checking it out.
`)
}
func (cmd *upCmd) Run(
ctx context.Context,
log *log.Logger,
view ui.View,
repo *git.Repository,
store *state.Store,
svc *spice.Service,
) error {
current, err := repo.CurrentBranch(ctx)
if err != nil {
// TODO: handle not a branch
return fmt.Errorf("get current branch: %w", err)
}
var branch string
outer:
for range cmd.N {
aboves, err := svc.ListAbove(ctx, current)
if err != nil {
return fmt.Errorf("list branches above %v: %w", current, err)
}
switch len(aboves) {
case 0:
if branch != "" {
// If we've already moved up once,
// and there are no branches above the current one,
// we're done.
break outer
}
return fmt.Errorf("%v: no branches found upstack", current)
case 1:
branch = aboves[0]
default:
desc := "There are multiple branches above the current branch."
if !ui.Interactive(view) {
log.Error(desc)
return errNoPrompt
}
items := make([]widget.BranchTreeItem, len(aboves))
for i, b := range aboves {
items[i] = widget.BranchTreeItem{
Branch: b,
Base: current,
}
}
prompt := widget.NewBranchTreeSelect().
WithValue(&branch).
WithItems(items...).
WithTitle("Pick a branch").
WithDescription(desc)
if err := ui.Run(view, prompt); err != nil {
return fmt.Errorf("a branch is required: %w", err)
}
}
current = branch
}
return (&branchCheckoutCmd{
checkoutOptions: cmd.checkoutOptions,
Branch: branch,
}).Run(ctx, log, view, repo, store, svc)
}