class Beaker::Aixer

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

def initialize(aix_hosts, options)
  @options = options
  @logger = options[:logger]
  @hosts = aix_hosts
  #aix machines are reverted to known state, not a snapshot
  @fog_file = nil
  if File.exists?( @options[:dot_fog] )
    @fog_file = YAML.load_file( @options[:dot_fog] )
  end
  raise "Cant load #{@options[:dot_fog]} config" unless @fog_file
end

def provision
  # Running the rake task on rpm-builder
  hypername = @fog_file[:default][:aix_hypervisor_server]
  hyperopts = @options.dup
  hyperopts['HOSTS'] = {
      hypername => { 'platform' => 'el-6-x86_64' }
  }
  @logger.notify "Connecting to hypervisor at #{hypername}"
  hypervisor = Beaker::Host.create( hypername, hyperopts, @options )
  hypervisor[:user] = @fog_file[:default][:aix_hypervisor_username] || hypervisor[:user]
  hypervisor[:ssh][:keys] = [@fog_file[:default][:aix_hypervisor_keyfile]] || hypervisor[:ssh][:keys]
  @hosts.each do |host|
    vm_name = host['vmname'] || host.name
    @logger.notify "Reverting #{vm_name} to aix clean state"
    start = Time.now
    # Restore AIX image, ID'd by the hostname
    hypervisor.exec(Command.new("cd pe-aix && rake restore:#{host.name}"))
    time = Time.now - start
    @logger.notify "Spent %.2f seconds reverting" % time
  end
  hypervisor.close
end