zda-update

#!/bin/bash

function exit_with_error() {
	echo "$1" >&2
	exit 2
}

function print_usage() {
	echo "${usagestring}"
	exit 0
}

progname=$(basename $0)
usagestring="usage: ${progname} -d <directory> [ -s -k <key> ]"

# Execute getopt
ARGS=$(getopt d:hk:s $*)

#Bad arguments
if [ $? -ne 0 ]
then
	exit_with_error "${usagestring}"
fi

eval set -- "$ARGS"

while true
do
	case "$1" in
	-d|--directory)
		shift
		if [ -n "$1" ]
		then
			articles_dir=$1
			shift
		fi
		;;
	-k|--key)
		shift
		if [ -n "$1" ]
		then
			keyfile=$1
			shift
		fi
		;;
	-s|--ssh)
		shift
		ssh=true
		;;
	-h|--help)
		shift
		print_usage
		;;
    	--)
		shift
		break
		;;
	esac
done

if [ -z "${articles_dir}" ]
then
	exit_with_error "Please specify a directory containing Zendesk articles with the -d flag."
elif [ ! -e "${articles_dir}" ]
then
	exit_with_error "${articles_dir}: no such file or directory"
elif [ ! -d "${articles_dir}" ]
then
	exit_with_error "${articles_dir}: not a directory"
fi

cwd=$(pwd -P)
exitcode=0

# Add an SSH key to ssh-agent if possible and required.
if [ "${ssh}" = true ]
then
	is_registered=$(ssh-add -l | grep ${keyfile})
	if [[ -z "${is_registered}" ]]
	then
		ssh-add ${keyfile} || exit_with_error "Cannot register SSH identity with ssh-agent"
	fi
fi

serverbranch="master"

for localrepo in ${articles_dir}
do
	echo "Updating local repository in ${localrepo}"
	cd ${localrepo}

	if [[ ! -d ".git" ]]
	then
		echo "Warning: ${localrepo}: not a git repository. Skipping."
		exitcode=1
		continue
	fi

	git fetch --prune --all
	allremotebranches=$(git branch -r | awk '{print $1}')
	# Note that the sed command on a Mac (which this script is designed for)
	# observes different syntax to that on Linux.
	current_branch=$(git branch | grep "^*" | sed -E -e 's/^\*[[:space:]]+//')

	if [[ "${current_branch}" != "${serverbranch}" ]]
	then
		git checkout "${serverbranch}"
		if [ $? -ne 0 ]
		then
			echo "Warning: ${localrepo}: cannot switch to \"${serverbranch}\" branch. Skipping."
			exitcode=1
			continue
		fi
	fi

	git pull
	if [ $? -ne 0 ]
	then
		echo "Warning: ${localrepo}: cannot update \"${serverbranch}\" branch from remote. Skipping."
		exitcode=1
		continue
	fi

	for branch in $(git branch | sed -e 's/^\**\s*//')
	do
		if [[ "${branch}" == "${serverbranch}" ]]
		then
			continue
		fi
		remote_branch=$(git branch -vv | sed -E -e 's/^\**[[:space:]]*//' | grep "^${branch}" | awk '{print $3}' | grep "^\[.*[]:]$" | sed -e 's/^\[\([^]:]*\)[]:]$/\1/')
		if [[ -z "${remote_branch}" ]]
		then
			echo "Warning: local branch \"${branch}\" is not tracking a remote branch. Skipping."
			exitcode=1
			continue
		fi
		remote_exists=false
		for rb in ${allremotebranches}
		do
			if [[ "${remote_branch}" == "${rb}" ]]
			then
				remote_exists=true
			fi
		done
		if [[ "${remote_exists}" == false ]]
		then
			echo "Warning: remote branch \"${remote_branch}\", tracked by local branch \"${branch}\", does not exist. Skipping."
			exitcode=1
			continue
		fi

		remote=$(echo ${remote_branch} | sed -e 's,/.*$,,')
		actual_remote_branch=$(echo ${remote_branch} | sed -e 's,^.*/,,')

		git checkout "${branch}"
		if [ $? -ne 0 ]
		then
			echo "Warning: ${localrepo}: cannot switch to \"${branch}\" branch. Skipping."
			exitcode=1
			continue
		fi

		git merge "${serverbranch}"
		if [ $? -ne 0 ]
		then
			echo "Warning: could not merge from \"${serverbranch}\" into \"${branch}\". Skipping."
			exitcode=1
			continue
		fi

		branch_status=$(git status | tail -n 1)
		echo "${branch}: ${branch_status}"
		if [[ ! "${branch_status}" =~ "nothing to commit" ]]
		then
			echo "Warning: local branch \"${branch}\" has uncommitted changes. Skipping."
			exitcode=1
			continue
		fi

		git push ${remote} ${actual_remote_branch}
	done

	git checkout ${current_branch}
done

cd ${cwd}

if [[ "${exitcode}" -gt 0 ]]
then
	echo "There were some warnings. Please review the output."
fi
exit ${exitcode}