#!/bin/sh

# You can put local mirrors here if you want
kali_mirror=http://archive.kali.org/kali
kalisec_mirror=http://archive.kali.org/kali-security

### DO NOT EDIT THE REST OF THIS FILE ###

public_kali_mirror=http://http.kali.org/kali
public_kalisec_mirror=http://security.kali.org/kali-security

# Detect target architecture and filter args
temp=""
arch=$(dpkg --print-architecture)
dist="kali"
lb_opts=""
while [ $# -gt 0 ]; do
   	arg="$1"
	case "$arg" in
        -a|--arch|--architecture|--architectures)
		arch="$2"
		temp="$temp "'"'"$arg"'"'
		temp="$temp "'"'"$2"'"'
		shift
		;;
	-p|--proposed-updates)
		enable_pu="1"
		;;
	-d|--kali-dev)
		dist="kali-dev"
		# No security repo
		lb_opts="--security false --initsystem systemd"
		;;
	-r|--kali-rolling)
		dist="kali-rolling"
		lb_opts="--initsystem systemd"
		;;
	--)
		# Skip the separator, it was added so that "lb config"
		# doesn't barf on our own options, but now we are
		# filtering them away assuming that the remaining ones
		# are intended for lb config !
		;;
	*)
		temp="$temp "'"'"$arg"'"'
		;;
	esac
	shift
done
eval set -- "$temp"

# live-build doesn't work if --parent-debian-distribution is unknown of
# debian-cd => we have to put a symlink so that it deals with kali like wheezy
if [ ! -e ${LIVE_BUILD:-/usr/share/live/build}/data/debian-cd/$dist ]; then
	if [ -w ${LIVE_BUILD:-/usr/share/live/build}/data/debian-cd ]; then
		ln -sf wheezy ${LIVE_BUILD:-/usr/share/live/build}/data/debian-cd/$dist
	else
		echo "ERROR: Run this first:"
		echo "ln -sf wheezy ${LIVE_BUILD:-/usr/share/live/build}/data/debian-cd/$dist"
		exit 1
	fi
fi

case "$arch" in
    amd64)
	lb_opts="$lb_opts --debian-installer live"
    ;;
    i386)
	lb_opts="$lb_opts --debian-installer live --linux-flavours 686-pae"
    ;;
    armel|armhf)
	lb_opts="$lb_opts --binary-images hdd --binary-filesystem ext4 --chroot-filesystem none"
    ;;
    *)
	echo "WARNING: configuration not tested on arch $arch" >&2
    ;;
esac

rm -f config/archives/kali-proposed-updates.list.*
if [ -n "$enable_pu" ]; then
	mkdir -p config/archives
	echo "deb $kali_mirror kali-proposed-updates main contrib non-free" \
		> config/archives/kali-proposed-updates.list.chroot
	echo "deb $public_kali_mirror kali-proposed-updates main contrib non-free" \
		> config/archives/kali-proposed-updates.list.binary
fi

if [ -e config/includes.installer/preseed.cfg ]; then
	sed -i -e '/^# KALI OPTIONS/,$ d' config/includes.installer/preseed.cfg
fi

cat >>config/includes.installer/preseed.cfg <<END
# KALI OPTIONS
# Default repository information
d-i mirror/country string enter information manually
d-i mirror/suite string $dist
d-i mirror/http/hostname string http.kali.org
d-i mirror/http/directory string /kali

# Add our own security mirror
d-i apt-setup/local0/repository string http://security.kali.org/kali-security $dist/updates main contrib non-free
d-i apt-setup/local0/comment string Security updates
d-i apt-setup/local0/source boolean false
END

lb config noauto \
	--distribution "$dist" \
	--debian-installer-distribution "$dist" \
	--archive-areas "main contrib non-free" \
	--bootstrap cdebootstrap \
	--cdebootstrap-options "--keyring=/usr/share/keyrings/kali-archive-keyring.gpg --suite-config=wheezy" \
	--keyring-packages kali-archive-keyring \
	--updates false \
	--backports false \
	--source false \
	--firmware-binary true \
	--firmware-chroot true \
	--mirror-bootstrap "$kali_mirror" \
	--mirror-debian-installer "$kali_mirror" \
	--mirror-chroot-security "$kalisec_mirror" \
	--mirror-binary "$public_kali_mirror" \
	--mirror-binary-security "$public_kalisec_mirror" \
	--iso-application "Kali Linux" \
	--iso-publisher "Kali" \
	--iso-volume "Kali Live" \
	--linux-packages linux-image \
	--bootappend-live "boot=live noconfig=sudo username=root hostname=kali" \
	$lb_opts \
	"$@"

if [ $dist = kali ]; then
	# Don' use systemd for now
	sed -i -e 's/systemd/sysvinit/' config/package-lists/live.list.chroot
fi