#!/usr/bin/env bash
# libredbdiff
#
# Copyright (C) 2014 Esteban Carnevale <alfplayer@mailoo.org>
# Copyright (C) 2014, 2017-2018, 2024 Luke T. Shumaker <lukeshu@parabola.nu>
# Copyright (C) 2018 Omar Vega Ramos <ovruni@gnu.org.pe>
#
# License: GNU GPLv3+
#
# 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; either version 3 of the License, or
# (at your option) any later version.
#
# 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.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

set -euE -o pipefail
. "$(librelib messages)"
. "$(librelib conf)"
setup_traps

declare -r name="Libredbdiff"
declare -r cmd="${0##*/}"

declare -r field_pkgname_prbl=30
declare -r field_pkgname_arch=30
declare -r field_pkgname_total="$((field_pkgname_prbl + field_pkgname_arch))"

declare -r printf_format="%s  %-${field_pkgname_prbl}s%-${field_pkgname_arch}s   %s | %s\n"
declare -r printf_format_noarch="%s  %-${field_pkgname_total}s   %s\n"

init_conf() {
	declare -rg conffile_prbl="$statedir/pacman.conf.parabola"
	declare -rg conffile_arch="$statedir/pacman.conf.archlinux"

	declare -rg dbpath_prbl="$statedir/pacman.parabola"
	declare -rg dbpath_arch="$statedir/pacman.archlinux"

	declare -rg mirrorlist_prbl="$statedir/mirrorlist.parabola"
	declare -rg mirrorlist_arch="$statedir/mirrorlist.archlinux"
}

enablerepo() {
	local repo="$1"
	local conffile_arg="$2"

	msg2 "Enabling repo %q in %q" "$repo" "$conffile_arg"
	sed -i "s/\#\[$repo\]/[$repo]/" "$conffile_arg"
	sed -i "\/\[$repo\]/,+1 s/#Include/Include/" "$conffile_arg"
}

createdir() {
	local dir=$1

	if ! [[ -d $dir ]]; then
		msg "Creating directory %q" "$dir"
		mkdir -- "$dir" || die "Failed to create directory %q.  Exiting." "$dir"
	fi
}

