capawsext 0.1 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 762e84774e300c85a73a55b0c6aea0c31fcad919
-  data.tar.gz: 148f79e31294a8ad2d0fae088761a8aa6614ab3e
+  metadata.gz: 462db8dd528f2d58989e39afe65e6487081bf2aa
+  data.tar.gz: d5870ce98e2533c6eb5f209f5851301b10e3faa3
 SHA512:
-  metadata.gz: b694827848eae4a164ce673874358f4b2c4eef31c35ad50bb2532f0ec76b6c81153e4570a632696d1c545765332f29f7c6f8a6b967ceb5a0790d7a82ed175956
-  data.tar.gz: 859bfeccca344579b30fd280f7603c3e1d576c793829dfde0a5b9498621cde1b24c7c9309ff3fe6aca56ad213083e32095baab9d021aa3c7d284c49c9312c70d
+  metadata.gz: 26267629a07b077072df8e3db3098c27f0d96195b21a65963f9fd240c81c1f60810c44d09059094c670feba9ef6ff0e660f12ad1157b16c1ec0cd1524d54601f
+  data.tar.gz: d276dfa3f31027c5fdd65ccbde83aa4c61654abebf0881487eb39a2fdf3faabfffe37c0b8ae0fb82052fffa658eaa00bb92aadbb4f09aba349551996604b887d

data/lib/capawsext/mockcloud.yml

@@ -0,0 +1,2 @@
+:aws_access_key_id: "mock"
+:aws_secret_access_key: "mock"

data/lib/capawsext/tasks.rb

@@ -7,8 +7,8 @@ end
 require 'capawsext/whoec2helper'
 
 Capistrano::Configuration.instance(:must_exist).load do
