#!/usr/bin/env bash

usage() {
    echo "$0 [-h] [-c <configuration>] [-i <iso image>]"
    echo ""
    echo "Options:"
    echo "  -c: Configuration: simplex, duplex, controllerstorage, dedicatedstorage"
    echo "  -i: StarlingX ISO image"
    echo ""
}

usage_destroy() {
    echo "$0 [-h] [-c <configuration>]"
    echo ""
    echo "Options:"
    echo "  -c: Configuration: simplex, duplex, controllerstorage, dedicatedstorage"
    echo ""
}

iso_image_check() {
    local ISOIMAGE=$1
    if ! file ${ISOIMAGE} | grep "DOS/MBR" > /dev/null; then
        echo "$ISOIMAGE is not an ISO type"
        exit -1
    fi
}

configuration_check() {
    local CONFIGURATION=$1
    if [ $CONFIGURATION != "simplex" ] && [ $CONFIGURATION != "duplex" ] && [ $CONFIGURATION != "controllerstorage" ] && [ $CONFIGURATION != "dedicatedstorage" ]; then
        echo "Please check your configuration name, available configurations:"
        echo "simplex, duplex, controllerstorage, dedicatedstorage"
        exit 1
    fi
}

# delete a node's disk file in a safe way
delete_disk() {
    local fpath="$1"

    if [ ! -f "$fpath" ]; then
        echo "file to delete is not a regular file: $fpath" >&2
        return 1
    fi

    file -b "$fpath" | grep -q "^QEMU QCOW Image (v3),"
    if [ $? -ne 0 ]; then
        echo "file to delete is not QEMU QCOW Image (v3): $fpath" >&2
        return 1
    fi

    sudo rm "$fpath"
}

# delete an xml file in a safe way
delete_xml() {
    local fpath="$1"

    if [ ! -f "$fpath" ]; then
        echo "file to delete is not a regular file: $fpath" >&2
        return 1
    fi

    file -b "$fpath" | grep -q "^ASCII text$"
    if [ $? -ne 0 ]; then
        echo "file to delete is not ASCII text: $fpath" >&2
        return 1
    fi

    sudo rm "$fpath"
}

# Create a Controller node
create_controller() {
    local CONFIGURATION=$1
    local CONTROLLER=$2
    local BRIDGE_INTERFACE=$3
    local ISOIMAGE=$4
    local DOMAIN_FILE
    if ([ "$CONFIGURATION" == "simplex" ]); then
        CONTROLLER_NODE_NUMBER=0
    else
        CONTROLLER_NODE_NUMBER=1
    fi
    for ((i=0; i<=$CONTROLLER_NODE_NUMBER; i++)); do
        CONTROLLER_NODE=${CONFIGURATION}-${CONTROLLER}-${i}
        DOMAIN_FILE=${DOMAIN_DIRECTORY}/${CONTROLLER_NODE}.xml
        if ([ "$CONFIGURATION" == "simplex" ] || [ "$CONFIGURATION" == "duplex" ]); then
            DISK_0_SIZE=600
            cp controller_allinone.xml ${DOMAIN_FILE}
        else
            DISK_0_SIZE=250
            cp controller.xml ${DOMAIN_FILE}
        fi
        sed -i -e "
            s,NAME,${CONTROLLER_NODE},
            s,DISK0,/var/lib/libvirt/images/${CONTROLLER_NODE}-0.img,
            s,DISK1,/var/lib/libvirt/images/${CONTROLLER_NODE}-1.img,
            s,%BR1%,${BRIDGE_INTERFACE}1,
            s,%BR2%,${BRIDGE_INTERFACE}2,
            s,%BR3%,${BRIDGE_INTERFACE}3,
            s,%BR4%,${BRIDGE_INTERFACE}4,
        " ${DOMAIN_FILE}
        sudo qemu-img create -f qcow2 /var/lib/libvirt/images/${CONTROLLER_NODE}-0.img ${DISK_0_SIZE}G
        sudo qemu-img create -f qcow2 /var/lib/libvirt/images/${CONTROLLER_NODE}-1.img 200G
        if ([ "$CONFIGURATION" == "simplex" ] || [ "$CONFIGURATION" == "duplex" ]); then
            sed -i -e "
                s,DISK2,/var/lib/libvirt/images/${CONTROLLER_NODE}-2.img,
            " ${DOMAIN_FILE}
            sudo qemu-img create -f qcow2 /var/lib/libvirt/images/${CONTROLLER_NODE}-2.img 200G
        fi
        if [ $i -eq 0 ]; then
            sed -i -e "s,ISO,${ISOIMAGE}," ${DOMAIN_FILE}
        else
            sed -i -e "s,ISO,," ${DOMAIN_FILE}
        fi
        sudo virsh define ${DOMAIN_FILE}
        if [ $i -eq 0 ]; then
            sudo virsh start ${CONTROLLER_NODE}
        fi
    done
}

