module Beaker 
  class Fusion < Beaker::Hypervisor

    def initialize(fusion_hosts, options)
      require 'rubygems' unless defined?(Gem)
      begin
        require 'fission'
      rescue LoadError
        raise "Unable to load fission, please ensure it is installed!"
      end
      @logger = options[:logger]
      @options = options
      @fusion_hosts = fusion_hosts
      #check preconditions for fusion
      @fusion_hosts.each do |host|
        raise "You must specify a snapshot for Fusion instances, no snapshot defined for #{host.name}!" unless host["snapshot"]
      end
      @fission = Fission::VM
    end

    def provision
      available = @fission.all.data.collect{|vm| vm.name}.sort.join(", ")
      @logger.notify "Available VM names: #{available}"

      @fusion_hosts.each do |host|
        vm_name = host["vmname"] || host.name
        vm = @fission.new vm_name
        raise "Could not find VM '#{vm_name}' for #{host.name}!" unless vm.exists?

        vm_snapshots = vm.snapshots.data
        if vm_snapshots.nil? or vm_snapshots.empty?
          raise "No snapshots available for VM #{host.name} (vmname: '#{vm_name}')" 
        end

        available_snapshots = vm_snapshots.sort.join(", ")
        @logger.notify "Available snapshots for #{host.name}: #{available_snapshots}"
        snap_name = host["snapshot"] 
        raise "Could not find snapshot '#{snap_name}' for host #{host.name}!" unless vm.snapshots.data.include? snap_name

        @logger.notify "Reverting #{host.name} to snapshot '#{snap_name}'"
        start = Time.now
        vm.revert_to_snapshot snap_name
        while vm.running?.data
          sleep 1
        end
        time = Time.now - start
        @logger.notify "Spent %.2f seconds reverting" % time

        @logger.notify "Resuming #{host.name}"
        start = Time.now
        vm.start :headless => true
        until vm.running?.data
          sleep 1
        end
        time = Time.now - start
        @logger.notify "Spent %.2f seconds resuming VM" % time
      end
      end #revert_fusion

      def cleanup
        @logger.notify "No cleanup for fusion boxes"
      end

  end
end