#!/bin/bash
# $Id$
#
# Simple Installation Script
# merged from alxinstall-ng and mcore-installer
#
# Niels Rogalla <niro@magellan-linux.de>
#

# setup locales
TEXTDOMAIN=installer

# include dir
INSTALLER_LIBDIR="%LIBDIR%"

# TOTALLINES=linecount of ${CDIMAGENAME}-tarball -1 !
# -> now in images.conf
CURRENTLINE=0

die()
{
	echo "Error: $@"
	exit 1
}

# load common includes
for inc in %SYSCONFDIR%/installer.conf \
	${INSTALLER_LIBDIR}/functions/common.sh \
	${INSTALLER_LIBDIR}/functions/common-dialogs.sh \
	${INSTALLER_LIBDIR}/functions/hwdetection.sh \
	${INSTALLER_LIBDIR}/functions/bootloader.sh \
	${INSTALLER_LIBDIR}/functions/initrd-tools.sh \
	${INSTALLER_LIBDIR}/functions/installer-dialogs.sh
do
	if [ -e ${inc} ]
	then
		source "${inc}"
	else
		die "'${inc}' not found"
	fi
done

# fetch cmdline
for argv in $*
do
	case $1 in
		--liveroot) shift; cmdliveroot="$1" ;;
		--imageroot) shift; cmdimageroot="$1" ;;
		--installroot) shift; cmdinstallroot="$1" ;;
		--filesystem) shift; cmdformatfilesystem="$1" ;;
		--help)
			echo "$0 options:"
			echo " --liveroot [path]     - override default liveroot"
			echo " --imageroot [path]    - override default imageroot"
			echo " --installroot [path]  - override default installroot"
			echo " --filesystem [fstype] - override default filesystem used to format the disk"
			exit 0 ;;
	esac
	shift
done

### System/Config Version
VERSION="%VERSIONTAG%"
TITLE="${DEFAULT_TITLE} - ${VERSION}"

# initialize global variables so they are exportable
# some sane defaults
INSTALL_METHOD=""
if [[ -n ${cmdliveroot} ]]
then
	LIVEROOT="${cmdliveroot}"
else
	LIVEROOT="${DEFAULT_LIVEROOT}"
fi
if [[ -n ${cmdimageroot} ]]
then
	IMAGEROOT="${cmdimageroot}"
else
	IMAGEROOT="${DEFAULT_IMAGEROOT}"
fi
if [[ -n ${cmdinstallroot} ]]
then
	INSTALLROOT="${cmdinstallroot}"
else
	INSTALLROOT="${DEFAULT_INSTALLROOT}"
fi
KERNELPKG="${DEFAULT_KERNELPKG}"
KERNELOPTS="${DEFAULT_KERNELOPTS}"
GRUBLEGACYOPTS=""
GRUB2OPTS=""
FDISKPARTIONBELOW256MB=0
SPECIALDEVICE=""
FORMFACTOR="${DEFAULT_FORMFACTOR}"
if [[ -n ${cmdformatfilesystem} ]]
then
	FORMAT_FILESYSTEM="${cmdformatfilesystem}"
else
	FORMAT_FILESYSTEM="${DEFAULT_FILESYSTEM}"
fi
FLASHDISK=0

# TOTALLINES=linecount of ${CDIMAGENAME}-tarball -1 !
# -> now in images.conf
CDIMAGENAME=""
TOTALLINES=""
CURRENTLINE=0
if [ -e ${IMAGEROOT}/images.conf ]
then
	source ${IMAGEROOT}/images.conf
	# check if all required variables are set
	[[ -z ${CDIMAGENAME} ]] && die "CDIMAGENAME is empty in ${IMAGEROOT}/images.conf"
	[[ -z ${TOTALLINES} ]] && die "TOTALLINES is empty in ${IMAGEROOT}/images.conf"
else
	die "${IMAGEROOT}/images.conf not found"
fi

### helper scripts ###

