#!/bin/bash

VM_NAME=${VM_NAME:-TEST}
VM_MEMORY_SIZE=${VM_MEMORY_SIZE:-1024}
VM_NET_MAC=${VM_NET_MAC:-52:54:00:c1:0d:ff}
VM_NET_INT=${VM_NET_INT:-eth0}
VN_NET_TYPE=${VM_NET_TYPE:-direct}
LIB_VIRT_VER=`virsh -v`
LIB_VIRT_VER_BASE="0.9"
REDHAT_RELEASE="$( cat /etc/redhat-release | awk '{print $1}' )"

# devices
VM_HW_HDD=${VM_HW_HDD:-ide}
VM_HW_NET=${VM_HW_NET:-rtl8139}

if [ "${REDHAT_RELEASE}" = "Fedora" ]
then VM_EMULATOR="/usr/bin/qemu-kvm"
else VM_EMULATOR="/usr/libexec/qemu-kvm"
fi

#VM_DISK_1="/var/lib/libvirt/images/grid_cloud-win_xp.img"
#VM_DISK_2="/var/lib/libvirt/images/win_xp.img"
#VM_CD_DISK_1="/home/Barracuda/Documents/Work/development/tftp/pxelinux/iso/f16-x86_64-netinst.iso"
#VM_CD_DISK_2="/home/Barracuda/Documents/Work/development/tftp/pxelinux/iso/f16-i386-netinst.iso"

function gen_network_direct {
local iface=$1
local mac=$2
cat << END
<interface type='direct'>
  <mac address='$mac'/>
  <source dev='$iface' mode='vepa'/>
  <model type='rtl8139'/>
  <address type='pci' domain='0x0000' bus='0x00' slot='0x03' function='0x0'/>
</interface>
END
}

function gen_network_bridge {
local iface=$1
local mac=$2
if [ "${REDHAT_RELEASE}" = "Fedora" ]
then cat << END
    <interface type='direct'>
      <mac address='$mac'/>
      <source dev='$iface' mode='bridge'/>
      <model type='$VM_HW_NET'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x03' function='0x0'/>
    </interface>
END
else cat << END
    <interface type='bridge'>
      <mac address='$mac'/>
      <source bridge='$iface'/> 
      <model type='$VM_HW_NET'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x03' function='0x0'/>
    </interface>
END
fi
}


cat << END > vm.xml
<domain type='kvm'>
  <name>$VM_NAME</name>
END
if [ $LIB_VIRT_VER_BASE = `echo -e "$LIB_VIRT_VER_BASE\n$LIB_VIRT_VER" | sort -n | head -n1` ] # if True then LIB_VIRT_VER_BASE smaler then LIB_VIRT_VER
then cat << END >> vm.xml
  <memory unit='MB'>$VM_MEMORY_SIZE</memory>
  <currentMemory unit='MB'>$VM_MEMORY_SIZE</currentMemory>
END
else 
VM_MEMORY_SIZE=$((VM_MEMORY_SIZE*1024))
cat << END >> vm.xml
  <memory>$VM_MEMORY_SIZE</memory>
  <currentMemory>$VM_MEMORY_SIZE</currentMemory>
END
fi

cat << END >> vm.xml
  <vcpu>1</vcpu>
  <os>
    <type arch='i686' machine='pc'>hvm</type>
    <boot dev='hd'/>
  </os>
  <features>
    <acpi/>
    <apic/>
    <pae/>
  </features>
  <clock offset='localtime'/>
  <on_poweroff>destroy</on_poweroff>
  <on_reboot>restart</on_reboot>
  <on_crash>restart</on_crash>
  <devices>
    <emulator>$VM_EMULATOR</emulator>
END

#DANGER!!!! magic begins

printf -v alphabet "%c" {a..z}

BUS="0"
var=""
ind=1
i=1
#++++++++++++++++++++++HDD+++++++++++++++++++++++++++++++++++
eval var=\$VM_DISK_$ind

while [ -n "$var" ]
do
unit_id=$[ind-1]
DEV=`echo hd${alphabet[@]:$unit_id:1}`
cat << END >> vm.xml
    <disk type='file' device='disk'>
      <driver name='qemu' type='raw'/>
      <source file='$var'/>
      <target dev='$DEV' bus='$VM_HW_HDD'/>
      <address type='drive' controller='0' bus='$BUS' target='0' unit='$unit_id'/>
    </disk>
END

ind=$[ind+1]
i=$[i+1]
eval var=\$VM_DISK_$ind

done

#++++++++++++++++++++++CD++++++++++++++++++++++++++++++++++++

BUS=$[BUS+1] # CD use another bus
ind=1 # unit ID and variable name begins from start
var="" # set to default
i=$[i-1] # need shif to use DEV in oder

eval var=\$VM_CD_DISK_$ind

while [ -n "$var" ]

do
unit_id=$[ind-1]
DEV=`echo hd${alphabet[@]:$i:1}`

cat << END >> vm.xml
    <disk type='file' device='cdrom'>
      <driver name='qemu' type='raw'/>
      <source file='$var'/>
      <target dev='$DEV' bus='ide'/>
      <readonly/>
      <address type='drive' controller='0' bus='$BUS' target='0' unit='$unit_id'/>
    </disk>
END

ind=$[ind+1]
i=$[i+1]
eval var=\$VM_CD_DISK_$ind

done

#end of magic

if [ $LIB_VIRT_VER_BASE = `echo -e "$LIB_VIRT_VER_BASE\n$LIB_VIRT_VER" | sort -n | head -n1` ] # if True then LIB_VIRT_VER_BASE smaler then LIB_VIRT_VER
then
cat << END >> vm.xml
    <controller type='usb' index='0'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x01' function='0x2'/>
    </controller>
END
fi
cat << END >> vm.xml
    <controller type='ide' index='0'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x01' function='0x1'/>
    </controller>
END

#
# network
#
if [ "$VN_NET_TYPE" = "direct" ]; then 
    gen_network_direct "$VM_NET_INT" "$VM_NET_MAC" >> vm.xml
elif [  "$VN_NET_TYPE" = "bridge" ]; then
    gen_network_bridge "$VM_NET_INT" "$VM_NET_MAC" >> vm.xml
fi


cat << END >> vm.xml
    <input type='mouse' bus='ps2'/>
    <graphics type='vnc' port='-1' autoport='yes'/>
    <sound model='ac97'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x04' function='0x0'/>
    </sound>
    <video>
      <model type='cirrus' vram='9216' heads='1'/>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x02' function='0x0'/>
    </video>
    <memballoon model='virtio'>
      <address type='pci' domain='0x0000' bus='0x00' slot='0x05' function='0x0'/>
    </memballoon>
  </devices>
</domain>
END

function check_vm_running {
    virsh -r -c qemu:///system list | grep -qe "\s${VM_NAME}\s"
    return $?
}

function destroy_vm {
    virsh -c qemu:///system destroy ${VM_NAME}
}

virsh -c qemu:///system create vm.xml