configure

#!/usr/bin/bash
#
# CDDL HEADER START
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source.  A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.
#
# CDDL HEADER END
#

#
# Copyright (c) 2012 Joyent, Inc.
#
shopt -s xpg_echo
unalias -a

conf_arg0=$(basename $0)
conf_arg0_dir=$(dirname $0)
conf_root=$PWD
conf_ips=
conf_priv="pfexec"

function fatal
{
	local msg="$*"
	[[ -z "$msg" ]] && msg="failed"
	echo "$conf_arg0: $msg" >&2
	exit 1
}

function check_loc
{
	[[ "$(readlink -f "$PWD")" != "$(readlink -f "$conf_arg0_dir")" ]] && \
	    fatal "configure must be run from inside $conf_arg0_dir"
	[[ "$(uname -s)" != "SunOS" ]] && fatal "build only works on SunOS"
	if [[ "$(uname -v)" =~ "joyent_" ]]; then
		[[ "$(zonename)" == "global" ]] && fatal \
		    "building in a GZ on SmartOS is not supported"
	else	
		echo "Not on SmartOS, assuming IPS and sudo"
		conf_ips=1
		conf_priv="sudo"
	fi
}

function create_projects
{
	local dir="projects"
	[[ -d "$dir" ]] || mkdir -p "$dir" || fatal "failed to create $dir"
}

function source_vars
{
	for file in "$(ls configure.*)"; do
		source "$file" || fatal "failed to source $file"
	done
}

function install_pkgin
{
	for pkg in gmake binutils autoconf automake bison flex libtool-base \
	    python24 libxslt gcc-compiler sun-jre6 sun-jdk6 gmp \
	    mpfr nodejs p5-XML-Parser; do
		if ! pkg_info -qe $pkg; then
			$conf_priv pkgin -y install $pkg || fatal \
		    "failed to install $pkg"
		fi
	done
}

function install_ips
{
	fatal "Building on ips based systems has yet to be implemented"
}

function install_packages
{
	if [[ -z "$conf_ips" ]]; then
		install_pkgin
	else
		install_ips
	fi
	[[ $? -eq 0 ]] || fatal "failed to install packages"
}

function fetch_studio
{
	[[ -f "/opt/SUNWspro/prod/bin/cc" ]] && return
	[[ -z "$SUNW_SPRO12_URL" ]] && fatal \
	    "SUNW_SPRO12_URL not set in configure.*"
	curl -k "$SUNW_SPRO12_URL" | $conf_priv tar jxf - -C /opt
	[[ $? -eq 0 ]] || fatal "failed to fetch Studio"
	[[ -e /opt/SUNspro/sunstudio12.1 ]] && return
	$conf_priv ln -s /opt/SUNWspro /opt/SUNWspro/sunstudio12.1 || fatal \
	    "cannot create Studio symlink"
}

function fetch_illumos
{
	local tfile="projects/illumos/usr/src/tools/env/illumos.sh"
	[[ -f "$tfile" ]] && return
	[[ -z "$GET_ILLUMOS" ]] && fatal "configure must define GET_ILLUMOS"
	cd projects >/dev/null || fatal "failed to cd into projects directory"
	/bin/bash -c "$GET_ILLUMOS " || fatal "fetching illumos failed"
	cd - >/dev/null || fatal "failed to cd out of projects directory"
	[[ -f "$tfile" ]] || fatal "fetched illumos appears to be missing files"
}

function fetch_illumos_extra
{
	local tfile="projects/illumos-extra/Makefile"
	[[ -f "$tfile" ]] && return
	[[ -z "$GET_ILLUMOS_EXTRA" ]] && fatal \
	    "configure must define GET_ILLUMOS_EXTRA"
	cd projects >/dev/null || fatal "failed to cd into projects directory"
	/bin/bash -c  "$GET_ILLUMOS_EXTRA" || fatal \
	    "fetching illumos-extra failed"
	cd - >/dev/null || fatal "failed to cd out of projects directory"
	[[ -f "$tfile" ]] || fatal \
	    "fetched illumos-extra appears to be missing files"
}

function fetch_adjuncts
{
	local tgz
	[[ -z "$ILLUMOS_ADJUNCT_TARBALL_URL" ]] && fatal \
	    "ILLUMOS_ADJUNCT_TARBALL_URL missing from configure"
	tgz=$(curl -k "$ILLUMOS_ADJUNCT_TARBALL_URL" | grep href | tail -n1 | \
	    cut -d '"' -f2)
	[[ -z "$tgz" ]] && fatal \
	    "Unable to get adjuncts from ILLUMOS_ADJUNCT_TARBALL_URL"
	curl -kO $ILLUMOS_ADJUNCT_TARBALL_URL/$tgz
	[[ $? -eq 0 ]] || fatal "failed to fetch adjuncts tarball"
}

