#!/bin/bash

function pause {
echo "Pulse enter para continuar..."
read
}

function mostrardialog {
local titled
local msgd
titled=$1
msgd=$2
dialog --title "$titled" --msgbox "$msgd" 15 60
}

function fatalerror {
razon=$1
mostrardialog "Error fatal" "Ha occurrido un error fatal, la instalacion se va a detener \n Razon: $razon"
exit
}

function particiones {
clear
mostrardialog "Preguntas acerca de las unidades donde instalar el sistema" "Esta parte es la mas importante, antes de seguir se mostrarán los discos duros. Si esta vacío o es nuevo, recomendamos formatear en DOS/MBR. Si sabe lo que hace, o ya usaba GPT/UEFI, tendrá que seleccionar la partición EFI, recuerdelo \n (Si lo instala junto a windows 8.1/10 de 64 bits con UEFI activado, entonces usa EFI y tendrás que seleccionar la unidad EFI que utiliza luego)"
clear
if [[ ! $(lsblk --output KNAME | cut -c1-3 | sort | uniq | sed '1d' | grep 'sd\|hd') ]];then
	mostrardialog "ERROR" "ERROR: No tienes discos duros, no se puede continuar. Por favor enchufe/inserte un disco duro y reinicie. \n Al pulsar aceptar se reiniciara el sistema"
	reboot
fi
optpa=`dialog --stdout --nocancel --menu "¿Quieres editar las particiones?¿De modo gráfico, o por la terminal?" 0 0 5 1 "Grafico con Gparted (Recomendado)" 2 "Terminal (Pequeño asistente o por tu cuenta)" 3 "No editar nada"`
case $optpa in
	1) 
		#Abrir gparted grafico
		mostrardialog "Gparted" "Se abrira Gparted. Con el puedes editar, borrar,añadir y poner a tu gusto graficamente las particiones. \n Recuerda que al menos debe haber una particion ext4 primaria para el sistema. \n (Nota: No monte ninguna particion)"
		startx;;
	2)
		#Terminal
		echo "Seleccione el disco duro para editar"
		echo "AVISO, GPT/UEFI ES EXPERMIENTAL, FALTA PROBARLO MAS, PUEDES QUEDARTE SIN ARRANQUE"
		lsblk --output NAME,KNAME,FSTYPE,SIZE,LABEL
		options=($(lsblk --output KNAME | cut -c1-3 | sort | uniq | sed '1d' | grep 'sd\|hd') "No editar" "Terminal")
		select opt in "${options[@]}"
		do
			case $opt in
			"Terminal") 
			echo "Haga los cambios pertinentes, asegurese, y escriba exit continuar la instalación"
			bash;;
			"No edit") break;;
			*) cfdisk /dev/$opt;break;;
			esac
		done
		;;
esac
clear
lsblk --output NAME,KNAME,FSTYPE,SIZE,LABEL
echo "Elige la particion a usar como sistema raíz ('/')"
options=($(lsblk --output KNAME | sed '1d' | grep 'sd\|hd' | grep '[1-9]$'))
select opt in "${options[@]}"
do
    case $opt in
    *) particionraiz=$opt;break;;
    esac
done

echo "Escoje el formato para la particion. EXT4 recomendado"
options=("EXT4" "BTRFS" "No formatear")
select opt in "${options[@]}"
do
    case $opt in
        "EXT4")
            formatoraiz="mkfs.ext4"
            break
            ;;
        "BTRFS")
            formatoraiz="mkfs.btrfs"
            break
            ;;
		"No formatear")
			#Asi no formateará
            formatoraiz="No formatear"
            break
            ;;
        *) echo invalid option;;
    esac
done

