asdf/internal/completions/asdf.zsh
DeedleFake 518a0fa442 feat(golang-rewrite): refactor completion code and move all packages except cmd into internal
* Refactor and move `completion` package to `internal/completions`
* Move `cli` and `repotest` packages to `internal/cli`
* Use passed version information for `--version` flag
* Add simple test for completions package
2024-12-18 20:01:10 -05:00

228 lines
6.9 KiB
Bash
Executable File

#compdef asdf
compdef _asdf asdf
#description tool to manage versions of multiple runtimes
local curcontext="$curcontext" state state_descr line subcmd
local asdf_dir="${ASDF_DATA_DIR:-$HOME/.asdf}"
local -a asdf_commands
asdf_commands=( # 'asdf help' lists commands with help text
# plugins
'plugin:plugin management sub-commands'
'plugin-add:add plugin from asdf-plugins repo or from git URL'
'plugin-list:list installed plugins (--urls with URLs)'
'plugin-list-all:list all plugins registered in asdf-plugins repo'
'plugin-remove:remove named plugin and all packages for it'
'plugin-update:update named plugin (or --all)'
# packages
'install:install plugin at stated version, or all from .tools-versions'
'uninstall:remove a specific version of a package'
'current:display current versions for named package (else all)'
'latest:display latest version available to install for a named package'
'where:display install path for given package at optional specified version'
'which:display path to an executable'
'shell:via env vars, set package to version in current shell'
'local:set package local version'
'global:set package global version'
'list:list installed versions of a package'
'list-all:list all available (remote) versions of a package'
# utils
'exec:executes the command shim for the current version'
'env:prints or runs an executable under a command environment'
'info:print os, shell and asdf debug information'
'reshim:recreate shims for version of a package'
'shim:shim management sub-commands'
'shim-versions:list for given command which plugins and versions provide it'
'update:update ASDF to the latest stable release (unless --head)'
)
_asdf__available_plugins() {
local plugin_dir="${asdf_dir:?}/repository/plugins"
if [[ ! -d "$plugin_dir" ]]; then
_wanted asdf-available-plugins expl 'ASDF Installable Plugins' \
compadd -x "no plugins repository found"
return
fi
local -a plugins
plugins=( "$plugin_dir"/*(:t) )
_wanted asdf-available-plugins expl 'ASDF Installable Plugins' \
compadd -a plugins
}
_asdf__installed_plugins() {
local plugin_dir="${asdf_dir:?}/plugins"
if [[ ! -d "$plugin_dir" ]]; then
_wanted asdf-plugins expl 'ASDF Plugins' \
compadd -x "no plugins dir, none installed yet"
return
fi
local -a plugins
plugins=( "$plugin_dir"/*(:t) )
_wanted asdf-plugins expl 'ASDF Plugins' \
compadd -a plugins
}
_asdf__installed_versions_of() {
local plugin_dir="${asdf_dir:?}/installs/${1:?need a plugin version}"
if [[ ! -d "$plugin_dir" ]]; then
_wanted "asdf-versions-$1" expl "ASDF Plugin ${(q-)1} versions" \
compadd -x "no versions installed"
return
fi
local -a versions
versions=( "$plugin_dir"/*(:t) )
_wanted "asdf-versions-$1" expl "ASDF Plugin ${(q-)1} versions" \
compadd -a versions
}
_asdf__installed_versions_of_plus_system() {
local plugin_dir="${asdf_dir:?}/installs/${1:?need a plugin version}"
if [[ ! -d "$plugin_dir" ]]; then
_wanted "asdf-versions-$1" expl "ASDF Plugin ${(q-)1} versions" \
compadd -x "no versions installed"
return
fi
local -a versions
versions=( "$plugin_dir"/*(:t) )
versions+="system"
_wanted "asdf-versions-$1" expl "ASDF Plugin ${(q-)1} versions" \
compadd -a versions
}
_asdf() {
local -i IntermediateCount=0
if (( CURRENT == 2 )); then
_arguments -C : '--version[version]' ':command:->command'
fi
case "$state" in
(command)
_describe -t asdf-commands 'ASDF Commands' asdf_commands
return
;;
esac
subcmd="${words[2]}"
curcontext="${curcontext%:*}=$subcmd:"
# Handle 'foo bar' == 'foo-bar'
_asdf__dash_commands() {
if (( CURRENT == 3 + IntermediateCount )); then
local -a sub_commands
sub_commands=(${${(M)asdf_commands:#${subcmd}-*}#${subcmd}-})
_describe -t asdf-commands 'ASDF Commands' sub_commands
else
IntermediateCount+=1
subcmd="${subcmd}-${words[2+IntermediateCount]}"
fi
}
case "$subcmd" in
(plugin|shim|list)
_asdf__dash_commands
;;
esac
case "$subcmd" in
(plugin-list)
_asdf__dash_commands
;;
esac
case "$subcmd" in
(plugin-add)
if (( CURRENT == 3 + IntermediateCount )); then
_asdf__available_plugins
else
# Optional URL
curcontext="${curcontext/=plugin-add:/=plugin-add-${words[3]}:}"
if (( CURRENT == 4 + IntermediateCount )); then
_arguments "*:${words[3]} package url:_urls"
fi
fi
;;
(plugin-remove|current|list|list-all)
(( CURRENT == 3 + IntermediateCount )) && _asdf__installed_plugins
;;
(plugin-update)
(( CURRENT == 3 + IntermediateCount )) && _alternative \
'all:all:(--all)' \
'asdf-available-plugins:Installed ASDF Plugins:_asdf__installed_plugins'
;;
(install)
if (( CURRENT == 3 + IntermediateCount )); then
_asdf__installed_plugins
elif (( CURRENT == 4 + IntermediateCount )); then
local pkg="${words[3+IntermediateCount]}"
local ver_prefix="${words[4+IntermediateCount]}"
if [[ $ver_prefix == latest:* ]]; then
_wanted "latest-versions-$pkg" \
expl "Latest version" \
compadd -- latest:${^$(asdf list-all "$pkg")}
else
_wanted "latest-tag-$pkg" \
expl "Latest version" \
compadd -- 'latest' 'latest:'
_wanted "remote-versions-$pkg" \
expl "Available versions of $pkg" \
compadd -- $(asdf list-all "$pkg")
fi
fi
;;
(latest)
if (( CURRENT == 3 + IntermediateCount )); then
_alternative \
'all:all:(--all)' \
'asdf-available-plugins:Installed ASDF Plugins:_asdf__installed_plugins'
elif (( CURRENT == 4 + IntermediateCount )); then
local pkg="${words[3+IntermediateCount]}"
local query=${words[4+IntermediateCount]}
[[ -n $query ]] || query='[0-9]'
_wanted "latest-pattern-$pkg" \
expl "Pattern to look for in matching versions of $pkg" \
compadd -- $(asdf list-all "$pkg" "$query")
fi
;;
(uninstall|reshim)
compset -n 2
_arguments '1:plugin-name: _asdf__installed_plugins' '2:package-version:{_asdf__installed_versions_of ${words[2]}}'
;;
(shell|local|global)
compset -n 2
_arguments '1:plugin-name: _asdf__installed_plugins' '2:package-version:{_asdf__installed_versions_of_plus_system ${words[2]}}'
;;
(where)
# version is optional
compset -n 2
_arguments '1:plugin-name: _asdf__installed_plugins' '2::package-version:{_asdf__installed_versions_of ${words[2]}}'
;;
(which|shim-versions)
_wanted asdf-shims expl "ASDF Shims" compadd -- "${asdf_dir:?}/shims"/*(:t)
;;
(exec)
# asdf exec <shim-cmd> [<shim-cmd args ...>]
if (( CURRENT == 3 )); then
_wanted asdf-shims expl "ASDF Shims" compadd -- "${asdf_dir:?}/shims"/*(:t)
else
compset -n 3
_normal -p "asdf-shims-${words[3]}"
fi
;;
(env)
# asdf exec <shim-name> <arbitrary-cmd> [<cmd args ...>]
if (( CURRENT == 3 )); then
_wanted asdf-shims expl "ASDF Shims" compadd -- "${asdf_dir:?}/shims"/*(:t)
else
compset -n 4
_normal -p "asdf-shims-${words[3]}"
fi
;;
(update)
(( CURRENT == 3 )) && compadd -- --head
;;
esac
}