#!/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_compiler_settings(){
	local tarch="$1" conf
	conf=${make_conf_dir}/$tarch.conf

	[[ -f $conf ]] || return 1

	info "Loading compiler settings: %s" "$tarch"
	source $conf

	return 0
}

get_makepkg_conf(){
	local conf="${tmp_dir}/makepkg-$1.conf"

	cp "${DATADIR}/makepkg.conf" "$conf"

	load_compiler_settings "$1"

	sed -i "$conf" \
		-e "s|@CARCH[@]|$carch|g" \
		-e "s|@CHOST[@]|$chost|g" \
		-e "s|@CFLAGS[@]|$cflags|g"

	echo "$conf"
}

# $1: target_arch
prepare_conf(){
	if ! is_valid_arch_pkg "$1";then
		die "%s is not a valid arch!" "$1"
	fi

	local pac_arch='default'

	if [[ "$1" == 'multilib' ]];then
		pac_arch='multilib'
		is_multilib=true
	fi

	pacman_conf="${DATADIR}/pacman-$pac_arch.conf"

	work_dir="${chroots_pkg}/${target_branch}/$1"
	pkg_dir="${cache_dir_pkg}/${target_branch}/$1"

	makepkg_conf=$(get_makepkg_conf "$1")

	[[ "$pac_arch" == 'multilib' ]] && target_arch='x86_64'
}

pkgver_equal() {
	local left right

	if [[ $1 = *-* && $2 = *-* ]]; then
		# if both versions have a pkgrel, then they must be an exact match
		[[ $1 = "$2" ]]
	else
		# otherwise, trim any pkgrel and compare the bare version.
		[[ ${1%%-*} = "${2%%-*}" ]]
	fi
}

get_full_version() {
	# set defaults if they weren't specified in buildfile
	pkgbase=${pkgbase:-${pkgname[0]}}
	epoch=${epoch:-0}
	if [[ -z $1 ]]; then
		if [[ $epoch ]] && (( ! $epoch )); then
			echo $pkgver-$pkgrel
		else
			echo $epoch:$pkgver-$pkgrel
		fi
	else
		for i in pkgver pkgrel epoch; do
			local indirect="${i}_override"
			eval $(declare -f package_$1 | sed -n "s/\(^[[:space:]]*$i=\)/${i}_override=/p")
			[[ -z ${!indirect} ]] && eval ${indirect}=\"${!i}\"
		done
		if (( ! $epoch_override )); then
			echo $pkgver_override-$pkgrel_override
		else
			echo $epoch_override:$pkgver_override-$pkgrel_override
		fi
	fi
}

find_cached_package() {
	local searchdirs=("$PWD" "$PKGDEST") results=()
	local targetname=$1 targetver=$2 targetarch=$3
	local dir pkg pkgbasename pkgparts name ver rel arch size r results

	for dir in "${searchdirs[@]}"; do
		[[ -d $dir ]] || continue

		for pkg in "$dir"/*.pkg.tar.xz; do
			[[ -f $pkg ]] || continue

			# avoid adding duplicates of the same inode
			for r in "${results[@]}"; do
				[[ $r -ef $pkg ]] && continue 2
			done

			# split apart package filename into parts
			pkgbasename=${pkg##*/}
			pkgbasename=${pkgbasename%.pkg.tar?(.?z)}

			arch=${pkgbasename##*-}
			pkgbasename=${pkgbasename%-"$arch"}

			rel=${pkgbasename##*-}
			pkgbasename=${pkgbasename%-"$rel"}

			ver=${pkgbasename##*-}
			name=${pkgbasename%-"$ver"}

			if [[ $targetname = "$name" && $targetarch = "$arch" ]] &&
				pkgver_equal "$targetver" "$ver-$rel"; then
				results+=("$pkg")
			fi
		done
	done

	case ${#results[*]} in
		0)
		return 1
		;;
		1)
		printf '%s\n' "$results"
		return 0
		;;
		*)
		error 'Multiple packages found:'
		printf '\t%s\n' "${results[@]}" >&2
		return 1
		;;
	esac
}

check_build(){
	find_pkg $1
	[[ ! -f $1/PKGBUILD ]] && die "Directory must contain a PKGBUILD!"
}

find_pkg(){
	local result=$(find . -type d -name "$1")
	[[ -z $result ]] && die "%s is not a valid package or build list!" "$1"
}

