#!/usr/bin/env bash

_asdf () {

local cur=${COMP_WORDS[COMP_CWORD]}

local cmd=${COMP_WORDS[1]}

local prev=${COMP_WORDS[COMP_CWORD-1]}

local plugins=$(asdf plugin-list | tr '\n' ' ')

COMPREPLY=()

case "$cmd" in

plugin-update)

COMPREPLY=($(compgen -W "$plugins --all" -- $cur))

;;

plugin-remove|which|list|list-all)

COMPREPLY=($(compgen -W "$plugins" -- $cur))

;;

install)

if [[ "$plugins" == *"$prev"* ]] ; then

local versions=$(asdf list-all $prev)

COMPREPLY=($(compgen -W "$versions" -- $cur))

else

COMPREPLY=($(compgen -W "$plugins" -- $cur))

fi

;;

uninstall|where|reshim)

if [[ "$plugins" == *"$prev"* ]] ; then

local versions=$(asdf list $prev)

COMPREPLY=($(compgen -W "$versions" -- $cur))

else

COMPREPLY=($(compgen -W "$plugins" -- $cur))

fi

;;

*)

local cmds='plugin-add plugin-list plugin-remove plugin-update install uninstall which where list list-all reshim'

COMPREPLY=($(compgen -W "$cmds" -- $cur))

;;

esac

return 0

}

complete -F _asdf asdf