#!/bin/sh
# ESXi 5.1 host automated shutdown script

# these are the VM IDs to shutdown in the order specified
# use the SSH shell, run "vim-cmd vmsvc/getallvms" to get ID numbers
# specify IDs separated by a space
SERVERIDS=$(vim-cmd vmsvc/getallvms | sed -e '1d' -e 's/ \[.*$//' | awk '$1 ~ /^[0-9]+$/ {print $1}')

# New variable to allow script testing, assuming the vim commands all work to issue shutdowns
# can be "0" or "1"
TEST=0

# script waits WAIT_TRYS times, WAIT_TIME seconds each time
# number of times to wait for a VM to shutdown cleanly before forcing power off.
WAIT_TRYS=20

# how long to wait in seconds each time for a VM to shutdown.
WAIT_TIME=10

# ------ DON'T CHANGE BELOW THIS LINE ------

validate_shutdown(){
    vim-cmd vmsvc/power.getstate $SRVID | grep -i "off\|Suspended" > /dev/null 2<&1
    STATUS=$?

    if [ $STATUS -ne 0 ]; then
        if [ $TRY -lt $WAIT_TRYS ]; then
            # if the vm is not off, wait for it to shut down
            TRY=$((TRY + 1))
            echo "Waiting for guest VM ID $SRVID to suspend (attempt #$TRY)..."
            sleep $WAIT_TIME
            validate_shutdown
        else
            # shutdown and wait a little (you could use vmsvc/power.suspend here instead)
            echo "Unable to suspend guest VM ID $SRVID... shutting down instead."
            if [ $TEST -eq 0 ]; then
                vim-cmd vmsvc/power.shutdown $SRVID
            fi
            sleep $WAIT_TIME
        fi
    fi
}

# enter maintenance mode immediately
echo "Entering maintenance mode..."
if [ $TEST -eq 0 ]; then
    esxcli system maintenanceMode set -e true -t 0 &
fi

# read each line as a server ID and suspend/shutdown
for SRVID in $SERVERIDS; do
    TRY=0

    vim-cmd vmsvc/power.getstate $SRVID | grep -i "off\|Suspended" > /dev/null 2<&1
    STATUS=$?

    if [ $STATUS -ne 0 ]; then
        echo "Shutting Down guest VM ID $SRVID..."
        if [ $TEST -eq 0 ]; then
            vim-cmd vmsvc/power.shutdown $SRVID
        fi
        validate_shutdown
    else
        echo "Guest VM ID $SRVID already off..."
    fi
done

# guest vm shutdown complete
echo "Guest VM shutdown complete..."

# shutdown the ESXi host
echo "Shutting down ESXi host after 10 seconds..."
if [ $TEST -eq 0 ]; then
    esxcli system shutdown poweroff -d 10 -r "Power-Monitor script initiated system shutdown"
fi

# exit maintenance mode immediately before server has a chance to shutdown/power off
# NOTE: it is possible for this to fail, leaving the server in maintenance mode on reboot!
echo "Exiting maintenance mode..."
if [ $TEST -eq 0 ]; then
    esxcli system maintenanceMode set -e false -t 0
fi

# exit the session
exit