Merge pull request #97 from asdf-vm/tb/self-contained-completions

feat(golang-rewrite): self-contained completion code

cli/cli.go

@@ -2,6 +2,7 @@
 package cli
 
 import (
+	_ "embed"
 	"errors"
 	"fmt"
 	"io"
@@ -71,6 +72,13 @@ func Execute(version string) {
 					return extensionCommand(logger, args)
 				},
 			},
+			{
+				Name: "completion",
+				Action: func(cCtx *cli.Context) error {
+					shell := cCtx.Args().Get(0)
+					return completionCommand(logger, shell)
+				},
+			},
 			{
 				Name: "current",
 				Flags: []cli.Flag{
@@ -307,6 +315,47 @@ func Execute(version string) {
 	}
 }
 
+//go:embed completions/asdf.bash
+var bashCompletions string
+
+//go:embed completions/asdf.zsh
+var zshCompletions string
+
+//go:embed completions/asdf.fish
+var fishCompletions string
+
+//go:embed completions/asdf.nu
+var nuCompletions string
+
+//go:embed completions/asdf.elv
+var elvishCompletions string
+
+func completionCommand(l *log.Logger, shell string) error {
+	switch shell {
+	case "bash":
+		fmt.Print(bashCompletions)
+		return nil
+	case "zsh":
+		fmt.Print(zshCompletions)
+		return nil
+	case "fish":
+		fmt.Print(fishCompletions)
+		return nil
+	case "nushell":
+		fmt.Print(nuCompletions)
+		return nil
+	case "elvish":
+		fmt.Print(elvishCompletions)
+		return nil
+	default:
+		fmtString := `No completions available for shell with name %s
+Completions are available for: bash, zsh, fish, nushell, elvish`
+		msg := fmt.Sprintf(fmtString, shell)
+		l.Print(msg)
+		return errors.New(msg)
+	}
+}
+
 // This function is a whole mess and needs to be refactored
 func currentCommand(logger *log.Logger, tool string, noHeader bool) error {
 	conf, err := config.LoadConfig()

cli/completions/asdf.elv

@@ -0,0 +1,148 @@
+# Setup argument completions
+fn arg-completer {|@argz|
+  set argz = $argz[1..-1]  # strip 'asdf' and trailing empty string
+  var num = (count $argz)
+  if (== $num 0) {
+    # list all subcommands
+    find $asdf_dir'/lib/commands' -name 'command-*' | each {|cmd|
+      put (re:replace '.*/command-(.*)\.bash' '${1}' $cmd)
+    }
+    put 'plugin'
+  } else {
+    if (match $argz 'current') {
+      # asdf current <name>
+      asdf plugin-list
+    } elif (match $argz 'env') {
+      # asdf env <command>
+      ls-shims
+    } elif (match $argz 'env' '.*') {
+      # asdf env <command> [util]
+      ls-executables
+    } elif (match $argz 'exec') {
+      # asdf exec <command>
+      ls-shims
+    } elif (match $argz 'global') {
+      # asdf global <name>
+      asdf plugin-list
+    } elif (match $argz 'global' '.*') {
+      # asdf global <name> <version>
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'install') {
+      # asdf install <name>
+      asdf plugin-list
+    } elif (match $argz 'install' '.*') {
+      # asdf install <name> <version>
+      ls-all-versions $argz[-1]
+    } elif (match $argz 'install' '.*' '.*') {
+      # asdf install <name> <version> [--keep-download]
+      put '--keep-download'
+    } elif (match $argz 'latest') {
+      # asdf latest <name>
+      asdf plugin-list
+    } elif (match $argz 'latest' '.*') {
+      # asdf latest <name> [<version>]
+      ls-all-versions $argz[-1]
+    } elif (match $argz 'list-all') {
+      # asdf list all <name>
+      asdf plugin-list
+    } elif (match $argz 'list-all' '.*') {
+      # asdf list all <name> [<version>]
+      ls-all-versions $argz[-1]
+    } elif (match $argz 'list') {
+      # asdf list <name>
+      asdf plugin-list
+    } elif (match $argz 'list' '.*') {
+      # asdf list <name> [<version>]
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'local') {
+      # asdf local <name> [-p|--parent]
+      asdf plugin-list
+      put '-p'
+      put '--parent'
+    } elif (match $argz 'local' '(-p|(--parent))') {
+      # asdf local <name> [-p|--parent] <version>
+      asdf plugin-list
+    } elif (match $argz 'local' '.*') {
+      # asdf local <name> [-p|--parent]
+      # asdf local <name> <version>
+      ls-installed-versions $argz[-1]
+      put '-p'
+      put '--parent'
+    } elif (match $argz 'local' '(-p|(--parent))' '.*') {
+      # asdf local [-p|--parent] <name> <version>
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'local' '.*' '(-p|(--parent))') {
+      # asdf local <name> [-p|--parent] <version>
+      ls-installed-versions $argz[-2]
+    } elif (match $argz 'local' '.*' '.*') {
+      # asdf local <name> <version> [-p|--parent]
+      put '-p'
+      put '--parent'
+    } elif (or (match $argz 'plugin-add') (match $argz 'plugin' 'add')) {
+      # asdf plugin add <name>
+      asdf plugin-list-all | each {|line|
+        put (re:replace '([^\s]+)\s+.*' '${1}' $line)
+      }
+    } elif (or (match $argz 'plugin-list') (match $argz 'plugin' 'list')) {
+      # asdf plugin list
+      put '--urls'
+      put '--refs'
+      put 'all'
+    } elif (or (match $argz 'plugin-push') (match $argz 'plugin' 'push')) {
+      # asdf plugin push <name>
+      asdf plugin-list
+    } elif (or (match $argz 'plugin-remove') (match $argz 'plugin' 'remove')) {
+      # asdf plugin remove <name>
+      asdf plugin-list
+    } elif (and (>= (count $argz) 3) (match $argz[..3] 'plugin-test' '.*' '.*')) {
+      # asdf plugin-test <plugin-name> <plugin-url> [--asdf-tool-version <version>] [--asdf-plugin-gitref <git-ref>] [test-command*]
+      put '--asdf-plugin-gitref'
+      put '--asdf-tool-version'
+      ls-executables
+      ls-shims
+    } elif (and (>= (count $argz) 4) (match $argz[..4] 'plugin' 'test' '.*' '.*')) {
+      # asdf plugin test <plugin-name> <plugin-url> [--asdf-tool-version <version>] [--asdf-plugin-gitref <git-ref>] [test-command*]
+      put '--asdf-plugin-gitref'
+      put '--asdf-tool-version'
+      ls-executables
+      ls-shims
+    } elif (or (match $argz 'plugin-update') (match $argz 'plugin' 'update')) {
+      # asdf plugin update <name>
+      asdf plugin-list
+      put '--all'
+    } elif (match $argz 'plugin') {
+      # list plugin-* subcommands
+      find $asdf_dir'/lib/commands' -name 'command-plugin-*' | each {|cmd|
+        put (re:replace '.*/command-plugin-(.*)\.bash' '${1}' $cmd)
+      }
+    } elif (match $argz 'reshim') {
+      # asdf reshim <name>
+      asdf plugin-list
+    } elif (match $argz 'reshim' '.*') {
+      # asdf reshim <name> <version>
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'shim-versions') {
+      # asdf shim-versions <command>
+      ls-shims
+    } elif (match $argz 'uninstall') {
+      # asdf uninstall <name>
+      asdf plugin-list
+    } elif (match $argz 'uninstall' '.*') {
+      # asdf uninstall <name> <version>
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'update') {
+      if (== $num 1) {
+        # asdf update
+        put '--head'
+      }
+    } elif (match $argz 'where') {
+      # asdf where <name>
+      asdf plugin-list
+    } elif (match $argz 'where' '.*') {
+      # asdf where <name> [<version>]
+      ls-installed-versions $argz[-1]
+    } elif (match $argz 'which') {
+      ls-shims
+    }
+  }
+}