#!/usr/bin/env bash
# libredbdiff
#
# Copyright (C) 2014 Esteban Carnevale <alfplayer@mailoo.org>
# Copyright (C) 2014, 2017 Luke Shumaker <lukeshu@sbcglobal.net>
#
# 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/>.

. "$(librelib messages)"
. "$(librelib conf)"
load_files libredbdiff || exit 1

conffile_prbl="$statedir/pacman.conf.parabola"
conffile_arch="$statedir/pacman.conf.archlinux"

dbpath_prbl="$statedir/pacman.parabola"
dbpath_arch="$statedir/pacman.archlinux"

mirrorlist_prbl="$statedir/mirrorlist.parabola"
mirrorlist_arch="$statedir/mirrorlist.archlinux"

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

field_pkgname_prbl=30
field_pkgname_arch=30
field_pkgname_total="$((field_pkgname_prbl + field_pkgname_arch))"

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

downloadfile() {
	local outfile=$1
	local url=$2
	local mesg=("${@:3}")
	if [[ ! -e "$outfile" ]] ; then
		msg "${mesg[@]}"
		if wget -q "$url" -O "$outfile"; then
			return 255
		else
			die "Failed to download %q.  Exiting." "$outfile"
		fi
	elif [[ -n "$init" ]]; then
		warning "%q already exists.  Skipping." "$outfile"
	fi
}

enablerepo() {
	repo="$1"
	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 [[ ! -e "$dir" ]] ; then
		msg "Creating directory %q" "$dir"
		mkdir -- "$1" || die "Failed to create directory %q.  Exiting." "$dir"
	elif [[ -n "$init" ]]; then
		warning "%q already exists.  Skipping." "$dir"
	fi
}

setmirror() {
	local distro="$1"
	local mirror="$2"
	local mirrorlist="$3"
	if [[ -n "$init" ]] && [[ -n "$mirror" ]]; then
		mirrorescaped="${mirror//./\\.}"
		mirrorescaped="${mirrorescaped//\$/\\$}"
		msg2 "Setting %s as the only enabled %s mirror." "$mirror" "$distro"
		sed -i 's|^#\(Server = '"$mirrorescaped"'\)$|\1|' "$mirrorlist"
	fi
}

filenotfound() {
	local file=$1
	if [[ ! -r "$1" ]]; then
		error "Could not read %q." "$file"
		die "Nothing done.  It may be necessary to run %q without \
arguments as root to initialize %s." "$cmd" "$name"
	fi
}

initialize() {
	createdir "$statedir"

	downloadfile \
		"$conffile_prbl" \
		"https://projects.parabola.nu/abslibre.git/plain/libre/pacman/pacman.conf.x86_64" \
		"Downloading Parabola %q" \
		pacman.conf
	if [[ $? == 255 ]] ; then
		msg2 "Setting DBPath in %q" "$conffile_prbl"
		sed -i "s|^#DBPath .*|DBPath = $dbpath_prbl|" "$conffile_prbl"
		enablerepo nonprism "$conffile_prbl"
		enablerepo pcr "$conffile_prbl"
		enablerepo libre-multilib "$conffile_prbl"
		enablerepo multilib "$conffile_prbl"
	fi

	downloadfile \
		"$conffile_arch" \
		"https://projects.archlinux.org/svntogit/packages.git/plain/pacman/trunk/pacman.conf.x86_64" \
		"Downloading Arch %q" \
		pacman.conf
	if [[ $? == 255 ]] ; then
		msg2 "Setting DBPath in %q" "$conffile_arch"
		sed -i "s|^#DBPath .*|DBPath = $dbpath_arch|" "$conffile_arch"
		msg2 "Setting Arch mirrorlist file in %q" "$conffile_arch"
		sed -i "s|/etc/pacman\.d/mirrorlist$|$mirrorlist_arch|" \
			"$conffile_arch"
		enablerepo multilib "$conffile_arch"
	fi

	downloadfile \
		"$mirrorlist_prbl" \
		"https://repo.parabola.nu/mirrorlist.txt" \
		"Downloading Parabola %q" \
		mirrorlist
	if [[ $? == 255 ]] ; then
		sed -i 's|^Server|#Server|' "$mirrorlist_prbl"
		setmirror "Parabola" "$mirror_prbl" "$mirrorlist_prbl"
	fi

	downloadfile \
		"$mirrorlist_arch" \
		"https://projects.archlinux.org/svntogit/packages.git/plain/pacman-mirrorlist/trunk/mirrorlist" \
		"Downloading Arch %q" \
		mirrorlist
	if [[ $? == 255 ]] ; then
		setmirror "Arch" "$mirror_arch" "$mirrorlist_arch"
	fi
}

