#!/bin/bash

dirn=$(dirname "$0")
source $dirn/common.sh

if [[ ! -f $dirn/SECRET.sh ]]; then
	error "need to create SECRET.sh... see Readme"
	exit 1
fi

source $dirn/SECRET.sh

# set in SECRET.sh
if [[ ! $USERN ]]; then
	error "\$USERN not set"
	exit 1
fi

# set in SECRET.sh
if [[ ! $PASSP ]]; then
	error "\$PASSP not set"
	exit 1
fi

function vms_avail() {
	sudo virsh list --all --name | sort
}

function vms_on() {
	sudo virsh list --name | sort
}

function vms_off() {
	sudo virsh list --inactive --name | sort
}

function get_vm_ip() {
	if ! [[ $(vms_avail) =~ $1 ]]; then
		log "VM $1 doesnt exist"
		return 1
	fi

	if ! [[ $(vms_on) =~ $1 ]]; then
		log "VM $1 is off"
		return 1
	fi

	sudo virsh net-dhcp-leases default | grep $1 | awk '{print $5}' | rev | cut -c 4- | rev
}

function turn_on_vm_and_wait() {
	if [[ ! $1 ]]; then
		log "no VM specified"
		return 1
	fi

	if ! [[ $(vms_avail) =~ $1 ]]; then
		log "VM $1 doesnt exist"
		return 1
	fi

	if ! [[ $(vms_off) =~ $1 ]]; then
		log "VM $1 already on"
		return 0
	fi

	sudo virsh start $1 >/dev/null
	log "Started VM $1. Please standby"


	# wait for an IP
	while ! [[ $(get_vm_ip $1) ]]; do
		sleep 0.5
	done
	log "Got IP for VM $1"

	# wait for successful ssh
	# ret set by get_vm_ip above
	while ! sshpass -p $PASSP \
		ssh -o PreferredAuthentications=password \
			-o StrictHostKeyChecking=no $USERN@$(get_vm_ip $1) \
				exit; do
		sleep 0.1
	done
	log "Got SSH on VM $1"
}

function turn_off_vm() {
	if [[ ! $1 ]]; then
		log "no VM specified"
		return 1
	fi

	if ! [[ $(vms_avail) =~ $1 ]]; then
		log "VM $1 doesnt exist"
		return 1
	fi

	if ! [[ $(vms_on) =~ $1 ]]; then
		log "VM $1 already off"
		return 0
	fi

	sudo virsh shutdown $1
}

function sync_dir_to_vm(){
	if [[ ! -d $2 ]]; then
		log "directory $2 does not exist."
		return 1
	fi

	if ! [[ $(vms_avail) =~ $1 ]]; then
		log "VM $1 doesnt exist"
		return 1
	fi

	if ! [[ $(vms_on) =~ $1 ]]; then
		log "VM $1 is off"
		return 1
	fi

	# wait for an IP
	while [[ ! $(get_vm_ip $1) ]]; do
		sleep 0.5
	done

	sshpass -p $PASSP \
		rsync -avze "ssh -o PreferredAuthentications=password -o StrictHostKeyChecking=no" \
		$2 $USERN@$(get_vm_ip $1):
}

function vm_cmd() {
	if ! [[ $(vms_avail) =~ $1 ]]; then
		log "VM $1 doesnt exist"
		return 1
	fi

	if ! [[ $(vms_on) =~ $1 ]]; then
		log "VM $1 is off"
		return 1
	fi

	if [[ ! $2 ]]; then
		log "wont execute empty command"
		return 1
	fi

	# wait for an IP
	while [[ ! $(get_vm_ip $1) ]]; do
		sleep 0.5
	done

	# wait for successful ssh
	sshpass -p $PASSP \
		ssh -o PreferredAuthentications=password \
			-o StrictHostKeyChecking=no \
			$USERN@$(get_vm_ip $1) "$2"
}