Module: EC2Launcher::SecurityGroupHandler

Included in:
DSL::Application, DSL::Environment
Defined in:
lib/ec2launcher/security_group_handler.rb

Overview

Helper module for all objects that support EC2 Security Groups.

Instance Method Summary collapse

Instance Method Details

#security_groups(*groups) ⇒ Hash, self

Add or retrieve security groups. Defines the @security_groups instance variable, which contains a Hash of environment names to Arrays of security group names. May define a “default” environment.

Can be defined several different ways:

* String - Adds the named security group to the "default" environment.
* Array - Adds the entire array of security groups to the "default" environment.
* Hash - Keys are environment names (Strings) to security groups. Values of the
         hash can be either a String or an Array. Both are appended to any
         security groups already defined for the environment.

Parameters:

  • groups (Array, nil) —

    Array of security_group definitions. See above. Returns the entire Hash of security groups if empty.

Returns:

  • (Hash, self) —

    Either returns the Hash of security groups (if groups parameter is empty) or returns a reference to self.



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/ec2launcher/security_group_handler.rb', line 24

def security_groups(*groups)
	if groups.empty?
		@security_groups
	else
		@security_groups = Hash.new if @security_groups.nil?
		if groups[0].kind_of? Array
			@security_groups[:default] = [] if @security_groups[:default].nil?
			@security_groups[:default] += groups[0]
		elsif groups[0].kind_of? Hash
			groups[0].keys.each do |env_name|
				@security_groups[env_name] = [] if @security_groups[env_name].nil?
				if groups[0][env_name].kind_of? Array
					@security_groups[env_name] += groups[0][env_name]
				else
					@security_groups[env_name] << groups[0][env_name]
				end
			end
		else
			@security_groups[:default] = [] if @security_groups[:default].nil?
			@security_groups[:default] << groups[0].to_s
		end
		self
	end
end

#security_groups_for_environment(environment) ⇒ Object

Retrieves the list of security groups for a given environment. Returns the default set of security groups if the environment isn’t defined. Returns an empty array if the default security group is empty.



52
53
54
55
56
57
58
 
# File 'lib/ec2launcher/security_group_handler.rb', line 52

def security_groups_for_environment(environment)
	if @security_groups[environment].nil?
		# Environment not found - check default
		return @security_groups[:default].nil? ? [] : @security_groups[:default]
	end
	@security_groups[environment]
end