HOSTNAME="ELFLaptop"
UEFI_DISK="/dev/sdb"
DISK="/dev/sda"
USERNAME="freenen"

UEFI_PARTITION="${UEFI_DISK}1"
SWAP_PARTITION="${DISK}3"
ROOT_PARTITION="${DISK}4"

function run(){
    echo "[    ] $1"
    echo "# $1" >>install.log
    echo "> $2" >>install.log
    bash -c "$2" &>>install.log \
        && echo -e "\e[1A\e[K[ \e[32mOK\e[0m ] $1" \
        || { 
            echo -e "\e[1A\e[K[\e[31mFAIL\e[0m] $1"
            bash -c "$3"
            exit 10
        }
    echo >>install.log
}

function run-chroot(){
    echo "[    ] $1"
    echo "# $1" >>install.log
    echo "> chroot /mnt $2" >>install.log
    arch-chroot /mnt bash -c "$2" &>>install.log \
        && echo -e "\e[1A\e[K[ \e[32mOK\e[0m ] $1" \
        || { 
            echo -e "\e[1A\e[K[\e[31mFAIL\e[0m] $1"
            bash -c "$3"
            exit 20
        }
    echo >>install.log
}

echo "" >install.log

echo -n "disk encryption password: "
read -s PASS
echo
echo -n "retype password: "
read -s PASSRE
echo

if [ "$PASS" != "$PASSRE" ]; then
  echo "password do not match"
  exit 1
fi

echo -n "password for freenen: "
read -s PASS
echo
echo -n "retype password: "
read -s PASSRE
echo

if [ "$PASS" != "$PASSRE" ]; then
  echo "password do not match"
  exit 2
fi
echo

echo
echo === setup localisation
echo

run "enable ntp"                  "timedatectl set-ntp true"
run "set timezone"                "timedatectl set-timezone Europe/Amsterdam"

echo
echo === setup partitions
echo

#run "applly partion table"        "sfdisk $DISK <./elfLaptop.sfdisk"	
run "format boot partition"       "mkfs.fat -F 32 ${UEFI_PARTITION}"
run "format swap partition"       "mkswap ${SWAP_PARTITION}"
echo -n "$PASS" >keyfile.luks
run "encrypt root partition"      "cryptsetup luksFormat --batch-mode --key-file keyfile.luks ${ROOT_PARITION}"      "rm keyfile.luks"
run "map root partitaion"         "cryptsetup open --batch-mode --key-file keyfile.luks ${ROOT_PARTITION} cryptroot" "rm keyfile.luks"
rm keyfile.luks
run "format root partition"       "mkfs.btrfs /dev/mapper/cryptroot"

run "mount root partition"        "mount /dev/mapper/cryptroot /mnt"
run "create root btrfs subvolume" "btrfs subvolume create /mnt/@"
run "create home btrfs subvolume" "btrfs subvolume create /mnt/@home"
run "unmount btrfs"               "umount /mnt"
run "mount root subvolume"        "mount -o subvol=@ /dev/mapper/cryptroot /mnt"
run "mount boot partition"        "mount --mkdir ${UEFI_PARITION} /mnt/boot"
run "mount home subvolume"        "mount --mkdir -o subvol=@home /dev/mapper/cryptroot /mnt/home"
run "enable swap"                 "swapon ${SWAP_PARTITION}"

echo
echo === install arch
echo

run "install base of arch"        "pacstrap /mnt base linux linux-firmware"
run "intall utitlities"           "pacstrap /mnt btrfs-progs man vim sudo"
run "generate fstab"              "genfstab -U /mnt >>/mnt/etc/fstab"
run "copy in-root script"         "cp in-root.sh /mnt/root"

echo
echo === set locals
echo

run-chroot "set timezone"                "ln -sf /usr/share/zoneinfo/Europe/Amsterdam /etc/localtime"
run-chroot "set hwclock to UTC"          "hwclock --systohc"
run-chroot "generate locals"             "locale-gen"
run-chroot "config locals"               "echo \"LANG=en_GB.UTF-8\" >/etc/locale.conf"
run-chroot "set hostname"                "echo \"$HOSTNAME\" >/etc/hostname"
run-chroot "create hosts file"           "cat >/etc/hosts <<EOF
127.0.0.1     localhost
::1           localhost
127.0.1.1     $HOSTNAME
EOF
"

echo
echo === install bootloader
echo

run-chroot "config initramfs"            "sed --in-place -e 's/HOOKS=(.*)/HOOKS=(base udev autodetect microcode modconf kms keyboard block encrypt filesystems fsck)/' /etc/mkinitcpio.conf"
run-chroot "generate initramfs"          "mkinitcpio -P"

DISKUUID=$(blkid --output export ${DISK}3 | grep '^UUID' | sed 's/UUID=//')
run-chroot "install systemd-boot"        "bootctl install"
run-chroot "create boot config for main" "cat >/boot/loader/entries/arch.conf <<EOF
title      Elf OS
linux      /vmlinuz-linux
initrd     /initramfs-linux.img
options    cryptdevice=UUID=$DISKUUID:cryptroot:allow-discards root=/dev/mapper/cryptroot rootflags=subvol=@ rd.luks.options=discard rw
EOF
"
run-chroot "create boot config for fallback" "cat >/boot/loader/entries/arch-fallback.conf <<EOF
title      Elf OS fallback
linux      /vmlinuz-linux
initrd     /initramfs-linux-fallback.img
options    cryptdevice=UUID=$DISKUUID:cryptroot:allow-discards root=/dev/mapper/cryptroot rootflags=subvol=@ rd.luks.options=discard rw
EOF
"
run-chroot "create main boot config"     "cat >/boot/loader/loader.conf <<EOF
default      arch.conf
timeout      4
console-mode max
editor       no
EOF
"

echo
echo === setup user
echo

run-chroot "add .ssh dir to skel"        "mkdir -p /etc/skel/.ssh"
run-chroot "create admin group"          "groupadd admin"
run-chroot "update polkit"               "sed --in-place -e 's/wheel/admin/g' /usr/share/polkit-1/rules.d/50-default.rules"
run-chroot "create user"                 "useradd --home-dir /home/${USERNAME} --create-home --skel /etc/skel ${USERNAME} --groups ${USERNAME},admin"
run-chroot "set password for user"       "echo \"${USERNAME}:$PASS\" | chpasswd"
run-chroot "add user to sudoers"         "echo \"%admin ALL=(ALL:ALL) ALL\" >>/etc/sudoers"
run-chroot "touch authoized keys"        "touch /home/${USERNAME}/.ssh/authorized_keys"
run-chroot "add sshkeys for new user"    "curl -o /home/${USERNAME}/.ssh/authorized_keys https://github.com/FReenen.keys"

echo
echo === umount and reboot
echo

run "umount drives"               "umount -R /mnt"
run "close cryptroot"             "cryptsetup close cryptroot"
run "reboot"                      "shutdown -r now"