#-
# Copyright (c) 2008-2011 Juan Romero Pardines.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#-

#
# Script to install packages into a sandbox in masterdir.
# Actually this needs the xbps-base-chroot package installed.
#
trap "_umount && return $?" 0 INT QUIT TERM

_mount()
{
	MASTERDIR="${XBPS_MASTERDIR}" ${sudo_cmd} \
		@@XBPS_INSTALL_LIBEXECDIR@@/xbps-src-chroot-helper mount
	return $?
}

_umount()
{
	MASTERDIR="${XBPS_MASTERDIR}" ${sudo_cmd} \
		@@XBPS_INSTALL_LIBEXECDIR@@/xbps-src-chroot-helper umount
	return $?
}

[ -n "$base_chroot" ] && return 0

if [ "${chroot_cmd}" = "chroot" ]; then
	if [ "$(id -u)" -ne 0 ]; then
		echo "Root permissions are required for the chroot, try again."
		exit 1
	fi
fi

. $XBPS_SHUTILSDIR/builddep_funcs.sh
check_installed_pkg xbps-base-chroot-0.11
if [ $? -ne 0 ]; then
	echo "${XBPS_MASTERDIR} has not been prepared for chroot operations."
	echo "Please install 'xbps-base-chroot>=0.11' and try again."
	exit 1
fi

msg_normal "Entering into the chroot on $XBPS_MASTERDIR.\n"

if [ ! -d $XBPS_MASTERDIR/usr/local/etc ]; then
	mkdir -p $XBPS_MASTERDIR/usr/local/etc
fi

XBPSSRC_CF=$XBPS_MASTERDIR/usr/local/etc/xbps-src.conf

cat > $XBPSSRC_CF <<_EOF
# Generated configuration file by xbps-src, DO NOT EDIT!
XBPS_DISTRIBUTIONDIR=/xbps
XBPS_MASTERDIR=/
XBPS_CFLAGS="$XBPS_CFLAGS"
XBPS_CXXFLAGS="$XBPS_CFLAGS"
XBPS_LDFLAGS="$XBPS_LDFLAGS"
XBPS_FETCH_CMD="xbps-uhelper.static fetch"
XBPS_COMPRESS_CMD="$XBPS_COMPRESS_CMD"
_EOF

if [ -n "$XBPS_MAKEJOBS" ]; then
	echo "XBPS_MAKEJOBS=$XBPS_MAKEJOBS" >> $XBPSSRC_CF
fi
if [ -n "$XBPS_PREFER_BINPKG_DEPS" ]; then
	echo "XBPS_PREFER_BINPKG_DEPS=$XBPS_PREFER_BINPKG_DEPS" >> $XBPSSRC_CF
fi
if [ -n "$XBPS_COMPRESS_LEVEL" ]; then
	echo "XBPS_COMPRESS_LEVEL=$XBPS_COMPRESS_LEVEL" >> $XBPSSRC_CF
fi
echo "# End of configuration file." >> $XBPSSRC_CF