trap_exit()
{
	is_mounted --location "${INSTALLROOT}/dev" && umount ${INSTALLROOT}/dev
	is_mounted --location "${INSTALLROOT}/proc" && umount ${INSTALLROOT}/proc
	is_mounted --location "${INSTALLROOT}/sys" && umount ${INSTALLROOT}/sys
	is_mounted --location "${INSTALLROOT}/boot" && umount ${INSTALLROOT}/boot
	is_mounted --location "${INSTALLROOT}" && umount ${INSTALLROOT}
	[[ -n ${SWAPHDD} ]] && swapoff ${SWAPHDD}

	echo $"Installation aborted."
	exit 1
}

install_meter()
{
	while [[ ${CURRENTLINE} != ${TOTALLINES} ]]
	do
		CURRENTLINE=$(grep -c . /tmp/install.log)
		PERCENT=$(( ${CURRENTLINE} * 100 / ${TOTALLINES}))
		echo ${PERCENT}
		sleep 1
	done
	rm -f /tmp/install.log
	return 0
}

mount_rootfs()
{
	local retval

	if [[ -n ${SWAPHDD} ]]
	then
		swapon ${SWAPHDD} || dialog_die $"Could not enable swap space '${SWAPHDD}'"
	fi

	if [[ -n ${ROOTHDD} ]]
	then
		if is_mounted --location "${INSTALLROOT}"
		then
			echo $"${INSTALLROOT} already mounted" >&2
		else
			mount -t "${FORMAT_FILESYSTEM_ROOTHDD}" "${ROOTHDD}" "${INSTALLROOT}" || dialog_die $"Could not mount rootfs - drive '${ROOTHDD}' -> '${INSTALLROOT}'"
		fi
	fi

	[[ -d ${INSTALLROOT}/boot ]] || install -d ${INSTALLROOT}/boot

	if [[ -n ${BOOTHDD} ]]
	then
		if is_mounted --location "${INSTALLROOT}"/boot
		then
			echo $"${INSTALLROOT}/boot already mounted" >&2
		else
			mount -t "${FORMAT_FILESYSTEM_BOOTHDD}" "${BOOTHDD}" "${INSTALLROOT}"/boot || dialog_die $"Could not mount bootfs - drive '${BOOTHDD}' -> '${INSTALLROOT}/boot'"
		fi
	fi
}

umount_rootfs()
{
	is_mounted --location ${INSTALLROOT}/boot && umount ${INSTALLROOT}/boot
	is_mounted --location ${INSTALLROOT} && umount ${INSTALLROOT}

	if [[ -n ${SWAPHDD} ]]
	then
		swapoff ${SWAPHDD} || die
	fi
}

install_do_reboot()
{
	reboot
}

