#!/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

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

# some sane defaults
LIVEROOT="${DEFAULT_LIVEROOT}"
IMAGEROOT="${DEFAULT_IMAGEROOT}"
INSTALLROOT="${DEFAULT_INSTALLROOT}"
KERNELPKG="${DEFAULT_KERNELPKG}"
KERNELOPTS="${DEFAULT_KERNELOPTS}"
GRUBLEGACYOPTS=""
GRUB2OPTS=""
FDISKPARTIONBELOW256MB=0
SPECIALDEVICE=""
FORMFACTOR="${DEFAULT_FORMFACTOR}"
FORMAT_FILESYSTEM="${DEFAULT_FILESYSTEM}"

# 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}

	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 is_mounted --location "${INSTALLROOT}"
	then
		echo $"${INSTALLROOT} already mounted" >&2
	else
		mount "${ROOTHDD}" "${INSTALLROOT}" || dialog_die $"Could not mount rootfs - drive '${ROOTHDD}' -> '${INSTALLROOT}'"
	fi
	[[ -d ${INSTALLROOT}/boot ]] || install -d ${INSTALLROOT}/boot
}

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

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
			#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 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_auto()
{
	ROOTHDD="${HDD}1"

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

	# 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 [[ ${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
}

setup_hdd_partitions_manual()
{
	ROOTHDD="${HDD}1"
	SWAPHDD=""
	BOOTHDD=""
	cfdisk ${HDD} || dialog_die
}

setup_hdd_format()
{
	# 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}" -q "${ROOTHDD}" || dialog_die
}

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
	addconfig -e "UUID=$(get_uuid ${ROOTHDD})\t/\t${FORMAT_FILESYSTEM}\tnoatime,noauto\t1 1"
	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" ;;
		*) message+=$"\n\ZnCommon device detected.\Zn" ;;
	esac
	if [[ ${FORMFACTOR} = laptop ]]
	then
		message+=$"\n\ZnFormfactor Laptop, activating 'ondemand' powersaving mode.\Zn"
	fi

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

task_setup_hdd_partitions_manual()
{
	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_manual
		fi
	fi
}

task_setup_hdd_partitions_auto()
{
	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_auto
			dialog_setup_hdd_create_partitions
			setup_hdd_partitions_auto
		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_normal()
{
	task_hardware_detection

	task_setup_hdd_partitions_manual
	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
}

run_install_auto()
{
	task_hardware_detection

	task_setup_hdd_partitions_auto
	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