#!/bin/bash

# Autor: Branislav Zeljak v0.8
# Beschreibung: Dieses Skript erstellt VirtualBox-VMs und klont Festplatten parallel.
# Verwendung: Führen Sie das Skript aus und befolgen Sie die Eingabeaufforderungen.
clear

# %--------------------start/Funktionen--------------------%

# Funktion zum Überprüfen der Benutzerexistenz
check_user_existence() {
    local VAR_USER_IN_GROUP="$1"
    if ! grep -q "^$VAR_USER_IN_GROUP:" /etc/passwd; then
        echo "Der Benutzer $VAR_USER_IN_GROUP existiert nicht auf diesem System."
        exit 1
    fi
}

get_group_users() {
    local groupname="$1"
    local members

    members=$(getent group "$groupname" | cut -d: -f4)

    if [[ -n "$members" ]]; then
        echo "$members" | tr ',' '\n'
    else
        echo "Die Gruppe $groupname existiert nicht oder hat keine Mitglieder."
    fi
}

# Funktion, um einen zufälligen Host-Namen aus der Liste auszuwählen
get_random_name() {
    # Liste der Namen
    local names=("frodo" "samweis" "gandalf" "aragorn" "legolas" "gimli" "boromir" "meriadoc" "peregrin" "arwen" "galadriel" "celeborn" "elrond" "glorfindel" "faramir" "eowyn" "theoden" "treebeard" "gollum" "sauron" "saruman" "bilbo" "denethor" "radagast" "thranduil" "bard" "dori" "frodo")

    # Zufälligen Index auswählen
    local random_index=$((RANDOM % ${#names[@]}))
    echo "${names[random_index]}"
}

# Funktion zur Generierung eines eindeutigen RDP-Ports
generate_rdp_port() {
    local rdp_port
    rdp_port=$(shuf -i 3390-49151 -n 1)
    echo "$rdp_port"
}

# Funktion zum Suchen einer ISO-Datei basierend auf einem Muster
find_iso_file() {
    local iso_pattern="$1"
    local iso_file
    iso_file=$(find "$ISO_DIR" -type f -name "$iso_pattern")
    echo "$iso_file"
}

# %--------------------end/Funktionen--------------------%

echo "User / Passwort: user00/Start1234"
echo "Welches VDI-OS soll installiert werden?"
echo

echo "Für wen soll es sein?"
echo
COLUMNS=10
PS3="Ihre Auswahl: "
options=("Aktueller Benutzer" "Anderer Benutzer" "Mitglieder einer Gruppe" "Beenden")
select opt in "${options[@]}"
do
    case $opt in
        "Aktueller Benutzer")
            VAR_GROUP_MEMBERS=$(whoami)
            ;;
        "Anderer Benutzer")
            read -p "Bitte geben Sie einen Benutzernamen ein: " VAR_GROUP_MEMBERS
            if grep -q "^$VAR_GROUP_MEMBERS:" /etc/passwd; then
                echo "Der Benutzer $VAR_GROUP_MEMBERS existiert auf diesem System."
            else
                echo "Der Benutzer $VAR_GROUP_MEMBERS existiert nicht auf diesem System."
                exit 1
            fi
            ;;
        "Mitglieder einer Gruppe")
            read -p "Bitte geben Sie den Gruppennamen ein: " VAR_GROUP
            if getent group "$VAR_GROUP" &>/dev/null; then
                echo "Mitglieder der Gruppe $VAR_GROUP:"
                get_group_users "$VAR_GROUP"
                VAR_GROUP_MEMBERS=($(get_group_users "$VAR_GROUP"))
            else
                echo "Die Gruppe $VAR_GROUP existiert nicht auf diesem System."
                exit 1
            fi
            ;;
        "Beenden")
            exit
            ;;
    esac
    break
done

# Variable für das Verzeichnis
VAR_VB="/storage/vbox_storage"

# Verzeichnis mit ISO-Dateien
ISO_DIR="/srv/data/ISO/"

echo "Welches Betriebssystem soll installiert werden?"
echo
COLUMNS=10
PS3="Bitte Ihre Auswahl eingeben: "

# Betriebssysteme und ISO-Dateimuster
declare -A WieOS_MAP=(
    ["Ubuntu LTS SRV"]="Ubuntu_64 ubuntu-*-live-server-amd64.iso"
    ["Ubuntu LTS GUI"]="Ubuntu_64 ubuntu-*-desktop-amd64.iso"
    ["Debian"]="Debian_64 debian-*.iso"
    ["RHEL"]="RedHat_64 rhel-*.iso"
    ["Fedora-Workstation"]="Fedora_64 Fedora-Workstation*.iso"
    ["Fedora-Server"]="Fedora_64 Fedora-Server*.iso"
    ["kali-linux"]="Linux_64 kali-linux-*.iso"
    ["linuxmint"]="Linux_64 linuxmint-*.iso"
    ["openSUSE"]="OpenSUSE_64 openSUSE-*.iso"
    ["Manjaro"]="ArchLinux_64 manjaro-xfce-*.iso"
    ["Quit"]="Quit"
)

