#!/bin/bash
depend_procedure core base # esp for auto_network, intro and set_clock workers


# This is a port of the original /arch/setup script.  It doesn't use aif phases but uses it's own menu-based flow (phase) control

BLOCK_ROLLBACK_USELESS=1

# clock
HARDWARECLOCK=
TIMEZONE=

# default filesystem specs (the + is bootable flag)
# <mountpoint>:<partsize>:<fstype>[:+]
DEFAULTFS="/boot:32:ext2:+ swap:256:swap /:7500:ext3 /home:*:ext3"

# this is a workaround for bash <4.2, where associative arrays are inherently local
# once we have 4.2 we can just declare this array global. eval $settitles where needed
settitles="declare -A workertitles
workertitles['select_source']='Select source'
workertitles['set_editor']='Set editor'
workertitles['runtime_network']='Setup network'
workertitles['set_clock']='Set clock'
workertitles['prepare_disks']='Prepare hard drive(s)'
workertitles['package_list']='Select packages'
workertitles['install_packages']='Install packages'
workertitles['configure_system']='Configure system'
workertitles['install_bootloader']='Install bootloader'
workertitles['msg_report']='Exit Install'"

menu_workers=(select_source set_editor set_clock prepare_disks package_list install_packages configure_system install_bootloader msg_report)

start_process ()
{
	#####################
	## begin execution ##

	execute worker configure
	execute worker intro
	execute worker sysprep

	default=1
	while true
	do
		mainmenu
	done
}


mainmenu()  
{
	eval "$settitles"
	menu=()
	for i in ${!menu_workers[@]}
	do
		worker=${menu_workers[$i]}
		title=${workertitles[$worker]}
		let i+=1 # in menu, we count from 1 onwards, not from 0
		menu+=($i "$title")
	done

	ask_option $default "MAIN MENU" '' required "${menu[@]}"
	case $ANSWER_OPTION in
	"1")                                         execute worker select_source       && \
	                                             execute worker runtime_packages    && default=2 ;;
	"2")                                         execute worker set_editor          && default=3 ;;
	"3")                                         execute worker set_clock           && default=4 ;;
	"4")                                         execute worker prepare_disks       && default=5 ;;
	"5") check_depend worker prepare_disks    && \
	     check_depend worker select_source    && execute worker package_list        && default=6 ;;
	"6") check_depend worker package_list     && \
	     check_depend worker select_source    && execute worker install_packages    && default=7 ;;
	"7") check_depend worker install_packages && execute worker configure_system    && default=8 ;;
	"8") check_depend worker configure_system && execute worker install_bootloader  && default=9 ;;
	"9") execute worker msg_report ;;
	*)   execute worker abort_installer;;
	esac
}


worker_configure_system()
{
	interactive_configure_system
}


worker_prepare_disks()
{
	get_possible_fs && interactive_prepare_disks
}


worker_select_source ()
{
	interactive_select_source
	local ret=$?;
	if [ $ret -eq 0 -a "$var_PKG_SOURCE_TYPE" = net ]
	then
		select_source_extras_menu || ret=$?
	fi
	# base/automatic procedure run a worker runtime_repositories, maybe we could do an interactive repo adder
	return $ret
}

select_source_extras_menu ()
{
	local default=no
	while true; do
		ask_option $default "NET (HTTP/FTP) Installation" "Make sure the network is ok and you've selected a mirror before continuing the installer" required \
		"1" "${workertitles['runtime_network']}" \
		"2" "Select mirror" \
		"3" "Return to Main Menu" || return 1
		case $ANSWER_OPTION in
			"1") execute worker runtime_network && default=2 || return 1 ;;
			"2") interactive_select_mirror && default=3 || return 1 ;;
			"3") break ;;
		esac
	done
	return 0
}

worker_set_editor ()
{
	seteditor force
}

worker_intro ()
{
	notify "Welcome to the Parabola GNU/Linux-Libre Installation program.\n
You're about to install a Free as in Freedom Arch system!\
\n
\n
The install process is fairly straightforward, and you should run \
through the options in the order they are presented. If you are \
unfamiliar with partitioning/making filesystems, you may want to \
consult some documentation before continuing. You can view all output \
from commands by viewing your tty7 console (ALT-F7). ALT-F1 will \
bring you back here.\n\n$DISCLAIMER"

}


worker_configure ()
{
	var_UI_TYPE=${arg_ui_type:-dia}
	ui_init
}


# select_packages()
# prompts the user to select packages to install
worker_package_list() {
	# if selection has been done before, warn about loss of input and let the user exit gracefully
	ended_ok worker package_list && ! ask_yesno "WARNING: Running this stage again will result in the loss of previous package selections.\n\nDo you wish to continue?" && return 0

	interactive_select_packages
}


worker_install_packages ()
{
	installpkg
}


# Hand-hold through setting up networking
worker_runtime_network() {
	interactive_runtime_network
}

worker_install_bootloader ()
{
	interactive_install_bootloader
}

# override show_report because this procedure does not use phases, but it's own menu
show_report ()
{
	eval "$settitles"
	data="Execution Report:"
	data="$data\n-----------------"
	for worker in ${menu_workers[@]}
	do
		[ "$worker" = 'msg_report' ] && continue # this worker is the one calling show_report, we can't know our exitcode yet
		title=${workertitles[$worker]}
		exit_var=exit_worker_$worker
		ret=${!exit_var}
		case "$ret" in
			"")  data="$data\n > $title: Skipped";;
			"0") data="$data\n > $title: Sucess";;
			*)   data="$data\n > $title: Failed";;
		esac
	done
	notify "$data\nIf the install finished successfully, you can type 'reboot' to restart the system." || return $?
}

worker_msg_report () {
	show_report && stop_installer
}

worker_abort_installer ()
{
	ret=0
	ask_yesno "Abort Installation?" || return
	maybe_interactive_rollback_filesystems || ret=$?
	stop_installer || ret=$?
	return $ret
}