From 6b616fe2c7edf0193e037d94572880cd0be5941a Mon Sep 17 00:00:00 2001
From: Mike <mscalora@users.noreply.github.com>
Date: Fri, 21 Aug 2015 04:50:00 -0600
Subject: [PATCH] Add update and auto-update for customization directory

---
 lib/functions.zsh          | 4 ++++
 tools/check_for_upgrade.sh | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/lib/functions.zsh b/lib/functions.zsh
index 0d632a268..5f87f27c1 100644
--- a/lib/functions.zsh
+++ b/lib/functions.zsh
@@ -8,6 +8,10 @@ function uninstall_oh_my_zsh() {
 
 function upgrade_oh_my_zsh() {
   env ZSH=$ZSH /bin/sh $ZSH/tools/upgrade.sh
+  if [ -x $ZSH/custom/tools/upgrade.sh ]
+  then
+    env ZSH=$ZSH /bin/sh $ZSH/custom/tools/upgrade.sh
+  fi
 }
 
 function take() {
diff --git a/tools/check_for_upgrade.sh b/tools/check_for_upgrade.sh
index 5afd83c1e..188b09bfb 100644
--- a/tools/check_for_upgrade.sh
+++ b/tools/check_for_upgrade.sh
@@ -14,6 +14,10 @@ function _upgrade_zsh() {
   env ZSH=$ZSH /bin/sh $ZSH/tools/upgrade.sh
   # update the zsh file
   _update_zsh_update
+  if [ -x $ZSH/custom/tools/upgrade.sh ]
+  then
+    env ZSH=$ZSH /bin/sh $ZSH/custom/tools/upgrade.sh
+  fi
 }
 
 epoch_target=$UPDATE_ZSH_DAYS