#!/bin/ksh93 -p
#
# CDDL HEADER START
#
# The contents of this file are subject to the terms of the
# Common Development and Distribution License (the "License").
# You may not use this file except in compliance with the License.
#
# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
# or http://www.opensolaris.org/os/licensing.
# See the License for the specific language governing permissions
# and limitations under the License.
#
# When distributing Covered Code, include this CDDL HEADER in each
# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
# If applicable, add the following below this CDDL HEADER, with the
# fields enclosed by brackets "[]" replaced with your own identifying
# information: Portions Copyright [yyyy] [name of copyright owner]
#
# CDDL HEADER END
#

#
# Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
#

PATH=/usr/bin:/usr/sbin
export PATH

DEFAULTONURI="http://ipkg.sfbay/on-nightly"
DEFAULTONPUB="on-nightly"
DEFAULTONEXTRAURI="http://ipkg.sfbay/on-extra"
DEFAULTONEXTRAPUB="on-extra"

usage()
{
	echo "usage: $0 [opts] [-s beName] -t beName"
	echo "usage: $0 [opts] -r"
	echo
	echo "\t-d repodir : directory for repositories"
	echo "\t-e uri : origin URI for extra repository"
	echo "\t-E prefix : prefix for extra repository"
	echo "\t-O : open mode, no extra repository will be used"
	echo "\t-r : start repository servers only"
	echo "\t-s : source BE to clone"
	echo "\t-t : new BE name"
	echo "\t-u uri : origin URI for redist repository"
	echo "\t-U prefix:  prefix for redist repository"
	echo "\t-v : verbose"
	echo "\t-Z : skip updating zones"
	echo
	echo "Update to an ON build:"
	echo "\tonu -t newbe -d /path/to/my/ws/packages/\`uname -p\`/nightly"
	echo
	echo "Update to the nightly build:"
	echo "\tonu -t newbe"
	echo
	echo "Re-enable the publishers, and start any pkg.depotd servers"
	echo "necessary in the current BE:"
	echo "\tonu -r -d /path/to/my/ws/packages/\`uname -p\`/nightly"
	exit 1
}

exit_error()
{
	echo $*
	cleanup
	exit 2
}

cleanup()
{
	[ $redistpid -gt 0 ] && kill $redistpid
	[ $extrapid -gt 0 ] && kill $extrapid
	[ -d /tmp/redist.$$ ] && /bin/rm -rf /tmp/redist.$$
	[ -d /tmp/extra.$$ ] && /bin/rm -rf /tmp/extra.$$
}

do_cmd()
{
	[ $verbose -gt 0 ] && echo $*
	$*
	exit_code=$?
	[ $exit_code -eq 0 ] && return
	# pkg(1) returns 4 if "nothing to do", which is safe to ignore
	[ $1 = "pkg" -a $exit_code -eq 4 ] && return
	exit_error "$*" failed: exit code $exit_code
}

configure_publishers()
{
	root=$1

	do_cmd pkg -R $root set-publisher --no-refresh --non-sticky opensolaris.org
	do_cmd pkg -R $root set-publisher -e --no-refresh -P -O $uri $redistpub
	[ $open -eq 0 ] && {
		do_cmd pkg -R $root set-publisher -e \
		    --no-refresh -O $extrauri $extrapub
	}
	do_cmd pkg -R $root refresh --full
}

#
# If we're working from a repodir, disable the new publishers in the new
# BE; they won't work without further configuration, in which case the
# -r option should be used.
#
unconfigure_publishers()
{
	root=$1

	if [ -n "$repodir" ]; then
		do_cmd pkg -R $root set-publisher -P opensolaris.org
		do_cmd pkg -R $root set-publisher -d $redistpub
		[ $open -eq 0 ] && {
			do_cmd pkg -R $root set-publisher -d $extrapub
		}
	fi
}

update()
{
	root=$1

	pkg -R $root list entire > /dev/null 2>&1
	[ $? -eq 0 ] && do_cmd pkg -R $root uninstall entire

	configure_publishers $root

	do_cmd pkg -R $root image-update

	unconfigure_publishers $root
}

update_zone()
{
	zone=$1

	name=`echo $zone | cut -d: -f 2`
	if [ $name = "global" ]; then
		return
	fi

	brand=`echo $zone | cut -d: -f 6`
	if [ $brand != "ipkg" ]; then
		return
	fi

	if [ "$zone_warned" = 0 ]; then
		echo "WARNING: Use of onu(1) will prevent use of zone attach in the new BE" >&2
		echo "See onu(1)" >&2
		zone_warned=1
	fi

	state=`echo $zone | cut -d: -f 3`

	case "$state" in 
	configured|incomplete)
		return
		;;
	esac

	zoneroot=`echo $zone | cut -d: -f 4`

	echo "Updating zone $name"
	update $zoneroot/root
}