run_hardware_detection()
{
	local hwinfo
	hwinfo="$(hwinfo --bios --storage --pci --gfxcard --sys)"

	# check for special devices/clients:
	# if zotac a zotac and the disk is a removeable device, then add rootdelay to kernelcmd
	local removable=0
	if [[ ! -z $(echo "${hwinfo}" | grep -i zotac) ]]
	then
		for i in /sys/block/[hs]d*/removable
		do
			if [[ $(< ${i}) = 1 ]]
			then
				removable=1
				# we assume that all removable disks are flash disks on a zotac
				export FLASHDISK=1
			fi
		done
		# smartcard = udevadm info -n /dev/sda -a | grep -i 'configuration.*card'

		# only add this for grub legacy, grub2 detect these settings on its own
		export GRUBLEGACYOPTS="rootdelay=8"
		# there are to zotac types in the wild, nvidia based gfx and intel
		if [[ ! -z $(echo "${hwinfo}" | grep -i nouveau) ]]
		then
			export SPECIALDEVICE="zotac_nvidia"
		else
			export SPECIALDEVICE="zotac_intel"
		fi
	fi

	# check for special devices/clients:
	# if a rangee and disk ist smaller then 256mb move partion one block further ahead
	if [[ ! -z $(echo "${hwinfo}" | grep -i CLE266) ]]
	then
		# for a rangee always define partion startblock +1 
		export FDISKPARTIONBELOW256MB="1"
		export SPECIALDEVICE="rangee"
		export GRUBLEGACYOPTS=""
	fi

	# check for special devices/clients:
	# check for maxdata / i810/ i815 Chipsets and disable KMS and use i810fb frambuffer
	if [[ ! -z $(echo "${hwinfo}" | grep -i i810) ]] || [[ ! -z $(echo "${hwinfo}" | grep -i i815) ]]
	then
		export SPECIALDEVICE="maxdata"
		export GRUBLEGACYOPTS=""
	fi

	# check for i845 Chipsets and enable KMS and use 915 drm driver later in initrd
	if [[ ! -z $(echo "${hwinfo}" | grep -i i845) ]]
	then
		export SPECIALDEVICE="i845"
		# unset default video=1024x768 opt or the drm driver breaks
		export KERNELOPTS="quiet"
		export GRUBLEGACYOPTS=""
		# enable full kms support
		export GRUB2GFXPAYLOAD="keep"
	fi

	# check for radeon gfxcards
	if [[ ! -z $(echo "${hwinfo}" | grep -i radeon) ]]
	then
		# enable full kms support
		export GRUB2GFXPAYLOAD="keep"
	fi

	# requires nomsi to prevent massive IRQ error spam
	# see: http://ubuntuforums.org/showthread.php?t=1234983
	if [[ ! -z $(echo "${hwinfo}" | grep -i 'P5VD2-X') ]] || [[ ! -z $(echo "${hwinfo}" | grep -i 'VT8237') ]] || [[ ! -z $(echo "${hwinfo}" | grep -i 'VX700') ]]
	then
		export SPECIALDEVICE="nomsi"
		export KERNELOPTS="${KERNELOPTS} pci=nomsi,noaer"
	fi

	# check for special devices/clients:
	# check for laptops and activate cpufreq scaling
	if [[ $(echo "${hwinfo}" | grep 'Formfactor:' | sed 's:.*Formfactor\:\ \"\(.*\)\":\1:') = laptop ]]
	then
		export FORMFACTOR="laptop"
		export KERNELOPTS="${KERNELOPTS} cpufreq.governor=ondemand"
	fi
}

run_hardware_detection_disks()
{
	local bootdev

	# all disks but exclude ramdisks
	export ALL_DISKS=$(get_hwinfo disk | sed '/\/dev\/ram[0-9].*/d')
	# remove the boot device from ALL_DISKS if it was an usbstick
	if [[ $(grep '[[:space:]]${LIVEROOT}[[:space:]]' /proc/mounts | cut -d' ' -f3) != iso9660 ]]
	then
		bootdev=$(grep "[[:space:]]${LIVEROOT}[[:space:]]" /proc/mounts | cut -d' ' -f1 | sed 's:[0-9]::g')
		export ALL_DISKS=$(echo "${ALL_DISKS}" | grep -v "${bootdev}")
	fi
	export ALL_CDROMS="$(get_hwinfo cdrom)"
}