function setup_overlays
{
	local of="overlay/order"
	[[ -z "$OVERLAYS" ]] && fatal \
	    "OVERLAYS variable missing from configure.*"
	rm -f $of
	touch $of || fail "failed to create $of"
	for o in $OVERLAYS; do
		echo -n "$conf_root/overlay/$o " >> $of || fatal \
		    "failed to append to $of"
	done
}

function fetch_closed
{
	local ildir="projects/illumos"
	local cld="on-closed-bin.i386.tar.bz2"
	local clnd="on-closed-bin-nd.i386.tar.bz2"
	[[ -z "$ON_CLOSED_BINS_URL" ]] && fatal \
	    "missing ON_CLOSED_BINS_URL from configure"
	[[ -z "$ON_CLOSED_BINS_ND_URL" ]] && fatal \
	    "missing ON_CLOSED_BINS_ND_URL from configure"
	if [[ ! -f $ildir/$cld ]]; then
		curl -k $ON_CLOSED_BINS_URL -o $ildir/$cld || fatal \
		"failed to fetch closed bins (debug)"
	fi
	if [[ ! -f $ildir/$clnd ]]; then
		curl -k $ON_CLOSED_BINS_ND_URL -o $ildir/$clnd || fatal \
		"failed to fetch closed bins (non-debug)"
	fi
	cd $ildir >/dev/null 2>&1 || fatal "failed to cd into $ildir"
	tar xpjf $cld || fatal \
	    "failed to extract closed bins (debug)"
	tar xpjf $clnd || fatal \
	    "failed to extract closed bins (non-debug)"
	cd - >/dev/null 2>&1
}

