#!/bin/bash

VEID=$1



# test DRBD
if grep -q 'UpToDate/UpToDate' /proc/drbd; then
  echo
  echo "DRBD is OK.  Continuing..."
  echo
else
  echo
  cat /proc/drbd
  echo
  echo "*** DRBD is NOT OK.  Quitting..."
  echo
  exit
fi



# test VEID
RUNNING=`vzctl status $VEID | awk '{print $5}'`
if [ "$RUNNING" != "running" ]; then
  echo "*** $VEID is not running or does not exist.  Online migration is pointless in this situation."
  echo
  exit 1
fi



# freeze and dump
echo "REMOVING OLD DUMPS (if any)..."
rm -f "/var/lib/vz/dump/$VEID"

echo "LOCALLY FREEZING $VEID..."
if ! vzctl chkpnt $VEID --suspend; then
  echo "*** FAILED to suspend $VEID."
  exit
fi

echo "LOCALLY DUMPING $VEID..."
if ! vzctl chkpnt $VEID --dump --dumpfile /var/lib/vz/dump/$VEID; then
  echo "*** FAILED to dump $VEID.  Resuming $VEID on current HN."
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  exit
fi



# stop and umount
echo "LOCALLY STOPPING $VEID..."
if ! vzctl chkpnt $VEID --kill; then
  echo "*** FAILED to kill $VEID on current HN.  Resuming $VEID on current HN."
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  exit
fi

echo "LOCALLY UMOUNTING $VEID..."
if ! vzctl umount $VEID; then
  echo "*** FAILED to umount $VEID on current HN.  Resuming $VEID on current HN."
  if ! vzctl chkpnt $VEID --undump --dumpfile /var/lib/vz/dump/$VEID; then
    echo "*** FAILED to undump $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
fi



# undump and thaw
echo "REMOTELY ADDING CONF FOR $VEID..."
if ! ssh root@vzpeer ln -s /var/lib/vz/conf/$VEID.* /etc/vz/conf/; then
  echo "*** FAILED adding $VEID conf files to vzpeer.  Resuming $VEID on current HN."
  if ! vzctl chkpnt $VEID --undump --dumpfile /var/lib/vz/dump/$VEID; then
    echo "*** FAILED to undump $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  exit
fi

echo "REMOTELY UNDUMPING $VEID..."
if ! ssh root@vzpeer vzctl restore $VEID --undump --dumpfile /var/lib/vz/dump/$VEID; then
  echo "*** FAILED undumping $VEID on vzpeer.  Removing $VEID conf on vzpeer and resuming $VEID on current HN."
  ssh root@vzpeer rm -f /etc/vz/conf/$VEID.*
  if ! vzctl chkpnt $VEID --undump --dumpfile /var/lib/vz/dump/$VEID; then
    echo "*** FAILED to undump $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  exit
fi

echo "REMOTELY THAWING $VEID..."
if ! ssh root@vzpeer vzctl restore $VEID --resume; then
  echo "*** FAILED to resume $VEID on vzpeer.  Removing $VEID conf on vzpeer and resuming $VEID on current HN."
  ssh root@vzpeer rm -f /etc/vz/conf/$VEID.*
  if ! vzctl chkpnt $VEID --undump --dumpfile /var/lib/vz/dump/$VEID; then
    echo "*** FAILED to undump $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  if ! vzctl chkpnt $VEID --resume; then
    echo "*** FAILED to resume $VEID on current HN.  I'll leave it to you to figure it out."
    exit
  fi
  exit
fi



# cleanup
echo "REMOVING DUMP..."
if ! rm -f "/var/lib/vz/dump/$VEID"; then
  echo "*** FAILED to remove $VEID dump file.  Maybe there isn't a problem.  I'll let you decide."
fi

echo "REMOVING LOCAL CONF FOR $VEID..."
if ! rm -f /etc/vz/conf/$VEID.*; then
  echo "*** FAILED to remove $VEID conf symlinks on current HN.  If they still exist, you might have the VE running on two machines at once on reboot.  I'll let you figure it out."
fi



# done
echo "All done!"