load_group(){
	local _multi \
		_space="s| ||g" \
		_clean=':a;N;$!ba;s/\n/ /g' \
		_com_rm="s|#.*||g" \
		devel_group='' \
		file=${DATADIR}/base-devel-udev

        info "Loading custom group: %s" "$file"

	if ${is_multilib}; then
		_multi="s|>multilib||g"
	else
		_multi="s|>multilib.*||g"
	fi

	devel_group=$(sed "$_com_rm" "$file" \
			| sed "$_space" \
			| sed "$_multi" \
			| sed "$_clean")

        echo ${devel_group}
}

init_base_devel(){
	if ${udev_root};then
		base_packages=( "$(load_group)" )
	else
		if ${is_multilib};then
			base_packages=('base-devel' 'multilib-devel')
		else
			base_packages=('base-devel')
		fi
	fi
}

chroot_create(){
	msg "Creating chroot for [%s] (%s)..." "${target_branch}" "${target_arch}"
	mkdir -p "${work_dir}"
	mkchroot_args+=(-L)
	setarch "${target_arch}" \
		mkchroot ${mkchroot_args[*]} \
		"${work_dir}/root" \
		${base_packages[*]} || abort
}

chroot_clean(){
	msg "Cleaning chroot for [%s] (%s)..." "${target_branch}" "${target_arch}"
	for copy in "${work_dir}"/*; do
		[[ -d ${copy} ]] || continue
		msg2 "Deleting chroot copy %s ..." "$(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 "${work_dir}"
}

chroot_update(){
	msg "Updating chroot for [%s] (%s)..." "${target_branch}" "${target_arch}"
	chroot-run ${mkchroot_args[*]} \
			"${work_dir}/${OWNER}" \
			pacman -Syu --noconfirm || abort

}

clean_up(){
	msg "Cleaning up ..."
	msg2 "Cleaning [%s]" "${pkg_dir}"
	find ${pkg_dir} -maxdepth 1 -name "*.*" -delete #&> /dev/null
	if [[ -z $SRCDEST ]];then
		msg2 "Cleaning [source files]"
		find $PWD -maxdepth 1 -name '*.?z?' -delete #&> /dev/null
	fi
}

sign_pkg(){
	su ${OWNER} -c "signfile ${pkg_dir}/$1"
}

move_to_cache(){
	local src="$1"
	[[ -n $PKGDEST ]] && src="$PKGDEST/$1"
	msg2 "Moving [%s] -> [%s]" "${src##*/}" "${pkg_dir}"
	mv $src ${pkg_dir}/
	${sign} && sign_pkg "${src##*/}"
	[[ -n $PKGDEST ]] && rm "$1"
	user_own "${pkg_dir}"
}

archive_logs(){
	local archive name="$1" ext=log.tar.xz ver src=${tmp_dir}/archives.list target='.'
	ver=$(get_full_version "$name")
	archive="${name}-${ver}-${target_arch}"
	if [[ -n $LOGDEST ]];then
            target=$LOGDEST
            find $target -maxdepth 1 -name "$archive*.log" -printf "%f\n" > $src
	else
            find $target -maxdepth 1 -name "$archive*.log" > $src
	fi
	msg2 "Archiving log files [%s] ..." "$archive.$ext"
	tar -cJf ${log_dir}/$archive.$ext  -C "$target" -T $src
	msg2 "Cleaning log files ..."

	find $target -maxdepth 1 -name "$archive*.log" -delete
}

post_build(){
	source PKGBUILD
	local ext='pkg.tar.xz' tarch ver src
	for pkg in ${pkgname[@]};do
		case $arch in
			any) tarch='any' ;;
			*) tarch=${target_arch}
		esac
		local ver=$(get_full_version "$pkg") src
		src=$pkg-$ver-$tarch.$ext
		move_to_cache "$src"
	done
	local name=${pkgbase:-$pkgname}
	archive_logs "$name"
}

chroot_init(){
	local timer=$(get_timer)
	if ${clean_first} || [[ ! -d "${work_dir}" ]]; then
		chroot_clean
		chroot_create
	else
		chroot_update
	fi
	show_elapsed_time "${FUNCNAME}" "${timer}"
}

build_pkg(){
	setarch "${target_arch}" \
		mkchrootpkg ${mkchrootpkg_args[*]} || return 1
		if [ $? -eq 0 ]; then
			post_build
		fi
}

make_pkg(){
	check_build "$1"
	msg "Start building [%s]" "$1"
	cd $1
		build_pkg
	cd ..
	msg "Finished building [%s]" "$1"
	show_elapsed_time "${FUNCNAME}" "${timer_start}"
}