# {{ A -lt B }} is like [ A -lt B ], but for pacman version numbers
#
# We wrap this using `eval` because otherwise Emacs auto-indent gets
# confused by the line "{{() {".
eval '{{() { test_vercmp "$@"; }'
test_vercmp() {
	[[ $# == 4 ]] || panic
	[[ $4 == '}}' ]] || panic
	local a=$1
	local op=$2
	local b=$3

	local cmp
	cmp="$(vercmp "$a" "$b")" || die "vercmp failed"
	[ "$cmp" "$op" 0 ]
}

# Globals:
#  - pkgname  : the Parabola pkgname
#  - ver_prbl : a map of pkgname->arch_pkgver
#  - ver_arch : a map of pkgname->prbl_pkgver
#  - provides : a map of pkgname->provides
compare_pkgs() {
	if [[ -n ${ver_arch[$pkgname]:-} ]]; then
		if {{ "${ver_prbl[$pkgname]}" -lt "${ver_arch[$pkgname]}" }}; then
			printf "$printf_format" \
				'=' \
				"$pkgname" \
				"" \
				"${ver_prbl[$pkgname]}" \
				"${ver_arch[$pkgname]}"
		fi
	elif [[ -n ${provides[$pkgname]:-} ]]; then
		local _provides provide
		read -r -a _provides <<<"${provides[$pkgname]}"
		for provide in "${_provides[@]}"; do
			if [[ -n ${ver_arch[$provide]:-} ]]; then
				if {{ "${ver_prbl[$pkgname]}" -lt "${ver_arch[$provide]}" }}; then
					printf "$printf_format" \
						'p' \
						"$pkgname" \
						"$provide" \
						"${ver_prbl[$pkgname]}" \
						"${ver_arch[$provide]}"
				fi
			fi
		done
	else
		printf "$printf_format_noarch" \
			'o' \
			"$pkgname" \
			"${ver_prbl[$pkgname]}"
	fi
}

# Globals:
#  - prbl_packages_tmp
#  - ver_prbl
#  - ver_arch (transitively through compare_pkgs)
#  - provides
print_cmp() {
	local repo="$1"

	<"$prbl_packages_tmp" awk -F/ -v repo="$repo" '$1 == repo {print $2}' |
		while read -r pkgname _pkgver _provides; do
			ver_prbl["$pkgname"]="$_pkgver"
			provides["$pkgname"]="$_provides"
			compare_pkgs
		done
}

usage() {
	print "Usage: sudo %q" "$cmd"
	print "   or: %s -n [REPO]" "$cmd"
	print "   or: %s {-h|--help}" "$cmd"
	print 'Show packages that need to be updated from Arch repositories.'
	echo
	prose "Compares packages in Parabola repositories.  Packages from
	       all configured Parabola repositories are compared.  A Parabola
	       repository name can be specified as argument to compare only
	       packages in that repository."
	echo
	prose 'The default mode of operation requires root, and downloads or
	       updates all necessary files for comparison, but does not compare
	       them.  After that mode has been run, run the tool again without
	       root with the the `-n` flag to not download anything, but to
	       compare the already downloaded files.'
	echo
	prose 'In the compare mode (if the `-n` flag is given), by default it
	       will iterate over each of the configured repositories, prefixing
	       each section with a line in the format `[REPO]`.  However, if a
	       REPO is specified on the command line, it scan for packages only
	       in that repository, and no `[REPO]` line will be printed.'
	echo
	print 'Options:'
	flag \
		'-n' "Don't update anything, just compare already downloaded files." \
		'-h' 'Show this message'
	echo
	prose "In the compare mode the output is in the format:"
	echo
	print '\t[repo]'
	print '\tpkgline'
	print '\tpkgline'
	print '\t...'
	print '\t[repo]'
	print '\tpkgline'
	print '\tpkgline'
	print '\t...'
	echo
	prose "That is, the output is organized by Parabola repository, and
	       within each repository section, there is one line for each
	       package in that Parabola repo that seems to be behind the
	       equivalent package in Arch."
	echo
	print "The line-format for packages is:"
	printf "\t$printf_format" \
		status parabola_pkgname '[arch_pkgname]' parabola_pkgver arch_pkgver
	echo
	print "Where 'status' is one of the characters:"
	flag \
		'=' "An Arch package with the same pkgname as a Parabola package
		     was found, and Arch has a higher pkgver than Parabola.  In
		     this case, arch_pkgname is not present (as it is the same as
		     the parabola_pkgname)." \
		'p' "An Arch package with a pkgname that is provide'd by a Parabola
		     package was found, and the pkgver of the Arch package is
		     higher than the pkgver of the parabola package." \
		'o' "It appears that this Parabola package is not a replacement for
		     an Arch package."
	echo
	print "Options:"
	flag \
		'-h, --help' 'Show this message'
}

main() {
	local args mode=update
	if ! args="$(getopt -n "$cmd" -o nh -l noupdate,help -- "$@")"; then
		mode=errusage
	else
		eval "set -- $args"
		local flag
		while true; do
			flag=$1
			shift
			case "$flag" in
				-n | --noupdate) [[ $mode == usage ]] || mode=compare ;;
				-h | --help) mode=usage ;;
				--) break ;;
				*) panic 'unhandled flag: %q' "$flag" ;;
			esac
		done
		case "$mode" in
			update)
				if [[ $# -gt 0 ]]; then
					gnuerror 'found non-flag arguments; a repo list may only be specified with the -n flag: %s' "$*"
					mode=errusage
				fi
				;;
			compare)
				if [[ $# -gt 1 ]]; then
					gnuerror 'found too many non-flag arguments; only one repo may be specified: %s' "$*"
					mode=errusage
				fi
				;;
		esac
	fi
	case "$mode" in
		errusage)
			print "Try '%s --help' for more information." "$cmd" >&2
			exit $EXIT_INVALIDARGUMENT
			;;
		usage)
			usage
			exit $EXIT_SUCCESS
			;;
		update)
			if [[ $EUID != 0 ]]; then
				error "Must run as root to initialize or update %s databases." "$cmd"
				exit $EXIT_NOPERMISSION
			fi

			load_conf libredbdiff.conf statedir mirror_prbl mirror_arch || exit
			init_conf

			main_update
			;;
		compare)
			load_conf libredbdiff.conf statedir repos || exit
			init_conf

			if [[ $# == 1 ]]; then
				if ! in_array "$1" "${repos[@]}"; then
					die "The specified Parabola repo \"%s\" cannot be compared.  It's not in the list of repos in the configuration variable \"repos\"." "$1"
				fi
			fi

			main_compare "$@"
			;;
		*) panic 'invalid mode: %q' "$mode" ;;
	esac
}

