capify-ec2 1 → 1.1.0

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in capify-ec2.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/capify-ec2.gemspec

@@ -1,18 +1,23 @@
-#-*- encoding: utf-8 -*-
+# -*- encoding: utf-8 -*-
 $:.push File.expand_path("../lib", __FILE__)
+require "capify-ec2/version"
 
 Gem::Specification.new do |s|
-  s.name = "capify-ec2"
-  s.version = "1"
-  s.authors = ["Noah Cantor"]
-  s.email = "noah.cantor@forward.co.uk"
-  s.homepage = "http://www.forward.co.uk"
-  s.summary = "Grabs roles from ec2's tags and autogenerates capistrano tasks"
-  s.platform = Gem::Platform::RUBY
-  s.files = %w(lib).map {|d| Dir.glob("#{d}/**/*")}.flatten << "capify-ec2.gemspec"
-  s.require_path = "lib"
-  s.has_rdoc = false
+  s.name        = "capify-ec2"
+  s.version     = Capify::Ec2::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Noah Cantor", "Siddharth Dawara"]
+  s.email       = ["noah.cantor@forward.co.uk", "siddharth.dawara@forward.co.uk"]
+  s.homepage    = "http://github.com/forward/capify-ec2"
+  s.summary     = %q{Grabs roles from ec2's tags and autogenerates capistrano tasks}
+  s.description = %q{Grabs roles from ec2's tags and autogenerates capistrano tasks}
+
+  s.rubyforge_project = "capify-ec2"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
   s.add_dependency('activesupport', '>= 3.0.0')
   s.add_dependency('fog')
 end
-

data/lib/capify-ec2.rb

@@ -3,9 +3,11 @@ require 'fog'
 
 class CapifyEc2
 
+  attr_accessor :ec2_config, :instance, :elb_name, :elb
+  
   def self.running_instances
-    ec2_config = YAML.load(File.new("config/ec2.yml"))
-    ec2 = Fog::Compute.new(:provider => 'AWS', :aws_access_key_id => ec2_config[:aws_access_key_id], :aws_secret_access_key => ec2_config[:aws_secret_access_key], :region => ec2_config[:aws_params][:region])
+    @ec2_config = YAML.load(File.new("config/ec2.yml"))
+    ec2 = Fog::Compute.new(:provider => 'AWS', :aws_access_key_id => @ec2_config[:aws_access_key_id], :aws_secret_access_key => @ec2_config[:aws_secret_access_key], :region => @ec2_config[:aws_params][:region])
     running_instances = ec2.servers.select {|instance| instance.state == "running"}
     running_instances.each do |instance|
       instance.instance_eval do
@@ -27,7 +29,7 @@ class CapifyEc2
     end
   end  
   
-  def self.get_instances_by_name(name)
+  def self.get_instance_by_name(name)
     selected_instances = running_instances.select do |instance|
       value = instance.case_insensitive_tag("Name")
       value == name.to_s
@@ -36,6 +38,25 @@ class CapifyEc2
   
   def self.server_names
     running_instances.map {|instance| instance.name}
-  end 
+  end
   
+  def self.get_elb_name_by_instance(instance_id)
+    @elb = Fog::AWS::ELB.new(:aws_access_key_id => @ec2_config[:aws_access_key_id], :aws_secret_access_key => @ec2_config[:aws_secret_access_key], :region => @ec2_config[:aws_params][:region])
+    @elb.load_balancers.each do |load_balancer|
+      load_balancer.instances.each {|instance| return load_balancer.id if instance_id == instance}
+    end
+    return nil
+  end
+     
+  def self.deregister_instance_from_elb(instance_name)
+    return unless @ec2_config[:load_balanced]
+    @instance = get_instance_by_name(instance_name).first
+    @elb_name = get_elb_name_by_instance(@instance.id)
+    @elb.deregister_instances_from_load_balancer(@instance.id, @elb_name) unless @elb_name.nil?
+  end
+  
+  def self.register_instance_in_elb
+    return unless @ec2_config[:load_balanced]
+    @elb.register_instances_with_load_balancer(@instance.id, @elb_name) unless @elb_name.nil?
+  end
 end

data/lib/capify-ec2/capistrano.rb

@@ -2,15 +2,10 @@ require File.join(File.dirname(__FILE__), '../capify-ec2')
 
 Capistrano::Configuration.instance(:must_exist).load do
   def ec2_role(role_to_define)
-    subroles = {}
-    if role_to_define.is_a?(Hash)
-      defined_role = role_to_define[:name]
-      subroles = role_to_define[:options]
-    else
-      defined_role = role_to_define
-    end
+    defined_role = role_to_define.is_a?(Hash) ? role_to_define[:name] : role_to_define
     instances = CapifyEc2.get_instances_by_role(defined_role)
-
+    subroles = role_to_define.is_a?(Hash) ? role_to_define[:options] : {}
+    
     instances.each do |instance|
       task instance.name.to_sym do
         set :server_address, instance.dns_name
@@ -31,16 +26,20 @@ Capistrano::Configuration.instance(:must_exist).load do
       end        
     end 
   end  
-
-  # task :add_tag do
-  #   instance_name = variables[:logger].instance_variable_get("@options")[:actions].first
-  #   CapifyEc2.add_tag(instance_name, tag, value)
-  # end
   
   def ec2_roles(*roles)
     roles.each {|role| ec2_role(role)}
   end
   
+  task :deregister_instance do
+    servers = variables[:logger].instance_variable_get("@options")[:actions].first
+    CapifyEc2.deregister_instance_from_elb(servers)
+  end
+  
+  task :register_instance do
+    CapifyEc2.register_instance_in_elb
+  end
+  
   task :date do
     run "date"
   end
@@ -48,4 +47,9 @@ Capistrano::Configuration.instance(:must_exist).load do
   task :server_names do
     puts CapifyEc2.server_names.sort
   end
+  
+  namespace :deploy do
+    before "deploy", "deregister_instance"
+    after "deploy", "register_instance"
+  end
 end

data/lib/capify-ec2/version.rb

@@ -0,0 +1,5 @@
+module Capify
+  module Ec2
+    VERSION = "1.1.0"
+  end
+end

data/readme.md

@@ -0,0 +1,100 @@
+Capify Ec2
+====================================================
+
+capify-ec2 is used to generate capistrano namespaces using ec2 tags. 
+
+eg: If you have three servers on amazon's ec2.
+
+    server-1 Tag: Role => "web"
+    server-2 Tag: Role => "db"
+    server-3 Tag: Role => "web"
+
+Installing
+
+    gem install capify-ec2
+
+In your deploy.rb:
+
+    require "capify-ec2/capistrano"
+    ec2_roles :web
+
+Will generate
+
+    task :server-1 do
+      role :web, {server-1 public dns fetched from Amazon}
+    end
+    
+    task :server-3 do
+      role :web, {server-1 public dns fetched from Amazon}
+    end
+    
+    task :web do
+      role :web, {server-1 public dns fetched from Amazon}
+      role :web, {server-3 public dns fetched from Amazon}
+    end
+
+Additionally
+
+    require "capify-ec2/capistrano"
+    ec2_roles :db
+
+Will generate
+
+    task :server-2 do
+      role :web, {server-2 public dns fetched from Amazon}
+    end
+    
+    task :db do
+      role :db, {server-2 public dns fetched from Amazon}
+    end
+
+Running
+
+    cap web date
+
+will run the date command on all server's tagged with the web role
+
+More options
+====================================================
+
+    ec2_roles {:name=>"web", :options=>{:cron=>"server-1"}}
+  
+Will generate
+
+    task :server-1 do
+      role :web, {server-1 public dns fetched from Amazon}, :cron=>true
+    end
+
+    task :server-3 do
+      role :web, {server-1 public dns fetched from Amazon}
+    end
+
+    task :web do
+      role :web, {server-1 public dns fetched from Amazon}, :cron=>true
+      role :web, {server-3 public dns fetched from Amazon}
+    end
+
+Which is cool if you want a task like this in deploy.rb
+
+    task :update_cron => :web, :only=>{:cron} do
+      Do something to a server with cron on it
+    end
+
+Ec2 config
+====================================================
+
+This gem requires 'config/ec2.yml' in your project.
+The yml file needs to look something like this:
+
+    	:aws_access_key_id: "YOUR ACCESS KEY"
+    	:aws_secret_access_key: "YOUR SECRET"
+    	:aws_params:
+    	  :region: 'eu-west-1'
+		:load_balanced: true
+
+The :aws_params are optional.
+If :load_balanced is set to true, the gem will deregister 
+the instance from the load balancer before deploying and
+reregister it after, using before and after deploy hooks.
+:load_balanced only works for individual instances, not
+for roles.

metadata

@@ -1,19 +1,23 @@
 --- !ruby/object:Gem::Specification 
 name: capify-ec2
 version: !ruby/object:Gem::Version 
-  hash: 1
+  hash: 19
   prerelease: 
   segments: 
   - 1
-  version: "1"
+  - 1
+  - 0
+  version: 1.1.0
 platform: ruby
 authors: 
 - Noah Cantor
+- Siddharth Dawara
 autorequire: 
 bindir: bin
 cert_chain: []
 
-date: 2011-05-11 00:00:00 Z
+date: 2011-05-20 00:00:00 +01:00
+default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: activesupport
@@ -45,8 +49,10 @@ dependencies:
         version: "0"
   type: :runtime
   version_requirements: *id002
-description: 
-email: noah.cantor@forward.co.uk
+description: Grabs roles from ec2's tags and autogenerates capistrano tasks
+email: 
+- noah.cantor@forward.co.uk
+- siddharth.dawara@forward.co.uk
 executables: []
 
 extensions: []
@@ -54,10 +60,16 @@ extensions: []
 extra_rdoc_files: []
 
 files: 
-- lib/capify-ec2/capistrano.rb
-- lib/capify-ec2.rb
+- .gitignore
+- Gemfile
+- Rakefile
 - capify-ec2.gemspec
-homepage: http://www.forward.co.uk
+- lib/capify-ec2.rb
+- lib/capify-ec2/capistrano.rb
+- lib/capify-ec2/version.rb
+- readme.md
+has_rdoc: true
+homepage: http://github.com/forward/capify-ec2
 licenses: []
 
 post_install_message: 
@@ -85,8 +97,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: "0"
 requirements: []
 
-rubyforge_project: 
-rubygems_version: 1.7.2
+rubyforge_project: capify-ec2
+rubygems_version: 1.6.2
 signing_key: 
 specification_version: 3
 summary: Grabs roles from ec2's tags and autogenerates capistrano tasks