#compdef svm
# ------------------------------------------------------------------------------
# Copyright (c) 2011 Github zsh-users - https://github.com/zsh-users
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of the zsh-users nor the
#       names of its contributors may be used to endorse or promote products
#       derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL ZSH-USERS BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# ------------------------------------------------------------------------------
# Description
# -----------
#
#  Completion script for svm, Scala2 version manager (https://github.com/yuroyoro/svm)
#
# ------------------------------------------------------------------------------
# Authors
# -------
#
#  * Hideaki Miyake (https://github.com/mollifier)
#
# ------------------------------------------------------------------------------

_svm() {
  local context curcontext="$curcontext" state line ret=1
  typeset -A opt_args

  _arguments -C \
    '(- *)-h[show this usage information]' \
    '-c[show the currently use scala version]' \
    "-l[show the scala version installed in svm_path(default is ${HOME}/.svm)]" \
    '-v[show the available scala version not installed]' \
    '-i[install specific scala version]: :_svm_not_installed_scala_versions' \
    '-r[uninstall specific scala version and remove their sources]: :_svm_installed_scala_versions' \
    '(-s -u)'{-s,-u}'[setup to use a specific scala version]: :_svm_not_selected_scala_versions' \
    '1: :_svm_commands' \
    '*:: :->args' && ret=0

  case $state in
    (args)
      # scala version number
      case $words[1] in
        (install)
          # install not installed version
          _arguments \
            '--docs[download scala-devel-docs]' \
            '--sources[download scala-sources]' \
            '1: :_svm_not_installed_scala_versions' \
            && ret=0
          ;;
        (update-latest)
          # update nightly build scala version
          _arguments \
            '--docs[download scala-devel-docs]' \
            '--sources[download scala-sources]' \
            && ret=0
          ;;
        (remove|uninstall)
          # remove installed version
          _arguments \
            '1: :_svm_installed_scala_versions' \
            && ret=0
          ;;
        (switch|use)
          # use installed version
          _arguments \
            '1: :_svm_not_selected_scala_versions' \
            && ret=0
          ;;
      esac

      ;;  # end args
  esac

  return ret
}

(( $+functions[_svm_commands] )) ||
_svm_commands() {
  case $PREFIX in
    (u*)
      local -a synonyms=(
        'uninstall:uninstall specific scala version and remove their sources'
        'use:setup to use a specific scala version'
        'update-latest:install or update nightly build scala version'
      )

      _describe -t actions 'svm actions' synonyms
      ;;
    (*)
      local -a commands=(
        'help:show this usage information'
        'current:show the currently use scala version'
        "list:show the scala version installed in svm_path(default is ${HOME}/.svm)"
        "versions:show the available scala version not installed"
        'install:install specific scala version'
        'remove:uninstall specific scala version and remove their sources'
        'switch:setup to use a specific scala version'
        'update-latest:install or update nightly build scala version'
        'latest:setup to use nightly build scala version'
        'stable:setup to use stable(x.x.x.final) scala version'
        'self-update:update svm itself'
      )

      _describe -t actions 'svm actions' commands
      ;;
  esac
}

# installed scala versions
(( $+functions[_svm_installed_scala_versions] )) ||
_svm_installed_scala_versions() {
  # collect lines starts with digit
  local -a installed_versions=( ${(M)${(@f)"$(_call_program installed svm list)"}:#[[:digit:]]*} )
  _describe -t installed "installed versions" installed_versions
}

# installed and not selected scala versions
(( $+functions[_svm_not_selected_scala_versions] )) ||
_svm_not_selected_scala_versions() {
  local current_version=$(_call_program current svm current | sed 's/currently version is //')
  local -a not_selected_versions=( ${(M)${(@f)"$(_call_program installed svm list)"}:#[[:digit:]]*} )

  # remove current version
  not_selected_versions=( ${not_selected_versions:#$current_version})
  _describe -t installed "not selected versions" not_selected_versions
}

# not installed scala versions
(( $+functions[_svm_not_installed_scala_versions] )) ||
_svm_not_installed_scala_versions() {
  local -a not_installed_versions=( ${(M)${(@f)"$(_call_program installed svm versions)"}:#[[:digit:]]*} )

  _describe -t notinstalled "not installed versions" not_installed_versions
}

_svm "$@"

# Local Variables:
# mode: Shell-Script
# sh-indentation: 2
# indent-tabs-mode: nil
# sh-basic-offset: 2
# End:
# vim: ft=zsh sw=2 ts=2 et
