#!/bin/sh

# merge
# collects info from package file and copies it into another package file

# merge [ -d ZPMDB ] [-f $pkgfile] pkg
# -F include file content
# -S don't include script content

die() {
	printf 'zpm-merge:' 1>&2
	printf ' %s' "$@" 1>&2
	printf '\n' 1>&2
	exit 1
}

verbose=0
mergefiles=0
mergescripts=1
mergeall=1
update=0
remove_older=0
only_if_newer=0

target=${ZPMDB:=/var/lib/zpm/local.db}

# TODO option to only update if new
# new is hash is different or build time is later
# TODO option to merge all packages found in pkgfile
while getopts :f:vd:FSs:auon opt; do
	case $opt in
		f) pkgfile="$OPTARG" ;;
		v) verbose=1 ;;
		d) target="$OPTARG" ;;
		F) mergefiles=1 ;;
		S) mergescripts=0 ;;
		s) newstatus="$OPTARG" ;;
		a) mergeall=1 ;;
		u) update=1 ;;
		o) remove_older=1 ;;
		n) only_if_newer=1 ;;
		*) echo 'zpm-merge unknown option' $OPTARG; exit 1 ;;
	esac
done
shift $((OPTIND - 1))

# TODO test target
if [ ! -e "$target" ]; then
	die "target file $target does not exist"
fi

zpm test -v "$target" || exit 1

if [ $# -eq 0 ]; then
	mergeall=1
else
	mergeall=0
fi

# TODO if pkgfile is specified, allow multiple packages as args
if [ $mergeall -eq 0 ]; then
	pkgid=$1
	shift

	if [ -z "$pkgid" ]; then
		die "must specify pkgid"
	fi

	eval "$(zpm parse -E $pkgid)"

	if [ -z "$pkgfile" ]; then
		pkgfile=$ZPM_PACKAGE_FILE
	fi

	# calculate package id, pkgfile, etc
	# cases R = full package id, F = specified package file

	# immediate error
	# --- 000 error, must specify something
	if [ -z "$release" ] && [ -z "$pkgfile" ]; then
		die must specify package file or complete package id
	fi

	# try to get from package file
	if [ -z "$release" ]; then
		pkgid=$(zpm findpkg -f $pkgfile $pkgid)
		if [ -z "$pkgid" ]; then
			die cannot find package id
		fi
		eval "$(zpm parse -E $pkgid)"
	fi

	# --F 001 error, wouldn't know which pkgid to create, could derive from file?
	if [ -z "$release" ]; then
		die must specify complete package id
	fi

	# set file from pkgid
	# -R- 010 set file from pkgid, create in file, error if no file
	if [ -z "$pkgfile" ]; then
		pkgfile="$pkgid.zpm"
	fi

	if [ $verbose -gt 0 ]; then
		echo merging $pkgfile $pkgid into $target
	fi
	pkglist=$pkgid
else
	pkglist=$(zpm list -f "$pkgfile")
fi

zpm test -v "$pkgfile" || exit 1

if [ -n "$newstatus" ]; then
	newstatus=$(zpm quote "$newstatus")
fi

mergeone() {
	pkgid=$1
	where="where printf('%s-%s-%s', P.package, P.version, P.release) = '$pkgid'"
	if [ $update -eq 1 ]; then
		printf "delete from packages_pkgid where pkgid = '%s';\n" "$pkgid"
	fi
	if [ $remove_older -eq 1 ]; then
		printf "delete from packages_pkgid where pkgid < '%s' collate vercmp;\n" "$pkgid"
	fi
	printf "insert or rollback into packages select * from remote.packages P %s;\n" "$where"
	printf "insert or rollback into packagefiles select * from remote.packagefiles P %s;\n" "$where"
	printf "insert or rollback into scripts select * from remote.scripts P %s;\n" "$where"
	if [ $mergefiles -eq 1 ]; then
		printf "insert into files select F.* from remote.files F\n"
	       	printf "inner join remote.packagefiles_pkgid P on P.hash = F.hash %s\n" "$where"
		printf "on conflict (hash) do nothing;\n";
	fi
	# scripts
	if [ $mergescripts -eq 1 ]; then
		printf "insert into files select F.* from remote.files F\n"
		printf "inner join remote.scripts_pkgid P on P.hash = F.hash where P.pkgid = '%s'\n" "$pkgid"
		printf "on conflict (hash) do nothing;\n";
	fi
	if [ -n "$newstatus" ]; then
		newstatus=$(zpm quote "$newstatus")
		printf "update packages as P set status = '$newstatus' %s;\n" "$where"
	fi
}

# check for already merged packages
if [ $update -eq 0 ]; then
	efail=0

	for pkg in $pkglist; do
		merged=$(zpm shell "$target" "select 1 from packages_pkgid where pkgid = '$pkgid'")
		if [ -n "$merged" ]; then
			warn "$pkgid already exists in $target";
			efail=1
		fi
	done
	if [ $efail -eq 1 ]; then
		die "aborting merge"
	fi
fi

check_newer() {
	pkg=$1
	package=$(zpm parse -n "$pkg")
	newer=$(zpm shell "$target" "select 1 from packages_pkgid where '$pkg' > (select max(pkgid) collate vercmp from packages_pkgid where package = '$package') and package = '$package'")
	if [ "$newer" = '1' ]; then
		return 0;
	fi
	return 1
}

mlist="$pkglist"
if [ $only_if_newer -eq 1 ]; then
	mlist=
	for pkgid in $pkglist; do
		if check_newer "$pkgid" ; then
			mlist="$mlist $pkgid"
		fi
	done
	if [ -z "$mlist" ]; then
		exit 0
	fi
fi

# TODO file tags and package tags
{
	printf ".bail on\n"
	printf "attach '%s' as remote;\n" "$pkgfile"
	printf "begin;\n"

	for pkgid in $pkglist; do
		mergeone "$pkgid"
	done

	printf "insert or ignore into elfneeded select * from remote.elfneeded;\n"
	printf "insert or ignore into elflibraries select * from remote.elflibraries;\n"

	printf "commit;\n"
} | zpm shell $target

if [ $verbose -ne 0 ]; then
	printf "%s\n" $pkglist
fi