From 5ccf9a09a5d2ae09f533ae23783b0426c26569a2 Mon Sep 17 00:00:00 2001
From: Andrew Janke <janke@pobox.com>
Date: Sat, 28 Feb 2015 01:46:10 -0500
Subject: [PATCH] Give install.sh ability to install from forked & branched
 repos, instead of just main robbyrussel repo. This will support testing of
 changes to the core installation code: you'll be able to do a roundtrip test
 of install and uninstall using the working code on your branch. Controlled by
 passing $REPO and $BRANCH environment variables to install.sh.

---
 tools/install.sh | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/tools/install.sh b/tools/install.sh
index bd4c55749..a323365fb 100755
--- a/tools/install.sh
+++ b/tools/install.sh
@@ -1,8 +1,19 @@
+# install.sh - oh-my-zsh installation script
+#
+# Respects these environment variables for tweaking the installation process:
+#   REPO    - name of the GitHub repo to install from (default: robbyrussel/oh-my-zsh)
+#   BRANCH  - branch to check out immediately after install
+# These options are for use by oh-my-zsh developers, to facilitate testing of
+# changes to the core oh-my-zsh installation code. Normal users should not set them.
+
 set -e
 
 if [ ! -n "$ZSH" ]; then
   ZSH=~/.oh-my-zsh
 fi
+if [ ! -n "$REPO" ]; then
+  REPO=robbyrussell/oh-my-zsh
+fi
 
 if [ -d "$ZSH" ]; then
   echo "\033[0;33mYou already have Oh My Zsh installed.\033[0m You'll need to remove $ZSH if you want to install"
@@ -10,7 +21,10 @@ if [ -d "$ZSH" ]; then
 fi
 
 echo "\033[0;34mCloning Oh My Zsh...\033[0m"
-hash git >/dev/null 2>&1 && env git clone --depth=1 https://github.com/robbyrussell/oh-my-zsh.git $ZSH || {
+if [ -n "$BRANCH" ]; then
+  BR_OPT="--branch $BRANCH"
+fi
+hash git >/dev/null 2>&1 && env git clone --depth=1 $BR_OPT https://github.com/$REPO.git $ZSH || {
   echo "git not installed"
   exit
 }