clear
echo "¿Quieres/tienes una partición home separada?"
options=("Si" "No")
select opt in "${options[@]}"
do
    case $opt in
        "Si")
		#Selección de home
			lsblk --output NAME,KNAME,FSTYPE,SIZE,LABEL
			echo "Elige la particion a usar como home ('/home')"
			options=($(lsblk --output KNAME | sed '1d'| grep 'sd\|hd'))
			select opt in "${options[@]}"
			do
				case $opt in
				*) particionhome=$opt;break;;
				esac
			done
			
			echo "Escoje el formato para la particion. EXT4 recomendado"

			options=("EXT4" "BTRFS" "No formatear")
			select opt in "${options[@]}"
			do
				case $opt in
					"EXT4")
						formatohome="mkfs.ext4"
						break
						;;
					"BTRFS")
						formatohome="mkfs.btrfs"
						break
						;;
					"No formatear")
						#Asi no formateará
						formatohome="No formatear"
						break
						;;
					*) echo invalid option;;
				esac
			done
            break
            ;;
        "No")
            break
            ;;
        *) echo invalid option;;
    esac
done
clear

opt=`dialog --stdout --nocancel --menu "¿Usaste EFI/UEFI/GPT o es MBR/BIOS?" 0 0 5 1 "MBR/BIOS" 2 "EFI/UEFI/GPT"`
case $opt in
	2)
		clear
		lsblk --output NAME,KNAME,FSTYPE,SIZE,LABEL
		echo "Elige la particion EFI del sistema (Si usa windows, suele ser una particion de 300-400Mb en FAT32 llamada EFI, y suele estar en /dev/sda2)"
		options=($(lsblk --output KNAME | sed '1d'| grep 'sd\|hd'))
		select opt in "${options[@]}"
		do
			case $opt in
			*) partefi=$opt;break;;
			esac
		done
		;;
esac

}

function install {
echo "Instalando el sistema, espere.."
$formatoraiz /dev/$particionraiz
mount /dev/$particionraiz /mnt/
if [[ $particionhome ]];then
$formatohome /dev/$particionhome
mount /dev/$particionhome /mnt/home/
fi
if [[ $partefi ]];then
mkdir /mnt/boot/efi
mount /dev/$partefi /mnt/boot/efi/
fi

wget https://gitlab.com/JKANetwork/JKArch/raw/master/archinstall/pacman.conf -O /tmp/pacman.conf
if [[ $? != 0 ]];then
fatalerror "Error de conexion con internet, compruebe la red"
fi
pacstrap -C /tmp/pacman.conf /mnt/ base 
mkdir -p /mnt/root/inst/
wget https://gitlab.com/JKANetwork/JKArch/raw/master/archinstall/chrootinstall.sh -O /mnt/root/inst/chrootinstall.sh
pause
if [[ $? != 0 ]];then
fatalerror "Error de conexion con internet, compruebe la red"
fi
chmod +x /mnt/root/inst/chrootinstall.sh
pause
genfstab -U /mnt >> /mnt/etc/fstab
cp /etc/netctl/* /mnt/etc/netctl/
mv /tmp/pacman.conf /mnt/etc/pacman.conf
argpart=$(echo $particionraiz | cut -c1-3)
pause
arch-chroot /mnt /bin/bash -c "./root/inst/chrootinstall.sh $argpart $partefi"
}
clear
mostrardialog "Script de instalación simple de Arch (BETA)" "Mediante unas preguntas y avanzando, instalaremos arch en su ordenador \n Ha sido pensado para ser simple, y guiado por opciones, pero debes estar atento durante el proceso."
mostrardialog "JKArch" "El script esta en estado beta y se esta actualizando casi constantemente. Puede haber fallos importantes. En caso de fallo, vuelva a probar en unas horas"
echo "Desmontando posibles particiones"
umount /mnt/*
umount /mnt/
particiones
clear
echo "Ok, vamos a comenzar la instalacion, pero antes, un resumen de donde lo va a instalar."
lsblk --output NAME,KNAME,FSTYPE,SIZE,LABEL
echo "Partición raiz (/): $particionraiz , con formato $formatoraiz"
if [[ $particionhome ]];then
echo "Partición home (/home): $particionhome , con formato $formatohome"
fi
echo "Si estas de acuerdo, presione enter, en caso contrario, apague el ordenador a la fuerza"
pause
install
#clear
pause
mostrardialog "Fin" "Instalación finalizada, al pulsar aceptar, se reiniciará el sistema"
reboot