Class: VagrantPlugins::AWS::Action::ReadSSHInfo

Inherits:
Object
  • Object
show all
Defined in:
lib/vagrant-aws/action/read_ssh_info.rb

Overview

This action reads the SSH info for the machine and puts it into the ‘:machine_ssh_info` key in the environment.

Instance Method Summary collapse

Constructor Details

#initialize(app, env) ⇒ ReadSSHInfo

Returns a new instance of ReadSSHInfo.



9
10
11
12
# File 'lib/vagrant-aws/action/read_ssh_info.rb', line 9

def initialize(app, env)
  @app    = app
  @logger = Log4r::Logger.new("vagrant_aws::action::read_ssh_info")
end

Instance Method Details

#call(env) ⇒ Object



14
15
16
17
18
# File 'lib/vagrant-aws/action/read_ssh_info.rb', line 14

def call(env)
  env[:machine_ssh_info] = read_ssh_info(env[:aws_compute], env[:machine])

  @app.call(env)
end

#read_ssh_info(aws, machine) ⇒ Object



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# File 'lib/vagrant-aws/action/read_ssh_info.rb', line 20

def read_ssh_info(aws, machine)
  return nil if machine.id.nil?

  # Find the machine
  server = aws.servers.get(machine.id)
  if server.nil?
    # The machine can't be found
    @logger.info("Machine couldn't be found, assuming it got destroyed.")
    machine.id = nil
    return nil
  end

  # read attribute override
  ssh_host_attribute = machine.provider_config.
      get_region_config(machine.provider_config.region).ssh_host_attribute
  # default host attributes to try. NOTE: Order matters!
  ssh_attrs = [:dns_name, :public_ip_address, :private_ip_address]
  ssh_attrs = (Array(ssh_host_attribute) + ssh_attrs).uniq if ssh_host_attribute
  # try each attribute, get out on first value
  host_value = nil
  while !host_value and attr_name = ssh_attrs.shift
    begin
      host_value = server.send(attr_name)
    rescue NoMethodError
      @logger.info("SSH host attribute not found #{attr_name}")
    end
  end

  return { :host => host_value, :port => 22 }
end