#!/bin/bash

count=$1
VMs=$(sudo virsh list --state-running --name | sort -R)

sudo aa-teardown
for vm in $VMs
do
    path_vda=$(sudo virsh domblklist $vm | grep vda | cut -d" " -f 8)
    path_vdb=$(sudo virsh domblklist $vm | grep vdb | cut -d" " -f 8)

    if [ ! -z "$path_vda" ] && [[ ! "$path_vda" =~ ".raw" ]]
    then
        echo "-- BLOCKCOMMITTING $vm disk vda ..."
        retour=$(sudo virsh blockcommit $vm vda --active --verbose --pivot --delete 2>&1 | tee /dev/tty)
        if [[ "$retour" =~ 'block copy still active' ]]
        then
            echo "-- Finishing blockjob for $vm ..."
            sudo virsh blockjob $vm $path_vda --pivot --info
            echo "-- BLOCKCOMMITTING $vm ..."
            sudo virsh blockcommit $vm vda --active --verbose --pivot --delete 2>&1
        fi
    fi

    if [ ! -z "$path_vdb" ] && [[ ! "$path_vda" =~ ".raw" ]]
    then
        echo "-- BLOCKCOMMITTING $vm disk vdb ..."
        retour=$(sudo virsh blockcommit $vm vdb --active --verbose --pivot --delete 2>&1 | tee /dev/tty)

        if [[ "$retour" =~ 'block copy still active' ]]
        then
            echo "-- Finishing blockjob for $vm ..."
            sudo virsh blockjob $vm $path_vdb --pivot --info
            echo "-- BLOCKCOMMITTING $vm ..."
            sudo virsh blockcommit $vm vda --active --verbose --pivot --delete 2>&1
        fi
    fi
done
sudo systemctl restart apparmor.service