# Delete a Controller node
destroy_controller() {
    local CONFIGURATION=$1
    local CONTROLLER=$2
    if ([ "$CONFIGURATION" == "simplex" ]); then
        CONTROLLER_NODE_NUMBER=0
    else
        CONTROLLER_NODE_NUMBER=1
    fi
    for ((i=0; i<=$CONTROLLER_NODE_NUMBER; i++)); do
        CONTROLLER_NODE=${CONFIGURATION}-${CONTROLLER}-${i}
        DOMAIN_FILE=$DOMAIN_DIRECTORY/$CONTROLLER_NODE.xml
        if virsh list --all --name | grep ${CONTROLLER_NODE}; then
            STATUS=$(virsh list --all | grep ${CONTROLLER_NODE} | awk '{ print $3}')
            if ([ "$STATUS" == "running" ])
            then
                sudo virsh destroy ${CONTROLLER_NODE}
            fi
            sudo virsh undefine ${CONTROLLER_NODE}
            delete_disk /var/lib/libvirt/images/${CONTROLLER_NODE}-0.img
            delete_disk /var/lib/libvirt/images/${CONTROLLER_NODE}-1.img
            if ([ "$CONFIGURATION" == "simplex" ] || [ "$CONFIGURATION" == "duplex" ]); then
                delete_disk /var/lib/libvirt/images/${CONTROLLER_NODE}-2.img
            fi
            [ -e ${DOMAIN_FILE} ] && delete_xml ${DOMAIN_FILE}
        fi
    done
}

# Create a Node
create_node() {
    local IDENTITY=$1
    local NODE=$2
    local BRIDGE_INTERFACE=$3
    local DOMAIN_FILE=${DOMAIN_DIRECTORY}/${NODE}.xml
    sudo qemu-img create -f qcow2 /var/lib/libvirt/images/${NODE}-0.img 200G
    sudo qemu-img create -f qcow2 /var/lib/libvirt/images/${NODE}-1.img 200G
    cp ${IDENTITY}.xml ${DOMAIN_FILE}
    sed -i -e "
        s,NAME,${NODE},;
        s,DISK0,/var/lib/libvirt/images/${NODE}-0.img,;
        s,DISK1,/var/lib/libvirt/images/${NODE}-1.img,
        s,%BR1%,${BRIDGE_INTERFACE}1,
        s,%BR2%,${BRIDGE_INTERFACE}2,
        s,%BR3%,${BRIDGE_INTERFACE}3,
        s,%BR4%,${BRIDGE_INTERFACE}4,
    " ${DOMAIN_FILE}
    sudo virsh define ${DOMAIN_FILE}
}

# Delete a Node
destroy_node() {
    local IDENTITY=$1
    local NODE=$2
    local DOMAIN_FILE=$DOMAIN_DIRECTORY/$NODE.xml
    if virsh list --all --name | grep ${NODE}; then
        STATUS=$(virsh list --all | grep ${NODE} | awk '{ print $3}')
        if ([ "$STATUS" == "running" ])
        then
            sudo virsh destroy ${NODE}
        fi
        sudo virsh undefine ${NODE}
        delete_disk /var/lib/libvirt/images/${NODE}-0.img
        delete_disk /var/lib/libvirt/images/${NODE}-1.img
        [ -e ${DOMAIN_FILE} ] && delete_xml ${DOMAIN_FILE}
    fi
}