Initial support for favorite themes

lib/functions.zsh

@@ -50,3 +50,31 @@ function extract() {
   fi
 }
 
+function get_current_theme() {
+    echo `basename $RANDOM_THEME`
+}
+
+function like_theme() {
+    theme_name=`get_current_theme`
+    link_name="$FAVORITE_THEMES_DIR/$theme_name"
+
+    if [[ ! -L $link_name ]]; then
+        ln -s $RANDOM_THEME $link_name
+        echo "Added $theme_name to favorites"
+    else
+        echo "$theme_name already a favorite!"
+    fi
+}
+
+function unlike_theme() {
+    theme_name=`get_current_theme`
+    link_name="$FAVORITE_THEMES_DIR/$theme_name"
+
+    if [[ -L $link_name ]]; then
+        # FIXME: What happens if we remove the last theme we have?
+        rm $link_name
+        echo "Removed $theme_name from favorites"
+    else
+        echo "$theme_name isn't a favorite"
+    fi
+}

oh-my-zsh.sh

@@ -1,5 +1,11 @@
 # Initializes Oh My Zsh
 
+# Create another directory under $ZSH called favorite_themes
+# Every time a user 'likes' a theme:
+#   Create symlink to it in favorite_themes (if it doesn't already exist)
+# If user is using 'random' theme AND there are contents in the favorite_themes
+# directory, use that as the 'themes' variable, otherwise use all
+
 # add a function path
 fpath=($ZSH/functions $ZSH/completions $fpath)
 
@@ -27,16 +33,30 @@ done
 # Load all of your custom configurations from custom/
 for config_file ($ZSH/custom/*.zsh) source $config_file
 
+export FAVORITE_THEMES_DIR="$ZSH/themes/favorites/"
+
 # Load the theme
 # Check for updates on initial load...
 if [ "$ZSH_THEME" = "random" ]
 then
-  themes=($ZSH/themes/*zsh-theme)
+  # Allow users to randomize themes with only their favorites
+  if [ ! -d $FAVORITE_THEMES_DIR ]; then
+      mkdir $FAVORITE_THEMES_DIR
+  fi
+
+  # Only randomize with liked themes if there are any
+  if [ `ls $FAVORITE_THEMES_DIR | wc -w` -gt 0 ]; then
+    themes=($FAVORITE_THEMES_DIR/*zsh-theme)
+  else
+    themes=($ZSH/themes/*zsh-theme)
+  fi
+
   N=${#themes[@]}
   ((N=(RANDOM%N)+1))
   RANDOM_THEME=${themes[$N]}
   source "$RANDOM_THEME"
   echo "[oh-my-zsh] Random theme '$RANDOM_THEME' loaded..."
+  export RANDOM_THEME
 else
   source "$ZSH/themes/$ZSH_THEME.zsh-theme"
 fi