Merge pull request cli#3942 from dscho/complete--repo-flag

mislav · web-flow · commit 4b499be96bc3 · 2021-07-28T17:27:56.000+02:00
Allow auto-completing the `--repo` values
diff --git a/pkg/cmdutil/repo_override.go b/pkg/cmdutil/repo_override.go
@@ -2,6 +2,8 @@ package cmdutil
 
 import (
 	"os"
+	"sort"
+	"strings"
 
 	"github.com/cli/cli/internal/ghrepo"
 	"github.com/spf13/cobra"
@@ -19,6 +21,34 @@ func executeParentHooks(cmd *cobra.Command, args []string) error {
 
 func EnableRepoOverride(cmd *cobra.Command, f *Factory) {
 	cmd.PersistentFlags().StringP("repo", "R", "", "Select another repository using the `[HOST/]OWNER/REPO` format")
+	_ = cmd.RegisterFlagCompletionFunc("repo", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		remotes, err := f.Remotes()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		config, err := f.Config()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+		defaultHost, err := config.DefaultHost()
+		if err != nil {
+			return nil, cobra.ShellCompDirectiveError
+		}
+
+		var results []string
+		for _, remote := range remotes {
+			repo := remote.RepoOwner() + "/" + remote.RepoName()
+			if !strings.EqualFold(remote.RepoHost(), defaultHost) {
+				repo = remote.RepoHost() + "/" + repo
+			}
+			if strings.HasPrefix(repo, toComplete) {
+				results = append(results, repo)
+			}
+		}
+		sort.Strings(results)
+		return results, cobra.ShellCompDirectiveNoFileComp
+	})
 
 	cmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
 		if err := executeParentHooks(cmd, args); err != nil {
