#!/bin/bash
# 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 of the License.
#
# 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.

FORCE='n'
RUN=''
NOCOPY='n'

working_dir=''

APPNAME=$(basename "${0}")

# usage: usage <exitvalue>
usage() {
	echo "usage ${APPNAME} [options] working-dir [package-list | app]"
	echo ' options:'
	echo '    -r <app>      Run "app" within the context of the chroot'
	echo '    -u            Update the chroot via pacman'
	echo '    -f            Force overwrite of files in the working-dir'
	echo '    -C <file>     Location of a pacman config file'
	echo '    -M <file>     Location of a makepkg config file'
	echo '    -n            Do not copy config files into the chroot'
	echo "    -c <dir>      Set pacman cache. Default: /var/cache/pacman/pkg"
	echo '    -h            This message'
	exit $1
}

while getopts 'r:ufnhC:M:c:' arg; do
	case "${arg}" in
		r) RUN="$OPTARG" ;;
		u) RUN='pacman -Syu --noconfirm' ;;
		f) FORCE='y' ;;
		C) pac_conf="$OPTARG" ;;
		M) makepkg_conf="$OPTARG" ;;
		n) NOCOPY='y' ;;
		c) cache_dir="$OPTARG" ;;
		h|?) usage 0 ;;
		*) echo "invalid argument '${arg}'"; usage 1 ;;
	esac
done

if [ "$EUID" != '0' ]; then
	echo 'error: this script must be run as root.'
	exit 1
fi

shift $(($OPTIND - 1))

if [ "$RUN" = '' -a $# -lt 2 ]; then
	echo 'you must specify a directory and one or more packages'
	usage 1
elif [ $# -lt 1 ]; then
	echo 'you must specify a directory'
	usage 1
fi

working_dir="$(readlink -f ${1})"
shift 1

[ "${working_dir}" = '' ] && echo 'error: please specify a working directory' && usage 1

if [ -z "$cache_dir" ]; then
	cache_conf=${working_dir}/etc/pacman.conf
	[ ! -f $cache_conf ] && cache_conf=${pac_conf:-/etc/pacman.conf}
	cache_dir=$((grep -m 1 '^CacheDir' $cache_conf || echo 'CacheDir = /var/cache/pacman/pkg') | sed 's/CacheDir\s*=\s*//')
	unset cache_conf
fi

if [ -f /etc/pacman.d/mirrorlist ]; then
	host_mirror=$(grep -v '^#' -m1 /etc/pacman.d/mirrorlist | sed -E 's#/os/(i686|x86_64)#/os/\$arch#g')
fi
if [ -z "${host_mirror}" ]; then
	host_mirror='Server = http://mirrors.kernel.org/archlinux/$repo/os/$arch'
fi
if echo "${host_mirror}" | grep -q 'file://'; then
	host_mirror_path=$(echo "${host_mirror}" | sed -E 's#Server\s=\sfile://(/.*)/\$repo/os/(\$arch|i686|x86_64)#\1#g')
fi

# {{{ functions 

chroot_mount() {
	[ -e "${working_dir}/sys" ] || mkdir "${working_dir}/sys"
	mount -t sysfs sysfs "${working_dir}/sys"

	[ -e "${working_dir}/proc" ] || mkdir "${working_dir}/proc"
	mount -t proc proc "${working_dir}/proc"

	[ -e "${working_dir}/dev" ] || mkdir "${working_dir}/dev"
	mount -o bind /dev "${working_dir}/dev"

	[ -e "${working_dir}/dev/shm" ] || mkdir "${working_dir}/dev/shm"
	mount -t tmpfs shm "${working_dir}/dev/shm"

	[ -e "${working_dir}/dev/pts" ] || mkdir "${working_dir}/dev/pts"
	mount -t devpts devpts "${working_dir}/dev/pts"

	[ -e "${cache_dir}" ] || mkdir -p "${cache_dir}"
	[ -e "${working_dir}/${cache_dir}" ] || mkdir -p "${working_dir}/${cache_dir}"
	mount -o bind "${cache_dir}" "${working_dir}/${cache_dir}"

	if [ -n "${host_mirror_path}" ]; then
		[ -e "${working_dir}/${host_mirror_path}" ] || mkdir -p "${working_dir}/${host_mirror_path}"
		mount -o bind "${host_mirror_path}" "${working_dir}/${host_mirror_path}"
		mount -o remount,ro,bind "${host_mirror_path}" "${working_dir}/${host_mirror_path}"
	fi

	trap 'chroot_umount' 0 1 2 15
}

copy_hostconf () {
	chroot "${working_dir}" /bin/findmnt -rnu -o SOURCE,TARGET,FSTYPE,OPTIONS > "${working_dir}/etc/mtab"
	cp /etc/resolv.conf "${working_dir}/etc/resolv.conf"
	echo "${host_mirror}" > ${working_dir}/etc/pacman.d/mirrorlist
}

chroot_umount () {
	umount "${working_dir}/proc"
	umount "${working_dir}/sys"
	umount "${working_dir}/dev/pts"
	umount "${working_dir}/dev/shm"
	umount "${working_dir}/dev"
	umount "${working_dir}/${cache_dir}"
	[ -n "${host_mirror_path}" ] && umount "${working_dir}/${host_mirror_path}"
}
# }}}

umask 0022
if [ "$RUN" != "" ]; then
	# run chroot {{{
	#Sanity check
	if [ ! -f "${working_dir}/.arch-chroot" ]; then
		echo "error: '${working_dir}' does not appear to be a Arch chroot"
		echo '       please build the image using mkarchroot'
		exit 1
	fi

	chroot_mount
	copy_hostconf

	chroot "${working_dir}" ${RUN}

	# }}}
	else
	# {{{ build chroot
	if [ -e "${working_dir}" -a "${FORCE}" = "n" ]; then
		echo "error: working dir '${working_dir}' already exists - try using -f"
		exit 1
	fi

	mkdir -p "${working_dir}/var/lib/pacman/"
	mkdir -p "${working_dir}/etc/"

	chroot_mount

	pacargs="--noconfirm --root=${working_dir} --cachedir=${cache_dir}"
	if [ "$pac_conf" != "" ]; then
		pacargs="$pacargs --config=${pac_conf}"
	fi

	if [ $# -ne 0 ]; then
		op='-Sy'
		if [ "$FORCE" = "y" ]; then
			op="${op}f"
		fi
		if ! pacman ${op} ${pacargs} $@; then
			echo 'error: failed to install all packages'
			exit 1
		fi
	fi

	if [ -d "${working_dir}/lib/modules" ]; then
		ldconfig -r "${working_dir}"
	fi

	if [ "$pac_conf" != "" -a "$NOCOPY" = "n" ]; then
		cp ${pac_conf} ${working_dir}/etc/pacman.conf
	fi

	if [ "$makepkg_conf" != "" -a "$NOCOPY" = "n" ]; then
		cp ${makepkg_conf} ${working_dir}/etc/makepkg.conf
	fi

	if [ -e "${working_dir}/etc/locale.gen" ]; then
		echo -e 'en_US.UTF-8 UTF-8\nde_DE.UTF-8 UTF-8' > "${working_dir}/etc/locale.gen"
		chroot "${working_dir}" locale-gen
	fi

	if [ ! -e "${working_dir}/.arch-chroot" ]; then
		date +%s > "${working_dir}/.arch-chroot"
	fi
	# }}}
fi