Merge pull request #1646 from rsteube/tea-pullrequest

rsteube · web-flow · commit cf2b19fd6bb9 · 2023-04-09T12:40:14.000+02:00
tea: complete pull requests
diff --git a/completers/tea_completer/cmd/action/action.go b/completers/tea_completer/cmd/action/action.go
@@ -24,6 +24,24 @@ func ActionIssues(cmd *cobra.Command, open bool) carapace.Action {
 	})
 }
 
+func ActionPullrequests(cmd *cobra.Command, open bool) carapace.Action {
+	return carapace.ActionCallback(func(c carapace.Context) carapace.Action {
+		opts := tea.PullrequestOpts{}
+		if f := cmd.Flag("login"); f != nil {
+			opts.Login = f.Value.String()
+		}
+		if f := cmd.Flag("remote"); f != nil {
+			opts.Remote = f.Value.String()
+		}
+		if f := cmd.Flag("repo"); f != nil {
+			opts.Repo = f.Value.String()
+		}
+		opts.Open = open
+		opts.Closed = !open
+		return tea.ActionPullrequests(opts)
+	})
+}
+
 func ActionLabels(cmd *cobra.Command) carapace.Action {
 	return carapace.ActionCallback(func(c carapace.Context) carapace.Action {
 		opts := tea.RepoOpts{}
diff --git a/completers/tea_completer/cmd/pulls_approve.go b/completers/tea_completer/cmd/pulls_approve.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -28,4 +29,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_approveCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_approveCmd, true),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_checkout.go b/completers/tea_completer/cmd/pulls_checkout.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -29,4 +30,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_checkoutCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_checkoutCmd, true),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_clean.go b/completers/tea_completer/cmd/pulls_clean.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -28,4 +29,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_cleanCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_cleanCmd, false),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_close.go b/completers/tea_completer/cmd/pulls_close.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
 )
@@ -25,4 +26,8 @@ func init() {
 	carapace.Gen(pulls_closeCmd).FlagCompletion(carapace.ActionMap{
 		"output": tea.ActionOutputFormats(),
 	})
+
+	carapace.Gen(pulls_closeCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_closeCmd, true),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_merge.go b/completers/tea_completer/cmd/pulls_merge.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -32,4 +33,8 @@ func init() {
 		"remote": git.ActionRemotes(),
 		"style":  carapace.ActionValues("merge", "rebase", "squash", "rebase-merge"),
 	})
+
+	carapace.Gen(pulls_mergeCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_mergeCmd, true),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_reject.go b/completers/tea_completer/cmd/pulls_reject.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -27,4 +28,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_rejectCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_rejectCmd, true),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_reopen.go b/completers/tea_completer/cmd/pulls_reopen.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -28,4 +29,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_reopenCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_reopenCmd, false),
+	)
 }
diff --git a/completers/tea_completer/cmd/pulls_review.go b/completers/tea_completer/cmd/pulls_review.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/completers/tea_completer/cmd/action"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/git"
 	"github.com/rsteube/carapace-bin/pkg/actions/tools/tea"
 	"github.com/spf13/cobra"
@@ -27,4 +28,8 @@ func init() {
 		"output": tea.ActionOutputFormats(),
 		"remote": git.ActionRemotes(),
 	})
+
+	carapace.Gen(pulls_reviewCmd).PositionalCompletion(
+		action.ActionPullrequests(pulls_reviewCmd, true),
+	)
 }
diff --git a/pkg/actions/tools/tea/issue.go b/pkg/actions/tools/tea/issue.go
@@ -4,7 +4,7 @@ import (
 	"strconv"
 
 	"github.com/rsteube/carapace"
-	"github.com/rsteube/carapace/pkg/style"
+	"github.com/rsteube/carapace-bin/pkg/styles"
 )
 
 // ActionIssueFields completes issue fields
@@ -64,15 +64,7 @@ func ActionIssues(opts IssueOpts) carapace.Action {
 		return actionYamlQuery(repoOpts, &issues, args...)(func() carapace.Action {
 			vals := make([]string, 0)
 			for _, issue := range issues {
-				s := ""
-				switch issue.State {
-				case "open":
-					s = style.Green
-				case "closed":
-					s = style.Red
-				}
-
-				vals = append(vals, strconv.Itoa(issue.Index), issue.Title, s)
+				vals = append(vals, strconv.Itoa(issue.Index), issue.Title, styles.Tea.ForState(issue.State, c))
 			}
 			return carapace.ActionStyledValuesDescribed(vals...)
 		})
diff --git a/pkg/actions/tools/tea/pullrequest.go b/pkg/actions/tools/tea/pullrequest.go
@@ -1,6 +1,11 @@
 package tea
 
-import "github.com/rsteube/carapace"
+import (
+	"strconv"
+
+	"github.com/rsteube/carapace"
+	"github.com/rsteube/carapace-bin/pkg/styles"
+)
 
 // ActionPullrequestFields completes pullrequest fields
 //
@@ -30,3 +35,43 @@ func ActionPullrequestFields() carapace.Action {
 		"comments",
 	)
 }
+
+type PullrequestOpts struct {
+	Login  string
+	Remote string
+	Repo   string
+	Open   bool
+	Closed bool
+}
+
+// ActionPullrequests completes pull requests
+func ActionPullrequests(opts PullrequestOpts) carapace.Action {
+	return carapace.ActionCallback(func(c carapace.Context) carapace.Action {
+		var pullrequests []struct {
+			Index int    `yaml:"index"`
+			Title string `yaml:"title"`
+			State string `yaml:"state"`
+		}
+
+		args := []string{"pull", "list"}
+		switch {
+		case opts.Closed && opts.Open:
+			args = append(args, "--state", "all")
+		case opts.Closed:
+			args = append(args, "--state", "closed")
+		case opts.Open:
+			args = append(args, "--state", "open")
+		default:
+			return carapace.ActionValues()
+		}
+
+		repoOpts := RepoOpts{Login: opts.Login, Remote: opts.Remote, Repo: opts.Repo}
+		return actionYamlQuery(repoOpts, &pullrequests, args...)(func() carapace.Action {
+			vals := make([]string, 0)
+			for _, pr := range pullrequests {
+				vals = append(vals, strconv.Itoa(pr.Index), pr.Title, styles.Tea.ForState(pr.State, c))
+			}
+			return carapace.ActionStyledValuesDescribed(vals...)
+		})
+	})
+}
diff --git a/pkg/styles/tea.go b/pkg/styles/tea.go
@@ -0,0 +1,32 @@
+package styles
+
+import "github.com/rsteube/carapace/pkg/style"
+
+type tea struct {
+	StateClosed string `desc:"closed issues/pulls" tag:"issue styles"`
+	StateMerged string `desc:"merged pulls" tag:"issue styles"`
+	StateOpen   string `desc:"open issues/pulls" tag:"issue styles"`
+}
+
+var Tea = tea{
+	StateClosed: style.Red,
+	StateMerged: style.Magenta,
+	StateOpen:   style.Green,
+}
+
+func init() {
+	style.Register("tea", &Tea)
+}
+
+func (_tea *tea) ForState(s string, _ style.Context) string {
+	switch s {
+	case "closed":
+		return _tea.StateClosed
+	case "merged":
+		return _tea.StateMerged
+	case "open":
+		return _tea.StateOpen
+	default:
+		return style.Default
+	}
+}
