class VagrantPlugins::OVirtProvider::Action::WaitTillUp

# File lib/vagrant-ovirt4/action/wait_till_up.rb, line 16
def initialize(app, env)
  @logger = Log4r::Logger.new("vagrant_ovirt4::action::wait_till_up")
  @app = app
end

# File lib/vagrant-ovirt4/action/wait_till_up.rb, line 38
def call(env)
  # Initialize metrics if they haven't been
  env[:metrics] ||= {}

  # Get config.
  config = env[:machine].provider_config

  # Wait for VM to obtain an ip address.
  env[:metrics]["instance_ip_time"] = Util::Timer.time do
    env[:ui].info(I18n.t("vagrant_ovirt4.waiting_for_ip"))
    for i in 1..300
      # If we're interrupted don't worry about waiting
      next if env[:interrupted]

      # Get VM.
      server = env[:vms_service].vm_service(env[:machine].id)
      if server == nil
        raise Errors::NoVMError, :vm_id => env[:machine].id
      end

      nics_service = server.nics_service
      nics = nics_service.list
      ip_addr = nics.collect { |nic_attachment| env[:connection].follow_link(nic_attachment).reported_devices.collect { |dev| dev.ips.collect { |ip| ip.address if ip.version == 'v4' } unless dev.ips.nil? } }.flatten.reject { |ip| ip.nil? }.first rescue nil
      unless ip_addr.nil?
        env[:ui].info("Got IP: #{ip_addr}")
        # Check if SSH-Server is up
        if port_open?(ip_addr, 22)
          env[:ip_address] = ip_addr
          break
          @logger.debug("Got output #{env[:ip_address]}")
        end
      end
      sleep 5
    end
  end
  terminate(env) if env[:interrupted]
  if env[:ip_address].nil?
    raise Errors::NoIPError
  else
    @logger.info("Got IP address #{env[:ip_address]}")
    @logger.info("Time for getting IP: #{env[:metrics]["instance_ip_time"]}")
    
    @logger.info("Time for SSH ready: #{env[:metrics]["instance_ssh_time"]}")

    # Booted and ready for use.
    env[:ui].info(I18n.t("vagrant_ovirt4.ready"))
    
    @app.call(env)
  end
end

# File lib/vagrant-ovirt4/action/wait_till_up.rb, line 21
def port_open?(ip, port, seconds=10)
  # => checks if a port is open or not on a remote host
  Timeout::timeout(seconds) do
    begin
      TCPSocket.new(ip, port).close
      @logger.info("SSH Check OK for IP: #{ip}")
      true
    rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH, SocketError => e
      @logger.info("SSH Connection Failed for IP #{ip}: #{e}")
      false
    end
  end
  rescue Timeout::Error
    @logger.info("SSH Connection Failed: Timeout for IP: #{ip}" )
    false
end

# File lib/vagrant-ovirt4/action/wait_till_up.rb, line 89
def recover(env)
  return if env["vagrant.error"].is_a?(Vagrant::Errors::VagrantError)

  if env[:machine].provider.state.id != :not_created
    # Undo the import
    terminate(env)
  end
end

# File lib/vagrant-ovirt4/action/wait_till_up.rb, line 98
def terminate(env)
  destroy_env = env.dup
  destroy_env.delete(:interrupted)
  destroy_env[:config_validate] = false
  destroy_env[:force_confirm_destroy] = true
  env[:action_runner].run(Action.action_destroy, destroy_env)        
end