From b6ebc2f08bd36a8eeaac93719b6b1b3bae7d0f00 Mon Sep 17 00:00:00 2001
From: Jamis Buck <jamis@37signals.com>
Date: Wed, 3 Aug 2011 11:17:28 -0600
Subject: [PATCH] look for plugins to extend rehash and which

---
 libexec/rbenv-rehash | 28 +++++++++++++++++++++-------
 libexec/rbenv-which  |  8 ++++++++
 2 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/libexec/rbenv-rehash b/libexec/rbenv-rehash
index c0096616..96305478 100755
--- a/libexec/rbenv-rehash
+++ b/libexec/rbenv-rehash
@@ -1,14 +1,28 @@
 #!/usr/bin/env bash -e
 
-mkdir -p "${HOME}/.rbenv/shims"
-cd "${HOME}/.rbenv/shims"
-rm -f *
+make_shims() {
+  local glob="$@"
 
-for file in ../versions/*/bin/*; do
-  shim="${file##*/}"
-  cat > "$shim" <<SH
+  for file in $glob; do
+    local shim="${file##*/}"
+    cat > "$shim" <<SH
 #!/bin/sh
 exec rbenv exec $shim "\$@"
 SH
-  chmod +x "$shim"
+    chmod +x "$shim"
+  done
+}
+
+mkdir -p "${HOME}/.rbenv/shims"
+cd "${HOME}/.rbenv/shims"
+rm -f *
+
+make_shims ../versions/*/bin/*
+
+shopt -s nullglob
+RBENV_REHASH_PLUGINS=(/etc/rbenv.d/rehash/*.bash ${HOME}/.rbenv/rbenv.d/rehash/*.bash)
+shopt -u nullglob
+
+for script in $RBENV_REHASH_PLUGINS; do
+  source $script
 done
diff --git a/libexec/rbenv-which b/libexec/rbenv-which
index 7f308e2d..34ea534e 100755
--- a/libexec/rbenv-which
+++ b/libexec/rbenv-which
@@ -4,6 +4,14 @@ RBENV_VERSION="$(rbenv-version)"
 RBENV_COMMAND="$1"
 RBENV_COMMAND_PATH="${HOME}/.rbenv/versions/${RBENV_VERSION}/bin/${RBENV_COMMAND}"
 
+shopt -s nullglob
+RBENV_WHICH_PLUGINS=(/etc/rbenv.d/which/*.bash ${HOME}/.rbenv/rbenv.d/which/*.bash)
+shopt -u nullglob
+
+for script in $RBENV_WHICH_PLUGINS; do
+  source $script
+done
+
 if [ -x "$RBENV_COMMAND_PATH" ]; then
   echo "$RBENV_COMMAND_PATH"
 else