#!/usr/bin/ruby

# Assumes you have run ./bootstrap to create the /tmp/kernel,
# /tmp/initrd and appliance disk (/tmp/root)

require 'libvirt'
require 'guestfs'

# Create a dummy disk for libguestfs to edit.
File.open("/tmp/disk.img", "w") {
  |f| f.seek(500*1024*1024); f.write("\0")
}

xml = <<EOF
<domain type='kvm'>
  <name>guestfs_test0</name>
  <memory>500000</memory>
  <currentMemory>500000</currentMemory>
  <vcpu>1</vcpu>
  <os>
    <type arch='x86_64'>hvm</type>
    <kernel>/tmp/kernel</kernel>
    <initrd>/tmp/initrd</initrd>
    <cmdline>panic=1 console=ttyS0 udevtimeout=300 no_timer_check
             acpi=off printk.time=1 cgroup_disable=memory selinux=0
             guestfs_verbose=1 TERM=xterm</cmdline>
  </os>
  <features>
    <apic/>
    <pae/>
  </features>

  <clock offset='utc'/>

  <on_poweroff>destroy</on_poweroff>
  <on_reboot>destroy</on_reboot>
  <on_crash>destroy</on_crash>

  <devices>
    <!-- this is the disk we are editing -->
    <disk type='file' device='disk'>
      <driver name='qemu' type='raw'/>
      <source file='/tmp/disk.img'/>
      <target dev='vda' bus='virtio'/>
    </disk>

    <!-- the appliance disk, must be added last -->
    <disk type='file' device='disk'>
      <driver name='qemu' type='raw'/>
      <source file='/tmp/root'/>
      <target dev='vdb' bus='virtio'/>
    </disk>

    <!-- virtio serial to talk to the guest -->
    <channel type='unix'>
      <source mode='bind' path='/tmp/socket'/>
      <target type='virtio' name='org.libguestfs.channel.0'/>
    </channel>

    <!-- XXX console? -->
    <console type='pty'>
      <target port='0'/>
    </console>

  </devices>
</domain>
EOF

begin
  File.delete("/tmp/socket")
rescue Exception
end

# Run the guest, locally for now.
conn = Libvirt::open('qemu:///session')
# 2 == autodestroy, so the domain is killed when conn is closed
dom = conn.create_domain_xml(xml, 2)

# Run libguestfs and try to connect to the socket.
g = Guestfs::Guestfs.new()
g.set_attach_method("unix:/tmp/socket")
g.set_trace(1)
g.launch()

# Send a message to the daemon to check we're talking to it OK.
r = g.echo_daemon(["one", "two", "three"])
if r != "one two three"
  raise "bad response from daemon"
end

# Do some disk operations on /tmp/disk.img.
g.part_disk("/dev/sda", "mbr")
g.mkfs("ext3", "/dev/sda1")
g.mount_options("", "/dev/sda1", "/")
g.write("/hello_world.txt", "hello from libvirt")

# Close everything.  Transient guest should go away.
g.close()
conn.close()

puts "create test run OK"
puts "/tmp/disk.img should be a complete disk image"