#!/bin/bash

# Zorg ervoor dat het script als root wordt uitgevoerd
if [ "$EUID" -ne 0 ]; then
  echo "Dit script moet als root worden uitgevoerd. Gebruik 'sudo' of log in als root."
  exit 1
fi

echo "=================================================="
echo " STAP 1: GEBRUIKER 'hcc' AANMAKEN"
echo "=================================================="
if id "hcc" &>/dev/null; then
    echo "Gebruiker 'hcc' bestaat al. Stap overgeslagen."
else
    # Gebruiker aanmaken met wachtwoord 'hcc' via chpasswd
    useradd -m -s /bin/bash hcc
    echo "hcc:hcc" | chpasswd
    # Voeg hcc toe aan de sudo groep voor administratieve rechten
    usermod -aG sudo hcc
    echo "Gebruiker 'hcc' succesvol aangemaakt met wachtwoord 'hcc'."
fi

echo ""
echo "=================================================="
echo " STAP 2: DESKTOP, XRDP EN REMMINA INSTALLEREN"
echo "=================================================="
echo "Pakketbronnen bijwerken..."
apt update

echo "Installatie van XFCE4, XRDP en Remmina (dit kan even duren)..."
# Installatie van desktop, xserver, xrdp en remmina
apt install -y xfce4 xfce4-goodies xorg xrdp remmina chromium

# Zorg ervoor dat XRDP de XFCE-omgeving start bij het inloggen
echo "xfce4-session" > /home/hcc/.xsession
chown hcc:hcc /home/hcc/.xsession

# XRDP service starten en activeren bij boot
systemctl enable xrdp
systemctl restart xrdp

echo "Desktop en Remote Desktop (XRDP) succesvol geconfigureerd."

echo ""
echo "=================================================="
echo " STAP 3: BOOTLOADER CONTROLEREN & IOMMU INRICHTEN"
echo "=================================================="
# Controleer of het systeem GRUB of systemd-boot gebruikt
if [ -d "/sys/firmware/efi/efivars" ] && [ -d "/etc/kernel" ] && grep -q "initrd" /etc/kernel/cmdline 2>/dev/null; then
    BOOTLOADER="systemd-boot"
    echo "Gedetecteerde bootloader: systemd-boot (ZFS/EFI Setup)"
else
    BOOTLOADER="grub"
    echo "Gedetecteerde bootloader: GRUB"
fi

CMDLINE_PARAMS="intel_iommu=on iommu=pt"

if [ "$BOOTLOADER" = "grub" ]; then
    if grep -q "intel_iommu=on" /etc/default/grub; then
        echo "Intel IOMMU staat al in /etc/default/grub. Stap overgeslagen."
    else
        sed -i 's/GRUB_CMDLINE_LINUX_DEFAULT="/GRUB_CMDLINE_LINUX_DEFAULT="intel_iommu=on iommu=pt /' /etc/default/grub
        echo "Parameters succesvol toegevoegd aan /etc/default/grub."
        echo "Bootloader bijwerken..."
        update-grub
    fi
elif [ "$BOOTLOADER" = "systemd-boot" ]; then
    if grep -q "intel_iommu=on" /etc/kernel/cmdline; then
        echo "Intel IOMMU staat al in /etc/kernel/cmdline. Stap overgeslagen."
    else
        sed -i "s/$/ $CMDLINE_PARAMS/" /etc/kernel/cmdline
        echo "Parameters succesvol toegevoegd aan /etc/kernel/cmdline."
        echo "Bootloader bijwerken..."
        proxmox-boot-tool refresh
    fi
fi

echo ""
echo "=================================================="
echo " STAP 4: VFIO KERNELMODULES TOEVOEGEN"
echo "=================================================="
MODULES=("vfio" "vfio_iommu_type1" "vfio_pci" "vfio_virqfd")

for mod in "${MODULES[@]}"; do
    if grep -q "^$mod" /etc/modules; then
        echo "Module '$mod' is al aanwezig in /etc/modules."
    else
        echo "$mod" >> /etc/modules
        echo "Module '$mod' toegevoegd aan /etc/modules."
    fi
done

echo "Maak proxmox gratis"
echo "=================================================="
echo " SCRIPT VOLTOOID"
echo "=================================================="
echo "Alle acties zijn met succes uitgevoerd!"
echo "1. Gebruiker: hcc | Wachtwoord: hcc"
echo "2. RDP Server (XRDP) draait op poort 3389."
echo "3. IOMMU & VFIO zijn ingesteld voor uw Intel CPU."
echo ""
echo "BELANGRIJK: Start uw Proxmox-server opnieuw op om alles te activeren:"
echo "--> reboot <--"