#!/bin/sh

DISLIKED_PACKAGES="lxde icewm "
RELEASE=`lsb_release -c -s`
URI="/tmp/set-$PPID.uris"
CACHE_DIR="/var/cache/apt/archives"
DOWNLOAD_DIR="/tmp/apt-offline-tests-$PPID"
BUNDLE_FILE="/tmp/apt-offline-tests-$PPID.zip"
THREADS=5
APT_OFFLINE="./apt-offline "

set_features () {
	if [ ! -z $1 ]; then
		URI=$1
	fi
	echo "Executing command 'set $URI'"
	$APT_OFFLINE set $URI

	echo "Executing command 'set $URI --simulate '"
	$APT_OFFLINE set $URI --simulate 

	echo "Executing command 'set $URI --update'"
	$APT_OFFLINE set $URI --update

	echo "Executing command 'set $URI --upgrade'"
	$APT_OFFLINE set $URI --upgrade

	echo "Executing command 'set $URI --update --upgrade'"
	$APT_OFFLINE set $URI --update --upgrade

	echo "Executing command 'set $URI --update --upgrade --upgrade-type upgrade'"
	$APT_OFFLINE set $URI --update --upgrade --upgrade-type upgrade

	echo "Executing command 'set $URI --update --upgrade --upgrade-type upgrade --release $RELEASE'"
	$APT_OFFLINE set $URI --update --upgrade --upgrade-type upgrade --release $RELEASE 

	echo "Executing command 'set $URI --install-packages $DISLIKED_PACKAGES'"
	$APT_OFFLINE set $URI --install-packages $DISLIKED_PACKAGES

	echo "Executing command 'set $URI --install-packages $DISLIKED_PACKAGES --release $RELEASE'"
	$APT_OFFLINE set $URI --install-packages $DISLIKED_PACKAGES --release $RELEASE

	echo "Executing command 'set $URI --install-src-packages $DISLIKED_PACKAGES'"
	$APT_OFFLINE set $URI --install-src-packages $DISLIKED_PACKAGES

	echo "Executing command 'set $URI --install-src-packages $DISLIKED_PACKAGES --release $RELEASE'"
	$APT_OFFLINE set $URI --install-src-packages $DISLIKED_PACKAGES --release $RELEASE

	echo "Executing command 'set $URI --src-build-dep --install-src-packages $DISLIKED_PACKAGES'"
	$APT_OFFLINE set $URI --src-build-dep --install-src-packages $DISLIKED_PACKAGES

	echo "Executing command 'set $URI --src-build-dep --install-src-packages $DISLIKED_PACKAGES --release $RELEASE'"
	$APT_OFFLINE set $URI --src-build-dep --install-src-packages $DISLIKED_PACKAGES --release $RELEASE

}

get_features () {
	if [ ! -z $1 ]; then
		URI=$1
	fi
	echo "Executing command 'get $URI '"
	$APT_OFFLINE get $URI 

	echo "Executing command 'get $URI --threads $THREADS'"
	$APT_OFFLINE get $URI --threads $THREADS

	echo "Executing command 'get $URI --threads $THREADS --socket-timeout 30'"
	$APT_OFFLINE get $URI --threads $THREADS --socket-timeout 30

	echo "Executing command 'get $URI --threads $THREADS -d $DOWNLOAD_DIR'"
	$APT_OFFLINE get $URI --threads $THREADS -d $DOWNLOAD_DIR

	echo "Executing command 'get $URI --threads $THREADS -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR'"
	$APT_OFFLINE get $URI --threads $THREADS -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR

	echo "Executing command 'get $URI --no-checksum -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR'"
	$APT_OFFLINE get $URI --no-checksum -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR

	echo "Executing command 'get $URI --bug-reports --threads $THREADS -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR'"
	$APT_OFFLINE get $URI --threads $THREADS --bug-reports -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR

	echo "Executing command 'get $URI --bug-reports --threads $THREADS --bundle $BUNDLE_FILE -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR'"
	$APT_OFFLINE get $URI --threads $THREADS --bug-reports -d $DOWNLOAD_DIR --cache-dir $CACHE_DIR --bundle $BUNDLE_FILE

}

install_features () {
	if [ ! -z $1 ]; then
		DOWNLOAD_DIR=$1
		BUNDLE_FILE=$1
	fi
	echo "Executing command 'install $DOWNLOAD_DIR  --skip-bug-reports'"
	$APT_OFFLINE install $DOWNLOAD_DIR  --skip-bug-reports

	echo "Executing command 'install $DOWNLOAD_DIR --simulate --skip-bug-reports'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate --skip-bug-reports

	echo "Executing command 'install $DOWNLOAD_DIR --skip-bug-reports'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate --skip-bug-reports

	echo "Executing command 'install $DOWNLOAD_DIR --skip-bug-reports --allow-unauthenticated'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate --skip-bug-reports  --allow-unauthenticated

	echo "Executing command 'install $BUNDLE_FILE  --skip-bug-reports'"
	$APT_OFFLINE install $BUNDLE_FILE  --skip-bug-reports

	echo "Executing command 'install $BUNDLE_FILE --simulate --skip-bug-reports'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate --skip-bug-reports

	echo "Executing command 'install $BUNDLE_FILE --skip-bug-reports'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate --skip-bug-reports

	echo "Executing command 'install $BUNDLE_FILE --skip-bug-reports --allow-unauthenticated'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate --skip-bug-reports  --allow-unauthenticated
}

install_features_prompt () {
	if [ ! -z $1 ]; then
		DOWNLOAD_DIR=$1
		BUNDLE_FILE=$1
	fi
	echo "Executing command 'install $DOWNLOAD_DIR '"
	$APT_OFFLINE install $DOWNLOAD_DIR 

	echo "Executing command 'install $DOWNLOAD_DIR --simulate'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate

	echo "Executing command 'install $DOWNLOAD_DIR'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate

	echo "Executing command 'install $DOWNLOAD_DIR --allow-unauthenticated'"
	$APT_OFFLINE install $DOWNLOAD_DIR --simulate --allow-unauthenticated

	echo "Executing command 'install $BUNDLE_FILE '"
	$APT_OFFLINE install $BUNDLE_FILE 

	echo "Executing command 'install $BUNDLE_FILE --simulate'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate

	echo "Executing command 'install $BUNDLE_FILE'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate

	echo "Executing command 'install $BUNDLE_FILE --allow-unauthenticated'"
	$APT_OFFLINE install $BUNDLE_FILE --simulate --allow-unauthenticated
}

all_features () {
	echo "Executing function set_features"
	set_features

	echo "Executing function get_features"
	get_features

	echo "Executing function install_features"
	install_features
}

case $1 in
	"set")
		if [ ! -z $2 ]; then
			set_features $2
		else
			set_features
		fi
		;;
	"get")
		if [ ! -z $2 ]; then
			get_features $2
		else
			get_features
		fi
		;;
	"install_features_promptless")
		if [ ! -z $2 ]; then
			install_features $2
		else
			install_features
		fi
		;;
	"install")
		# With prompts for bug reports
		if [ ! -z $2 ]; then
			install_features_prompt $2
		else
			install_features_prompt
		fi
		;;
	"--help")
		echo "$0 [set || get || install_features_promptless || install]"
		exit 0;
		;;
	"-h")
		echo "$0 [set || get || install_features_promptless || install]"
		exit 0;
		;;
	*)
		all_features
		;;
esac

