#!/bin/bash
set -e

validate_unique()
{
	[ "$#" = "2" ]
	local duplicates=$(echo $2 | tr " " "\n" | sort | uniq -d)
	if [ "$duplicates" ]; then
		echo "EE: duplicate values in $1" 1>&2
		exit 1
	fi
}

# Pick out relevant version and package information including our predecessor
# packages: linux-restricted-generate -> linux-restricted-signatures -> linux-restricted-modules
src_package=$(LC_ALL=C dpkg-parsechangelog -SSource)
src_version=$(LC_ALL=C dpkg-parsechangelog -SVersion)
src_abi=$(echo "${src_version}" | sed -ne 's/\([0-9]*\.[0-9]*\.[0-9]*\-[0-9]*\)\..*/\1/p')
src_series=$(LC_ALL=C dpkg-parsechangelog -SDistribution | sed -e 's/-\(security\|updates\|proposed\)$//')

# linux/5.8.0-41.46
src_main_package=$(echo "${src_package}" | sed -e 's/-restricted-signature//')
src_main_version=$(echo ${src_version} | sed -e 's/+[0-9][0-9\.]*$//')

# linux-restricted-generate/5.8.0-41.46[+1]
src_genr_package=$(echo "${src_package}" | sed -e 's/-restricted-signatures/-restricted-generate/')
src_genr_version=${src_version}

# linux-restricted-signatures/5.8.0-41.46[+1]

# linux-restricted-modules/5.8.0-41.46[+1]
src_lrm_package=$(echo "${src_package}" | sed -e 's/-restricted-signatures/-restricted-modules/')
src_lrm_version=${src_version}

cat - "debian/rules.in" >"debian/rules.gen" <<EOL
#! /usr/bin/make -f

src_package := ${src_package}
src_version = ${src_version}
src_abi = ${src_abi}
src_series = ${src_series}
src_genr_package = ${src_genr_package}
src_genr_version = ${src_genr_version}

EOL

echo " ${src_genr_package} (= ${src_genr_version})," >"debian/control.interlock-up"

build_archs=
while read command flavour archs
do
	case "$command" in
	build)		;;
	*)		continue ;;
	esac

	for arch in $archs
	do
		case " $build_archs " in
		*\ $arch\ *)	;;
		*)		build_archs="$buildarchs $arch" ;;
		esac
	done
done <"debian/package.config"

while read command flavour archs
do
	case "$command" in
	build)		;;
	*)		continue ;;
	esac

	validate_unique "archs" "$archs"
	targets=$(echo "$archs" | sed -e 's/\</nvidia-/g')

	echo "II: build linux-signatures-nvidia-${src_abi}-${flavour} for ${archs}"

	cat - >>"debian/rules.gen" <<EOL

