From 4ee92fca43805a3d4a04f453e79a06e85b9810e9 Mon Sep 17 00:00:00 2001
From: Sam Stephenson <sam@37signals.com>
Date: Thu, 4 Aug 2011 00:45:40 -0500
Subject: [PATCH] Bare `rbenv init` prints a message; `rbenv init -` prints the
 init script

---
 libexec/rbenv-init | 35 ++++++++++++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 3 deletions(-)

diff --git a/libexec/rbenv-init b/libexec/rbenv-init
index d8a4a0bc..43e1339d 100755
--- a/libexec/rbenv-init
+++ b/libexec/rbenv-init
@@ -1,9 +1,14 @@
 #!/usr/bin/env bash -e
 
-shell=$1
+print=""
+if [ "$1" = "-" ]; then
+  print=1
+  shift
+fi
 
+shell="$1"
 if [ -z "$shell" ]; then
-  shell=$(basename $SHELL)
+  shell="$(basename "$SHELL")"
 fi
 
 abs_dirname() {
@@ -19,9 +24,33 @@ abs_dirname() {
   pwd
   cd "$cwd"
 }
+
 root="$(abs_dirname "$0")/.."
 
-echo 'export PATH="$HOME/.rbenv/shims:$PATH"'
+if [ -z "$print" ]; then
+  case "$shell" in
+  bash )
+    profile='~/.bash_profile'
+    ;;
+  zsh )
+    profile='~/.zshrc'
+    ;;
+  * )
+    profile='your profile'
+    ;;
+  esac
+
+  { echo "# Load rbenv automatically by adding"
+    echo "# the following to ${profile}:"
+    echo
+    echo 'eval "$(rbenv init -)"'
+    echo
+  } >&2
+
+  exit 1
+fi
+
+echo 'export PATH="${HOME}/.rbenv/shims:${PATH}"'
 
 if [ "$shell" = "bash" ]; then
   echo "source \"$root/completions/rbenv.bash\""