From 345600d4363458572cb9fa7a5a0ee69dc3a152fc Mon Sep 17 00:00:00 2001
From: Evan Sosenko <razorx@evansosenko.com>
Date: Wed, 21 Feb 2018 10:43:25 -0800
Subject: [PATCH] Respect RBENV_ROOT and PYENV_ROOT if set

---
 plugins/pyenv/pyenv.plugin.zsh | 4 ++++
 plugins/rbenv/rbenv.plugin.zsh | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/plugins/pyenv/pyenv.plugin.zsh b/plugins/pyenv/pyenv.plugin.zsh
index ec3ae9f5b..1c6e66c68 100644
--- a/plugins/pyenv/pyenv.plugin.zsh
+++ b/plugins/pyenv/pyenv.plugin.zsh
@@ -9,6 +9,10 @@ _pyenv-from-homebrew-installed() {
 FOUND_PYENV=0
 pyenvdirs=("$HOME/.pyenv" "/usr/local/pyenv" "/opt/pyenv")
 
+if [[ -n "$PYENV_ROOT" ]]; then
+  pyenvdirs=($PYENV_ROOT "${pyenvdirs[@]}")
+fi
+
 for pyenvdir in "${pyenvdirs[@]}" ; do
     if [ -d $pyenvdir/bin -a $FOUND_PYENV -eq 0 ] ; then
         FOUND_PYENV=1
diff --git a/plugins/rbenv/rbenv.plugin.zsh b/plugins/rbenv/rbenv.plugin.zsh
index 7430e9625..cb34b41a7 100644
--- a/plugins/rbenv/rbenv.plugin.zsh
+++ b/plugins/rbenv/rbenv.plugin.zsh
@@ -12,6 +12,10 @@ if _homebrew-installed && rbenv_homebrew_path=$(brew --prefix rbenv 2>/dev/null)
     fi
 fi
 
+if [[ -n "$RBENV_ROOT" ]]; then
+  rbenvdirs=($RBENV_ROOT "${rbenvdirs[@]}")
+fi
+
 for rbenvdir in "${rbenvdirs[@]}" ; do
   if [ -d $rbenvdir/bin -a $FOUND_RBENV -eq 0 ] ; then
     FOUND_RBENV=1