$targets::
	for stream in "${src_genr_version}/${src_abi}-${flavour}/signatures"/*; \\
	do \\
		nvidia=\$\$(basename "\$\$stream"); \\
		echo "\$\$stream/* /lib/modules/${src_abi}-${flavour}/kernel/\$\$nvidia/bits" >>debian/linux-signatures-nvidia-${src_abi}-${flavour}.install; \\
	done
EOL

	cat - >>"debian/control.signatures" <<EOL

Package: linux-signatures-nvidia-${src_abi}-${flavour}
Build-Profiles: <!stage1>
Architecture: ${archs}
Section: kernel
Priority: optional
Depends: linux-image-${src_abi}-${flavour} | linux-image-unsigned-${src_abi}-${flavour},
Built-Using: \${linux:BuiltUsing}
Description: Linux kernel signatures for nvidia modules for version ${src_abi}-${flavour}
 This package contains the Linux kernel signatures nvidia modules for
 version ${src_abi}-${flavour}.
 .
 You likely do not want to install this package directly.
EOL
done <"debian/package.config"

nvidia_desktop=
nvidia_server=
nvidia_ignore=
while read command arg
do
	case "$command" in
	option)		;;
	suppress)		nvidia_ignore="$nvidia_ignore $arg"; continue ;;
	*)		continue ;;
	esac

	case "$arg" in
	desktop)	nvidia_desktop=y ;;
	server)		nvidia_server=y ;;
	esac
done <"debian/package.config"

: >"debian/control.open"
while read command flavour archs
do
	case "$command" in
	build)		;;
	*)		continue ;;
	esac

	validate_unique "archs" "$archs"

	while read package version extra
	do
		case "$package" in
		nvidia-graphics-drivers-*-server)
			[ -z "$nvidia_server" ] && continue
			;;
		nvidia-graphics-drivers-*)
			[ -z "$nvidia_desktop" ] && continue
			;;
		*) continue ;;
		esac
		case " $nvidia_ignore " in
		*\ $package\ *)		continue ;;
		esac

		case " $extra " in
		*\ signonly\ *)		continue ;;
		esac

		# If package specifies arch= keys, only build it on
		# the intersection of kernel flavour arches & the
		# package arches, otherwise build for all kernel
		# flavour arches.
		pkg_archs=
		variant=
		for cmd in $extra
		do
			case "$cmd" in
			arch=*)		pkg_archs="$pkg_archs ${cmd#arch=}" ;;
			variant=*) variant="-${cmd#variant=}" ;;
			esac
		done

		if [ "$variant" != "-open" ]; then
			continue
		fi

		# No package arches set, build for everything
		if [ -z "$pkg_archs" ]; then
			pkg_archs=$archs
		else
			# Package arches set, build for intersection
			validate_unique "pkg_archs" "$pkg_archs"
			pkg_archs=$(sort <(echo $archs | tr " " "\n") <(echo $pkg_archs | tr " " "\n") | uniq -d | tr "\n" " ")
			# No intersection, skip
			if [ -z "$pkg_archs" ]; then
				continue
			fi
		fi

		targets=$(echo "$pkg_archs" | sed -e 's/\</nvidia-/g')
		suffix_minus=$(echo "$package$variant" | sed -e 's/nvidia-graphics-drivers-//')
		suffix_under=$(echo "$suffix_minus" | sed -e 's/-/_/g')
		suffix_short=$(echo "$suffix_minus" | sed -e 's/-server/srv/g')
		suffix_minus_common=$(echo "$package" | sed -e 's/nvidia-graphics-drivers-//')

		echo "II: build $package for $flavour $pkg_archs"

		cat - >>"debian/control.interlock-up" <<EOL
 linux-objects-nvidia-${suffix_minus}-${src_abi}-${flavour} (>= ${src_lrm_version}) [${pkg_archs}],
EOL

		# debian/rules.gen
		cat - >>"debian/rules.gen" <<EOL

# $package $version $suffix_minus $suffix_under
dkms_nvidia_${suffix_under}_version="$version"
nvidia_${suffix_under}_upstream_version=\$(shell echo "\$(dkms_nvidia_${suffix_under}_version)" | cut -d '-' -f 1)
gencontrol_flags += -Vdkms:nvidia-${suffix_minus}-modules="nvidia-dkms-${suffix_minus} (= \$(dkms_nvidia_${suffix_under}_version))"
gencontrol_flags += -Vnvk:nvidia-${suffix_minus}="nvidia-kernel-common-${suffix_minus_common} (<= \$(nvidia_${suffix_under}_upstream_version)-1), nvidia-kernel-common-${suffix_minus_common} (>= \$(nvidia_${suffix_under}_upstream_version))"
test::
	@echo $package $version $suffix_minus $suffix_under $suffix_short $targets
$targets::
	install -d debian/linux-modules-nvidia-${suffix_minus}-\$(src_abi)-${flavour}
	install -d debian/linux-modules-nvidia-${suffix_minus}-\$(src_abi)-${flavour}/usr/lib/linux/triggers
	\$(call install_control,linux-modules-nvidia-${suffix_minus}-\$(src_abi)-${flavour},${flavour},open,${suffix_short},postinst postrm); \\
	stream="${src_genr_version}/${src_abi}-${flavour}/open/nvidia-${suffix_short}"; \\
	nvidia="nvidia-${suffix_short}"; \\
	install -d debian/linux-modules-nvidia-${suffix_minus}-\$(src_abi)-${flavour}/lib/modules/kernel/\$\$nvidia; \\
	for sig in \$\$stream/*.ko.sig; \\
	do \\
		ko=\$\$(basename "\$\$sig" ".sig"); \\
		cat "/lib/modules/${src_abi}-${flavour}/kernel/\$\$nvidia/\$\$ko" "\$\$sig" >"\$\$stream/\$\$ko"; \\
		echo "\$\$stream/\$\$ko /lib/modules/${src_abi}-${flavour}/kernel/\$\$nvidia" >>debian/linux-modules-nvidia-${suffix_minus}-${src_abi}-${flavour}.install; \\
	done
EOL

		# debian/control.d/nvidia
		cat - >>"debian/control.open" <<EOL

Package: linux-modules-nvidia-${suffix_minus}-${src_abi}-${flavour}
Build-Profiles: <!stage1>
Architecture: ${pkg_archs}
Section: kernel
Priority: optional
Built-Using: \${dkms:nvidia-${suffix_minus}-modules}
Depends:
 \${misc:Depends},
 \${shlibs:Depends},
 linux-image-${src_abi}-${flavour} | linux-image-unsigned-${src_abi}-${flavour},
 \${nvk:nvidia-${suffix_minus}},
Description: Linux kernel nvidia modules for version ${src_abi}
 This package pulls together the Linux kernel nvidia modules for
 version ${src_abi} with the appropriate signatures.
 .
 You likely do not want to install this package directly. Instead, install the
 one of the linux-modules-nvidia-${suffix_minus}-${flavour}* meta-packages,
 which will ensure that upgrades work correctly, and that supporting packages are
 also installed.
EOL

	done <"debian/dkms-versions"
done <"debian/package.config"

cat "debian/control.common" "debian/control.signatures" "debian/control.open" | sed \
	-e "/@BUILD-INTERLOCK@/{"		\
	-e " r debian/control.interlock-up"	\
	-e " d"					\
	-e " }"					\
	-e "s/@SRCPKGNAME@/${src_package}/g"	\
	-e "s/@ABI@/${src_abi}/g"		\
	-e "s/@SERIES@/${src_series}/g"		\
    >"debian/control"

rm -f "debian/control.interlock-up" "debian/control.signatures" "debian/control.open"
