#!/bin/bash

# https://wizardzines.com/comics/bash-errors/bash-errors.png
set -euo pipefail

exec >> >(tee -i /tmp/install.log)
exec 2>&1

clear

# Check root
if [[ $(id -u) -ne 0 ]]; then
  echo "This script needs to be run as root!"
  exit 1
fi

# Friendly introduction.
echo "0. WELCOME"
echo "Welcome to the Arch Linux installation script!"
echo "This script will ERASE ALL DATA on the partition you will choose next!"
echo
read -p "Do you want to continue? Type [Y]es or [N]o. " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
  exit 1
fi
clear

# Selecting disk.
echo "1. PARTITIONING"
echo "Please select a disk to partition:"
echo

# Don't separate values by space.
IFS=$'\n'

# Set variable containing name and size of all disks.
declare -a dsks=( $(lsblk -d | tail -n+2 | awk '{print $1" "$4}') )

# Select value on array.
select dev in "${dsks[@]}"
do

break
done
echo

# Separate values by spaces.
IFS=' '

# Create new variable of selection.
array=(${dev})
dev="${array[0]}"

# TODO Check disk size
# Disk auswählen (Variable `dev` muss gesetzt sein)
#disk="/dev/${dev}"

# Festplattengröße in Gigabyte berechnen
#disk_size=$(lsblk -b -n -o SIZE "$disk" | awk '{print $1 / 1024 / 1024 / 1024}')

# Prüfen, ob die Festplatte größer als 10 GB ist
#if (( $(echo "$disk_size >= 10" | bc -l) )); then
#  echo "✅ Die Festplatte hat $disk_size GB und ist größer als 10 GB."
#else
#  echo "❌ Fehler: Die Festplatte ist zu klein ($disk_size GB). Bitte eine größere auswählen."
#  exit 1
#fi

clear

echo "2. ENCRYPTION"
echo

# Setting encryption password.
echo "Choose a strong password for encrypting the primary partition:"
password_encrypt=""
while [[ -z "${password_encrypt}" ]]; do
  #echo "Please enter a password: "
  read -rs -p "Please enter a password: " password_first
  read -rs -p "Retype a password: " password_second
    if [[ "${password_first}" == "${password_second}" ]];
    then
      password_encrypt="${password_first}"
      echo
      echo "Both passwords are the same. Continuing.."
      break
    else
      echo
      echo "You have entered different passwords. Try again.."
      echo
    fi
done

clear

echo "3. HOSTNAME"
echo

# Setting hostname
#echo "Before installing the system, please enter a hostname:"
read -r -p "Before installing the system, please enter a hostname: " host
while [[ -z "${host}" ]]; do
  echo "Hostname can not be empty!"
  read -r -p "Please enter a hostname: " host
done

clear

echo "4. ROOT PASSWORD"
echo

# Setting root password
echo "Choose a password for the root account:"
password_root=""
while [[ -z "${password_root}" ]]; do
  #echo "Please enter a password: "
  read -rs -p "Please enter a password: " password_first
  read -rs -p "Retype a password: " password_second

  if [[ "${password_first}" == "${password_second}" ]]; then
    password_root="${password_first}"
    echo
    echo "Both passwords are the same. Continuing.."
    break
  else
    echo
    echo "You have entered different passwords. Try again.."
    echo
  fi
done

clear

echo "4. USER"
echo

# Setting username
#echo "Please enter a username:"
read -r -p "Please enter a username: " user
while [[ -z "${user}" ]]; do
  echo "Benutzername darf nicht leer sein!"
  read -r -p "Please enter a username: " user
done

echo

# Setting user password.
echo "Also please choose a password for your user:"
password_user=""
while [[ -z "${password_user}" ]]; do
  #echo "Please enter a password: "
  read -rs -p "Please enter a password: " password_first
  read -rs -p "Retype a password: " password_second 

  if [[ "${password_first}" == "${password_second}" ]];
  then
    password_user="${password_first}"
    echo
    echo "Both passwords are the same. Continuing.."
    break
  else
    echo
    echo "You have entered different passwords. Try again.."
    echo
  fi
done

clear

echo "5. INSTALLING SYSTEM.."
echo

# Starting partitioning.
echo "Partitioning /dev/${dev}.."
echo

# Clearing partition table of selected disk.
echo "Clearing existing partitioning table."
sgdisk -Z "/dev/${dev}"
echo

# Creating boot partition.
echo "Creating boot partition of 512 MB."
sgdisk -n 1:0:+512M "/dev/${dev}"
echo

# Setting type for EFI.
echo "Setting partition type."
sgdisk -t 1:ef00 "/dev/${dev}"
echo

# Creating system partition.
echo "Creating system partition."
sgdisk -n 2:0:0 "/dev/${dev}"
echo

