Class: Bosh::AwsCliPlugin::EC2

Inherits:
Object
  • Object
show all
Defined in:
lib/bosh_cli_plugin_aws/ec2.rb

Constant Summary collapse

NAT_AMI_ID = 
{
  'us-east-1' => 'ami-f619c29f',      # ami-vpc-nat-1.1.0-beta
  'us-west-1' => 'ami-3bcc9e7e',      # ami-vpc-nat-1.0.0-beta
  'us-west-2' => 'ami-52ff7262',      # ami-vpc-nat-1.0.0-beta
  'eu-west-1' => 'ami-e5e2d991',      # ami-vpc-nat-1.1.0-beta
  'ap-southeast-1' => 'ami-02eb9350', # ami-vpc-nat-1.0.0-beta
  'ap-northeast-1' => 'ami-14d86d15', # ami-vpc-nat-1.0.0-beta
  'ap-southeast-2' => 'ami-ab990e91', # ami-vpc-nat-1.0.0-beta
  'sa-east-1' => 'ami-0039e61d',      # ami-vpc-nat-1.0.0-beta
}

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(credentials) ⇒ EC2

Returns a new instance of EC2.



18
19
20
21
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 18

def initialize(credentials)
  @aws_provider = AwsProvider.new(credentials)
  @elastic_ips = []
end

Instance Attribute Details

#elastic_ips ⇒ Object (readonly)

Returns the value of attribute elastic_ips.



16
17
18
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 16

def elastic_ips
  @elastic_ips
end

Instance Method Details

#add_key_pair(name, path_to_public_private_key) ⇒ Object 



159
160
161
162
163
164
165
166
167
168
169
170
171
172
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 159

def add_key_pair(name, path_to_public_private_key)
  private_key_path = path_to_public_private_key.gsub(/\.pub$/, '')
  public_key_path = "#{private_key_path}.pub"

  if !File.exist?(private_key_path)
    system "ssh-keygen", "-q", '-N', "", "-t", "rsa", "-f", private_key_path
  end

  unless key_pair_by_name(name).nil?
    err "Key pair #{name} already exists on AWS"
  end

  aws_ec2.key_pairs.import(name, File.read(public_key_path))
end

#allocate_elastic_ip ⇒ Object 



42
43
44
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 42

def allocate_elastic_ip
  aws_ec2.elastic_ips.allocate(vpc: true)
end

#allocate_elastic_ips(count) ⇒ Object 



35
36
37
38
39
40
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 35

def allocate_elastic_ips(count)
  count.times do
    @elastic_ips << allocate_elastic_ip.public_ip
  end
  #say "\tallocated #{eip.public_ip}".make_green
end

#create_instance(options) ⇒ Object 



70
71
72
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 70

def create_instance(options)
  aws_ec2.instances.create(options)
end

#create_internet_gateway ⇒ Object 



54
55
56
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 54

def create_internet_gateway
  aws_ec2.internet_gateways.create
end

#create_nat_instance(options) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 74

def create_nat_instance(options)
  name = options["name"]
  key_pair = select_key_pair_for_instance(name, options["key_name"])


  instance_options = {
      image_id: NAT_AMI_ID[aws_provider.region],
      instance_type: options.fetch("instance_type", "m1.medium"),
      subnet: options["subnet_id"],
      private_ip_address: options["ip"],
      security_groups: [options["security_group"]],
      key_name: key_pair
  }

  create_instance(instance_options).tap do |instance|
    Bosh::AwsCloud::ResourceWait.for_instance(instance: instance, state: :running)

    instance.add_tag("Name", {value: name})

    elastic_ip = allocate_elastic_ip

    ignorable_errors = [
      AWS::EC2::Errors::InvalidAddress::NotFound,
      AWS::EC2::Errors::InvalidAllocationID::NotFound,
    ]

    Bosh::Common.retryable(tries: 30, on: ignorable_errors) do
      instance.associate_elastic_ip(elastic_ip)
      true
    end

    disable_src_dest_checking(instance.id)
  end
end

#delete_all_security_groups ⇒ Object 



203
204
205
206
207
208
209
210
211
212
213
214
215
216
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 203

def delete_all_security_groups
  dsg = deletable_security_groups

  # Revoke all permissions before deleting because a permission can reference
  # another security group, causing a delete to fail
  dsg.each do |sg|
    sg.ingress_ip_permissions.map(&:revoke)
    sg.egress_ip_permissions.map(&:revoke)
  end

  dsg.each do |sg|
    sg.delete unless (sg.name == "default" && !sg.vpc_id)
  end