setup_hdd_partitions()
{
	# sanity check - should not happen
	if is_mounted --device "${ROOTHDD}"
	then
		echo "partition: device ${ROOTHDD} is already mounted, umount it" >&2
		umount "${ROOTHDD}"
	fi
	if [[ -n ${BOOTHDD} ]]
	then
		if is_mounted --device "${BOOTHDD}"
		then
			echo "partition: device ${BOOTHDD} is already mounted, umount it" >&2
			umount "${BOOTHDD}"
		fi
	fi

	if [[ ${INSTALL_METHOD} = auto ]]
	then
		# run this only if FDISKPARTITIONBELOW256MB is not already 1
		if [[ ${FDISKPARTIONBELOW256MB} != 1 ]]
		then
			if device_minimum_size "${HDD}" 256
			then
				FDISKPARTIONBELOW256MB=1
			else
				FDISKPARTIONBELOW256MB=0
			fi
		fi

		## delete disk
		dd if=/dev/zero of=${HDD} count=1 &> /dev/null || dialog_die

		if [[ ${FLASHDISK} = 1 ]]
		then
			if [[ ${FDISKPARTIONBELOW256MB} = 1 ]]
			then
				## setup one bootable partition
				# 1. n= new disk
				# 2. p= primary disk
				# 3. 1= first partition
				# 4. 2= default sector start // small disk needs more space for grub2 mbr sector
				# 5. ''= defaul sector end
				# 6. a= bootable flag
				# 7. 1= boot flag for partition 1
				# 8. n= new disk
				#10. p= primary disk
				#11. 2= second partition
				#12. ''= default sector start
				#13. ''= defaul sector end
				#14. w= write/quit
				fdisk ${HDD} &> /dev/null << EOF
n
p
1
2
+50M
a
1
n
p
2


w
EOF
			else
				## setup one bootable partition
				# 1. n= new disk
				# 2. p= primary disk
				# 3. 1= first partition
				# 4. ''= default sector start
				# 5. ''= defaul sector end
				# 6. a= bootable flag
				# 7. 1= boot flag for partition 1
				# 8. n= new disk
				#10. p= primary disk
				#11. 2= second partition
				#12. ''= default sector start
				#13. ''= defaul sector end
				#14. w= write/quit
				fdisk ${HDD} &> /dev/null << EOF
n
p
1

+50M
a
1
n
p
2


w
EOF
			fi
		else
			if [[ ${FDISKPARTIONBELOW256MB} = 1 ]]
			then
				## setup one bootable partition
				#1. n= new disk
				#2. p= primary disk
				#3. 1= first partition
				#4. 2= default sector start // small disk needs more space for grub2 mbr sector
				#5. ''= defaul sector end
				#6. a= bootable flag
				#7. 1= boot flag for partition 1
				#8. w= write/quit
				fdisk ${HDD} &> /dev/null << EOF
n
p
1
2

a
1
w
EOF
			else
				## setup one bootable partition
				#1. n= new disk
				#2. p= primary disk
				#3. 1= first partition
				#4. ''= default sector start
				#5. ''= defaul sector end
				#6. a= bootable flag
				#7. 1= boot flag for partition 1
				#8. w= write/quit
				fdisk ${HDD} &> /dev/null << EOF
n
p
1


a
1
w
EOF
			fi
		fi
	else
		cfdisk ${HDD} || dialog_die
	fi
}

setup_hdd_format()
{
	install -d /tmp
	:> /tmp/format.log

	if [[ -n ${SWAPHDD} ]]
	then
		# sanity check - should not happen
		if is_mounted --device "${SWAPHDD}"
		then
			echo "format: device ${SWAPHDD} is already mounted, umount it" >&2
			umount "${SWAPHDD}"
		fi
		mkswap ${SWAPHDD} || die
	fi

	if [[ -n ${BOOTHDD} ]]
	then
		# sanity check - should not happen
		if is_mounted --device "${BOOTHDD}"
		then
			echo "format: device ${BOOTHDD} is already mounted, umount it" >&2
			umount "${BOOTHDD}"
		fi

		mkfs."${FORMAT_FILESYSTEM_BOOTHDD}" "${BOOTHDD}" &>> /tmp/format.log || dialog_die
	fi

	if [[ -n ${ROOTHDD} ]]
	then
		# sanity check - should not happen
		if is_mounted --device "${ROOTHDD}"
		then
			echo "format: device ${ROOTHDD} is already mounted, umount it" >&2
			umount "${ROOTHDD}"
		fi

		mkfs."${FORMAT_FILESYSTEM_ROOTHDD}" "${ROOTHDD}" &>> /tmp/format.log || dialog_die
	fi
}

install_system_image()
{
	pushd ${INSTALLROOT} > /dev/null
	tar xvjpf ${IMAGEROOT}/${CDIMAGENAME} -C ${INSTALLROOT}
	popd > /dev/null
}

