#!/bin/bash
# $Id$

# setup exit trap for reboot
trap_exit()
{
	if [[ -f /.dist-upgrade ]]
	then
		rm -f /.dist-upgrade
		# force a reboot at this point here
		( sleep 3 && reboot -f & ); exit 0
	fi	
}
trap "trap_exit" EXIT

# get a suitable environment
source /etc/profile

MAGERC="/etc/mage.rc"
MAGEPROFILE="alx-060"
NEW_MAGE_DISTRIBUTION="unstable"
NEW_RSYNC="rsync://magellan-linux.de/mage-alx-0.6.0"
NEW_MIRRORS="http://magellan-linux.de/magellan/alx-0.6.0/${NEW_MAGE_DISTRIBUTION}"
NEW_PACKAGES_SERVER_PATH='packages/${ARCH}'
NEW_MAGE_UNINSTALL_TIMEOUT=0

CONFIGRC="/etc/alxconfig-ng/config.rc"
NEW_ALX_NETWORKING="dhcp"
NEW_ALX_DHCP_PROG="/sbin/udhcpc"
NEW_ALX_DHCP_START="-T 10 -t 5 -n -i"
NEW_ALX_DHCP_STOP=""
NEW_MIN_DISK_SPACE="85000"

read_value()
{
	local file=$1
	local var=$2
	local value

	[[ ! -e ${file} ]] && return 1

	value=$(source ${file}; echo $(eval echo \${${var}}))
	echo "${value}"
	return 0
}

updateconfig()
{
	local variables="$@"
	local value
	local i

	if [[ -z ${CONFIG} ]]
	then
		echo "You must define \$CONFIG varibale first!"
		return 1
	fi

	for i in ${variables}
	do
		value="$(eval echo \${NEW_${i}})"
		if [[ ! -z $(grep "^${i}=" ${CONFIG}) ]]
		then
			echo "fixing ${i} -> ${value}"
			sed -i "s|^\(${i}=\).*|\1\"${value}\"|" ${CONFIG}
		else
			echo "adding ${i}=${value}"
			echo "${i}=\"${value}\"" >> ${CONFIG}
		fi
	done
}

# fix mage.rc
CONFIG="${MAGERC}"
updateconfig RSYNC MIRRORS MAGE_DISTRIBUTION PACKAGES_SERVER_PATH MAGE_UNINSTALL_TIMEOUT

# fix config.rc
CONFIG="${CONFIGRC}"
updateconfig ALX_NETWORKING ALX_DHCP_PROG ALX_DHCP_START ALX_DHCP_STOP MIN_DISK_SPACE

# fix profile
if [[ $(readlink /etc/mage-profile) != */${MAGEPROFILE} ]]
then
	echo "fixing profile link -> /usr/mage/profiles/${MAGEPROFILE}"
	ln -snf /usr/mage/profiles/${MAGEPROFILE} /etc/mage-profile
fi

# update mage3 -> mage4
if [[ -z $(magequery -n mage) ]]
then
	# update mage tree
	mage update

	# mage3 has problems with md5
	rm -rf /usr/mage/app-mage/mage/md5
	if [[ ${NEW_MAGE_DISTRIBUTION} = unstable ]]
	then
		USE_UNSTABLE=true MAGE_DISTRIBUTION=unstable mage install mage
	elif [[ ${NEW_MAGE_DISTRIBUTION} = testing ]]
	then
		USE_TESTING=true MAGE_DISTRIBUTION=testing mage install mage
	else
		mage install mage
	fi

	# drop all virtuals
	:> /var/db/mage/virtuals

	# enable run of orphaned files check
	touch /.orphaned

	# tell that we're running an dist-upgrade
	touch /.dist-upgrade
fi

# only run this if X11R6 is a directory and not a symlink
if [[ -d /usr/X11R6 ]] && [[ ! -L /usr/X11R6 ]]
then
	# check for -f option
	if [[ -n $(magequery -h | grep -- -f) ]]
	then
		# uninstall all /usr/X11R6 packages first
		for i in $(magequery -f /usr/X11R6 | sed 's:.*/\(.*\)-.*-r.*:\1:')
		do
			mage uninstall ${i}
		done
	fi

	# if there are any files after uninstall, simply delete them
	[ -d /usr/X11R6 ] && rm -rf /usr/X11R6
