#!/bin/bash

# -------------------------------------------------------------------------- #
# Copyright 2002-2021, OpenNebula Project, OpenNebula Systems                #
#                                                                            #
# Licensed under the Apache License, Version 2.0 (the "License"); you may    #
# not use this file except in compliance with the License. You may obtain    #
# a copy of the License at                                                   #
#                                                                            #
# http://www.apache.org/licenses/LICENSE-2.0                                 #
#                                                                            #
# Unless required by applicable law or agreed to in writing, software        #
# distributed under the License is distributed on an "AS IS" BASIS,          #
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
# See the License for the specific language governing permissions and        #
# limitations under the License.                                             #
#--------------------------------------------------------------------------- #

#rbt: brest mode
DRIVER_PATH=$(dirname $0)
if [ "$ONE_BREST_MODE" == "1" ] && [ "$SVMBREST" != "1" ]; then #rbt

    $DRIVER_PATH/shutdown_brest $@

else

    source $(dirname $0)/../../etc/vmm/kvm/kvmrc
    source $(dirname $0)/../../scripts_common.sh

    count=0

    deploy_id=$1

    #rbt: vgpu
    HOST=$2
    VM_ID=$3

    if [ "$ONE_BREST_MODE" == "1" ]; then
        LIBVIRT_URI=qemu+ssh://$HOST/system #rbt
    fi

    shutdown_command="virsh --connect $LIBVIRT_URI shutdown $deploy_id"

    # Check if the domain is already shutdown
    virsh --connect $LIBVIRT_URI --readonly dominfo $deploy_id > /dev/null 2>&1 || exit 0

    #If VM does not have ACPI enabled do not try to use it, destroy instead shutdown
    if [ "x$CANCEL_NO_ACPI" = "xyes" ]; then
        virsh --connect $LIBVIRT_URI dumpxml $deploy_id | grep -q acpi

        if [ "x$?" != "x0" ]; then
        shutdown_command="virsh --connect $LIBVIRT_URI destroy $deploy_id"
        fi
    fi

    if [ -z "$SHUTDOWN_TIMEOUT" ]; then
        TIMEOUT=120
    else
        TIMEOUT=$SHUTDOWN_TIMEOUT
    fi

    HALF_LOOP=$(($TIMEOUT/2))

    function monitor
    {
        # Issue another shutdown to cover occasional libvirt lack of attention
        if [ "$count" -eq "$HALF_LOOP" ]
        then
            eval $shutdown_command
        fi

        let count=count+1

        export count

        virsh --connect $LIBVIRT_URI --readonly dominfo $deploy_id > /dev/null 2>&1

        [ "x$?" != "x0" ]
    }

    exec_and_log "$shutdown_command" \
        "Could not shutdown $deploy_id"

    retry $TIMEOUT monitor

    force_shutdown "$deploy_id" \
        "virsh --connect $LIBVIRT_URI destroy $deploy_id"

    #rbt: vgpu
    # Destroy vGPU for VM HA
    $DRIVER_PATH/vgpu_brest "DELETE" "$VM_CONF_DIR/$VM_ID/vm.xml" "$HOST"

    # Compact memory
    if [ "x$CLEANUP_MEMORY_ON_STOP" = "xyes" ]; then
        sudo -n sysctl vm.drop_caches=3 vm.compact_memory=1 &>/dev/null &
    fi

    sleep 4

fi