cmd: slightly improve stage auto complete

Author: sahib

autocomplete/zsh_autocomplete

@@ -1,10 +1,7 @@
 _cli_zsh_autocomplete() {
-
   local -a opts
   opts=("${(@f)$(_CLI_ZSH_AUTOCOMPLETE_HACK=1 ${words[@]:0:#words[@]-1} --generate-bash-completion)}")
-
   _describe 'values' opts
-
   return
 }
 

cmd/help.go

@@ -520,7 +520,7 @@ EXAMPLES:
 	"stage": {
 		Usage:     "Add a local file to the storage",
 		ArgsUsage: "(<local-path> [<path>]|--stdin <path>)",
-		Complete:  completeArgsUsage,
+		Complete:  completeLocalFile,
 		Flags: []cli.Flag{
 			cli.BoolFlag{
 				Name:  "stdin,i",

cmd/suggest.go

@@ -2,6 +2,8 @@ package cmd
 
 import (
 	"fmt"
+	"io/ioutil"
+	"os"
 	"sort"
 	"strings"
 
@@ -137,6 +139,41 @@ func completeArgsUsage(ctx *cli.Context) {
 	}
 }
 
+func completeLocalFile(ctx *cli.Context) {
+	if len(os.Args) >= 2 {
+		lastArg := os.Args[len(os.Args)-2]
+		cmd := findCurrentCommand(ctx)
+		if lastArg != cmd.FullName() {
+			return
+		}
+	}
+
+	// CAVEAT: We currently do not get partial words from bash/zsh.
+	// e.g. "brig stage /us" will pass the following os.Args:
+	// ["brig", "stage", "--generate-bash-completion"]
+	//
+	// Because of that we do no prefix completion right now.
+	// We can probably tweak autcomplete/{z,ba}sh_autcomplete to
+	// somehow do this, but after 30mins of googling I give up for now.
+	//
+	// If you read this, I challenge you to do it better.
+	dir, err := os.Getwd()
+	if err != nil {
+		// silent error.
+		return
+	}
+
+	children, err := ioutil.ReadDir(dir)
+	if err != nil {
+		// silent error.
+		return
+	}
+
+	for _, child := range children {
+		fmt.Println(child.Name())
+	}
+}
+
 func completeSubcommands(ctx *cli.Context) {
 	if command := findCurrentCommand(ctx); command != nil {
 		for _, subCmd := range command.Subcommands {