fi

# install new toolchain if not exist
TOOLCHAIN="$(< /etc/mage-profile/toolchain.defaults)"
if [[ -z $(magequery -n ${TOOLCHAIN}) ]]
then
	# export bootstrap to not start any services
	export MAGE_BOOTSTRAP=true
	mage install ${TOOLCHAIN}
	unset MAGE_BOOTSTRAP

	# enable run of orphaned files check
	touch /.orphaned

	# tell that we're running an dist-upgrade
	touch /.dist-upgrade
fi

# install new basesystem
BASESYSTEM="$(< /etc/mage-profile/basesystem.defaults)"
if [[ -z $(magequery -n ${BASESYSTEM}) ]]
then
	# first keep some important files
	
	# export bootstrap to not start any services
	export MAGE_BOOTSTRAP=true
	mage install ${BASESYSTEM}
	unset MAGE_BOOTSTRAP

	# enable run of orphaned files check
	touch /.orphaned

	# tell that we're running an dist-upgrade
	touch /.dist-upgrade
fi

# install remserial, if the plugin was enabled
if [[ ! -z $(magequery -n remserial-alx) ]]
then
	mage install remserial

	# enable run of orphaned files check
	touch /.orphaned
fi

# etc-update
#etc-update

# clean mage cache
mage clean

echo "Searching for deprecated packages ..."
for i in $(magequery -i | grep -- -alx | sed 's:.*/\(.*\)-.*-.*:\1:')
do
	# excludes
	case ${i} in
		alxconfig-ng) continue ;;
		alxinstall-ng) continue ;;
		kernel-alx) continue ;;
		kernel26-alx) continue ;;
		kernel-sources-alx) continue ;;
		kernel26-sources-alx) continue ;;
	esac

	echo "Removing deprecated mage-target '${i}'"
	mage uninstall ${i}
done

if [[ -f /.dist-upgrade ]]
then
	# array of wireless opts
	WIRELESS_OPTS=( WIRELESS_BITRATE WIRELESS_CHANNEL WIRELESS_ESSID WIRELESS_FREQUENCY WIRELESS_MODE WIRELESS_NICK WIRELESS_AUTH_MODE WIRELESS_KEY_LENGTH WIRELESS_KEY WIRELESS_KEY_ASCII WIRELESS_WPA_DRIVER )

	# fixing current dhcp network settings
	#for i in $(grep -irl NETWORKING=.*dhcp.* /etc/conf.d/net.*)
	for i in $(grep -irl DHCP_PROG=.*dhcpcd.* /etc/conf.d/net.*)
	do
		case ${i} in
			*/net.sample) continue ;;
			*/net.routes) continue ;;
		esac

		echo "fixing current dhcp network settings in '${i}'"
		mv ${i}{,.orig}
		onboot="$(read_value ${i}.orig ONBOOT)"
		echo "ONBOOT=\"${onboot}\"" > ${i}
		echo 'NETWORKING="dhcp"' >> ${i}
		# check wireless extensions
		for opt in ${WIRELESS_OPTS[*]}
		do
			value="$(read_value ${i}.orig ${opt})"
			if [[ ! -z ${value} ]]
			then
				echo "${opt}=\"${value}\"" >> ${i}
			fi
		done
		rm -f ${i}.orig
	done
	# search modules files
	for i in $(find /etc/modules.d -name net.\*)
	do
		install -d /etc/modprobe.d
		mv ${i} /etc/modprobe.d/${i%.conf}.conf
	done
fi

if [[ -f /.orphaned ]]
then
	echo "Searching for orphaned files and directories ..."
	bash /etc/mage-profile/prune-orphaned-files
	rm -f /.orphaned
fi

#if [[ -f /.dist-upgrade ]]
#then
#	rm -f /.dist-upgrade
#	# force a reboot at this point here
#	( reboot -f & ); exit 0
#fi