#!/bin/bash

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

VAR_ISO_PATH="/srv/data/ISO/"

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

# Funktion zum Überprüfen der Benutzerexistenz
check_user_existence() {
    local user="$1"
    if ! id "$user" &>/dev/null; then
        echo "Der Benutzer $user 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" "frodo")

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

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

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_USER=$(whoami)
            ;;
        "Anderer Benutzer")
            read -p "Bitte geben Sie einen Benutzernamen ein: " VAR_USER
            check_user_existence "$VAR_USER"
            ;;
        "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"
                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 den User

clear

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

declare -A OS_MAP
OS_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"]="Fedora_64 Fedora-*.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"
)

select opt in "${!OS_MAP[@]}"
do
    case $opt in
        "Quit")
            exit
            ;;
        *)
            VAR_OS="${OS_MAP[$opt]%% *}"    # Extrahiere den ersten Teil (Betriebssystem)
            VAR_ISO="${OS_MAP[$opt]#* }"    # Extrahiere den zweiten Teil (ISO-Muster)
            ;;
    esac

    default_name=$(get_random_name)

    # Benutzereingabe für VM-Konfiguration
    read -p "Geben Sie die Anzahl der VMs an (Standard: 1): " VAR_QUAN
    read -p "Präfix Host Name (Vorgabe: $default_name): " VAR_NAME_PREFIX
    read -p "CPUs (Standard: 2 CPU): " VAR_CPU
    read -p "Memory (Standard: 2048 MB-RAM) : " VAR_MEM
    read -p "Bitte die Anzahl der HDD (Standard: 1-HDD) : " VAR_HDD

    # Variablen:
    IFS="
    "
    # Standardwerte festlegen, wenn keine Eingabe erfolgt
    VAR_CPU=${VAR_CPU:-2}
    VAR_MEM=${VAR_MEM:-2048}
    VAR_VIDEO_RAM=64
    VAR_NIC1="nat"
    VAR_NIC2="bridged"
    VAR_NET_INTERFACE="eth0"
    VAR_RDP_PORT=339
    

    # Schleife zur Erzeugung der VMs für Gruppenmitglieder
    for VAR_USER_IN_GROUP in "${GROUP_MEMBERS[@]}"
    do
        for N in $(seq 1 $VAR_QUAN)
        do
            ORDER_NUM=$(printf "%02d" $N) # Zweistellige Ordnungszahl generieren
            VM_NAME="${VAR_NAME_PREFIX}-${VAR_USER_IN_GROUP}-${ORDER_NUM}"
            VM_DIR="/srv/vbox/$VAR_USER_IN_GROUP/$VM_NAME"

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

            # VM-Konfigurieren
            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" \
                --graphicscontroller vmsvga \
                --nic2 "$VAR_NIC2" \
                --bridgeadapter2 "$VAR_NET_INTERFACE" \
                --vrde on \
                --vrdemulticon on \
                --vrdeport "$VAR_RDP_PORT"

            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_PATH}${VAR_ISO}"

            # HDD erzeugen und zuweisen
            for j in $(seq 1 $VAR_HDD)
            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/$HDD_NAME" \
                    --size 102400 \
                    --format VDI

                sudo -u "$VAR_USER_IN_GROUP" VBoxManage storageattach "$VM_NAME" \
                    --storagectl "SATA Controller" \
                    --port "$j" \
                    --device 0 \
                    --type hdd \
                    --medium "$VM_DIR/$HDD_NAME"
            done
        done
    done
    echo "VMs wurden erstellt."
    break
done