import to github

Signed-off-by: Ava Affine <ava@sunnypup.io>

virt.sh

@@ -0,0 +1,209 @@
+#!/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() {
+	ret=$(sudo virsh list --all --name | sort)
+}
+
+function vms_on() {
+	ret=$(sudo virsh list --name | sort)
+}
+
+function vms_off() {
+	ret=$(sudo virsh list --inactive --name | sort)
+}
+
+function get_vm_ip() {
+	vms_avail
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 doesnt exist"
+		ret=""
+		return 1
+	fi
+
+	vms_on
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 already off"
+		ret=""
+		return 1
+	fi
+
+	ret=$(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"
+		ret=""
+		return 1
+	fi
+
+	vms_avail
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 doesnt exist"
+		ret=""
+		return 1
+	fi
+
+	vms_off
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 already on"
+		ret=""
+		return 0
+	fi
+
+	sudo virsh start $1 >/dev/null
+	log "Started VM $1. Please standby"
+
+
+	# wait for an IP
+	ret=""
+	while [[ "$ret" == "" ]]; do
+		get_vm_ip $1
+		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@$ret \
+			exit; do
+				sleep 0.1
+			done
+	log "Got SSH on VM $1"
+}
+
+function vm_shell() {
+	vms_avail
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 doesnt exist"
+		ret=""
+		return 1
+	fi
+
+	vms_on
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 is off"
+		ret=""
+		return 1
+	fi
+
+	if [[ "$2" == "" ]]; then
+		log "wont execute empty command"
+		ret=""
+		return 1
+	fi
+
+	# wait for an IP
+	ret=""
+	while [[ "$ret" == "" ]]; do
+		get_vm_ip $1
+		sleep 0.5
+	done
+
+	# wait for successful ssh
+	# ret set by get_vm_ip above
+	sshpass -p $PASSP \
+		ssh -o PreferredAuthentications=password \
+			-o StrictHostKeyChecking=no \
+			$USERN@$ret "$2"
+	return $?
+}
+
+function turn_off_vm() {
+	if [[ ! $1 ]]; then
+		log "no VM specified"
+		ret=""
+		return 1
+	fi
+
+	vms_avail
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 doesnt exist"
+		ret=""
+		return 1
+	fi
+
+	vms_on
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 already off"
+		ret=""
+		return 1
+	fi
+
+	ret=$(sudo virsh shutdown $1)
+}
+
+function sync_dir_to_vm(){
+	if [ ! -d $2 ]; then
+		log "directory $2 does not exist."
+		ret=""
+		return 1
+	fi
+
+	vms_avail
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 doesnt exist"
+		ret=""
+		return 1
+	fi
+
+	vms_on
+	if ! [[ $ret =~ $1 ]]; then
+		log "VM $1 is off"
+		ret=""
+		return 1
+	fi
+
+	# wait for an IP
+	ret=""
+	while [[ "$ret" == "" ]]; do
+		get_vm_ip $1
+		log "waiting for ip"
+		sleep 0.5
+	done
+
+	sshpass -p $PASSP \
+		rsync -avze "ssh -o PreferredAuthentications=password -o StrictHostKeyChecking=no" \
+		$2 $USERN@$ret:
+	return $?
+}
+
+# boots vm, enters SSH, shuts down VM
+function vsh() {
+	# takes care of error cases
+	if ! turn_on_vm_and_wait $1; then
+		exit 1
+	fi
+
+	get_vm_ip $1
+	sshpass -p $PASSP \
+		ssh -o PreferredAuthentications=password \
+			-o StrictHostKeyChecking=no \
+			$USERN@$ret
+
+	turn_off_vm $1
+}