# -*- mode: python; -*-
import sipb_xen_database.models as models
from sipb_xen_database import connect
import re
import tempfile
from subprocess import call

connect('postgres://sipb-xen@sipb-xen-dev.mit.edu/sipb_xen')
prefix = "d_"

# 'machine_name', and optionally 'cdrom_image', should get passed in
# from the xm create call

def check(b):
    if not b:
        import sys
        sys.exit(1)

machine = models.Machine.get_by(name=machine_name)
check(machine is not None)
machine_type = models.Type.get_by(type_id=machine.type_id)

memory = machine.memory
maxmem = memory
check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name))
name = prefix + machine.name
check(re.match('^[0-9a-f-]+$', machine.uuid))
uuid = machine.uuid

vcpus = machine.cpus

diskioemu = ""
viftype = ""

pae = machine_type.pae
acpi = machine_type.acpi
apic = machine_type.apic

vif = []

disk = []

if machine_type.hvm:
    codepath = 'hvm'
else:
    codepath = 'paravm'

if 'mirror' in locals(): #Installer
    disk.append('phy:/dev/xenvg/d_foobar3_hda,hdb,r')

    kernel = '/boot/vmlinuz-2.6.24-19-xen' #From hardy
    ramdisk = '/boot/initrd.img-2.6.24-19-xen'

    if not machine.nics:
        raise RuntimeError('You must have a nic to autoinstall')
    n = machine.nics[0]
    extra = 'ro noresume'
    extra += ' ip=%s::18.181.0.1:255.255.0.0:%s:eth0:off' % (n.ip, machine.name)
    extra += ' mirror=%s dist=%s' % (mirror, dist)
    extra += ' imagesize=%s' % imagesize
    if 'noinstall' in locals():
        extra += ' noinstall'
    root = '/dev/hdb1'
    codepath = None
elif 'cdrom_image' in locals():
    check(re.match('^[A-Za-z0-9][A-Za-z0-9_.-]*$', cdrom_image))
    disk.append('file:/srv/images/' + cdrom_image + '.iso,hdc:cdrom,r')
    boot = 'd'
    codepath = 'hvm'

if codepath == 'hvm':
    ioemu = "ioemu:"
    viftype = "type=ioemu, "
    kernel = 'hvmloader'
    builder = 'hvm'
    vnc = 1
    vncpasswd = 'moocow'
    device_model = '/usr/sbin/qemu-dm-sipb'
    serial = "pty"
elif codepath == 'paravm':
    bootloader = '/usr/lib/xen-default/bin/pygrub'


for n in machine.nics:
    check(re.match('^[0-9a-fA-F:]+$', n.mac_addr) and re.match('^[0-9.]*$', n.ip))
    d = '%smac=%s, ip=%s, bridge=xenbr0, script=vif-sipbroute' % (viftype, n.mac_addr, n.ip)
    vif.append(d)

for d in machine.disks:
    check(re.match('^[A-Za-z0-9]+$', d.guest_device_name))
    device = '/dev/xenvg/' + prefix + machine.name + '_' + d.guest_device_name
    dspec = 'phy:%s,%s%s,w' % (device, diskioemu, d.guest_device_name)
    disk.append(dspec)

usbdevice = 'tablet'

on_poweroff = 'destroy'
on_reboot = 'restart'
on_crash = 'destroy'
if machine.autorestart:
    on_crash = 'restart'