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

# ignore the environment
LC_ALL=C
# 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

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

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

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

#################################################
#		  DIALOG BOXEN			#
#################################################

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
}

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
}

dialog_setup_hdd_info()
{
	local SHDD="${HDD//\/dev\/}"

	messagebox -h $"[ Harddrive partitioning ]" \
		$"\nPlease create 1 partition.\n\n\[ \Z3${SHDD}1\Zn ] type: \Z3linux\Zn with the whole diskspace\n\Please mark ${SHDD}1 \Z3bootable\Zn."
}

dialog_setup_hdd_info_auto()
{
	local SHDD="${HDD//\/dev\/}"

	messagebox -h $"[ Harddrive partitioning ]" \
		$"\Z1Warning!\Zn\n\nAll data on disk [ \Z3${HDD}\Zn ] will be erased!"
}

dialog_setup_system_menu()
{
	local mode
	local retval

	mode=$(menubox -h $"[ Harddrive partitioning ]" $"\nSelect an installation mode" \
		$"1:Automatic setup (recommended)" \
		":" \
		$":\Z1Expert modes:\Zn" \
		$"2:Common IDE-disk (manual setup)")
	retval=$?
	[[ ${retval} -eq 1 ]] && return 1
	if [[ ${retval} -eq 0 ]]
	then
		case "${mode}" in
			"1") run_install_auto ;;
			"2") run_install_normal ;;
			"") dialog_setup_system_menu;;
		esac
	fi
}

dialog_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}"
}

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

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

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
}

dialog_main()
{
	local method=0
	local retval

	while [[ ${method} -le 2 ]]
	do
		method=$(CANCEL_LABEL=$"Exit "menubox $"Configuration:" \
			$"1:Install system" \
			$"2:Show detected harddrives" \
			$"3:Exit and reboot" \
			$"4:Exit and drop into a shell")
		retval=$?
		[[ ${retval} -eq 1 ]] && exit 1
		if [[ ${retval} -eq 0 ]]
		then
			case ${method} in
				"1") dialog_setup_system_menu ;;
				"2") dialog_hardware_detection ;;
				"3") install_do_reboot ;;
				"4") /bin/bash --login -i ;;
			esac
		fi
	done
}

#################################################
#		Install Komandos		#
#################################################
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:]]/mnt/cdrom[[:space:]]' /proc/mounts | cut -d' ' -f3) != iso9660 ]]
	then
		bootdev="$(grep '[[:space:]]/mnt/cdrom[[: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)"
}

hdd_size_below_256mb()
{
	local hdd="$1"
	local size
	local retval
	[[ -z ${hdd} ]] && dialog_die "Error: get_hdd_size() no \$hdd given!"

	size=$(fdisk -l ${hdd} | grep "Disk.*${hdd}" | sed 's:.*,\ \(.*\)\ byte.*:\1:')
	if [[ ${size} -le 257000000 ]]
	then
		retval="0"
	else
		retval="1"
	fi

	return "${retval}"
}

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 hdd_size_below_256mb ${HDD}
		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_mount_rootfs()
{
	is_mounted --location "${INSTALLROOT}" || mount "${ROOTHDD}" "${INSTALLROOT}"
	install -d ${INSTALLROOT}/boot || dialog_die
}

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