install_system_settings()
{
	local CONFIG

	# write fstab
	CONFIG="${INSTALLROOT}/etc/fstab"
	clearconfig
	if [[ -n ${BOOTHDD} ]]
	then
		addconfig -e "UUID=$(get_uuid ${BOOTHDD})\t/\t${FORMAT_FILESYSTEM_BOOTHDD}\tnoatime\t1 1"
	fi
	if [[ -n ${ROOTHDD} ]]
	then
		addconfig -e "UUID=$(get_uuid ${ROOTHDD})\t/\t${FORMAT_FILESYSTEM_ROOTHDD}\tnoatime\t0 0"
	fi
	if [[ -n ${SWAPHDD} ]]
	then
		addconfig -e "UUID='$(get_uuid ${SWAPHDD})'\tswap\tswap\tpri=1\t0 0"
	fi
	addconfig -e "proc\t/proc\tproc\tdefaults\t0 0"
	addconfig -e "shm\t/dev/shm\ttmpfs\tdefaults\t0 0"

	# install network config skeleton
	CONFIG="${INSTALLROOT}/etc/conf.d/net.eth0"
	clearconfig
	addconfig 'ONBOOT="yes"'
	addconfig 'NETWORKING="dhcp"'

	# intel framebuffer quirk
	CONFIG="${INSTALLROOT}/etc/splash/splash.conf"
	if [ -e ${CONFIG} ] && [ -e /proc/fb ]
	then
		if [[ ! -z $(grep 'inteldrmfb' /proc/fb) ]]
		then
			fbdev=$(grep 'inteldrmfb' /proc/fb | sed 's:\([0-9]\).*:\1:')
			[[ ${fbdev} != 0 ]] && sed -i "s:^\(SPLASH_DEV=\).*:\1\"/dev/fb${fbdev}\":" ${CONFIG} || dialog_die
		fi
	fi
}


### installer dialogs ###

dialog_die()
{
	ERROR="$1"
	RETVAL="$?"
	dialog_install_failure
	exit 1
}

dialog_warning()
{
	local retval

	yesnobox $"\Z1 !!! Warning !!! \Zn\n\n\This harddrive will be irreversibly erased.\n\n\Do you want to continue ?"
	retval=$?
	if [[ ${retval} -eq 1 ]]
	then
		clear
		echo $"The process was aborted."
		exit 1
	fi
}


### installer tasks ###

task_setup_system_menu()
{
	local mode
	local retval

	mode="$(dialog_setup_system_menu)"
	retval=$?
	[[ ${retval} -eq 1 ]] && return 1
	if [[ ${retval} -eq 0 ]]
	then
		case "${mode}" in
			"1") run_install auto ;;
			"2") run_install normal ;;
			"") task_setup_system_menu;;
		esac
	fi
}

task_hardware_detection()
{
	local message

	run_hardware_detection_disks

	message+=$"Harddrives:\n"

	if [[ ! -z ${ALL_DISKS} ]]
	then
		for i in ${ALL_DISKS}
		do
			message+="\Z3${i}\Zn "
		done
		message+="\n"
	fi

	if [[ ! -z ${ALL_CDROMS} ]]
	then
		message+="\n"
		message+=$"Optical disk drives:\n"
		for i in ${ALL_CDROMS}
		do
			message+="\Z3${i}\Zn"
		done
		message+="\n"
	fi

	# other devices
	run_hardware_detection
	case "${SPECIALDEVICE}" in
		zotac*) message+=$"\n\Z2Zotac device detected.\Zn" ;;
		rangee) message+=$"\n\Z2Rangee device detected.\Zn" ;;
		maxdata) message+=$"\n\Z2Maxdata device detected.\Zn" ;;
		i845) message+=$"\n\Z2Intel i845 VGA device detected.\Zn" ;;
		nomsi) message+=$"\n\Z2Mainboard with P5VD2-X/VT8237/VX700 chipset detected.\Zn"
			message+=$"\n\Z2Disabling Message Signaled Interrupts (MSI) capability of the kernel.\Zn" ;;
		*) message+=$"\n\ZnCommon device detected.\Zn" ;;
	esac
	if [[ ${FORMFACTOR} = laptop ]]
	then
		message+=$"\n\ZnFormfactor Laptop, activating 'ondemand' powersaving mode.\Zn"
	fi
	if [[ ${FLASHDISK} = 1 ]]
	then
		message+=$"\n\ZnFlash memory detected.\Zn"
		message+=$"\n\ZnF2FS will be used as default filesystem withn the auto installation mode.\Zn"
	fi

	messagebox -y 12 -h $"Detected hardware:" "${message}"
}

