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

shopt -s nullglob

[[ -r @libdir@/util-msg.sh ]] && source @libdir@/util-msg.sh
[[ -r @libdir@/util.sh ]] && source @libdir@/util.sh
[[ -r @libdir@/util-build.sh ]] && source @libdir@/util-build.sh

chroot_create(){
    mkdir -p "${chrootdir}"
    setarch "${arch}" mkchroot \
	    ${mkchroot_args[*]} \
	    "${chrootdir}/root" \
	    "${base_packages[*]}" || abort
}

chroot_update(){
    lock 9 "${chrootdir}/root.lock" "Locking clean chroot"
    chroot-run ${mkchroot_args[*]} \
	      "${chrootdir}/root" \
	      pacman -Syu --noconfirm || abort
}

chroot_build(){
    if ${is_profile};then
	msg "Start building profile: [${profile}]"
	for pkg in $(cat ${profiledir}/${profile}.set); do
	    cd $pkg
	    if [[ $pkg == 'eudev' ]] || [[ $pkg == 'lib32-eudev' ]]; then
		blacklist_pkg "${chrootdir}"
	    fi
	    setarch "${arch}" \
		mkchrootpkg ${mkchrootpkg_args[*]} -- ${makepkg_args[*]} || break
	    move_pkg "${pkg_owner}"
	    cd ..
	done
	msg "Finished building profile: [${profile}]"
    else
	cd ${profile}
	if [[ ${profile} == 'eudev' ]] || [[ ${profile} == 'lib32-eudev' ]]; then
	    blacklist_pkg "${chrootdir}"
	fi
	setarch "${arch}" \
	    mkchrootpkg ${mkchrootpkg_args[*]} -- ${makepkg_args[*]} || abort
	move_pkg "${pkg_owner}"
	cd ..
    fi
}

chroot_init(){
    if ${clean_first} || [[ ! -d "${chrootdir}" ]]; then
	msg "Creating chroot for [${branch}] (${arch})..."
	chroot_clean "${chrootdir}"
	chroot_create
    else
	msg "Updating chroot for [${branch}] (${arch})..."
	chroot_update
    fi
}

display_settings(){
    msg "manjaro-tools version: ${version}"

    msg "OPTIONS:"
    msg2 "arch: ${arch}"
    msg2 "branch: ${branch}"
    msg2 "chroots: ${chroots}"
    
    msg "ARGS:"
    msg2 "mkchrootpkg_args: ${mkchrootpkg_args[*]}"
    msg2 "makepkg_args: ${makepkg_args[*]}"
    
    msg "PATHS:"
    msg2 "chrootdir: ${chrootdir}"
    msg2 "profiledir: ${profiledir}"
    msg2 "pkgdir: ${pkgdir}"
    msg2 "pacman_conf: ${pacman_conf}"
    msg2 "makepkg_conf: ${makepkg_conf}"
    msg2 "pm_conf: ${pm_conf}"

    if ${clean_first};then
	msg "PKG:"
	msg2 "base_packages: ${base_packages[*]}"
    fi

    msg "SETS:"
    msg2 "profiles: $(get_profiles)"
    msg2 "profile: ${profile}"
    msg2 "is_profile: ${is_profile}"

    if ${is_profile};then
	msg "Build queue:"
	local list=$(cat ${profiledir}/${profile}.set)
	for item in ${list[@]}; do
	    msg2 "$item"
	done
    else
	msg "Build queue:"
	msg2 "${profile}"
    fi
}

export LC_MESSAGES=C

arch=$(uname -m)

pacman_conf_arch='default'

clean_first=false
wipe_clean=false
namcap=false
pretend=false
is_profile=false
sign=false

base_packages=('base-devel')

mkchroot_args=()
mkchrootpkg_args=()
makepkg_args=()

pkg_owner=${SUDO_USER:-$USER}

if [[ -n $SUDO_USER ]]; then
    eval "USER_HOME=~$SUDO_USER"
else
    USER_HOME=$HOME
fi

load_vars "$USER_HOME/.makepkg.conf"
load_vars /etc/makepkg.conf

usage() {
    echo "Usage: ${0##*/} [options] [--] [makepkg args]"
    echo "    -p <profile>       Set profile or pkg [default: ${profile}]"
    echo "    -a <arch>          Set arch  [default: ${arch}]"
    echo "    -b <branch>        Set branch [default: ${branch}]"
    echo "    -r <dir>           Chroots directory [default: ${chroots}]"
    echo "    -t <repo>          Set repo [default: ${repo}]"
    echo '    -c                 Recreate chroot'
    echo '    -w                 Clean up'
    echo '    -n                 Install and run namcap check'
    echo '    -s                 Sign packages'
    echo '    -q                 Query settings and pretend build'
    echo '    -h                 This help'
    echo ''
    echo ''
    exit 1
}

orig_argv=("$@")

opts='p:a:b:r:t:cwnsqh'

while getopts "${opts}" arg; do
    case "${arg}" in
	p) profile="$OPTARG" ;;
	a) arch="$OPTARG" ;;
	b) branch="$OPTARG" ;;
	r) chroots="$OPTARG" ;;
	t) repo="$OPTARG" ;;
	c) clean_first=true ;;
	w) wipe_clean=true ;;
	n) namcap=true; mkchrootpkg_args+=(-n) ;;
	s) sign=true ;;
	q) pretend=true ;;
	h) usage ;;
    esac
done

if [[ "$arch" == 'multilib' ]]; then
    pacman_conf_arch='multilib'
    arch='x86_64'
    base_packages+=('multilib-devel')
fi

chrootdir=${chroots}/${branch}/${arch}

pacman_conf="@pkgdatadir@/pacman-${pacman_conf_arch}.conf"
makepkg_conf="@pkgdatadir@/makepkg-${arch}.conf"
pm_conf="@pkgdatadir@/pacman-mirrors-${branch}.conf"

load_config '@sysconfdir@'

mkchroot_args+=(-C ${pacman_conf} -M ${makepkg_conf} -S ${pm_conf})

mkchrootpkg_args+=(-r ${chrootdir})

makepkg_args+=("${@:$OPTIND}")

check_root "$0" "${orig_argv[@]}"

prepare_dir "${pkgdir}" "${pkg_owner}"

eval_profile "${profile}"

if ${pretend}; then
    display_settings
    exit $?
else
    ${wipe_clean} && clean_up
    display_settings
    chroot_init
    chroot_build
    ${sign} && sign_pkgs "${pkg_owner}"
fi