From 730111f468476af8bcc814e43a9520ee1d81ae0b Mon Sep 17 00:00:00 2001
From: "Joe H. Rahme" <joehakimrahme@gmail.com>
Date: Fri, 13 Nov 2015 12:52:08 +0100
Subject: [PATCH] Restore custom dir hierarchy in theme plugin

Themes are now searched for in the following directories (in order):

* $ZSH/custom/themes
* $ZSH/custom
* $ZSH/themes

Search will always return first match found.
---
 plugins/themes/themes.plugin.zsh | 35 ++++++++++++++++++++++++--------
 1 file changed, 27 insertions(+), 8 deletions(-)

diff --git a/plugins/themes/themes.plugin.zsh b/plugins/themes/themes.plugin.zsh
index 1681f684..ac95ff4c 100644
--- a/plugins/themes/themes.plugin.zsh
+++ b/plugins/themes/themes.plugin.zsh
@@ -1,23 +1,42 @@
+THEMEPATH=("$ZSH_CUSTOM/themes" "$ZSH_CUSTOM" "$ZSH/themes")
+
 function theme
 {
     if [ -z "$1" ] || [ "$1" = "random" ]; then
-	themes=($ZSH/themes/*zsh-theme)
+	# Load every theme in the themepath
+	# TODO remove the shadowed themes from the list
+	# i.e: if I have a custom theme foo overshadowing
+	# "$ZSH/themes/foo.zsh-theme", the latter shouldn't
+	# be part of this list.
+	themes=($(find $THEMEPATH -name '*.zsh-theme'))
+
 	N=${#themes[@]}
 	((N=(RANDOM%N)+1))
-	RANDOM_THEME=${themes[$N]}
-	source "$RANDOM_THEME"
-	echo "[oh-my-zsh] Random theme '$RANDOM_THEME' loaded..."
+	random_theme=${themes[$N]}
+	source "$random_theme"
+	echo "[oh-my-zsh] Random theme '$random_theme' loaded."
     else
-	if [ -f "$ZSH_CUSTOM/themes/$1.zsh-theme" ]
+	theme_name="$1.zsh-theme"
+	match=""
+	for dir in $THEMEPATH
+	do
+	    # Keep looking until we find a matching theme
+	    if [ -f "$dir/$theme_name" ] && [ -z "$match" ]
+	    then
+		match=$dir/$theme_name
+	    fi
+	done
+
+	if [ -n "$match" ]
 	then
-	    source "$ZSH_CUSTOM/themes/$1.zsh-theme"
+	    source "$match"
 	else
-	    source "$ZSH/themes/$1.zsh-theme"
+	    echo "[oh-my-zsh]Couldn't find theme $1."
 	fi
     fi
 }
 
 function lstheme
 {
-    find "$ZSH"/themes "$ZSH_CUSTOM"/themes -name '*.zsh-theme' | sort -u | sed 's:.*/\(.*\)\.zsh-theme:\1:'
+    find $THEMEPATH -name '*.zsh-theme' | sort -u | sed 's:.*/\(.*\)\.zsh-theme:\1:'
 }