capistrano-elb 0.2

data/lib/capistrano-elb.rb

@@ -0,0 +1,80 @@
+require 'fog'
+require 'yaml'
+require 'capistrano'
+require 'pp'
+# require "dir"
+
+class CapELB
+  def initialize(configdir=File.join(Dir.pwd, 'config'))
+    ec2credentials = YAML::load(File.open(File.join(configdir, 'ec2credentials.yaml')))
+    aws = Fog::Compute.new(ec2credentials.merge({:provider=>'AWS'}))
+    @regions = aws.describe_regions.body["regionInfo"].map {|region| region["regionName"]}
+
+    @compute = {}
+    @regions.each do |region|
+      @compute.merge!(region => Fog::Compute.new(ec2credentials.merge({:provider=>'AWS',:region=>region})))
+    end
+
+    @elb = {}
+    @regions.each do |region|
+      @elb.merge!(region => Fog::AWS::ELB.new(ec2credentials.merge(:region=>region)))
+    end
+    
+    @lbsfile = File.join(configdir, 'lbs.yaml') 
+    
+    @lbs = load_config
+  end
+  
+  def config_from_aws
+    lbs = {}
+    @regions.each do |region| 
+      loadBalancerDescriptions = 
+        @elb[region].describe_load_balancers.body["DescribeLoadBalancersResult"]["LoadBalancerDescriptions"]
+      loadBalancerDescriptions.each do |lb|
+        lbs.merge!({region => {lb["LoadBalancerName"] => lb["Instances"]}})
+      end
+    end
+    lbs
+  end
+  
+  def save_config
+    File.open( @lbsfile, 'w' ) do |file|
+       YAML.dump( config_from_aws, file )
+    end
+  end
+  
+  def load_config
+    unless File.exists? @lbsfile
+       save_config
+     end
+    YAML::load(File.open(@lbsfile))
+  end
+  
+  def check_config
+    load_config == config_from_aws
+  end
+  
+  def add(serverlist)
+    each_server_by_lbs(serverlist) do |region, lbname, servers|
+      region.register_instances_with_load_balancer(servers, lbname)
+    end
+  end
+  
+  def remove(serverlist)
+    each_server_by_lbs(serverlist) do |region, lbname, servers|
+      region.deregister_instances_from_load_balancer(servers, lbname)
+    end
+  end
+  
+  def each_server_by_lbs(serverlist)
+    @lbs.each_pair do |region, lbs|
+      lbs.each_pair do |lbname, target_instances|
+        to_change = @compute[region].servers.select{|server| serverlist.include? server.dns_name}.map{|server| server.id}
+        yield(@elb[region], lbname, to_change) unless to_change.empty?
+      end
+    end
+  end
+end
+
+
+

data/lib/capistrano-elb/tasks.rb

@@ -0,0 +1,30 @@
+require "capistrano-elb"
+require "pp"
+
+Capistrano::Configuration.instance(:must_exist).load do
+  namespace :elb do
+    capELB = CapELB.new()
+    
+    task :remove do 
+      servers = roles[:web].servers.map {|server| server.host}
+      pp servers
+      capELB.remove servers
+    end
+
+    task :add do 
+      servers = roles[:web].servers.map {|server| server.host}
+      capELB.add servers
+    end
+    
+    task :save do
+      capELB.save_config
+    end
+    
+    task :check do 
+      puts capELB.check_config
+    end
+  end
+
+  before "deploy", "elb:remove"
+  after "deploy", "elb:add"
+end

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification
+name: capistrano-elb
+version: !ruby/object:Gem::Version
+  version: '0.2'
+  prerelease: !!null 
+platform: ruby
+authors:
+- thattommyhall
+autorequire: !!null 
+bindir: bin
+cert_chain: []
+date: 2011-04-11 00:00:00.000000000 +01:00
+default_executable: !!null 
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fog
+  requirement: &2156032200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156032200
+- !ruby/object:Gem::Dependency
+  name: excon
+  requirement: &2156031720 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156031720
+description: Remove servers automagically from ELB before deploy
+email: tom.hall@forward.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/capistrano-elb.rb
+- lib/capistrano-elb/tasks.rb
+has_rdoc: true
+homepage: https://github.com/thattommyhall/capistrano-elb
+licenses: []
+post_install_message: !!null 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: !!null 
+rubygems_version: 1.5.0
+signing_key: !!null 
+specification_version: 3
+summary: Automagically remove/readd servers from EC2 load balancers as you cap deploy
+test_files: []