#!/bin/sh

# lh_chroot_sources(1) - manage /etc/apt/sources.list
# Copyright (C) 2006-2008 Daniel Baumann <daniel@debian.org>
#
# live-helper comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
# This is free software, and you are welcome to redistribute it
# under certain conditions; see COPYING for details.

set -e

# Including common functions
LH_BASE="${LH_BASE:-/usr/share/live-helper}"

for FUNCTION in "${LH_BASE}"/functions/*.sh
do
	. "${FUNCTION}"
done

# Setting static variables
DESCRIPTION="manage /etc/apt/sources.list"
HELP=""
USAGE="${PROGRAM} {install|remove} [--force]"

Arguments "${@}"

# Reading configuration files
Read_conffile config/all config/common config/bootstrap config/chroot config/binary config/source
Set_defaults

# Requiring stage file
Require_stagefile .stage/bootstrap

case "${1}" in
	install)
		Echo_message "Configuring file /etc/apt/sources.list"

		# Checking stage file
		Check_stagefile .stage/chroot_sources

		# Checking lock file
		Check_lockfile .lock

		# Creating lock file
		Create_lockfile .lock

		# Configure custom sources.list
		echo "deb ${LH_MIRROR_CHROOT} ${LH_DISTRIBUTION} ${LH_SECTIONS}" > chroot/etc/apt/sources.list

		if [ "${LH_SOURCE}" = "enabled" ]
		then
			echo "deb-src ${LH_MIRROR_CHROOT} ${LH_DISTRIBUTION} ${LH_SECTIONS}" >> chroot/etc/apt/sources.list
		fi

		if [ "${LH_SECURITY}" = "enabled" ]
		then
			if [ "${LH_DISTRIBUTION}" != "sid" ] && [ "${LH_DISTRIBUTION}" != "unstable" ]
			then
				echo "deb ${LH_MIRROR_CHROOT_SECURITY} ${LH_DISTRIBUTION}/updates ${LH_SECTIONS}" >> chroot/etc/apt/sources.list

				if [ "${LH_SOURCE}" = "enabled" ]
				then
					echo "deb-src ${LH_MIRROR_CHROOT_SECURITY} ${LH_DISTRIBUTION}/updates ${LH_SECTIONS}" >> chroot/etc/apt/sources.list
				fi
			fi
		fi

		# Check local sources.list
		if ls config/chroot_sources/*.chroot > /dev/null 2>&1
		then
			echo "" >> chroot/etc/apt/sources.list
			echo "# Custom repositories" >> chroot/etc/apt/sources.list

			for FILE in config/chroot_sources/*.chroot
			do
				cat ${FILE} | grep -v "^##" >> chroot/etc/apt/sources.list
			done
		fi

		# Configure local package repository
		if ls config/chroot_local-packages/*.deb > /dev/null 2>&1
		then
			rm -rf chroot/root/local-packages
			mkdir -p chroot/root/local-packages

			# Copy packages
			if ls config/chroot_local-packages/*_"${LH_ARCHITECTURE}".deb > /dev/null 2>&1
			then
				cp -l config/chroot_local-packages/*_"${LH_ARCHITECTURE}".deb chroot/root/local-packages
			fi

			if ls config/chroot_local-packages/*_all.deb > /dev/null 2>&1
			then
				cp -l config/chroot_local-packages/*_all.deb chroot/root/local-packages
			fi

			if ls chroot/root/local-packages/*.deb > /dev/null 2>&1
			then
				# Generate Packages.gz
				echo "cd /root/local-packages && apt-ftparchive packages . > Packages" | Chroot sh
				gzip -9 chroot/root/local-packages/Packages

				# Add to sources.list
				echo "" >> chroot/etc/apt/sources.list
				echo "# Local packages" >> chroot/etc/apt/sources.list
				echo "deb file:/root/local-packages ./" >> chroot/etc/apt/sources.list
			else
				Echo_warning "Local packages must be named with suffix '_all.deb' or '_\$architecture.deb'."
			fi
		fi

		# Update indices from cache
		if [ "${LH_CACHE_INDICES}" = "enabled" ] && [ -d cache/indices_bootstrap ]
		then
			if ls cache/indices_bootstrap/secring.gpg* > /dev/null 2>&1
			then
				cp -f cache/indices_bootstrap/secring.gpg* chroot/etc/apt
			fi

			if ls cache/indices_bootstrap/trusted.gpg* > /dev/null 2>&1
			then
				cp -f cache/indices_bootstrap/trusted.gpg* chroot/etc/apt
			fi

			if [ -f cache/indices_bootstrap/pkgcache.bin ]
			then
				cp -f cache/indices_bootstrap/pkgcache.bin chroot/var/cache/apt
			fi

			if [ -f cache/indices_bootstrap/srcpkgcache.bin ]
			then
				cp -f cache/indices_bootstrap/srcpkgcache.bin chroot/var/cache/apt
			fi

			if ls cache/indices_bootstrap/*_Packages > /dev/null 2>&1
			then
				cp -f cache/indices_bootstrap/*_Packages chroot/var/lib/apt/lists
			fi

			if ls cache/indices_bootstrap/*_Sources > /dev/null 2>&1
			then
				cp -f cache/indices_bootstrap/*_Sources chroot/var/lib/apt/lists
			fi

			if ls cache/indices_bootstrap/*_Release* > /dev/null 2>&1
			then
				cp -f cache/indices_bootstrap/*_Release* chroot/var/lib/apt/lists
			fi

			if [ "${LH_APT}" = "aptitude" ] && [ ! -x /usr/bin/aptitude ]
			then
				Chroot "apt-get ${APT_OPTIONS} install aptitude"
			fi
		else # Get fresh indices
			# Check local gpg keys
			if ls config/chroot_sources/*.chroot.gpg > /dev/null 2>&1
			then
				for FILE in config/chroot_sources/*.chroot.gpg
				do
					cp ${FILE} chroot/root
					Chroot "apt-key add /root/$(basename ${FILE})"
					rm -f chroot/root/$(basename ${FILE})
				done
			fi

			# Check local keyring packages
			if ls config/chroot_sources/*.deb > /dev/null 2>&1
			then
				for PACKAGE in config/chroot_sources/*.deb
				do
					cp ${PACKAGE} chroot/root
					Chroot "dpkg -i /root/$(basename ${PACKAGE})"
					rm -f chroot/root/$(basename ${PACKAGE})
				done
			fi

			# Installing aptitude
			if [ "${LH_APT}" = "aptitude" ] && [ ! -x /usr/bin/aptitude ]
			then
				Chroot "apt-get ${APT_OPTIONS} update"
				Chroot "apt-get ${APT_OPTIONS} install aptitude"
			fi

			Apt update
			Apt upgrade
			Apt dist-upgrade

			# Installing keyring packages
			if [ -n "${LH_KEYRING_PACKAGES}" ]
			then
				Apt --force-yes install "${LH_KEYRING_PACKAGES}"
				Apt update
			fi

			if [ "${LH_CACHE_INDICES}" = "enabled" ]
			then
				mkdir -p cache/indices_bootstrap

				cp -f chroot/etc/apt/secring.gpg* cache/indices_bootstrap
				cp -f chroot/etc/apt/trusted.gpg* cache/indices_bootstrap

				cp -f chroot/var/cache/apt/pkgcache.bin cache/indices_bootstrap
				cp -f chroot/var/cache/apt/srcpkgcache.bin cache/indices_bootstrap

				cp -f chroot/var/lib/apt/lists/*_Packages cache/indices_bootstrap
				cp -f chroot/var/lib/apt/lists/*_Sources cache/indices_bootstrap
				cp -f chroot/var/lib/apt/*_Release* cache/indices_bootstrap
			fi
		fi

		# Creating stage file
		Create_stagefile .stage/chroot_sources
		;;

	remove)
		Echo_message "Deconfiguring file /etc/apt/sources.list"

		# Checking lock file
		Check_lockfile .lock

		# Creating lock file
		Create_lockfile .lock

		# Configure generic indices
		if [ "${LH_BINARY_INDICES}" = "enabled" ]
		then
			# Don't do anything if it's not required
			if [ "${LH_MIRROR_CHROOT}" = "${LH_MIRROR_BINARY}" ] && \
			[ "${LH_MIRROR_CHROOT_SECURITY}" = "${LH_MIRROR_BINARY_SECURITY}" ] && \
			[ ! -d chroot/root/local-packages ]
			then
				# Removing stage file
				rm -f .stage/chroot_sources

				exit 0
			fi

			# Cleaning apt list cache
			rm -rf chroot/var/lib/apt/lists
			mkdir -p chroot/var/lib/apt/lists/partial

			echo "deb ${LH_MIRROR_BINARY} ${LH_DISTRIBUTION} ${LH_SECTIONS}" > chroot/etc/apt/sources.list

			if [ "${LH_SOURCE}" = "enabled" ]
			then
				echo "deb-src ${LH_MIRROR_BINARY} ${LH_DISTRIBUTION} ${LH_SECTIONS}" >> chroot/etc/apt/sources.list
			fi

			if [ "${LH_SECURITY}" = "enabled" ]
			then
				if [ "${LH_DISTRIBUTION}" != "sid" ] && [ "${LH_DISTRIBUTION}" != "unstable" ]
				then
					echo "deb ${LH_MIRROR_BINARY_SECURITY} ${LH_DISTRIBUTION}/updates ${LH_SECTIONS}" >> chroot/etc/apt/sources.list

					if [ "${LH_SOURCE}" = "enabled" ]
					then
						echo "deb-src ${LH_MIRROR_BINARY_SECURITY} ${LH_DISTRIBUTION}/updates ${LH_SECTIONS}" >> chroot/etc/apt/sources.list
					fi
				fi
			fi

			# Check local sources.list
			if ls config/chroot_sources/*.binary > /dev/null 2>&1
			then
				echo "" >> chroot/etc/apt/sources.list
				echo "# Custom repositories" >> chroot/etc/apt/sources.list

				for FILE in config/chroot_sources/*.binary
				do
					cat ${FILE} | grep -v "^##" >> chroot/etc/apt/sources.list
				done
			fi

			# Check local gpg keys
			if ls config/chroot_sources/*.binary.gpg > /dev/null 2>&1
			then
				for FILE in config/chroot_sources/*.binary.gpg
				do
					cp ${FILE} chroot/root
					Chroot "apt-key add /root/$(basename ${FILE})"
					rm -f chroot/root/$(basename ${FILE})
				done
			fi

			# Updating indices
			Apt update
		fi

		# Cleaning apt packages cache
		rm -rf chroot/var/cache/apt
		mkdir -p chroot/var/cache/apt/archives/partial

		# Remove local package repository
		rm -rf chroot/root/local-packages

		# Removing stage file
		rm -f .stage/chroot_sources
		;;

	*)
		Usage
		;;
esac