-	def add_groups
-		ec2_helper = Ec2Helper.new
+	def add_groups(region)
+		ec2_helper = Ec2Helper.new(region)
 		groups = ec2_helper.get_groups
 		groups.each { |group|
 			task group.to_sym, :desc => "Run the task in all instances of the #{group}" do

data/lib/capawsext/version.rb

@@ -1,3 +1,3 @@
 module CapAwsExt
-	VERSION="0.1"
+	VERSION="0.2"
 end

data/lib/capawsext/whoec2helper.rb

@@ -1,50 +1,25 @@
 require 'rubygems'
 require 'fog'
 require 'colored'
-#require File.expand_path(File.dirname(__FILE__) + '/capify-cloud/server')
 
 class Ec2Helper
 	
-       def initialize(cloud_config = "config/cloud.yml")
+       def initialize(region, cloud_config = "config/cloud.yml")
 		@cloud_config = YAML.load_file cloud_config
 		@instances =[]
-		@cloud_providers = @cloud_config[:cloud_providers]
-    		@cloud_providers.each do |cloud_provider|
-		      config = @cloud_config[cloud_provider.to_sym]
-		      config[:provider] = cloud_provider
-		        regions = determine_regions(cloud_provider)
-			config.delete(:regions)
-			      if regions
-			          regions.each do |region|
-			            config.delete(:region)
-		        	    config[:region] = region
-			            populate_instances(config)
-			          end
-		        else populate_instances(config)
-			      end
-		 end
-
-        end
-	def determine_regions(cloud_provider = 'AWS')
-	    regions = @cloud_config[cloud_provider.to_sym][:regions]
-	  end
-
-
-	def populate_instances(config)
-    		@fog = Fog::Compute.new(config)
-		@fog.servers.each {|server| @instances << server }
+		populate_instances(region, @cloud_config)
 	end
 
-      	def get_zoo_srvs_private_dns(group)
-		zoosrvs = Array.new
-		instances = get_instances_byrole(group, 'zoo-srv')
-		instances.each {|instance|
-			zoosrvs << instance.private_dns_name
-		}
-		return zoosrvs
+	def populate_instances(region, config)
+		#Choose the defualt prover and region
+		raise "aws_access_key_id or aws_secret_access_key keys are not present in the config file" if config[:aws_access_key_id].nil? or config[:aws_secret_access_key].nil?
+		temp_config = {:provider => 'AWS'}
+		config[:region] = region
+    		@fog = Fog::Compute.new(temp_config.merge!(config))
+		@fog.servers.each {|server| @instances << server }
 	end
 		
-	def get_instances_byrole(group, role)
+	def get_instances_role(group, role)
 		ret_instances = Array.new
 		@instances.each {|instance|
 			if not instance.tags['role'].nil? and instance.ready?
@@ -56,22 +31,6 @@ class Ec2Helper
 		return ret_instances
 	end
 	
-	def get_solr_private_dns(group)
-		instances = get_instances_byrole(group, 'solr')
-		return instances[0].private_dns_name if instances.count > 0
-	end
-		
-	
-	def get_facade_srv_public_dns(group)
-		instance = get_instances_byrole(group, 'api-srv')
-		return instance[0].dns_name if instance.count > 0
-	end
-
-	def get_db_srv_private_dns(group)
-		instance = get_instances_byrole(group, 'db-srv')
-		return instance[0].private_dns_name if instance.count > 0
-	end
-
 	def printInstanceDetails()
 		 @instances.each_with_index do |instance, i|
 		      puts sprintf "%02d:  %-20s  %-20s %-20s  %-20s  %-25s  %-20s  (%s)  (%s) (%s)",
@@ -82,47 +41,14 @@ class Ec2Helper
 
 	end	
 
-	def get_server_details_by_pri_pub_dns(dns)
-		details={}
+	def get_instance_by_pri_pub_dns(dns)
 		@instances.each { |instance|
 			if instance.private_dns_name == dns or instance.dns_name == dns
-				details['name']=instance.tags['name_s']
-				details['orgid']=instance.tags['orgid']
-				details['group']=instance.tags['group']
-				details['private_dns_name'] = instance.private_dns_name
-				details['role']	= instance.tags['role']
-				details['public_dns_name'] = instance.dns_name
-				details['id'] = instance.id
-				details['zone'] = instance.availability_zone
-				return details
+				return instance
 			end
 		}
-		return details
 	end
 
-	def get_server_details_map(isrunning)
-		servers={}
-		@instances.each { |instance|
-				if not isrunning or (isrunning and instance.ready?)	
-				details={}
-				details['name']=instance.tags['name_s']
-				details['orgid']=instance.tags['orgid']
-				details['group']=instance.tags['group']
-				details['private_dns_name'] = instance.private_dns_name
-				details['role']	= instance.tags['role']
-				details['public_dns_name'] = instance.dns_name
-				details['id'] = instance.id
-				details['zone'] = instance.availability_zone
-				details['device_mappings'] = Array.new
-				instance.block_device_mapping.each { |mapping|
-					details['device_mappings'] << mapping['deviceName']
-				} 
-				servers[instance.private_dns_name] = details
-				end
-		}
-		return servers
-	end	
-
 	def get_all_stacks
 		stacks = {}
 		@instances.each {|instance|
@@ -137,14 +63,6 @@ class Ec2Helper
 	end
 
 		
-	def get_instance_on_dns(dns)
-		@instances.each { |instance|
-			if instance.private_dns_name == dns or instance.dns_name == dns 
-				return instance
-			end	
-		}
-	end
-
 	def get_groups
 		groups = Set.new
 		@instances.each { |instance|
@@ -170,6 +88,34 @@ class Ec2Helper
 		return nil
 	end
 
+	def add_tags(resource_id, tags = {})
+		tags.each { |key, value|
+			@fog.tags.create(:resource_id => resource_id, :key => key, :value => value)
+		}
+	end
+
+	def get_all_instances
+		return @instances
+	end
+
+	def lb_unregister_instances(lb_name, instances)
+		lb = get_lb(lb_name)
+		lb.deregister_instances(instances.map{|i| i.id })
+	end
+
+	def lb_register_instances(lb_name, instances)
+		lb = get_lb(lb_name)
+		lb.register_instances(instances.map{|i| i.id })
+	end
+
+	private
+	def get_lb(lb_name)
+		@lbs ||= Fog::AWS::ELB.new(@cloud_config).load_balancers
+		lb = @lbs.get(lb_name)
+		raise "lb #{lb_name} not found in the region #{@cloud_config[:region]}" if lb.nil?
+		return lb
+	end
+	#Not used.
 	def get_zone(roles, sec_group)
 		zones = Hash["us-west-1b" =>0,"us-west-1c" =>0]
 		@instances.each { |instance|
@@ -186,16 +132,27 @@ class Ec2Helper
 		return min_zone
 	end
 	
-
-	def add_tags(resource_id, tags = {})
-		tags.each { |key, value|
-			puts "#{resource_id}, #{key}, #{value}"
-			@fog.tags.create(:resource_id => resource_id, :key => key, :value => value)
+	#Not used anywhere...I just kept this code to check the instance properties to access block_device_mapping
+	def get_server_details_map(isrunning)
+		servers={}
+		@instances.each { |instance|
+				if not isrunning or (isrunning and instance.ready?)	
+				details={}
+				details['name']=instance.tags['name_s']
+				details['orgid']=instance.tags['orgid']
+				details['group']=instance.tags['group']
+				details['private_dns_name'] = instance.private_dns_name
+				details['role']	= instance.tags['role']
+				details['public_dns_name'] = instance.dns_name
+				details['id'] = instance.id
+				details['zone'] = instance.availability_zone
+				details['device_mappings'] = Array.new
+				instance.block_device_mapping.each { |mapping|
+					details['device_mappings'] << mapping['deviceName']
+				} 
+				servers[instance.private_dns_name] = details
+				end
 		}
-	end
-
-	def get_all_instances
-		return @instances
-	end
+		return servers
+	end	
 end
-

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: capawsext
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: '0.2'
 platform: ruby
 authors:
 - Sumit vij
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-19 00:00:00.000000000 Z
+date: 2013-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -62,6 +62,7 @@ files:
 - README
 - capawsext.gemspec
 - lib/capawsext.rb
+- lib/capawsext/mockcloud.yml
 - lib/capawsext/tasks.rb
 - lib/capawsext/version.rb
 - lib/capawsext/whoec2helper.rb