#!/bin/sh

## live-build(7) - System Build Scripts
## Copyright (C) 2006-2011 Daniel Baumann <daniel@debian.org>
##
## live-build 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
. "${LB_BASE:-/usr/share/live/build}"/scripts/build.sh

# Setting static variables
DESCRIPTION="$(Echo 'manage /etc/hosts')"
HELP=""
USAGE="${PROGRAM} {install|remove} [--force]"

Arguments "${@}"

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

# Requiring stage file
Require_stagefile .stage/config .stage/bootstrap

case "${1}" in
	install)
		Echo_message "Configuring file /etc/hosts"

		# Checking stage file
		Check_stagefile .stage/chroot_hosts

		# Checking lock file
		Check_lockfile .lock

		# Creating lock file
		Create_lockfile .lock

		if [ -f chroot/etc/hosts ]
		then
			# Save hosts file
			mv chroot/etc/hosts chroot/etc/hosts.orig
		fi

		# Creating hosts file

cat > chroot/etc/hosts << EOF
127.0.0.1	localhost localhost.localdomain
127.0.1.1	debian
EOF

		if [ -f /etc/hosts ]
		then
			# Append hosts file
			#grep -e "127.0.0.1" -e "127.0.1.1" /etc/hosts >> chroot/etc/hosts
			cat /etc/hosts >> chroot/etc/hosts
		fi

		# Creating stage file
		Create_stagefile .stage/chroot_hosts
		;;

	remove)
		Echo_message "Deconfiguring file /etc/hosts"

		# Checking lock file
		Check_lockfile .lock

		# Creating lock file
		Create_lockfile .lock

		if [ -e config/chroot_local-includes/etc/hosts ]
		then
			# Copying local hosts
			cp -a config/chroot_local-includes/etc/hosts chroot/etc/hosts
			rm -f chroot/etc/hosts.orig
		elif [ -f chroot/etc/hosts.orig ]
		then
			# Restore hosts file
			mv chroot/etc/hosts.orig chroot/etc/hosts
		else
			# Blank out hosts file, don't remove in case
			# its a symlink, as in the case of exposedroot mode
			Truncate chroot/etc/hosts
		fi

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

	*)
		Usage
		;;
esac