S58_rvm

#!/bin/zsh

get_rvm_prompt_info () {
  if test -s "$HOME/.rvm/scripts/rvm" && ( test -s "Rakefile" || test -s "Gemfile" || test -s ".rvmrc" ); then
    res=`rvm current`
    echo "${res} " 
  fi
}

#compdef rvm
#stolen from https://github.com/nono/dotfiles/blob/master/zsh/Completion/_rvm
_rvm () {
  local curcontext="$curcontext" state line cmds ret=1

  _arguments -C \
    '(- 1 *)'{-v,--version}'[display version information]' \
    '(-l|--level)'{-l,--level}'+[patch level to use with rvm use / install]:number' \
    '(--prefix)--prefix[path for all rvm files (~/.rvm/), with trailing slash!]:path:_files' \
    '(--bin)--bin[path for binaries to be placed (~/.rvm/bin/)]:path:_files' \
    '(--source)--source[src directory to use (~/.rvm/src/)]:path:_files' \
    '(--archives)--archives[directory for downladed files (~/.rvm/archives/)]:path:_files' \
    '-S[Specify a script file to attempt to load and run (rubydo)]:file:_files' \
    '-e[Execute code from the command line]:code' \
    '(-G)-G[root gem path to use]:path:_files' \
    '(--gems)--gems[Used to set the gems_flag, use with remove to remove gems]' \
    '(--archive)--archive[Used to set the archive_flag, use with remove to remove archive]' \
    '(--patch)--patch[With MRI Rubies you may specify one or more full paths to patches]' \
    '(-C|--configure)'{-C,--configure}'=[custom configure options]' \
    '(--nice)--nice[process niceness (for slow computers, default 0)]:number' \
    '(--ree)--ree-options[Options passed directly to ree ./installer on the command line]:options' \
    '(--head)--head[with update, updates rvm to git head version]' \
    '(--rubygems)--rubygems[with update, updates rubygems for selected ruby]' \
    '(--default)--default[with ruby select, sets a default ruby for new shells]' \
    '(--debug)--debug[Toggle debug mode on for very verbose output]' \
    '(--trace)--trace[Toggle trace mode on to see EVERYTHING rvm is doing]' \
    '(--force)--force[Force install, removes old install & source before install]' \
    '(--summary)--summary[Used with rubydo to print out a summary of the commands run]' \
    '(--latest)--latest[with gemset --dump skips version strings for latest gem]' \
    '(--gems)--gems[with uninstall/remove removes gems with the interpreter]' \
    '(--docs)--docs[with install, attempt to generate ri after installation]' \
    '(--reconfigure)--reconfigure[Force ./configure on install even if Makefile already exists]' \
    '1: :->cmds' \
    '*: :->args' && ret=0

  case $state in
    cmds)
      cmds=( ${(f)"$(_call_program commands rvm help 2> /dev/null | sed -e '/^Action/,/^Implementation/!d; / - /!d; s/^\W*\([-_a-zA-Z0-9]*\) *- /\1:/')"} )
      cmds+=( $(rvm list strings) )
      _describe -t commands 'rvm command' cmds && ret=0
      ;;
    args)
      case $line[1] in
        (use|uninstall|remove|list)
          _values -S , 'rubies' $(rvm list strings | sed -e 's/ruby-\([^) ]*\)-\([^) ]*\)/ruby-\1-\2 \1-\2 \1/g') default system && ret=0
          ;;
        (install|fetch)
          _values -S , 'rubies' $(rvm list known_strings) && ret=0
          ;;
        gemset)
          if (( CURRENT == 3 )); then
            _values 'gemset_commands' $(rvm gemset | sed -e '/where/!d; s/^.*[{]\(.*\)[}].*$/\1/; s/,/ /g')
          else
            _values -S , 'gemsets' $(rvm gemset list | grep -v gemset 2>/dev/null)
          fi
          ret=0
          ;;
        package)
          if (( CURRENT == 3 )); then
            _values 'package_commands' $(rvm package | sed -e '/Usage/!d; s/^.*[{]\(.*\)[}] [{].*$/\1/; s/,/ /g')
          else
            _values 'packages' $(rvm package | sed -e '/Usage/!d; s/^.*[}] [{]\(.*\)[}].*$/\1/; s/,/ /g')
          fi
          ret=0
          ;;
        *)
          (( ret )) && _message 'no more arguments'
          ;;
      esac
      ;;
  esac

  return ret
}

compdef _rvm rvm