# -*- mode: sh; sh-shell-file: bash -*-

_scoop() {

  local tasks='alias      (Manage scoop aliases)

bucket     (Manage Scoop buckets)

cache      (Show or clear the download cache)

cat        (Show content of specified manifest)

checkup    (Check for potential problems)

cleanup    (Cleanup apps by removing old versions)

config     (Get or set configuration values)

create     (Create a custom app manifest)

depends    (List dependencies for an app)

download   (Download apps in the cache folder and verify hashes)

export     (Exports (an importable) list of installed apps)

help       (Show help for a command)

hold       (Hold an app to disable updates)

home       (Opens the app homepage)

info       (Display information about an app)

install    (Install apps)

list       (List installed apps)

prefix     (Returns the path to the specified app)

reset      (Reset an app to resolve conflicts)

search     (Search available apps)

shim       (Manipulate Scoop shims)

status     (Show status and check for new app versions)

unhold     (Unhold an app to enable updates)

uninstall  (Uninstall an app)

update     (Update apps, or Scoop itself)

virustotal (Look for apps hash on virustotal.com)

which      (Locate a shim/executable)'

  local cur prev

  cur=${COMP_WORDS[COMP_CWORD]}

  [[ $COMP_CWORD -gt 1 ]] && prev=${COMP_WORDS[COMP_CWORD-1]}

  if [[ $COMP_CWORD = 1 ]]; then

    local OIFS="$IFS"

    local IFS=$'\n'

    COMPREPLY=( $(compgen -W '$tasks' -- "$cur") )

    if [[ ${#COMPREPLY[@]} -le 1 ]]; then

      COMPREPLY=( ${COMPREPLY[0]%% *} )

    fi

    IFS=$OIFS

    return

  fi

  local cmd="${COMP_WORDS[1]}"

  case "$cmd" in

    cat|depends|download|export|hold|home|info|install|prefix|reset|status|unhold|uninstall|update|virustotal|which)

      local dir=`command -v scoop`

      dir=${dir%/shims/*}/buckets

      [[ -d "$dir" ]] || return

      local olddir="$PWD"

      cd $dir

      local -a jsons=( */bucket/"$cur"*.json )

      local -a pkgs=( "${jsons[@]##*/}" )

      COMPREPLY=( "${pkgs[@]%.json}" )

      cd "$olddir"

      return

    ;;

    bucket)

      if [[ $COMP_CWORD = 2 ]]; then

        COMPREPLY=( $(compgen -W 'add list known rm' -- "$cur") )

        return

      fi

      local subcmd="${COMP_WORDS[2]}"

      case "$subcmd" in

        list|known)

          COMPREPLY=( )

          return

          ;;

        add|rm)

          local -a buckets=( `scoop bucket known` )

          local cr=$'\r'

          buckets=( "${buckets[@]/$cr/}" )

          COMPREPLY=( $(compgen -W '${buckets[@]}' -- "$cur") )

          return

          ;;

      esac

  esac

}

complete -F _scoop scoop