function generate_env
{
	local nopts="-CimNnt"
	local lint debug lprefix

	if [[ -n "$ILLUMOS_ENABLE_LINT" ]]; then
		lint=1
		debug=1
	fi

	if [[ "$ILLUMOS_ENABLE_DEBUG" == "exclusive" ]]; then
		debug=2
	elif [[ "$ILLUMOS_ENABLE_DEBUG" == "yes" ]]; then
		debug=1
	elif [[ -n "$ILLUMOS_ENABLE_DEBUG" ]]; then
		fatal "Unknown value for $ILLUMOS_ENABLE_DEBUG"
	fi

	[[ -n "$lint" ]] && nopts="${nopts}l"
	[[ "$debug" -eq 1 ]] && nopts="${nopts}D"
	[[ "$debug" -eq 2 ]] && nopts="${nopts}DF"

	lprefix=$(echo $conf_root | tr / _)
	[[ $? -eq 0 ]] || fatal "failed to create lock prefix"

cat > "projects/illumos/illumos.sh" <<EOF
NIGHTLY_OPTIONS="$nopts";			export NIGHTLY_OPTIONS
GATE="${RELEASE_VER}";				export GATE
CODEMGR_WS="$conf_root/projects/illumos";	export CODEMGR_WS
maxjobs() {
        ncpu=\`kstat -p cpu_info:::state | grep -c on-line\`
        expr \$ncpu + 2
}
DMAKE_MAX_JOBS=\`maxjobs\`;			export DMAKE_MAX_JOBS

PARENT_WS="";					export PARENT_WS
CLONE_WS="http://hg.illumos.org/illumos-gate"	export CLONE_WS
STAFFER="nobody";				export STAFFER
BUILD_PROJECT="";				export BUILD_PROJECT
LOCKNAME="\`whoami\`_${lprefix}_nightly.lock";	export LOCKNAME
ATLOG="\$CODEMGR_WS/log";			export ATLOG
LOGFILE="\$ATLOG/nightly.log";			export LOGFILE
MACH=\`uname -p\`;				export MACH
ON_CLOSED_BINS="\$CODEMGR_WS/closed";		export ON_CLOSED_BINS
REF_PROTO_LIST="\$PARENT_WS/usr/src/proto_list_\${MACH}";
						export REF_PROTO_LIST
ROOT="$conf_root/proto";			export ROOT
ADJUNCT_PROTO="$conf_root/proto.strap";		export ADJUNCT_PROTO
NATIVE_ADJUNCT="/opt/local";			export NATIVE_ADJUNCT
SRC="\$CODEMGR_WS/usr/src";			export SRC
VERSION="\$GATE";				export VERSION
PARENT_ROOT="$conf_root/proto";			export PARENT_ROOT
PARENT_TOOLS_ROOT="\$PARENT_WS/usr/src/tools/proto/root_\$MACH-nd"
						export PARENT_TOOLS_ROOT
PKGARCHIVE="\${CODEMGR_WS}/packages/\${MACH}/nightly";
						export PKGARCHIVE
PKGPUBLISHER_REDIST="${PUBLISHER}";		export PKGPUBLISHER_REDIST
MAKEFLAGS=k;					export MAKEFLAGS
UT_NO_USAGE_TRACKING="1";			export UT_NO_USAGE_TRACKING
CW_NO_SHADOW="1";				export CW_NO_SHADOW
BUILD_TOOLS="\${CODEMGR_WS}/usr/src/tools/proto/root_\${MACH}-nd/opt";
						export BUILD_TOOLS
SPRO_ROOT=/opt/SUNWspro;			export SPRO_ROOT
SPRO_VROOT=\$SPRO_ROOT;				export SPRO_VROOT
GNU_ROOT="$conf_root/proto.strap/usr/gnu"		export GNU_ROOT
GCC_ROOT="$conf_root/proto.strap/usr";		export GCC_ROOT
# A temporary hack to allow bootstrap of cw(1)
CW_GCC_DIR="\${GCC_ROOT}/bin";			export CW_GCC_DIR
# Use GCC as the primary compiler
__GNUC="";					export __GNUC
# Use GCC4 specific flags
__GNUC4="";					export __GNUC4
JAVA_ROOT=/opt/local/java/sun6;			export JAVA_ROOT
FLEX=/opt/local/bin/flex;			export FLEX
EOF
	[[ $? -eq 0 ]] || fatal "failed to write illumos nightly env file"
}

function update_git
{
	local path=$1
	local branch=$2
	local cur

	[[ -d "$path" ]] || fatal "can't find directory $path"
	[[ -z "$branch" ]] && fatal "unspecified branch"

	cd $path >/dev/null 2>&1 || fatal "can't cd into $path"
	cur=$(git branch | grep "^*" | awk 'BEGIN{ FS=" " } { print $2 }')
	[[ $? -ne 0 || -z "$cur" ]] && fatal "failed to get current branch"
	[[ "$cur " == "$branch" ]] && return
	git checkout $branch >/dev/null || fatal "failed to checkout $branch"
	cd - >/dev/null 2>&1 || fatal "can't return to previous directory"
}

function update_branches
{	
	local line repo branch

	[[ ! -f "configure-branches" ]] && return

	while read line; do
		repo=${line%%:*}
		branch=${line##*:}

		[[ -z "$repo" ]] && fatal \
		    "missing repository entry in line: [$line]"
		[[ -z "$branch" ]] && fatal \
		    "missing branch entry in line: [$branch]"
		
		#
		# Handle the legacy name of smartos-live
		#
		[[ "$repo" == "smartos-live" || \
		    "$repo" == "illumos-live" ]] && fatal \
		    "configure-branches cannot be used to update smartos-live"
		if [[ "$repo" == "illumos-joyent" || \
		    "$repo" == "illumos" ]]; then
			update_git "projects/illumos" $branch && continue
		elif  [[ "$repo" == "illumos-extra" ]]; then
			update_git "projects/illumos-extra" $branch && continue
		else
			[[ -d "projects/local/$repo" ]] && update_git \
			    "projects/local/$repo" $branch && continue
		fi
		fatal "failed to update $repo to $branch"
	done < "configure-branches"
}

echo "Doing pre-flight checks... \c "
check_loc
echo "done."
echo "Creating projects directory ... \c "
create_projects
echo "done."
echo "Sourcing configure.* files ... \c "
source_vars
echo "done."
echo "Installing packages ... \c "
install_packages
echo "done."
echo "Fetching studio... \c "
fetch_studio
echo "done."
echo "Fetching illumos... \c "
fetch_illumos
echo "done."
echo "Fetching illumos-extra... \c "
fetch_illumos_extra
echo "done."
echo "Fetching adjuncts tgz... \c "
fetch_adjuncts
echo "done."
echo "Configuring overlays... \c "
setup_overlays
echo "done."
echo "Fetching and extracting closed bins ... \c "
fetch_closed
echo "done."
echo "Generating illumos environment file... \c "
generate_env
echo "done."
echo "Setting up branches... \c "
update_branches
echo "done."


cat <<EOF
Configuration complete. To build the live image run:
gmake world && gmake live
EOF

# Congrats, we made it
exit 0