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

load_sets(){
    local prof temp
    for item in $(ls ${profiledir}/*.set); do
	temp=${item##*/}
	prof=${prof:-}${prof:+|}${temp%.set}
    done
    echo $prof
}

eval_profile(){
    eval "case $1 in
	    $(load_sets)) is_profile=true ;;
	    *) is_profile=false ;;
	esac"
}

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

chroot_clean(){
    for copy in "${chrootdir}"/*; do
	[[ -d ${copy} ]] || continue
	msg2 "Deleting chroot copy '$(basename "${copy}")'..."

	lock 9 "${copy}.lock" "Locking chroot copy '${copy}'"

	if [[ "$(stat -f -c %T "${copy}")" == btrfs ]]; then
	    { type -P btrfs && btrfs subvolume delete "${copy}"; } &>/dev/null
	fi
	rm -rf --one-file-system "${copy}"
    done
    exec 9>&-
    
    rm -rf --one-file-system "${chrootdir}"
}

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

clean_up(){
    msg "Cleaning up ..."
    
    local query=$(find ${pkg_dir} -maxdepth 1 -name "*.*")
    
    [[ -n $query ]] && rm -v $query
    
    if [[ -z $LOGDEST ]];then
	query=$(find $PWD -maxdepth 2 -name '*.log')
	[[ -n $query ]] && rm -v $query
    fi
    
    if [[ -z $SRCDEST ]];then
	query=$(find $PWD -maxdepth 2 -name '*.?z?')
	[[ -n $query ]] && rm -v $query
    fi
}

blacklist_pkg(){
    msg "Removing ${blacklist[@]}..."
    for item in "${blacklist[@]}"; do
	chroot-run $1/root pacman -Rdd "$item" --noconfirm
    done
}

prepare_cachedir(){
    mkdir -p "${pkg_dir}"
    chown -R "${pkg_owner}:users" "${pkg_dir_loaded}"
}

move_pkg(){
    local ext='pkg.tar.xz'
    if [[ -n $PKGDEST ]];then
	mv $PKGDEST/*{any,$arch}.${ext} ${pkg_dir}/
    else
	mv *.${ext} ${pkg_dir}
    fi
    chown -R "${pkg_owner}:users" "${pkg_dir_loaded}"
}

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

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

sign_pkgs(){
    cd $pkg_dir
    su "${pkg_owner}" <<'EOF'
signpkgs
EOF
}