#!/bin/sh
#
# Script run on the clients, to set them up in preparation for building
# packages.  This includes setting up parts of the /a/portbuild
# directory hierarchy, the portbuild script and the bindist.tar file
# for populating the build chroots.

if [ $# -lt 8 ]; then
    echo "usage: $0 portbuilddir arch branch buildid tmpdir portsmd5 srcmd5 phase [-nocopy] [-force]"
    exit 1
fi

pbd=$1
arch=$2
branch=$3
buildid=$4
tmpdir=$5
portsmd5=$6
srcmd5=$7
bindistmd5=$8
phase=$9
shift 9

precopy() {
    # Create directories and symlinks for later population

    # Timestamp of finished builds
    mkdir -p ${tmpdir}/stamp/

    # Prepare all directories, they will be populated by a rsync
    # push from the master
    mkdir -p ${pbd}/${arch}/clients/

    if [ "${buildid}" != "-" -a "${branch}" != "-" ]; then
	if [ ${nocopy} -eq 0 ]; then
	    mkdir -p ${builddir}
	fi
    fi
}

postcopy() {
    if [ "${buildid}" != "-" -a "${branch}" != "-" ]; then
        # Extract ports trees and cleanup
	if [ ${nocopy} -eq 0 ]; then

	    cd ${builddir} || return 1

	    # Unpack bindist
	    if [ -f bindist.tbz.md5 -a "${force}" -ne 1 ]; then
		localbindistmd5=$(awk '{print $4}' bindist.tbz.md5)
	    else
		localbindistmd5=0
	    fi

	    if [ ${localbindistmd5} != ${bindistmd5} -o ! -f bindist.tar ]; then
		if [ -f bindist.tar ]; then
		    rm -f bindist.tar
		fi
      		bzcat bindist.tbz > bindist.tar || return 1
	    fi

	    # Unpack ports
	    if [ -f ports-${buildid}.tbz.md5 -a "${force}" -ne 1 ]; then
		localportsmd5=$(awk '{print $4}' ports-${buildid}.tbz.md5)
	    else
		localportsmd5=0
	    fi

	    if [ ${localportsmd5} != ${portsmd5} -o ! -d ports ]; then
		if [ -d ports ]; then
		    mv ports ports~
		    mkdir ports
		    rm -rf ports~ &
		fi
      		tar xfj ports-${buildid}.tbz || return 1
	    fi

	    # Unpack src
	    if [ -f src-${buildid}.tbz.md5 -a "${force}" -ne 1 ]; then
		localsrcmd5=$(awk '{print $4}' src-${buildid}.tbz.md5)
	    else
		localsrcmd5=0
	    fi

	    if [ ${localsrcmd5} != ${srcmd5} -o ! -d src ]; then
		if [ -d src ]; then
		    mv src src~
		    mkdir src
		    rm -rf src~ &
		fi
      		tar xfj src-${buildid}.tbz || return 1
	    fi
	    touch .ready
	fi

        # Clean up the tmpdir
        # By now the portbuild.conf files are in place so we can source them

	. ${pbd}/${arch}/portbuild.conf
	if [ -f ${pbd}/${arch}/${branch}/builds/${buildid}/portbuild.conf ]; then
	    . ${pbd}/${arch}/${branch}/builds/${buildid}/portbuild.conf
	fi
	me=$(hostname)
	if [ -f ${pbd}/${arch}/portbuild.${me} ] ; then
	    . ${pbd}/${arch}/portbuild.${me}
	fi

	if [ "${buildid}" != "-" -a "${branch}" != "-" ]; then
	    mkdir -p ${wrkdir}/chroot
	    if [ "${use_zfs}" -eq 1 ]; then
		zbase=zroot/a
		zfs create ${zbase}/${branch} || true
		zfs create ${zbase}/${branch}/${buildid} || true
		zfs create ${zbase}/${branch}/${buildid}/world || true
		zfs create ${zbase}/${branch}/${buildid}/chroot || true
		tar xfpC ${builddir}/bindist.tar ${scratchdir}/${branch}/${buildid}/world
		tar xfpC ${pbd}/${arch}/clients/bindist-$(hostname).tar ${scratchdir}/${branch}/${buildid}/world
		zfs snapshot ${zbase}/${branch}/${buildid}/world@base

	    else
		mkdir -p ${wrkdir}/tarballs
		if [ ${nocopy} -eq 0 ]; then
		    ln -sf ${pbab}/builds/${buildid}/bindist.tar ${wrkdir}/tarballs
		    ln -sf ${pbd}/${arch}/clients/bindist-$(hostname).tar ${wrkdir}/tarballs
		fi
	    fi
	fi

	for i in ${wrkdir}/chroot/*; do
	    ${sudo_cmd} /tmp/${buildid}/scripts/clean-chroot ${arch} ${branch} ${buildid} ${i} 2
	done
	wait
    else
	# Client may have been waiting for us to set it up, so finish
	# the job.
	touch /tmp/.boot_finished
    fi

}

if [ "${branch}" != "-" -a "${buildid}" != "-" ]; then
    pbab=${pbd}/${arch}/${branch}
    builddir=${pbab}/builds/${buildid}

    wrkdir=${tmpdir}/${branch}/${buildid}
fi

nocopy=0
force=0
while [ $# -ge 1 ]; do
    case $1 in
	-nocopy)
	    nocopy=1
	    ;;
	-force)
	    force=1
	    ;;
    esac
    shift
done

case ${phase} in
    pre-copy)
	precopy
	;;
    post-copy)
	postcopy
	;;
    *)
	echo "Invalid phase ${phase}"
	exit 1
esac