#!/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.

version=@version@

LIBDIR='@libdir@'

[[ -r ${LIBDIR}/util-msg.sh ]] && source ${LIBDIR}/util-msg.sh

import ${LIBDIR}/util.sh
import ${LIBDIR}/util-mount.sh

working_dir=''
keep_mirrors=false

usage() {
    echo "Usage: ${0##*/} [options] working-dir [run arguments]"
    echo "A wrapper around chroot. Provides support for pacman."
    echo
    echo ' options:'
    echo '    -C <file>     Location of a pacman config file'
    echo '    -M <file>     Location of a makepkg config file'
    echo '    -S <file>     Location of a pacman-mirrors config file'
    echo '    -c <dir>      Set pacman cache'
    echo '    -r <list>     Bind mountargs ro'
    echo '    -w <list>     Bind mountargs rw'
    echo '                  List format [src1:target1,...,srcN:targetN]'
    echo '    -B            Use custom  build mirror'
    echo '    -K            Keep mirrorlist (-B)'
    echo '    -h            This message'
    exit 1
}

orig_argv=("$@")

opts='hKC:M:S:c:r:w:B:'

while getopts ${opts} arg; do
    case "${arg}" in
        C) pac_conf="$OPTARG" ;;
        M) makepkg_conf="$OPTARG" ;;
        S) mirrors_conf="$OPTARG" ;;
        c) cache_dir="$OPTARG" ;;
        r) mountargs_ro="$OPTARG" ;;
        w) mountargs_rw="$OPTARG" ;;
        B) build_mirror="$OPTARG" ;;
        K) keep_mirrors=true ;;
        h|?) usage ;;
        *) error "invalid argument '$arg'"; usage ;;
    esac
done
shift $(($OPTIND - 1))

(( $# < 1 )) && die 'You must specify a directory.'
check_root "$0" "${orig_argv[@]}"

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

[[ -z $working_dir ]] && die 'Please specify a working directory.'

if [[ -z $cache_dir ]]; then
    cache_dirs=($(pacman -v 2>&1 | grep '^Cache Dirs:' | sed 's/Cache Dirs:\s*//g'))
else
    cache_dirs=("$cache_dir")
fi

copy_hostconf () {
    cp -a /etc/pacman.d/gnupg "$1/etc/pacman.d"

    [[ -n $pac_conf ]] && cp $pac_conf "$1/etc/pacman.conf"

    [[ -n $makepkg_conf ]] && cp $makepkg_conf "$1/etc/makepkg.conf"
    [[ -n $mirrors_conf ]] && cp ${mirrors_conf} "$1/etc/pacman-mirrors.conf"

    if [[ -n ${build_mirror} ]]; then
        build_mirror=${build_mirror}'/$repo/$arch'
        if ${keep_mirrors}; then
            set_branch "$1" "$(get_branch $1)"
        else
            echo "Server = ${build_mirror}" > "$1/etc/pacman.d/mirrorlist"
        fi
    else
        set_branch "$1" "$(get_branch $1)"
    fi

    sed -r "s|^#?\\s*CacheDir.+|CacheDir = $(echo -n ${cache_dirs[@]})|g" -i "$1/etc/pacman.conf"
}

chroot_extra_umount() {
    chroot_mount "/etc/resolv.conf" "$1/etc/resolv.conf" -B
    chroot_mount "${cache_dirs[0]}" "$1${cache_dirs[0]}" -B

    for cache_dir in ${cache_dirs[@]:1}; do
        chroot_mount "$cache_dir" "$1${cache_dir}" -Br
    done

    if [[ -n ${mountargs_ro[@]} ]]; then
        local IFS=','
        for m in ${mountargs_ro[@]}; do
            chroot_mount "${m%%:*}" "$1${m##*:}" -Br
        done
        unset IFS
    fi

    if [[ -n ${mountargs_rw[@]} ]]; then
        local IFS=','
        for m in ${mountargs_rw[@]}; do
            chroot_mount "${m%%:*}" "$1${m##*:}" -B
        done
        unset IFS
    fi
}

umask 0022

# Sanity check
if [[ ! -f "$working_dir/.manjaro-tools" ]]; then
    die "'%s' does not appear to be a Manjaro chroot." "$working_dir"
elif [[ $(cat "$working_dir/.manjaro-tools") != $version ]]; then
    die "chroot '%s' is not at version %s. Please rebuild." "$working_dir" "$version"
fi

chroot_api_mount "${working_dir}" || die "failed to setup API filesystems in chroot %s" "${working_dir}"

chroot_extra_umount "${working_dir}"

copy_hostconf "${working_dir}"

eval $(grep '^CARCH=' "$working_dir/etc/makepkg.conf")

${CARCH:+setarch "$CARCH"} chroot "${working_dir}" "$@"

kill_chroot_process "${working_dir}"