# Alphabetisch sortieren, mit Quit am Ende
sorted_keys=($(for key in "${!WieOS_MAP[@]}"; do echo "$key"; done | sort | grep -v "Quit"))
sorted_keys+=("Quit")

# Auswahlmenü für die Betriebssysteme anzeigen
select opt in "${sorted_keys[@]}"
do
    if [ "$opt" == "Quit" ]; then
        exit
    fi
    
    VAR_OS="${WieOS_MAP[$opt]%% *}"         # Extrahiere den ersten Teil (Betriebssystem)
    VAR_ISO_PATTERN="${WieOS_MAP[$opt]#* }" # Extrahiere den zweiten Teil (ISO-Muster)
    
    # Finden der tatsächlichen ISO-Datei
    VAR_ISO=$(find_iso_file "$VAR_ISO_PATTERN")
    
    if [ -z "$VAR_ISO" ]; then
        echo "Die ISO-Datei mit dem Muster '$VAR_ISO_PATTERN' wurde nicht gefunden."
        exit 1
    fi
    
    default_name=$(get_random_name)

    # Benutzereingaben für VM-Konfiguration
    read -p "Wie viele VMs werden benötigt? [1]: " VAR_QUANTITY
    VAR_QUANTITY=${VAR_QUANTITY:-1}
    read -p "Präfix Host Name (Vorgabe: $default_name): " VAR_NAME_PREFIX
    read -p "CPUs [2]: " VAR_CPU
    VAR_CPU=${VAR_CPU:-2}
    read -p "Memory [2048]: " VAR_MEM
    VAR_MEM=${VAR_MEM:-2048}
    read -p "Anzahl der HDDs (Standard: 1): " VAR_ADDITIONAL_HDD
    VAR_ADDITIONAL_HDD=${VAR_ADDITIONAL_HDD:-1}

    # Standardwerte
    VAR_VIDEO_RAM=64
    VAR_NIC1="nat"
    VAR_NIC2="bridged"
    VAR_NET=$(ip addr show | grep "^2: " | awk '{print $2}' | sed 's/://')
    VAR_NAME_PREFIX=${VAR_NAME_PREFIX:-$default_name}

    # Schleife zur Erzeugung der VMs für jeden Benutzer in der Gruppe
    for VAR_USER_IN_GROUP in "${VAR_GROUP_MEMBERS[@]}"
    do
        # Generiere einen eindeutigen RDP-Port
        VAR_RDP_PORT=$(generate_rdp_port)

        # Schleife zur Erzeugung der angeforderten Anzahl von VMs
        for ((i = 1; i <= VAR_QUANTITY; i++))
        do
            # VM erstellen
            VM_NAME="${VAR_NAME_PREFIX}_${VAR_USER_IN_GROUP}_VM$(printf "%02d" $i)"
            VM_DIR="$VAR_VB/$VAR_USER_IN_GROUP"

            sudo -u "$VAR_USER_IN_GROUP" VBoxManage createvm \
                --name "$VM_NAME" \
                --ostype "$VAR_OS" \
                --register \
                --basefolder "$VM_DIR"

            # VM-Konfiguration
            sudo -u "$VAR_USER_IN_GROUP" VBoxManage modifyvm "$VM_NAME" \
                --ioapic on \
                --memory "$VAR_MEM" \
                --cpus "$VAR_CPU" \
                --vram "$VAR_VIDEO_RAM" \
                --nic1 $VAR_NIC1 \
                --nic2 $VAR_NIC2 \
                --bridgeadapter2 ${VAR_NET} \
                --graphicscontroller "vboxsvga" \
                --vrde on \
                --vrdemulticon on \
                --vrdeport "$VAR_RDP_PORT"

            # Festplattencontroller hinzufügen
            sudo -u "$VAR_USER_IN_GROUP" VBoxManage storagectl "$VM_NAME" \
                --name "SATA Controller" \
                --add sata \
                --controller IntelAhci

            sudo -u "$VAR_USER_IN_GROUP" VBoxManage storagectl "$VM_NAME" \
                --name "IDE Controller" \
                --add ide \
                --controller PIIX4

            sudo -u "$VAR_USER_IN_GROUP" VBoxManage storageattach "$VM_NAME" \
                --storagectl "IDE Controller" \
                --port 1 \
                --device 0 \
                --type dvddrive \
                --medium "$VAR_ISO"

            # Schleife zur Erzeugung von zusätzlichen Festplatten
            for ((j = 1; j <= VAR_ADDITIONAL_HDD; j++))
            do
                HDD_NUM=$(printf "%02d" $j)
                HDD_NAME="${VM_NAME}-disk${HDD_NUM}.vdi"

                sudo -u "$VAR_USER_IN_GROUP" VBoxManage createhd \
                    --filename "$VM_DIR/$VM_NAME/$HDD_NAME" \
                    --size 102400 \
                    --format VDI

                sudo -u "$VAR_USER_IN_GROUP" VBoxManage storageattach "$VM_NAME" \
                    --storagectl "SATA Controller" \
                    --port "$HDD_NUM" \
                    --device 0 \
                    --type hdd \
                    --medium "$VM_DIR/$VM_NAME/$HDD_NAME"
            done
        done
    done
    
    echo "VMs wurden für alle wie angegeben erstellt."
    break
done