#!/bin/zsh
echo "SteamOS 3 Installer (x86_64)"
echo "Start time: $(date)"
CMD_PACMAN_INSTALL=(/usr/bin/pacman --noconfirm -S --needed)
CMD_PACMAN_UPDATE=(/usr/bin/pacman -Sy)
SYSTEM_LOCALE="${SYSTEM_LOCALE:-en_US.UTF-8 UTF-8}"
HOLO_INSTALL_DIR="${HOLO_INSTALL_DIR:-/mnt}"

# Installation begins here
echo "Starting installation..."
${CMD_PACMAN_UPDATE}
echo "db update complete!"
lsblk
read "?Enter your desired drive node here: " DRIVEDEVICE
read "?WARNING: This drive is going to be erased fully. Press enter to continue, or CTRL+Z to terminate"

DEVICE="/dev/${DRIVEDEVICE}"

INSTALLDEVICE="${DEVICE}"
echo ${DEVICE} | grep -q -P "^/dev/(nvme|loop)"
if [ $? -eq 0 ]; then
    INSTALLDEVICE="${DEVICE}p"
fi

echo "Creating partitions..."
parted ${DEVICE} mklabel gpt
parted ${DEVICE} mkpart primary fat32 2M 256M
parted ${DEVICE} set 1 boot on
parted ${DEVICE} set 1 esp on
parted ${DEVICE} mkpart primary btrfs 256M 100%
root_partition="${INSTALLDEVICE}2"
mkfs -t vfat ${INSTALLDEVICE}1
fatlabel ${INSTALLDEVICE}1 HOLOEFI
mkfs -t btrfs ${root_partition}
btrfs filesystem label ${root_partition} holo-root

echo "Partition creating complete, mounting and pacstrapping..."
mount -t btrfs -o subvol=/,compress-force=zstd:1,discard,noatime,nodiratime ${root_partition} ${HOLO_INSTALL_DIR}
pacstrap -i ${HOLO_INSTALL_DIR} base base-devel amd-ucode intel-ucode linux-neptune linux-firmware
echo "Base system installation done, generating fstab..."
genfstab -U -p /mnt >> /mnt/etc/fstab
cp /etc/pacman.conf /mnt/etc/pacman.conf
cp /etc/pacman.d/mirrorlist /mnt/etc/pacman.d/mirrorlist
cp /etc/holoinstall/steamos-session-select /mnt/usr/bin/steamos-session-select
cp /etc/holoinstall/steamos-update /mnt/usr/bin/steamos-update
arch-chroot ${HOLO_INSTALL_DIR} chmod +x /usr/bin/steamos-session-select
arch-chroot ${HOLO_INSTALL_DIR} chmod +x /usr/bin/steamos-update
read "?Enter your hostname for this installation: " HOLOHOSTNAME
echo ${HOLOHOSTNAME} > ${HOLO_INSTALL_DIR}/etc/hostname
read "?Enter your username for this installation: " HOLOUSER
read "?Enter your user password for this installation: " HOLOPASS
echo "Creating user ${HOLOUSER}..."
echo -e "root\nroot" | arch-chroot ${HOLO_INSTALL_DIR} passwd root
arch-chroot ${HOLO_INSTALL_DIR} useradd --create-home ${HOLOUSER}
echo -e "${HOLOPASS}\n${HOLOPASS}" | arch-chroot ${HOLO_INSTALL_DIR} passwd ${HOLOUSER}
echo "${HOLOUSER} ALL=(root) NOPASSWD:ALL" > ${HOLO_INSTALL_DIR}/etc/sudoers.d/${HOLOUSER}
chmod 0440 ${HOLO_INSTALL_DIR}/etc/sudoers.d/${HOLOUSER}
echo "127.0.1.1    ${HOLOHOSTNAME}" >> ${HOLO_INSTALL_DIR}/etc/hosts
echo "Installing bootloader..."
arch-chroot ${HOLO_INSTALL_DIR} ${CMD_PACMAN_UPDATE}
mkdir ${HOLO_INSTALL_DIR}/boot/efi
mount -t vfat ${INSTALLDEVICE}1 ${HOLO_INSTALL_DIR}/boot/efi
arch-chroot ${HOLO_INSTALL_DIR} ${CMD_PACMAN_INSTALL} efibootmgr sddm neofetch core/grub mkinitcpio networkmanager inetutils
arch-chroot ${HOLO_INSTALL_DIR} systemctl enable NetworkManager systemd-timesyncd
arch-chroot ${HOLO_INSTALL_DIR} grub-install --target=x86_64-efi --efi-directory=/boot/efi --bootloader-id=holo --removable
arch-chroot ${HOLO_INSTALL_DIR} grub-mkconfig -o /boot/grub/grub.cfg
echo "Installing DE and dependencies..."
# Updates pacman db for chroot
arch-chroot ${HOLO_INSTALL_DIR} ${CMD_PACMAN_UPDATE}
# Installs DE
arch-chroot ${HOLO_INSTALL_DIR} ${CMD_PACMAN_INSTALL} archlinux-keyring curl tar yay git go plasma-meta plasma-nm ffmpegthumbs kdegraphics-thumbnailers gwenview glxinfo dolphin konsole steamdeck-kde-presets chromium steam-jupiter-stable gamescope mangohud vulkan-radeon lib32-vulkan-radeon jupiter-hw-support pipewire-alsa pipewire-jack lib32-pipewire-jack pipewire-pulse pipewire-v4l2 spectacle lib32-pipewire-v4l2 pipewire lib32-pipewire pipewire-media-session
# Installs more stuff
echo "Installing Bluetooth dependencies..."
arch-chroot ${HOLO_INSTALL_DIR} ${CMD_PACMAN_INSTALL} bluez bluez-utils cheeze blueman bluez-qt
arch-chroot ${HOLO_INSTALL_DIR} systemctl enable bluetooth
arch-chroot ${HOLO_INSTALL_DIR} usermod -a -G rfkill ${HOLOUSER}
echo "Configuring Steam Deck UI by default..."
mkdir ${HOLO_INSTALL_DIR}/etc/sddm.conf.d
echo "[Autologin]
User=${USER}
Session=gamescope-wayland.desktop" >> ${HOLO_INSTALL_DIR}/etc/sddm.conf.d/autologin.conf
# Services
echo "Setting up locale..."
echo "${SYSTEM_LOCALE}" >> ${HOLO_INSTALL_DIR}/etc/locale.gen
arch-chroot ${HOLO_INSTALL_DIR} locale-gen
echo "LANG=$(echo ${SYSTEM_LOCALE} | cut -d' ' -f1)" > ${HOLO_INSTALL_DIR}/etc/locale.conf
arch-chroot ${HOLO_INSTALL_DIR} systemctl enable sddm
echo "Installation complete! You may reboot now!"
echo "End time: $(date)"