summaryrefslogtreecommitdiffstats
path: root/lib/aws_helper.rb
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