source zsh autocomplete when using zshell

RiotGamesCookbooks · Aug 29, 2013 · 903222d · 903222d
1 parent 9c624d3
commit 903222d
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/templates/default/rbenv.sh.erb b/templates/default/rbenv.sh.erb
@@ -1,4 +1,16 @@
 export RBENV_ROOT=<%= @rbenv_root %>
 export PATH=$RBENV_ROOT/bin:<%= @ruby_build_bin_path %>:$PATH
-source $RBENV_ROOT/completions/rbenv.bash
+
+case $SHELL in
+*/zsh)
+  source $RBENV_ROOT/completions/rbenv.zsh
+  ;;
+*/bash)
+  source $RBENV_ROOT/completions/rbenv.bash
+  ;;
+*)
+  source $RBENV_ROOT/completions/rbenv.bash
+  ;;
+esac
+
 eval "$(rbenv init -)"