#!/bin/bash
ORIGINAL_ARGS=("$@")

#  Includes
. $(miniade) || { echo "${0##*/}: ERROR: miniade failed (hint: run 'miniade' to see error)" >&2; exit 1; }

#  Configurable stuff

#  Other globals

main()
{
    local MY_ARGS
    local PROGNAME

    #  Defaults for options
    REMOTE_HOST=
    MEM_SIZE=1
    DISK_TUPLES=()
    NIC_TUPLES=()
    VM_NAME=
    CPU_COUNT=1
    RELEASE=

    #  Process options
    special_opts_handler()
    {
        case $1 in
            --name=*)    VM_NAME=${1#*=} ;;
            --os=*)      miniade_warning "'--os=<release>' has been replaced by '--release=<release>'"
                         RELEASE=${1#*=} ;;
            --release=*) RELEASE=${1#*=} ;;
            --remote=*)  REMOTE_HOST=${1#*=} ;;
            --mem=*)     MEM_SIZE=${1#*=} ;;
            --nic=*)     NIC_TUPLES=( "${NIC_TUPLES[@]}" ${1#*=} ) ;;
            --cpus=*)    CPU_COUNT=${1#*=} ;;
            --disk=*)    DISK_TUPLES=( "${DISK_TUPLES[@]}" ${1#*=} ) ;;
            *)            return 1 ;;
        esac
    }
    miniade_process_options --help-handler=help --special-opts-handler=special_opts_handler MY_ARGS "$@" && set -- "${MY_ARGS[@]}"

    #  Process arguments
    [ $# = 0 ] || miniade_bad_usage

    #  Sanity checks and derivations
    miniade_get_progname PROGNAME
    HISTFILE=$HOME/.${PROGNAME}_history
    for VAR in REMOTE_HOST; do
        eval "[ \"X\$$VAR\" != Xunset ] || $VAR="
    done
    for VAR in VM_NAME; do
        eval "[ \"X\$$VAR\" != X ]" || miniade_bad_usage
    done
    export LIBVIRT_DEFAULT_URI=qemu:///system
    [ ${#DISK_TUPLES[*]} -ge 1 ] || miniade_bad_usage
    [ "X$REMOTE_HOST" = X ] || miniade_check_ssh_ok $REMOTE_HOST || miniade_error "$REMOTE_HOST: no unfettered ssh access"
    { ! virsh dominfo $VM_NAME > /dev/null 2>&1; } || miniade_error "$VM_NAME: VM defined locally already (hint: maybe run 'virsh undefine $VM_NAME'?)"
    if [ "X$REMOTE_HOST" != X ]; then
        miniade_check_ssh_ok $REMOTE_HOST
        { ! ssh -n $REMOTE_HOST "export LIBVIRT_DEFAULT_URI=$LIBVIRT_DEFAULT_URI; virsh dominfo $VM_NAME" > /dev/null 2>&1; } || miniade_error "$VM_NAME: VM defined remotely already (hint: maybe run 'ssh -n $REMOTE_HOST \"export LIBVIRT_DEFAULT_URI=$LIBVIRT_DEFAULT_URI; virsh undefine $VM_NAME\"'?)"
    fi
    #  Disk checks
    for ((I=0; I<${#DISK_TUPLES[*]}; I++)); do
        [[ ${DISK_TUPLES[$I]} =~ ^(file|block):(/.*) ]] || miniade_error "${DISK_TUPLES[$I]}: failed to parse (hint: should be {file|block}:<absolute-path>)"
        DISK_TYPE=${BASH_REMATCH[1]}
        DISK_IMAGE_OR_DEVICE=${BASH_REMATCH[2]}
        STAT_OUTPUT=$(stat -L -c %F $DISK_IMAGE_OR_DEVICE 2>&1) || true
        miniade_debug 10 "main: DISK_TYPE=$DISK_TYPE, DISK_IMAGE_OR_DEVICE=$DISK_IMAGE_OR_DEVICE, STAT_OUTPUT=[$STAT_OUTPUT]"
        if [ $DISK_TYPE = file ] && [ "$STAT_OUTPUT" != "regular file" ]; then
            miniade_error "$DISK_IMAGE_OR_DEVICE: not a file"
        elif [ $DISK_TYPE = block  ] && [ "$STAT_OUTPUT" != "block special file" ]; then
            miniade_error "$DISK_IMAGE_OR_DEVICE: not a block device"
        fi
        { ! [[ $DISK_IMAGE_OR_DEVICE =~ /dev/drbd.* ]]; } || [ "X$(drbdadm status ${DISK_IMAGE_OR_DEVICE#/dev/} | sed -n '1s/.*://p')" = XPrimary ] || miniade_error "$DISK_IMAGE_OR_DEVICE: appears to be a DRBD device but not in primary mode"
    done
    DISK_LETTER_MAP=( {a..z} )
    #  NIC checks
    for ((I=0; I<${#NIC_TUPLES[*]}; I++)); do
        [[ ${NIC_TUPLES[$I]} =~ ^(bridge|network):(.+) ]] || miniade_error "${NIC_TUPLES[$I]}: failed to parse (hint: should be {bridge|network}:<bridge-or-network-name>)"
        NIC_TYPE=${BASH_REMATCH[1]}
        NIC_BRIDGE_OR_NETWORK=${BASH_REMATCH[2]}
    done
    #  Memory checks
    if [[ $MEM_SIZE =~ ^([1-9][0-9]*)([GMK]?)$ ]]; then
        MEM_SIZE=${BASH_REMATCH[1]}
        MEM_SIZE_UNIT=${BASH_REMATCH[2]:-G}
    else
        miniade_error "$MEM_SIZE: invalid memory size (hint: format is <num>[unit])"
    fi
    #  Release checks
    [ "X$RELEASE" != X ] || { RELEASE=bullseye; miniade_warning "release defaulted to $RELEASE (hint: use option '--release=$RELEASE' to avoid this warning or '--release=<release>' to use another release instead)"; }
    case $RELEASE in
        bullseye) : ;;
        debian11) RELEASE=bullseye ;;
        bookworm) : ;;
        debian12) RELEASE=bookworm ;;
        trixie) : ;;
        debian13) RELEASE=trixie ;;
        stable|unstable|testing|sid) miniade_error "$RELEASE: moving-target release name (hint: use a less ambiguous release name (e.g. 'debian12' or 'bookworm')" ;;
        *)        miniade_internal "$RELEASE: invalid release" ;;
    esac
    case $RELEASE in
        bullseye) ISO_FILE=/vol/local/images/debian-11.6.0-amd64-netinst.iso ;;
        bookworm) ISO_FILE=/vol/local/images/debian-12.6.0-amd64-netinst.iso ;;
        trixie)   ISO_FILE=/vol/local/images/debian-trixie-DI-alpha1-amd64-netinst.iso ;;
        *)        miniade_internal "this should not happen (1, RELEASE=$RELEASE)" ;;
    esac
    #  Misc
    UUID=$(uuidgen)

    #  Guts
    echo "$(date --iso=seconds): $0" "${ORIGINAL_ARGS[@]}" >> $HISTFILE
    write_xml > /tmp/$PROGNAME.$$.xml
    define_vm /tmp/$PROGNAME.$$.xml
}

