Class: AmiSpec::WaitForRC

Inherits:
Object
  • Object
show all
Defined in:
lib/ami_spec/wait_for_rc.rb

Class Method Summary collapse

Class Method Details

.wait(ip_address, user, key, port = 22) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/ami_spec/wait_for_rc.rb', line 5

def self.wait(ip_address, user, key, port=22)
  Net::SSH.start(ip_address, user, keys: [key], :verify_host_key => :never, port: port) do |ssh|
    distrib_stdout = ""
    # Determine the OS family
    ssh.exec!("source /etc/*release && echo -n $DISTRIB_ID && echo -n $ID") do |channel, stream, data|
      distrib_stdout << data if stream == :stdout
    end
    if distrib_stdout == "Ubuntu"
      codename_stdout = ""
      ssh.exec!("source /etc/*release && echo -n $DISTRIB_CODENAME") do |channel, stream, data|
        codename_stdout << data if stream == :stdout
      end
      if codename_stdout == "trusty"
        ssh.exec 'while /usr/sbin/service rc status | grep -q "^rc start/running, process"; do sleep 1; done'
      elsif codename_stdout == "xenial"
        ssh.exec 'while /usr/sbin/service rc status >/dev/null; do sleep 1; done'
      else
        puts "WARNING: Only Ubuntu trusty and xenial supported and we detected '#{codename_stdout}'. --wait-for-rc has no effect."
      end
    elsif distrib_stdout == "amzn"
      version_stdout = ""
      ssh.exec!("source /etc/*release && echo -n $VERSION_ID") do |channel, stream, data|
        version_stdout << data if stream == :stdout
      end
      if version_stdout =~ %r{201[0-9]{1}\.[0-9]+}
        ssh.exec 'while initctl status rc |grep -q "^rc start/running"; do sleep 1; done'
      else
        puts "WARNING: Only Amazon Linux 1 is supported and we detected '#{version_stdout}'. --wait-for-rc has no effect."
      end
    else
      puts "WARNING: Only Ubuntu and Amazon linux are supported and we detected '#{distrib_stdout}'. --wait-for-rc has no effect."
    end
  end
end