qs-init.sh

#!/bin/sh

##
## USAGE: ./qs-init.sh --client UNIQUE_SLUG --git-repo GIT_REMOTE [--wxr WXR_TO_IMPORT] [--theme DIRECTORY_NAME] [--up]
## Include `--up` when setting up a fresh instance of this VM.
##

usage() {
	echo "
Usage: $(basename "$0") --client UNIQUE_SLUG --git-repo GIT_REMOTE [--wxr WXR_TO_IMPORT] [--theme DIRECTORY_NAME] [--up]

	--client	The client slug of the site we're going to work on next
	--git-repo	The location of the git repo we're going to work on next
	--wxr		The location of a WXR if we have one to import
	--up		Start the VM
	--help		Output usage information
	"
}

# Kicking things off!
printf '\nPreparing to initialize the VIP Go Quickstart environment...\n\n'

# Get the local and Git versions to compare
QS_VERSION=`cat .version.txt`
if [ -f .version-current.txt ]; then
	QS_VERSION_CURRENT=`cat .version-current.txt`
else
	QS_VERSION_CURRENT=0
fi

if [ "$QS_VERSION" -gt "$QS_VERSION_CURRENT" ]; then
	vagrant provision
fi

# Note the latest version we've updated to
echo $QS_VERSION > .version-current.txt

# Parse args and set necessary envars
printf '1) Parsing arguments...\n\n'
client=''
client_git_repo=''
client_git_branch='master'
theme=0
wxr=0
needs_to_up=0

# This while loop will go forever until we break out of it or exit, 
# as the : operator aliases to `true`.
while :; do
	case $1 in
		--client)
			if [ -n "$2" ]; then
				client=$2
				shift 2
				continue
			else
				printf 'ERROR: "--client" is a required argument.\n' >&2
				usage
				exit 1
			fi
			;;

		--git-repo)
			if [ -n "$2" ]; then
				client_git_repo=$2
				shift 2
				continue
			else
				printf 'ERROR: "--git-repo" is a required argument.\n' >&2
				usage
				exit 1
			fi
			;;

		--git-branch)
			if [ -n "$2" ]; then
				client_git_branch=$2
				shift 2
				continue
			else
				client_git_branch="master"
				shift 2
				continue
			fi
			;;

		--wxr)
			if [ -n "$2" ]; then
				rm data/import.xml
				cp "$2" data/import.xml
				printf 'CONTENT: Using specified WXR to initialize environment.\n\n'
				wxr=1
				shift 2
				continue
			fi
			;;

		--theme)
			if [ -n "$2" ]; then
				theme=$2
				shift 2
				continue
			fi
			;;

		--up)
			needs_to_up=1
			;;
			
		--help)
			usage
			exit
			;;

		*)
			if [ "$1" ]; then
				echo "\n\nERROR: Unknown parameter or bad parameter value detected: $1 $2\n"
				usage
				exit 1
			fi
			break
	esac

	shift
done

if [ -z "$client" ]; then
	printf 'ERROR: "--client" is a required argument.\n' >&2
	usage
	exit 1
fi

if [ -z "$client_git_repo" ]; then
	printf 'ERROR: "--git-repo" is a required argument.\n' >&2
	usage
	exit 1
fi

if [ "$wxr" = 0 ]; then
	git checkout -- data/import.xml
	printf 'CONTENT: Using default WXR to initialize environment.\n\n'
fi

export VIP_GO_CLIENT=$client
export VIP_GO_CLIENT_GIT=$client_git_repo
export VIP_GO_CLIENT_GIT_BRANCH=$client_git_branch
export VIP_GO_THEME=$theme

printf "CLIENT: %s\nGIT REMOTE: %s\nGIT BRANCH: %s\nTHEME: %s\nVAGRANT UP REQUESTED: %d\n\n" "$client" "$client_git_repo" "$client_git_branch" "$theme" "$needs_to_up"

# Require confirmation before destructive actions that follow
read -p "Continue (y/n)? " CONT
if [ "$CONT" != "y" ]; then
	printf 'Aborting at your request.\n';
	exit 0;
fi

# Load client's custom code into VM's NFS shares
# Can't simply delete `go-client-repo` and replace it, as removing the synced folders sends Vagrant into a tizzy
printf '2) Loading new client code...\n\n'
rm -rf go-client-repo-new/
rm -rf go-client-repo/.git/
rm -rf go-client-repo/languages/*
rm -rf go-client-repo/plugins/*
rm -rf go-client-repo/themes/*
git clone "$client_git_repo" go-client-repo-new
EXIT_CODE=$?
if [ 0 != $EXIT_CODE ]; then
	echo "\n\nERROR: Cloning git repository at $client_git_repo failed with the error code $EXIT_CODE."
	exit 210
fi
cd go-client-repo-new
git checkout $client_git_branch
cd ..
EXIT_CODE=$?
if [ 0 != $EXIT_CODE ]; then
	echo "\n\nERROR: Checkout out branch $client_git_branch failed with the error code $EXIT_CODE."
	exit 220
fi
cp -r go-client-repo-new/. go-client-repo
rm -rf go-client-repo-new/

# Ensure theme directory exists, otherwise exit with an error
if [ "$theme" != 0 ] && [ ! -d "./go-client-repo/themes/$theme" ]; then
	printf '\n\nERROR: Theme directory "%s" not found. Please check your entry, or omit the "--theme" argument, and try again.\n' "$theme" >&2
	usage
	exit 1
fi

# Ensure optional languages directory exists
# Vagrant doesn't respond well if the directory exists sometimes but not always
if [ ! -d "./go-client-repo/languages/" ]; then
	mkdir ./go-client-repo/languages/
fi

# Provision the VM
if [ $needs_to_up = 1 ]; then
	printf '\n3) Starting to provision VM. vagrant up can take some time...\n\n'
	vagrant up
else
	printf '\n3) Starting to re-provision VM...\n\n'
	vagrant provision
fi

# Done!
printf '\n4) DONE!\nYour VIP Go Quickstart is ready at http://go-vip.local/\n'
printf 'Synchronized git checkout is at ./go-client-repo/\n'