blob: 053f5267ecb55bb5462624fc5b42aeec607a3d19 (
plain)
1
2
3
4
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
require 'fileutils'
module OpenShift
module Ops
class AwsHelper
MYDIR = File.expand_path(File.dirname(__FILE__))
def self.get_list()
cmd = "#{MYDIR}/../inventory/aws/ec2.py --list"
hosts = %x[#{cmd} 2>&1]
raise "Error: failed to list hosts\n#{hosts}" unless $?.exitstatus == 0
return JSON.parse(hosts)
end
def self.get_hosts()
hosts = get_list()
retval = []
hosts['_meta']['hostvars'].each do |host, info|
retval << OpenStruct.new({
:name => info['ec2_tag_Name'],
:env => info['ec2_tag_environment'] || 'UNSET',
:public_ip => info['ec2_ip_address'],
:public_dns => info['ec2_public_dns_name'],
:state => info['ec2_state'],
:created_by => info['ec2_tag_created-by']
})
end
retval.sort_by! { |h| [h.env, h.state, h.name] }
return retval
end
def self.get_host_details(host)
hosts = get_list()
dns_names = hosts["tag_Name_#{host}"]
raise "Error: host not found [#{host}]" if dns_names.nil?
return hosts['_meta']['hostvars'][dns_names.first]
end
def self.check_creds()
raise "AWS_ACCESS_KEY_ID environment variable must be set" if ENV['AWS_ACCESS_KEY_ID'].nil?
raise "AWS_SECRET_ACCESS_KEY environment variable must be set" if ENV['AWS_SECRET_ACCESS_KEY'].nil?
end
def self.clear_inventory_cache()
path = "#{ENV['HOME']}/.ansible/tmp"
cache_files = ["#{path}/ansible-ec2.cache", "#{path}/ansible-ec2.index"]
FileUtils.rm(cache_files)
end
def self.generate_env_tag(env)
return { "environment" => env }
end
def self.generate_env_tag_name(env)
h = generate_env_tag(env)
return "tag_#{h.keys.first}_#{h.values.first}"
end
def self.generate_host_type_tag(host_type)
return { "host-type" => host_type }
end
def self.generate_host_type_tag_name(host_type)
h = generate_host_type_tag(host_type)
return "tag_#{h.keys.first}_#{h.values.first}"
end
def self.generate_env_host_type_tag(env, host_type)
return { "env-host-type" => "#{env}-#{host_type}" }
end
def self.generate_env_host_type_tag_name(env, host_type)
h = generate_env_host_type_tag(env, host_type)
return "tag_#{h.keys.first}_#{h.values.first}"
end
end
end
end
|