main_update() {
	createdir "$statedir"
	{
		msg "Generating %s %q" Parabola pacman.conf
		cp -T /usr/share/pacman/defaults/pacman.conf.x86_64 "$conffile_prbl"
		sed -E \
			-e "s|/etc/pacman\.d/mirrorlist$|$mirrorlist_prbl|" \
			-e "s|^#?DBPath\s*=.*|DBPath = $dbpath_prbl|" \
			-e "s|^#?Architecture\s*=.*|Architecture = x86_64|" \
			-i "$conffile_prbl"
		local repo
		for repo in multilib {libre,pcr,nonprism}{,-multilib}; do
			enablerepo "$repo" "$conffile_prbl"
		done

		msg "Generating %s %q" Arch pacman.conf
		wget -q \
			-O "$conffile_arch" \
			"https://git.archlinux.org/svntogit/packages.git/plain/pacman/repos/core-x86_64/pacman.conf" ||
			die "Failed to download %q.  Exiting." "$conffile_arch"
		sed -E \
			-e "s|/etc/pacman\.d/mirrorlist$|$mirrorlist_arch|" \
			-e "s|^#?DBPath\s*=.*|DBPath = $dbpath_arch|" \
			-e "s|^#?Architecture\s*=.*|Architecture = x86_64|" \
			-i "$conffile_arch"
		enablerepo multilib "$conffile_arch"

		printf 'Server = %s\n' "$mirror_prbl" >"$mirrorlist_prbl"

		printf 'Server = %s\n' "$mirror_arch" >"$mirrorlist_arch"
	}

	createdir "$dbpath_prbl"
	{
		msg "Synchronizing %s pacman databases for %s" "$name" "Parabola"
		pacman --config "$conffile_prbl" -Sy ||
			die "Failed to synchronize pacman database for %s.  Exiting." Parabola
	}

	createdir "$dbpath_arch"
	{
		msg "Synchronizing %s pacman databases for %s" "$name" "Arch"
		pacman --config "$conffile_arch" -Sy ||
			die "Failed to synchronize pacman database for %s.  Exiting." Arch
	}

	msg "%s pacman databases are updated.  %s is ready.  Run %q -n to print results." \
		"$name" "$name" "$cmd"
}

main_compare() {
	if ! [[ -f $conffile_prbl &&
		-f $conffile_arch &&
		-f "$dbpath_prbl/sync/libre.db" &&
		-f "$dbpath_arch/sync/core.db" ]]; then
		error "You must run %q with out arguments as root to initialize." "$cmd"
		usage >&2
		exit $EXIT_INVALIDARGUMENT
	fi

	local tmpdir
	tmpdir="$(mktemp --tmpdir -d "$cmd.XXXXXXXXXX")" || die "Could not create temporary working directory"
	trap "rm -rf -- ${tmpdir@Q}" RETURN
	local arch_packages_tmp="$tmpdir/arch-packages"
	local prbl_packages_tmp="$tmpdir/parabola-packages"

	unset provides ver_prbl ver_arch
	declare -gA provides ver_prbl ver_arch

	expac --config "$conffile_arch" -S '%n %v' \
		>"$arch_packages_tmp" ||
		die "expac command to get %s package data has failed.  Exiting." Arch

	local pkgname pkgver
	while read -r pkgname pkgver; do
		ver_arch["$pkgname"]="$pkgver"
	done <"$arch_packages_tmp"

	expac --config "$conffile_prbl" -S '%r/%n %v %S' \
		>"$prbl_packages_tmp" ||
		die "expac command to get %s package data has failed.  Exiting." Parabola

	if [[ $# == 1 ]]; then
		print_cmp "$1"
	else
		for repo in "${repos[@]}"; do
			echo "[$repo]"
			print_cmp "$repo"
		done
	fi
}

main "$@"
