From 283e67b57e8ab0bbbe504aab6866729b0035186a Mon Sep 17 00:00:00 2001
From: Sam Stephenson <sam@37signals.com>
Date: Thu, 27 Dec 2012 13:41:55 -0600
Subject: [PATCH] When the ruby shim is invoked with a script, set RBENV_DIR to
 the script's dirname

---
 libexec/rbenv-rehash | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/libexec/rbenv-rehash b/libexec/rbenv-rehash
index eebc4d3c..2b5ffecc 100755
--- a/libexec/rbenv-rehash
+++ b/libexec/rbenv-rehash
@@ -37,8 +37,25 @@ create_prototype_shim() {
   cat > "$PROTOTYPE_SHIM_PATH" <<SH
 #!/usr/bin/env bash
 set -e
+[ -n "\$RBENV_DEBUG" ] && set -x
+
+program="\${0##*/}"
+if [ "\$program" = "ruby" ]; then
+  for arg; do
+    case "\$arg" in
+    -e* | -- ) break ;;
+    */* )
+      if [ -f "\$arg" ]; then
+        export RBENV_DIR="\${arg%/*}"
+        break
+      fi
+      ;;
+    esac
+  done
+fi
+
 export RBENV_ROOT="$RBENV_ROOT"
-exec rbenv exec "\${0##*/}" "\$@"
+exec rbenv exec "\$program" "\$@"
 SH
   chmod +x "$PROTOTYPE_SHIM_PATH"
 }