#! /bin/bash

set -e

# runmirrors script for Debian
# Based losely on existing scripts, written by an unknown number of
# different people over the years.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; version 2.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

# Read our config file
NAME="`basename $0`"

# In case we are called with an argument we look for a different configuration.
if [ -n "$1" ]; then
	NAME="${NAME}-$1"
fi

. ${HOME}/etc/${NAME}.conf

# Make sure some variables are always in the environment
export HOME LOGNAME USER PATH

# Source our common functions
. ${HOME}/etc/common

# Set sane defaults if the configfile didn't do that for us.
# The directory for our logfiles
LOGDIR=${LOGDIR:-"${HOME}/log"}
# Our own logfile
LOG=${LOG:-"${LOGDIR}/${NAME}.log"}
# How to rotate the log
SAVELOG=${SAVELOG:-"savelog -t -c 14"}
# Amount of delay between mirrors if the mirror file contains a DELAY
DELAY=${DELAY:-"60"}
# Our mirrorfile
MIRRORS=${MIRRORS:-"${HOME}/etc/${NAME}.mirror"}
# used by log()
PROGRAM=${PROGRAM:-"${NAME}-$(hostname -s)"}

# Make sure we have our log directory
mkdir -p ${LOGDIR}

# Some sane defaults
cd ${HOME}
umask 002

trap 'log "Mirrorpush done" >> ${LOG}; ${SAVELOG} "${LOG}" > /dev/null' EXIT

log "Pushing leaf mirrors" >> ${LOG}

if [ -n "${HOOK1}" ]; then
	log "Running hook1: ${HOOK1}" >> ${LOG}
	${HOOK1}
	result=$?
	log "Back from hook1, got returncode ${result}" >> ${LOG}
fi

# From here on we do *NOT* want to exit on errors. We don't want to
# stop pushing mirrors just because we can't reach one atm.
set +e

# Now read our mirrorfile and push the mirrors defined in there.
# We use grep to easily sort out all lines having a # in front of them or are empty.
egrep -v '^[[:space:]]*(#|$)' ${MIRRORS} |
while read MLNAME MHOSTNAME MUSER MPROTO MKEYFILE; do
	if [ "x${MLNAME}x" = "xDELAYx" ]; then
		# We should wait a bit.
		log "Delay of ${DELAY} seconds requested, sleeping" >> ${LOG}
		sleep ${DELAY}
		continue
	fi

	# Process the two options that can be left blank in the config
	if [ -z ${MPROTO} ]; then
		MPROTO=2
	fi
	if [ -z ${MKEYFILE} ]; then
		MKEYFILE="${HOME}/${KEYFILE}"
	fi
	# Now, people can do stupid things and leave out the protocol, but
	# define a keyfile...
	if [ ${MPROTO} -ne 1 ] && [ ${MPROTO} -ne 2 ]; then
		error "Need a correct ssh protocol version for ${MLNAME}, skipping" >> ${LOG}
		continue
	fi

	# And finally, push the mirror
	log "Trigger ${MLNAME}" >> ${LOG}
	signal ${MLNAME} ${MHOSTNAME} ${MUSER} ${MPROTO} ${MKEYFILE}
	log "Trigger for ${MLNAME} done" >> ${LOG}
	if [ -n "${HOOK2}" ]; then
		log "Running hook2: ${HOOK2}" >> ${LOG}
		${HOOK1}
		result=$?
		log "Back from hook2, got returncode ${result}" >> ${LOG}
	fi
done

if [ -n "${HOOK3}" ]; then
	log "Running hook3: ${HOOK3}" >> ${LOG}
	${HOOK3}
	result=$?
	log "Back from hook3, got returncode ${result}" >> ${LOG}
fi

exit 0