task_setup_hdd_partitions()
{
	local i
	local retval

	if [[ -z ${ALL_DISKS} ]]
	then
		dialog_no_harddrive_found
		exit 1
	else
		HDD=$(dialog_select_target_harddrive)
		retval=$?
		[[ ${retval} -eq 1 ]] && return 1
		if [[ ${retval} -eq 0 ]]
		then
			dialog_setup_hdd_info
			setup_hdd_partitions
		fi
	fi
}

task_main()
{
	local method=0
	local retval

	while [[ ${method} -le 2 ]]
	do
		method=$(dialog_main)
		retval=$?
		[[ ${retval} -eq 1 ]] && exit 1
		if [[ ${retval} -eq 0 ]]
		then
			case ${method} in
				"1") task_setup_system_menu ;;
				"2") task_hardware_detection ;;
				"3") install_do_reboot ;;
				"4") /bin/bash --login -i ;;
			esac
		fi
	done
}

run_install()
{
	local method="$1"

	task_hardware_detection

	case "${method}" in
		auto)
			if [[ ${FLASHDISK} = 1 ]]
			then
				export BOOTHDD="${HDD}1"
				export SWAPHDD=""
				export ROOTHDD="${HDD}2"
				export FORMAT_FILESYSTEM_BOOTHDD="f2fs"
				export FORMAT_FILESYSTEM_ROOTHDD="ext2"
			else
				export BOOTHDD=""
				export SWAPHDD=""
				export ROOTHDD="${HDD}1"
				export FORMAT_FILESYSTEM_BOOTHDD=""
				export FORMAT_FILESYSTEM_ROOTHDD="${FORMAT_FILESYSTEM}"
			fi
			export INSTALL_METHOD="${method}"
			;;
		normal)
			export BOOTHDD="${HDD}1"
			export SWAPHDD="${HDD}2"
			export ROOTHDD="${HDD}3"
			export FORMAT_FILESYSTEM_BOOTHDD="${FORMAT_FILESYSTEM}"
			export FORMAT_FILESYSTEM_ROOTHDD="${FORMAT_FILESYSTEM}"
			export INSTALL_METHOD="${method}"
			;;
		single)
			export BOOTHDD=""
			export SWAPHDD=""
			export ROOTHDD="${HDD}1"
			export FORMAT_FILESYSTEM_BOOTHDD=""
			export FORMAT_FILESYSTEM_ROOTHDD="${FORMAT_FILESYSTEM}"
			export INSTALL_METHOD="${method}"
			;;
		flash)
			export BOOTHDD="${HDD}1"
			export SWAPHDD=""
			export ROOTHDD="${HDD}2"
			export FORMAT_FILESYSTEM_BOOTHDD="${FORMAT_FILESYSTEM}"
			export FORMAT_FILESYSTEM_ROOTHDD="f2fs"
			export INSTALL_METHOD="${method}"
			;;
		*)
			die "Unknown install method '${method}', aborting."
			;;
	esac

	task_setup_hdd_partitions
	dialog_setup_hdd_format
	setup_hdd_format > /dev/null
	mount_rootfs
	(install_system_image > /tmp/install.log) 2> /tmp/install_errors.log | install_meter | dialog_install_system_image

	dialog_install_settings
	sleep 1
	install_system_settings
	if is_initrd_supported
	then
		dialog_install_initrd
		initrd_config
		initrd_install
	fi

	dialog_install_bootsector
	bootloader_config
	bootloader_install

	umount_rootfs
	dialog_install_successful
}

# set some proper traps
trap "trap_exit" SIGINT SIGQUIT

task_main

exit 0