sourcebe=""
targetbe=""
uri=""
extrauri=""
repodir=""
verbose=0
open=0
redistpid=0
extrapid=0
redistport=13000
extraport=13001
no_zones=0
zone_warned=0
reposonly=0

trap cleanup 1 2 3 15

while getopts :d:E:e:Ors:t:U:u:vZ i ; do
	case $i in
	d)
		repodir=$OPTARG
		;;
	E)
		extrapub=$OPTARG
		;;
	e)
		extrauri=$OPTARG
		;;
	O)
		open=1
		;;
	r)
		reposonly=1
		;;
	s)
		sourcebe=$OPTARG
		;;
	t)
		targetbe=$OPTARG
		;;
	U)
		redistpub=$OPTARG
		;;
	u)
		uri=$OPTARG
		;;
	v)
		verbose=1
		;;
	Z)
		no_zones=1
		;;
	*)
		usage
	esac
done
shift `expr $OPTIND - 1`

[ -n "$1" ] && usage

if [ "$reposonly" -eq 1 ]; then
	[ -n "$sourcebe" ] && usage
	[ -n "$targetbe" ] && usage
	[ "$no_zones" -eq 1 ] && usage
else
	[ -z "$targetbe" ] && usage
fi
[ -z "$uri" ] && uri=$ONURI
[ -z "$uri" ] && uri=$DEFAULTONURI
[ -z "$redistpub" ] && redistpub=$ONPUB
[ -z "$redistpub" ] && redistpub=$DEFAULTONPUB
[ -z "$extrauri" ] && extrauri=$ONEXTRAURI
[ -z "$extrauri" ] && extrauri=$DEFAULTONEXTRAURI
[ -z "$extrapub" ] && extrapub=$ONEXTRAPUB
[ -z "$extrapub" ] && extrapub=$DEFAULTONEXTRAPUB

if [ -n "$repodir" ]; then
	redistdir=$repodir/repo.redist
	[ -d $redistdir ] || exit_error "$redistdir not found"
	redistpub=$(python2.6 <<# EOF
		import ConfigParser
		p = ConfigParser.SafeConfigParser()
		p.read("$redistdir/cfg_cache")
		pp = p.get("publisher", "prefix")
		print "%s" % pp
		EOF)
	[ $verbose -gt 0 ] && echo "starting pkg.depotd -d $redistdir -p $redistport"
	ARGS="--readonly --writable-root"
	mkdir /tmp/redist.$$
	/usr/lib/pkg.depotd -d $redistdir -p $redistport $ARGS /tmp/redist.$$ >/dev/null &
	redistpid=$!
	uri="http://localhost:$redistport/"
	if [ $open -eq 0 ]; then
		extradir=$repodir/repo.extra
		[ -d $extradir ] || exit_error "$extradir not found"
		extrapub=$(python2.6 <<# EOF
			import ConfigParser
			p = ConfigParser.SafeConfigParser()
			p.read("$extradir/cfg_cache")
			pp = p.get("publisher", "prefix")
			print "%s" % pp
			EOF)
		[ $verbose -gt 0 ] && echo "starting pkg.depotd -d $extradir -p $extraport"
		mkdir /tmp/extra.$$
		/usr/lib/pkg.depotd -d $extradir -p $extraport $ARGS /tmp/extra.$$ >/dev/null &
		extrapid=$!
		extrauri="http://localhost:$extraport/"
	fi
fi

if [ "$reposonly" -eq 1 ]; then
	configure_publishers /
	if [ "$redistpid" -ne 0 ]; then
		echo "$redistpub pkg.depotd running with pid $redistpid"
	fi
	if [ "$extrapid" -ne 0 ]; then
		echo "$extrapub pkg.depotd running with pid $extrapid"
	fi
	exit 0
fi

createargs=""
[ -n "$sourcebe" ] && createargs="-e $sourcebe"

# ksh seems to have its own mktemp with slightly different semantics
tmpdir=`/usr/bin/mktemp -d /tmp/onu.XXXXXX`
[ -z "$tmpdir" ] && exit_error "mktemp failed"

do_cmd beadm create $createargs $targetbe
do_cmd beadm mount $targetbe $tmpdir
update $tmpdir
do_cmd beadm activate $targetbe

if [ "$no_zones" != 1 ]; then
	for zone in `do_cmd zoneadm -R $tmpdir list -cip`; do
		update_zone $zone
	done
fi

cleanup
exit 0