# Print partitions.
echo "This is your new partition table:"
lsblk | grep "${dev}"
echo

# Get new variable.
if [[ "${dev}" = "nvme0n1" ]]; then
  main="${dev}p2"
else
  main="${dev}2"
fi

# Encrypting partition.
echo "Encrypting system partition. This might take a while."
echo
echo -en "${password_encrypt}\n${password_encrypt}" | cryptsetup -c aes-xts-plain -s 512 luksFormat "/dev/${main}"
echo "Partition successfully encrypted."
echo

# Opening encrypted partition and mounting at /dev/mapper/main.
echo "Decrypting.. This also might take a while."
echo
echo -en "${password_encrypt}\n${password_encrypt}" | cryptsetup luksOpen "/dev/${main}" main
echo "Partition successfully opened."
echo

lsblk | grep "${dev}"
echo

echo "Creating the filesystem."

if [ "${dev}" = "nvme0n1" ]; then
  boot="${dev}p1"
else
  boot="${dev}1"
fi

mkfs.fat -F 32 -n UEFI "/dev/${boot}"
echo "Filesystem for boot successfully created."
echo

# Creating btrfs partition.
mkfs.btrfs "/dev/mapper/main"

mount "/dev/mapper/main" "/mnt"

btrfs subvolume create "/mnt/root"
btrfs subvolume create "/mnt/home"

sync
umount "/mnt"
echo

echo "Mounting.."
mount -o autodefrag,compress=zstd:3,subvol=root "/dev/mapper/main" "/mnt"
mkdir "/mnt/home"
mount -o autodefrag,compress=zstd:3,subvol=home "/dev/mapper/main" "/mnt/home"
mkdir "/mnt/boot"
mount "/dev/${boot}" "/mnt/boot"
echo "Mounting complete."
echo

lsblk -a
echo

# Updating mirrors using reflector.
echo "Updating mirrors.."
reflector --verbose --latest 10 --country Germany --age 24 --protocol https --sort rate --save /etc/pacman.d/mirrorlist
echo "Done."
echo

# Check if keyring is up-to-date
if checkupdates | grep -q archlinux-keyring; then 
  echo "Updating keyring.."
  pacman -Sy archlinux-keyring --noconfirm
  echo "Done."
fi
echo

# Installing base system.
echo "Installing basic packages.."

# Basis-Pakete
base_packages=(
    base
    btrfs-progs
    firefox
    konsole
    linux
    linux-firmware
    linux-zen
    nano
    networkmanager
    sddm
    sddm-kcm
    sudo
    xorg-xwayland
    plasma-desktop
    plasma-nm
)

# Notwendige Plasma-Integrationen
plasma_integrations=(
    plasma-pa
    breeze-gtk
    kde-gtk-config
    kdeplasma-addons
    kscreen
    plasma-browser-integration
    powerdevil
)

# KDE Utilities
kde_utils=(
  ark
  bluedevil
  discover
  dolphin
  filelight
  gwenview
  kate
  kcalc
  kdeconnect
  kinfocenter
  kmag
  plasma-systemmonitor
  okular
  partitionmanager
  spectacle
)

# Additional default software
default_software=(
  libreoffice
  pipewire
  pipewire-pulse
  pipewire-alsa
  wireplumber
)

# Alle Pakete zusammenführen
all_packages=(
    "${base_packages[@]}"
    "${plasma_integrations[@]}"
    "${kde_utils[@]}"
    "${default_software[@]}"
)

# Pakete installieren
pacstrap "/mnt" "${all_packages[@]}"

echo "Base system installed."
echo

# Generating fstab.
echo "Generating fstab.."
genfstab -Up "/mnt" > "/mnt/etc/fstab"
echo "Done."
echo

# Setting hostname.
echo "Setting hostname.."
echo "${host}" > "/mnt/etc/hostname"
echo "Done."
echo

# Setting locale.
echo "Setting and generating locale.."
cat << EOF > "/mnt/etc/locale.gen"
en_US.UTF-8 UTF-8
EOF

#echo "${localegen}" >>/mnt/etc/locale.gen
arch-chroot "/mnt" locale-gen
echo "LANG=en_US.UTF-8" > "/mnt/etc/locale.conf"
echo "KEYMAP=de-latin1" > "/mnt/etc/vconsole.conf"
echo "Done."
echo

# Set X locale and keyboard alyout.
echo "Setting locale and keyboard layout."

#arch-chroot /mnt localectl --no-convert set-keymap de-latin1-nodeadkeys
systemd-firstboot --root=/mnt --keymap=de-latin1-nodeadkeys