end

#delete_internet_gateways(ids) ⇒ Object 



62
63
64
65
66
67
68
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 62

def delete_internet_gateways(ids)
  Array(ids).each do |id|
    gw = aws_ec2.internet_gateways[id]
    gw.attachments.map(&:delete)
    gw.delete
  end
end

#delete_volumes ⇒ Object 



145
146
147
148
149
150
151
152
153
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 145

def delete_volumes
  unattached_volumes.each do |vol|
    begin
      vol.delete
    rescue AWS::EC2::Errors::InvalidVolume::NotFound
      # ignored
    end
  end
end

#dhcp_options ⇒ Object 



31
32
33
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 31

def dhcp_options
  aws_ec2.dhcp_options
end

#disable_src_dest_checking(instance_id) ⇒ Object 



109
110
111
112
113
114
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 109

def disable_src_dest_checking(instance_id)
  aws_ec2.client.modify_instance_attribute(
      :instance_id => instance_id,
      :source_dest_check => {:value => false}
  )
end

#force_add_key_pair(name, path_to_public_private_key) ⇒ Object 



182
183
184
185
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 182

def force_add_key_pair(name, path_to_public_private_key)
  remove_key_pair(name)
  add_key_pair(name, path_to_public_private_key)
end

#get_running_instance_by_name(name) ⇒ Object 



132
133
134
135
136
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 132

def get_running_instance_by_name(name)
  instances = aws_ec2.instances.select { |instance| instance.tags["Name"] == name && instance.status == :running }
  raise "More than one running instance with name '#{name}'." if instances.count > 1
  instances.first
end

#instance_names ⇒ Object 



125
126
127
128
129
130
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 125

def instance_names
  terminatable_instances.inject({}) do |memo, instance|
    memo[instance.instance_id] = instance.tags["Name"] || '<unnamed instance>'
    memo
  end
end

#instances_count ⇒ Object 



23
24
25
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 23

def instances_count
  terminatable_instances.size
end

#internet_gateway_ids ⇒ Object 



58
59
60
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 58

def internet_gateway_ids
  aws_ec2.internet_gateways.map(&:id)
end

#key_pair_by_name(name) ⇒ Object 



174
175
176
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 174

def key_pair_by_name(name)
  key_pairs.detect { |kp| kp.name == name }
end

#key_pairs ⇒ Object 



178
179
180
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 178

def key_pairs
  aws_ec2.key_pairs.to_a
end

#release_all_elastic_ips ⇒ Object 



50
51
52
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 50

def release_all_elastic_ips
  releasable_elastic_ips.map(&:release)
end

#release_elastic_ips(ips) ⇒ Object 



46
47
48
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 46

def release_elastic_ips(ips)
  aws_ec2.elastic_ips.each { |ip| ip.release if ips.include? ip.public_ip }
end

#remove_all_key_pairs ⇒ Object 



195
196
197
198
199
200
201
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 195

def remove_all_key_pairs
  aws_ec2.key_pairs.each(&:delete)

  Bosh::Common.retryable(tries: 10) do
    aws_ec2.key_pairs.to_a.empty?
  end
end

#remove_key_pair(name) ⇒ Object 



187
188
189
190
191
192
193
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 187

def remove_key_pair(name)
  key_pair = key_pair_by_name(name)
  key_pair.delete unless key_pair.nil?
  Bosh::Common.retryable(tries: 15) do
    key_pair_by_name(name).nil?
  end
end

#terminatable_instance_names ⇒ Object 



138
139
140
141
142
143
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 138

def terminatable_instance_names
  terminatable_instances.inject({}) do |memo, instance|
    memo[instance.instance_id] = instance.tags["Name"]
    memo
  end
end

#terminate_instances ⇒ Object 



116
117
118
119
120
121
122
123
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 116

def terminate_instances
  terminatable_instances.each(&:terminate)
  1.upto(100) do
    break if terminatable_instances.empty?
    sleep 4
  end
  terminatable_instances.empty?
end

#volume_count ⇒ Object 



155
156
157
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 155

def volume_count
  unattached_volumes.count
end

#vpcs ⇒ Object 



27
28
29
 
# File 'lib/bosh_cli_plugin_aws/ec2.rb', line 27

def vpcs
  aws_ec2.vpcs
end