# 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() {
	local cmp
	if [[ -n "${ver_arch[$pkgname]}" ]] ; then
		cmp=$(vercmp "${ver_prbl[$pkgname]}" "${ver_arch[$pkgname]}")
		if [[ "$cmp" -lt 0 ]]; 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
				cmp=$(vercmp "${ver_prbl[$pkgname]}" "${ver_arch[$provide]}")
				if [[ "$cmp" -lt 0 ]]; 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"
	awk -F/ -v repo="$repo" \
		'$1 == repo {print $2}' \
		"$prbl_packages_tmp" | \
			while read -a line ; do
				ver_prbl["${line[0]}"]="${line[1]}"
				provides["${line[0]}"]="${line[*]:2}"
				pkgname="${line[0]}"
				compare_pkgs
			done
}

usage() {
	print "Usage: sudo %q" "$cmd"
	print "   or: %q -n [REPO]" "$cmd"
	print "   or: %q -h" "$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 is to download/update all necessary
	       files for comparison, but not compare them.  Specify the \`-n\`
	       flag to not download anything, but to compare 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."
	flag '-h' 'Show this message'
	echo
	print "Output format:"
	print "type_character  parabola_pkgname  (arch_pkgname)   parabola_pkgver - (arch_pkgver)"
	echo
	print "Type characters:"
	flag '=' "Arch package with the same pkgname and greater pkgver was found"
	flag 'p' "Arch package with pkgname equal to a provide and greater
	          pkgver was found In this case arch_pkgname is a provide of
	          parabola_pkgname"
	flag 'o' "No Arch package with the same pkgname or with pkgname equal to
	          a provide was found"
}

main() {
	local args arg mode=update badargs=false
	args="$(getopt -n "$cmd" -o nh -l noupdate,help -- "$@")" || badargs=true
	eval "set -- $args"
	while [[ $# -gt 0 ]]; do
		arg=$1
		shift
		case "$arg" in
			-n|--noupdate)
				mode=compare
				;;
			-h|--help)
				usage
				return 0
				;;
			--)
				break
				;;
		esac
	done
	if $badargs; then
		usage >&2
		return 2
	fi

	case $mode in
		update)
			if [[ $# -gt 0 ]]; then
				print "%s: found non-flag arguments; a repo list may only be specified with the -n flag: %s" "$cmd" "$*" >&2
				usage >&2
				return 2
			fi
			main_update
			;;
		compare)
			if [[ $# -gt 1 ]]; then
				print "%s: found too many non-flag arguments; only one repo may be specified: %s" "$cmd" "$*" >&2
				usage >&2
				return 2
			fi
			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 "$@"
			;;
	esac
}

main_update() {
	if true; then
		if [[ $EUID != 0 ]]; then
			die "To initialize %s or update %s pacman databases, %s must be run as root (without arguments).  Nothing done." \
				"$name" \
				"$name" \
				"$cmd"
		fi
		check_vars libredbdiff statedir mirror_prbl mirror_arch || exit 1

		if ! [[ -e "$conffile_prbl" && \
		        -e "$conffile_arch" && \
		        -e "$mirrorlist_prbl" && \
		        -e "$mirrorlist_arch" ]]; then
			warning "At least one %s configuration file is missing." \
				"$name"
			msg "Downloading and preparing missing configuration files."
			init=1
		fi

		createdir "$statedir"

		initialize

		if ! [[ -d "$dbpath_prbl" && \
		        -d "$dbpath_arch" ]]; then
			warning "At least one %s pacman DB directory is missing.  Synchronizing %s DB files." \
				"$name" "$name"
		fi

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

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

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

main_compare() {
	if true; then
		check_vars libredbdiff statedir repos || exit 1

		filenotfound "$dbpath_prbl/sync/libre.db"
		filenotfound "$dbpath_arch/sync/core.db"
		filenotfound "$conffile_prbl"
		filenotfound "$conffile_arch"

		local tmpdir
		tmpdir="$(mktemp --tmpdir -d "$cmd.XXXXXXXXXX")" || die "Could not create temporary working directory"
		trap "rm -rf -- $(printf %q "$tmpdir")" 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

		pacman --config "$conffile_arch" -Ss | \
				grep -v '^ ' | \
				awk -F/ '{print $2}' \
			> "$arch_packages_tmp" || \
			die "pacman command to get Arch package data has failed.  Exiting."

		while read -a line; do
			ver_arch["${line[0]}"]="${line[1]}"
		done < "$arch_packages_tmp"

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

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

main "$@"