cat << EOF > "/mnt/etc/X11/xorg.conf.d/00-keyboard.conf"
# Written by systemd-localed(8), read by systemd-localed and Xorg. It's
# probably wise not to edit this file manually. Use localectl(1) to
# instruct systemd-localed to update it.
Section "InputClass"
        Identifier "system-keyboard"
        MatchIsKeyboard "on"
        Option "XkbLayout" "de"
        Option "XkbModel" "pc105"
        Option "XkbVariant" "deadgraveacute"
EndSection
EOF
echo "Done."
echo

# Editing mkinitcpio.conf.
echo "Editing /etc/mkinitcpio.conf.."
sed -i 's/^HOOKS=.*/HOOKS=(base udev autodetect microcode modconf kms block keyboard keymap encrypt filesystems fsck shutdown)/' /mnt/etc/mkinitcpio.conf
#sed -i '55s/.*/HOOKS=(base udev autodetect microcode modconf kms block keyboard keymap encrypt filesystems fsck shutdown)/' /mnt/etc/mkinitcpio.conf
echo "Done."
echo

# Editing taskbar.
#sed -i '65s/.*/launchers=applications:systemsettings.desktop,applications:org.kde.konsole.desktop,applications:firefox.desktop/' /mnt/home/${user}/.config/plasma-org.kde.plasma.desktop-appletsrc

# Creating inital ramdisk.
echo "Creating inital ramdisk for linux-zen.."
arch-chroot /mnt mkinitcpio -p linux-zen
echo "Done."
echo

# Enable network and display manager.
echo "Enabling Networkmanager and display manager (sddm).."
arch-chroot "/mnt" systemctl enable NetworkManager sddm
echo "Done."
echo

# Set root password
echo "Setting root password.."
echo -en "${password_root}\n${password_root}" | arch-chroot "/mnt" passwd
echo "Done."
echo

# Install bootloader.
echo "Installing systemd-boot.."
arch-chroot "/mnt" bootctl --path=/boot install
echo "Done."
echo

# Get UUID for primary partition
uuid=$(ls -l /dev/disk/by-uuid | grep "${main}" | awk '{print $9}')

# Set bootloader entry.
echo "Setting up bootloader.."
cat << EOF > "/mnt/boot/loader/entries/arch.conf"
title   Arch Linux
linux   /vmlinuz-linux-zen
#initrd  /amd-ucode.img 
#initrd  /intel-ucode.img
initrd  /initramfs-linux-zen.img
options cryptdevice=UUID=${uuid}:main
options root=/dev/mapper/main rw
options rootflags=subvol=root
options lang=de locale=de_DE.UTF-8
options init=/usr/lib/systemd/systemd
EOF

# Set bootloader fallback entry.
cat << EOF > "/mnt/boot/loader/entries/arch-fallback.conf"
title   Arch Linux Fallback
linux   /vmlinuz-linux
initrd  /initramfs-linux-fallback.img
options cryptdevice=UUID=${uuid}:main
options root=/dev/mapper/main rw
options rootflags=subvol=root
options lang=de locale=de_DE.UTF-8
options init=/usr/lib/systemd/systemd
EOF

# Set bootloader.
echo "timeout 1" >> "/mnt/boot/loader/loader.conf"
echo "default arch.conf" >> "/mnt/boot/loader/loader.conf"
echo "Done."
echo

# Set username, password and group.
echo "Adding user.."
arch-chroot /mnt useradd -m "${user}"
echo -en "${password_user}\n${password_user}" | arch-chroot /mnt passwd "${user}"
arch-chroot /mnt gpasswd -a "${user}" wheel
sed -i '82s/.*/%wheel ALL=(ALL) ALL/' /mnt/etc/sudoers
echo "Done."
echo

# Enabling sddm auto-login.
echo "Enabling auto-login.."
mkdir /mnt/etc/sddm.conf.d
cat << EOF > /mnt/etc/sddm.conf.d/autologin.conf
[Autologin]
User=${user}
Session=plasma
EOF
echo "Done."
echo

# Adding sddm config.
cat << EOF > /mnt/etc/sddm.conf.d/kde_settings.conf
[Autologin]
Relogin=false
Session=plasma
User=${user}

[General]
HaltCommand=/usr/bin/systemctl poweroff
RebootCommand=/usr/bin/systemctl reboot

[Theme]
Current=breeze

[Users]
MaximumUid=60513
MinimumUid=1000
EOF

# Unmounting and rebooting.
echo "Unmounting system.."
sync
umount "/mnt/boot"
umount "/mnt/home"
umount "/mnt"
echo "Done."
echo

clear

echo "Installation finished!"
#read -p "Do you want to read the log? Type [Y]es or [N]o. " -n 1 -r
#echo
#if [[ ! $REPLY =~ ^[Yy]$ ]]
#then
#    less /tmp/install.log
#fi
#echo

#echo "Press [Enter] to shut down, then remove the installation media and start your system."
#poweroff