if [ -d $XBPS_MASTERDIR/tmp ]; then
	if [ ! -f $XBPS_MASTERDIR/.xbps_mount_bind_done ]; then
		msg_normal "Cleaning up /tmp...\n"
		rm -rf $XBPS_MASTERDIR/tmp/*
	fi
fi

prepare_chroot()
{
	local f=

	# Create some required files.
	cp -f /etc/mtab $XBPS_MASTERDIR/etc
	cp -f /etc/resolv.conf $XBPS_MASTERDIR/etc
	[ -f /etc/localtime ] && cp -f /etc/localtime $XBPS_MASTERDIR/etc

	for f in run/utmp log/btmp log/lastlog log/wtmp; do
		touch -f $XBPS_MASTERDIR/var/$f
	done
	for f in run/utmp log/lastlog; do
		chmod 644 $XBPS_MASTERDIR/var/$f
	done

	cat > $XBPS_MASTERDIR/etc/passwd <<_EOF
root:x:0:0:root:/root:/bin/bash
nobody:x:99:99:Unprivileged User:/dev/null:/bin/false
$(whoami):x:$(id -u):$(id -g):$(whoami) user:/dev/null:/bin/bash
_EOF

	# Default group list as specified by LFS.
	cat > $XBPS_MASTERDIR/etc/group <<_EOF
root:x:0:
bin:x:1:
sys:x:2:
kmem:x:3:
wheel:x:4:
tty:x:5:
tape:x:6:
daemon:x:7:
floppy:x:8:
disk:x:9:
lp:x:10:
dialout:x:11:
audio:x:12:
video:x:13:
utmp:x:14:
usb:x:15:
cdrom:x:16:
optical:x:17:
mail:x:18:
storage:x:19:
scanner:x:20:
nogroup:x:99:
users:x:1000:
$(whoami):x:$(id -g):
_EOF

	# Default file as in Ubuntu.
	cat > $XBPS_MASTERDIR/etc/hosts <<_EOF
127.0.0.1	xbps	localhost.localdomain	localhost
127.0.1.1	xbps

# The following lines are desirable for IPv6 capable hosts
::1     ip6-localhost ip6-loopback
fe00::0 ip6-localnet
ff00::0 ip6-mcastprefix
ff02::1 ip6-allnodes
ff02::2 ip6-allrouters
ff02::3 ip6-allhosts
_EOF

	create_binsh_symlink

	touch $XBPS_MASTERDIR/.xbps_perms_done
}

create_binsh_symlink()
{
	if [ ! -h $XBPS_MASTERDIR/bin/sh ]; then
		cd $XBPS_MASTERDIR/bin && ln -sf bash sh
	fi
}

prepare_binpkg_repos()
{
	local repo

	if [ ! -f "$XBPS_MASTERDIR/.xbps_added_local_repo" ]; then
		if [ ! -f ${XBPS_PACKAGESDIR}/${xbps_machine}/pkg-index.plist ]; then
			:
		else
			msg_normal "Registering local repository...\n"
			${chroot_cmd} $XBPS_MASTERDIR \
			    ${XBPS_REPO_CMD} add /xbps_packagesdir 2>/dev/null
			[ $? -eq 0 ] && touch -f \
				$XBPS_MASTERDIR/.xbps_added_local_repo
		fi
	fi
	for repo in ${XBPS_REPO_LIST}; do
		${chroot_cmd} ${XBPS_MASTERDIR} ${XBPS_REPO_CMD} \
			add ${repo} 2>/dev/null
		[ $? -ne 0 ] && msg_warn "Failed to sync pkg-index from ${repo}\n"
	done
}

create_busybox_links()
{
	local lbindir=$XBPS_MASTERDIR/usr/local/bin

	[ -f $XBPS_MASTERDIR/.busybox_done ] && return 0

	[ ! -d ${lbindir} ] && mkdir -p ${lbindir}

	# Create other symlinks in /usr/local/bin
	cd ${lbindir} || return 1

	for f in $(${XBPS_MASTERDIR}/bin/busybox --list); do
		[ "$f" = "sh" ] && continue
		ln -s ../../../bin/busybox $f
	done

	touch -f $XBPS_MASTERDIR/.busybox_done
}

install_xbps_utils()
{
	local needed _cmd
	local xbps_prefix=$XBPS_MASTERDIR/usr/local

	for f in bin repo uhelper; do
		if [ ! -x $xbps_prefix/sbin/xbps-${f}.static ]; then
			needed=yes
		fi
	done

	if [ -n "$needed" ]; then
		echo "=> Installing static XBPS utils into masterdir."
		for f in bin repo uhelper; do
			_cmd=$(which xbps-${f}.static 2>/dev/null)
			if [ -z "${_cmd}" ]; then
				echo "Unexistent xbps-uhelper.static file!"
				exit 1
			fi
			cp -f ${_cmd} $xbps_prefix/sbin
		done
	fi
}

xbps_chroot_handler()
{
	local action="$1" pkg="$2" norm_builddir="$3" rv=0
	local path="/tools/bin:/usr/local/sbin:/usr/local/bin:/bin:/sbin:/usr/bin:/usr/sbin"

	[ -z "$action" -o -z "$pkg" ] && return 1

	[ "$action" != "configure" -a "$action" != "build" -a \
	  "$action" != "install" -a "$action" != "chroot" -a \
	  "$action" != "install-destdir" ] && return 1

	if [ ! -f $XBPS_MASTERDIR/.xbps_perms_done ]; then
		echo -n "==> Preparing chroot on $XBPS_MASTERDIR... "
		prepare_chroot
		echo "done."
	fi

	if [ ! -d "$XBPS_MASTERDIR/tmp" ]; then
		mkdir -p "$XBPS_MASTERDIR/tmp"
	fi

	create_binsh_symlink
	create_busybox_links
	install_xbps_utils

	_mount || return $?

	if [ -n "$XBPS_PREFER_BINPKG_DEPS" ]; then
		prepare_binpkg_repos
	fi

	# Reinstall xbps-src in the chroot
	msg_normal "Installing xbps-src in the masterdir...\n"
	env in_chroot=yes LANG=C PATH=$path \
	    ${chroot_cmd} $XBPS_MASTERDIR sh -c \
	    "cd /xbps/xbps-src && make IN_CHROOT=1 install clean" \
	    2>&1 >/dev/null || return $?

	# Update ld.so(8) cache
	msg_normal "Updating ld.so(8) cache...\n"
	${chroot_cmd} $XBPS_MASTERDIR sh -c "ldconfig" || return $?

	if [ "$action" = "chroot" ]; then
		env in_chroot=yes IN_CHROOT=1 LANG=C PATH=$path \
			${chroot_cmd} $XBPS_MASTERDIR /bin/sh || \
			rv=$?
	else
		[ "$norm_builddir" = "yes" ] && \
			action="-C $action"
		env in_chroot=yes LANG=C PATH=$path _ORIGINPKG="$pkg" \
			${chroot_cmd} $XBPS_MASTERDIR sh -c \
			"cd /xbps/srcpkgs/$pkg && xbps-src $action" || \
			rv=$?
	fi

	msg_normal "Exiting from the chroot on $XBPS_MASTERDIR.\n"

	return $rv
}