help()
{
    local PROGNAME

    miniade_get_progname PROGNAME
    echo "Usage: $PROGNAME [ <standard-options> ] [ --remote=<host> ] --name=<vm-name> [ --release=<release> ] [ --mem=<mem-size>[MGT] ] [ --cpus=<count> ] --disk={block|file}:<absolute-path> [ ... ] --nic={bridge|network}:<bridge-or-network-name> [ ... ]"
    exit 0
}

write_xml()
{
    miniade_info "generating XML ..."
    DISK_OR_NIC_BUS=
    echo "<domain type='kvm'>"
    echo "  <name>$VM_NAME</name>"
    echo "  <uuid>$UUID</uuid>"
    echo "  <metadata>"
    miniade_debug 10 "write_xml: writing mounthook stuff ..."
    miniade_debug 10 "write_xml: writing libosinfo stuff ..."
    echo "    <libosinfo:libosinfo xmlns:libosinfo=\"http://libosinfo.org/xmlns/libvirt/domain/1.0\">"
    case $RELEASE in
        bullseye) echo "      <libosinfo:os id=\"http://debian.org/debian/11\"/>" ;;
        bookworm) echo "      <libosinfo:os id=\"http://debian.org/debian/12\"/>" ;;
        trixie)   echo "      <libosinfo:os id=\"http://debian.org/debian/13\"/>" ;;
        *)        miniade_internal "this should not happen (2)" ;;
    esac
    echo "    </libosinfo:libosinfo>"
    echo "  </metadata>"
    miniade_debug 10 "write_xml: writing memory stuff ..."
    case $MEM_SIZE_UNIT in
        G) MEM_SIZE_KB=$((MEM_SIZE*1024*1024)) ;;
        M) MEM_SIZE_KB=$((MEM_SIZE*1024)) ;;
        K) MEM_SIZE_KB=$MEM_SIZE ;;
    esac
    echo "  <memory unit='KiB'>$MEM_SIZE_KB</memory>"
    echo "  <currentMemory unit='KiB'>$MEM_SIZE_KB</currentMemory>"
    miniade_debug 10 "write_xml: writing cpu stuff ..."
    echo "  <vcpu placement='static'>$CPU_COUNT</vcpu>"
    miniade_debug 10 "write_xml: writing cgroups stuff ..."
    miniade_debug 10 "write_xml: writing OS stuff ..."
    echo "  <os>"
    echo "    <type arch='x86_64' machine='pc-q35-3.1'>hvm</type>"
    case $RELEASE in
        bullseye) echo "    <loader readonly='yes' type='pflash'>/usr/share/OVMF/OVMF_CODE.fd</loader>" ;;
        bookworm) echo "    <loader readonly='yes' type='pflash'>/usr/share/OVMF/OVMF_CODE_4M.fd</loader>" ;;
        trixie)   echo "    <loader readonly='yes' type='pflash'>/usr/share/OVMF/OVMF_CODE_4M.fd</loader>" ;;
        *)        miniade_internal "this should not happen (3)" ;;
    esac
    echo "    <nvram>/var/lib/libvirt/qemu/nvram/${VM_NAME}_VARS.fd</nvram>"
    echo "    <bootmenu enable='yes'/>"
    echo "  </os>"
    miniade_debug 10 "write_xml: writing UID/GID remapping stuff ..."
    miniade_debug 10 "write_xml: writing features stuff ..."
    echo "  <features>"
    echo "    <acpi/>"
    echo "    <apic/>"
    echo "    <vmport state='off'/>"
    echo "  </features>"
    miniade_debug 10 "write_xml: writing cpu stuff (should be moved for better alignment) ..."
    echo "  <cpu mode='host-passthrough' check='partial'/>"
    miniade_debug 10 "write_xml: writing clock stuff ..."
    echo "  <clock offset='utc'>"
    echo "    <timer name='rtc' tickpolicy='catchup'/>"
    echo "    <timer name='pit' tickpolicy='delay'/>"
    echo "    <timer name='hpet' present='no'/>"
    echo "  </clock>"
    miniade_debug 10 "write_xml: writing switch-related stuff ..."
    echo "  <on_poweroff>destroy</on_poweroff>"
    echo "  <on_reboot>restart</on_reboot>"
    echo "  <on_crash>destroy</on_crash>"
    echo "  <pm>"
    echo "    <suspend-to-mem enabled='no'/>"
    echo "    <suspend-to-disk enabled='no'/>"
    echo "  </pm>"
    echo "  <devices>"
    miniade_debug 10 "write_xml: writing hypervisor stuff ..."
    echo "    <emulator>/usr/bin/qemu-system-x86_64</emulator>"
    miniade_debug 10 "write_xml: writing video stuff ..."
    echo "    <video>"
    #echo "      <model type='qxl' ram='65536' vram='65536' vgamem='16384' heads='1' primary='yes'/>"
    echo "      <model type='virtio' heads='1' primary='yes'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x01' function='0x0'/>"
    echo "    </video>"
    miniade_debug 10 "write_xml: writing controller stuff ..."
    #  This script reserves busses 0x00-0x04 for non-disk/non-nic stuff.
    echo "    <controller type='sata' index='0'>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x1f' function='0x2'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='0' model='pcie-root'/>"
    echo "    <controller type='pci' index='1' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='1' port='0x10'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x0' multifunction='on'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='2' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='2' port='0x11'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x1'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='3' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='3' port='0x12'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x2'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='4' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='4' port='0x13'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x3'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='5' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='5' port='0x14'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x4'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='6' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='6' port='0x15'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x5'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='7' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='7' port='0x16'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x6'/>"
    echo "    </controller>"
    echo "    <controller type='pci' index='8' model='pcie-root-port'>"
    echo "      <model name='pcie-root-port'/>"
    echo "      <target chassis='8' port='0x17'/>"
    echo "      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x7'/>"
    echo "    </controller>"
    echo "    <controller type='virtio-serial' index='0'>"
    echo "      <address type='pci' domain='0x0000' bus='0x01' slot='0x00' function='0x0'/>"
    echo "    </controller>"
    echo "    <controller type='usb' index='0' model='qemu-xhci' ports='15'>"
    echo "      <address type='pci' domain='0x0000' bus='0x02' slot='0x00' function='0x0'/>"
    echo "    </controller>"
    miniade_debug 10 "write_xml: writing CDROM stuff ..."
    #  CDROM is attached to the above SATA controller
    echo "    <disk type='file' device='cdrom'>"
    echo "      <driver name='qemu' type='raw'/>"
    echo "      <source file='$ISO_FILE'/>"
    if [ ! -f $ISO_FILE ]; then
        miniade_warning "$ISO_FILE: not found (hint: either remove the CD from the virtual CD drive or copy the file into place or update ISO_FILE in this script)"
    fi
    echo "      <target dev='sda' bus='sata'/>"
    echo "      <readonly/>"
    echo "      <shareable/>"
    echo "      <boot order='2'/>"
    echo "      <address type='drive' controller='0' bus='0' target='0' unit='0'/>"
    echo "    </disk>"
    miniade_debug 10 "write_xml: writing memballoon stuff ..."
    echo "    <memballoon model='virtio'>"
    echo "      <address type='pci' domain='0x0000' bus='0x03' slot='0x00' function='0x0'/>"
    echo "    </memballoon>"
    miniade_debug 10 "write_xml: writing RNG stuff ..."
    echo "    <rng model='virtio'>"
    echo "      <backend model='random'>/dev/urandom</backend>"
    echo "      <address type='pci' domain='0x0000' bus='0x04' slot='0x00' function='0x0'/>"
    echo "    </rng>"
    miniade_debug 10 "write_xml: writing disk stuff ..."
    #  This script reserves busses 0x05-upwards for disk/nic stuff.
    NEXT_DISK_OR_NIC_BUS=$(printf %d 0x05)
    for ((I=0; I<${#DISK_TUPLES[*]}; I++)); do 
        [[ ${DISK_TUPLES[$I]} =~ ^(file|block):(/.*) ]]
        DISK_TYPE=${BASH_REMATCH[1]}
        DISK_IMAGE_OR_DEVICE=${BASH_REMATCH[2]}
        if [ $DISK_TYPE = file ]; then
            echo "    <disk type='$DISK_TYPE' device='disk'>"
            echo "      <driver name='qemu' type='raw'/>"
            echo "      <source file='$DISK_IMAGE_OR_DEVICE'/>"
        else
            echo "    <disk type='$DISK_TYPE' device='disk'>"
            echo "      <driver name='qemu' type='raw'/>"
            echo "      <source dev='$DISK_IMAGE_OR_DEVICE'/>"
        fi
        echo "      <target dev='vd${DISK_LETTER_MAP[$I]}' bus='virtio'/>"
        if ((I==0)); then
            echo "      <boot order='1'/>"
        fi
        echo "      <address type='pci' domain='0x0000' bus='$(printf 0x%02X $NEXT_DISK_OR_NIC_BUS)' slot='0x00' function='0x0'/>"
        echo "    </disk>"
        ((NEXT_DISK_OR_NIC_BUS++))
    done
    miniade_debug 10 "write_xml: writing nic stuff ..."
    for ((I=0; I<${#NIC_TUPLES[*]}; I++)); do 
        [[ ${NIC_TUPLES[$I]} =~ ^(bridge|network):(.+) ]]
        NIC_TYPE=${BASH_REMATCH[1]}
        NIC_BRIDGE_OR_NETWORK=${BASH_REMATCH[2]}
        MACADDR=$(perl -e 'printf("%02x:%02x:%02x:%02x:%02x:%02x", 0x52, 0x54, 0x00, map { int(rand(256)) } 1..3);')
        echo "    <interface type='$NIC_TYPE'>"
        echo "      <mac address='$MACADDR'/>"
        echo "      <source $NIC_TYPE='$NIC_BRIDGE_OR_NETWORK'/>"
        echo "      <model type='virtio'/>"
        echo "      <address type='pci' domain='0x0000' bus='$(printf 0x%02X $NEXT_DISK_OR_NIC_BUS)' slot='0x00' function='0x0'/>"
        echo "    </interface>"
        ((NEXT_DISK_OR_NIC_BUS++))
    done
    miniade_debug 10 "write_xml: writing console stuff ..."
    echo "    <serial type='pty'>"
    echo "      <target type='isa-serial' port='0'>"
    echo "        <model name='isa-serial'/>"
    echo "      </target>"
    echo "    </serial>"
    echo "    <console type='pty'>"
    echo "      <target type='serial' port='0'/>"
    echo "    </console>"
    echo "    <channel type='spicevmc'>"
    echo "      <target type='virtio' name='com.redhat.spice.0'/>"
    echo "      <address type='virtio-serial' controller='0' bus='0' port='2'/>"
    echo "    </channel>"
    echo "    <input type='mouse' bus='ps2'/>"
    echo "    <input type='keyboard' bus='ps2'/>"
    echo "    <graphics type='spice' autoport='yes'>"
    echo "      <listen type='address'/>"
    echo "      <image compression='off'/>"
    echo "    </graphics>"
    #  20240828: I commented out the audio code as it failed when run on
    #  Debian 11 VM server to create Debian 12/bookworm KVM-based VM. My
    #  guess is that this will run on a Debian 12 VM server but not a
    #  Debian 11 VM server.
    #miniade_debug 10 "write_xml: writing audio stuff ..."
    #case $RELEASE in
    #    bullseye) : ;;
    #    bookworm) echo "<audio id='1' type='spice'/>" ;;
    #    *)        miniade_internal "this should not happen (4)" ;;
    #esac
    miniade_debug 10 "write_xml: writing USB redirector stuff ..."
    echo "    <redirdev bus='usb' type='spicevmc'>"
    echo "      <address type='usb' bus='0' port='2'/>"
    echo "    </redirdev>"
    echo "  </devices>"
    echo "</domain>"
}

define_vm()
{
    local XML_FILE

    [ $# = 1 ] || miniade_internal "define_vm: bad arg count ($#)"
    XML_FILE=$1

    miniade_info "importing XML ..."
    miniade_evaler "virsh --quiet define $XML_FILE"
    if [ "X$REMOTE_HOST" != X ]; then
        miniade_info "importing XML on remote host ..."
        miniade_evaler "ssh -T $REMOTE_HOST \"export LIBVIRT_DEFAULT_URI=$LIBVIRT_DEFAULT_URI; virsh --quiet define <(cat)\" < $XML_FILE"
        miniade_evaler "rm $XML_FILE"
    else
        miniade_warning "$XML_FILE: use this file to define the VM on the other node"
    fi
}

main "$@"