#!/bin/sh -e

echo "Adding APT repositories"

. /etc/os-release

debian_apt_repo () {
	repo="$(printf "$*" | cut -d' ' -f-3)"
	suites="$(printf "$*" | cut -d' ' -f4-)"
	# Potentially uncomment
	if grep -q "^# $repo" /etc/apt/sources.list; then
		sed -i "s,# $repo,$repo," /etc/apt/sources.list
	fi

	if grep -q "^$repo" /etc/apt/sources.list; then
		for suite in $suites; do
			if grep -q "^$repo .*$suite" /etc/apt/sources.list; then
				# We're good, do nothing
				echo "No new repository or suite necessary"
			else
				sed -i "s,^$repo \(.*\)\$,$repo \1 $suite," /etc/apt/sources.list
			fi
		done
	else
		echo "$*" >> /etc/apt/sources.list
	fi
}

if [ "$ID" = "ubuntu" ]; then
	echo "Adding ${UBUNTU_CODENAME}-security"
	add-apt-repository "deb http://security.ubuntu.com/ubuntu/ ${UBUNTU_CODENAME}-security main universe"
	echo "Adding ${UBUNTU_CODENAME}-updates"
	add-apt-repository "deb http://us.archive.ubuntu.com/ubuntu/ ${UBUNTU_CODENAME}-updates main universe"
elif [ "$ID" = "debian" ]; then
	echo "Adding ${VERSION_CODENAME}-updates"
	debian_apt_repo "deb http://deb.debian.org/debian ${VERSION_CODENAME}-updates main"
	echo "Adding ${VERSION_CODENAME} security"
	debian_apt_repo "deb http://security.debian.org/debian-security/ ${VERSION_CODENAME}/updates main"
fi