disabled_install_bootsector_chroot()
{
	local my_roothdd
	local grubconf="${INSTALLROOT}/boot/grub/grub.conf"
	local grub2conf="/boot/grub/grub.cfg"

	# check for grub2
	if [[ -f ${INSTALLROOT}/sbin/grub-mkconfig ]]
	then
		# needed by grub-mkconfig on the first run
		if [[ ! -f ${INSTALLROOT}/boot/grub/video.lst ]]
		then
			install -m0644 ${INSTALLROOT}/lib/grub/*/video.lst ${INSTALLROOT}/boot/grub/video.lst || dialog_die
		fi

		# set kernelopts
		if [[ -f ${INSTALLROOT}/etc/conf.d/grub ]]
		then
			sed -i "s:^\(export GRUB_CMDLINE_LINUX_DEFAULT=\).*:\1\"${KERNELOPTS}\":" ${INSTALLROOT}/etc/conf.d/grub || dialog_die
		else
			CONFIG="${INSTALLROOT}/etc/conf.d/grub"
			clearconfig
			addconfig "export GRUB_CMDLINE_LINUX_DEFAULT=\"${KERNELOPTS}\""
		fi
		CONFIG="${INSTALLROOT}/root/.installrc"
		clearconfig
		# only grub 1.99
		addconfig 'type -P grub-mkdevicemap && grub-mkdevicemap'
		addconfig "grub-install --no-floppy ${HDD} &> /dev/null"
		addconfig "LC_ALL=C grub-mkconfig -o ${grub2conf}  &> /dev/null"
		addconfig "exit 0"

	# grub-legacy
	else
		source ${INSTALLROOT}/boot/kernelversion

		if [ -e ${INSTALLROOT}/etc/alx_version ]
		then
			OLD_ALXVER="${ALXVER}"
			source ${INSTALLROOT}/etc/alx_version
			KRNVER="ALX-${ALXVER}"
			ALXVER="${OLD_ALXVER}"
		fi

		[[ -z ${KRNVER} ]] && KRNVER="AutoSta_LX"
		[[ -z ${KRNINITRD} ]] && KRNINITRD="initrd"
		[[ -z ${KRNIMG} ]] && KRNIMG="vmlinuz"

		# uuid support
		my_roothdd="UUID=$(get_uuid ${ROOTHDD})"

		CONFIG="${grubconf}"
		clearconfig
		addconfig "default 0"
		addconfig "timeout 3"
		# using current root password
		addconfig "password --md5 $(cat ${INSTALLROOT}/etc/shadow | grep root | cut -d: -f2)"

		addconfig
		addconfig "# normal boot"
		addconfig "title ${KRNVER}"
		addconfig "root (hd0,0)"
		addconfig "kernel /boot/${KRNIMG} root=${my_roothdd} ${KERNELOPTS} ${GRUBLEGACYOPTS}"
		if is_initrd_supported
		then
			addconfig "initrd /boot/${KRNINITRD}"
		fi

		addconfig
		addconfig "# admin boot"
		addconfig "title ${KRNVER} - Re-run hardware-detection"
		addconfig "lock"
		addconfig "root (hd0,0)"
		addconfig "kernel /boot/${KRNIMG} root=${my_roothdd} ${KERNELOPTS} ${GRUBLEGACYOPTS} hardware-auto-detection"
		if is_initrd_supported
		then
			addconfig "initrd /boot/${KRNINITRD}"
		fi

		addconfig
		addconfig "title ${KRNVER} - Reset *all* local settings"
		addconfig "lock"
		addconfig "root (hd0,0)"
		addconfig "kernel /boot/${KRNIMG} root=${my_roothdd} ${KERNELOPTS} ${GRUBLEGACYOPTS} alx-reset-settings"
		if is_initrd_supported
		then
			addconfig "initrd /boot/${KRNINITRD}"
		fi

		# bootsector schreiben chrooted schreiben (lfs/magellan)
		cat > ${INSTALLROOT}/root/.installrc << CHROOTEOF
/usr/sbin/grub --no-floppy --batch << "EOF" 1> /dev/null 2> /dev/null
root (hd0,0)
setup (hd0)
quit
EOF
exit 0
CHROOTEOF
	fi

	# run installrc
	chrooted /bin/bash --rcfile /root/.installrc -i
	rm ${INSTALLROOT}/root/.installrc
}

disabled_install_initrd_chroot()
{
	# only generate initrds if the cmd exists
	is_initrd_supported || return 0

	DISKMODS="sd_mod"
	OLDPATAMODS="amd74xx piix sis5513 via82cxxx"
	PATAMODS="ata_piix pata_amd pata_mpiix pata_oldpiix pata_sis pata_via"
	SATAMODS="sata_via sata_sis sata_nv"
	DRMMODS="i915 mga nouveau r128 radeon savage sis tdfx ttm via"
	OTHERMODS=""
	case ${SPECIALDEVICE} in
		zotac_intel) FBMODS=""; DRMMODS="i915" ;;
		zotac_nvidia) FBMODS=""; DRMMODS="nouveau" ;;
		rangee) FBMODS="" ;; ## fallback to vesafb, viafb does not run on all CLE266 boards
		# not working with kms enabled drivers -> segfaults
		#maxdata) FBMODS="i810fb" ;; ## check for maxdata / i810/ i815 Chipsets and disable KMS and use i810fb frambuffer
		maxdata) FBMODS="" ;;
		*) FBMODS="uvesafb" ;;
	esac

	if [[ ${FORMFACTOR} = laptop ]]
	then
		OTHERMODS="acpi-cpufreq cpufreq_ondemand cpufreq_conservative cpufreq_powersave"
	fi

	# install an appropriate uvesafb.conf
	install -d ${INSTALLROOT}/etc/modprobe.d || dialog_die
	CONFIG="${INSTALLROOT}/etc/modprobe.d/uvesafb.conf"
	clearconfig
	addconfig "options uvesafb mode_option=1024x768-32@60 scroll=ywrap"

	# install an appropriate viafb.conf
	CONFIG="${INSTALLROOT}/etc/modprobe.d/viafb.conf"
	clearconfig
	addconfig "options viafb viafb_mode=1024x768 viafb_refresh=60"

	# install an appropriate i810fb.conf
	CONFIG="${INSTALLROOT}/etc/modprobe.d/i810fb.conf"
	clearconfig
	addconfig "options i810fb xres=1024 yres=768 bpp=16 mtrr=1 hsync1=30 hsync2=62 vsync1=30 vsync2=60"

	cat > ${INSTALLROOT}/root/.installrc << CHROOTEOF
echo "MODULES=\"${FORMAT_FILESYSTEM} ${DISKMODS} ${OLDATAMODS} ${PATAMODS} ${SATAMODS} ${DRMMODS} ${FBMODS} ${OTHERMODS}\"" > /etc/conf.d/mkinitrd
mkinitrd -f /boot/$(readlink /boot/initrd) $(readlink /boot/vmlinuz | sed "s:kernel-::g") > /dev/null
exit 0
CHROOTEOF

	# run installrc
	chrooted /bin/bash --rcfile /root/.installrc -i
	rm ${INSTALLROOT}/root/.installrc
}

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
	if [ -e ${INSTALLROOT}/etc/splash/splash.conf ]
	then
		if [ -e /proc/fb ]
		then
			if [[ ! -z $(grep 'inteldrmfb' /proc/fb) ]]
			then
				fbdev=$(grep 'inteldrmfb' /proc/fb | sed 's:\([0-9]\).*:\1:')
				if [[ ${fbdev} != 0 ]]
				then
					sed -i "s:^\(SPLASH_DEV=\).*:\1\"/dev/fb${fbdev}\":" ${INSTALLROOT}/etc/splash/splash.conf || dialog_die
				fi
			fi
		fi
	fi
}

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

install_do_reboot()
{
	reboot
}

#################################################
#	     Install Ablauf Scripte		#
#################################################

run_install_normal()
{
	dialog_hardware_detection

	dialog_setup_hdd_partitions_manual
	dialog_setup_hdd_format
	setup_hdd_format > /dev/null
	install_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

	install_umount_rootfs
	dialog_install_successful
}

run_install_auto()
{
	dialog_hardware_detection

	dialog_setup_hdd_partitions_auto
	dialog_setup_hdd_format
	setup_hdd_format > /dev/null
	install_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

	install_umount_rootfs
	dialog_install_successful
}

# set some proper traps
trap "trap_exit" SIGINT SIGQUIT

dialog_main

exit 0