capify-ec2 1

data/capify-ec2.gemspec

@@ -0,0 +1,18 @@
+#-*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+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.add_dependency('activesupport', '>= 3.0.0')
+  s.add_dependency('fog')
+end
+

data/lib/capify-ec2.rb

@@ -0,0 +1,41 @@
+require 'rubygems'
+require 'fog'
+
+class CapifyEc2
+
+  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])
+    running_instances = ec2.servers.select {|instance| instance.state == "running"}
+    running_instances.each do |instance|
+      instance.instance_eval do
+        def case_insensitive_tag(key)
+          tags[key] || tags[key.downcase]
+        end
+        
+        def name
+          case_insensitive_tag("Name").split('-').reject {|portion| portion.include?(".")}.join("-")
+        end
+      end
+    end
+  end
+  
+  def self.get_instances_by_role(role)
+    selected_instances = running_instances.select do |instance| 
+      value = instance.case_insensitive_tag("Role")
+      value == role.to_s
+    end
+  end  
+  
+  def self.get_instances_by_name(name)
+    selected_instances = running_instances.select do |instance|
+      value = instance.case_insensitive_tag("Name")
+      value == name.to_s
+    end
+  end
+  
+  def self.server_names
+    running_instances.map {|instance| instance.name}
+  end 
+  
+end

data/lib/capify-ec2/capistrano.rb

@@ -0,0 +1,51 @@
+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
+    instances = CapifyEc2.get_instances_by_role(defined_role)
+
+    instances.each do |instance|
+      task instance.name.to_sym do
+        set :server_address, instance.dns_name
+        role :web, *server_address
+        role :app, *server_address
+      end
+    end
+    
+    task defined_role.to_sym do
+      instances.each do |instance|
+        new_options = {}
+        subroles.each {|key, value| new_options[key] = true if value.to_s == instance.name}
+        if new_options
+          role defined_role.to_sym, instance.dns_name, new_options
+        else
+          role defined_role.to_sym, instance.dns_name
+        end    
+      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 :date do
+    run "date"
+  end
+  
+  task :server_names do
+    puts CapifyEc2.server_names.sort
+  end
+end

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification 
+name: capify-ec2
+version: !ruby/object:Gem::Version 
+  hash: 1
+  prerelease: 
+  segments: 
+  - 1
+  version: "1"
+platform: ruby
+authors: 
+- Noah Cantor
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-11 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: activesupport
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        - 0
+        version: 3.0.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: fog
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: 
+email: noah.cantor@forward.co.uk
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/capify-ec2/capistrano.rb
+- lib/capify-ec2.rb
+- capify-ec2.gemspec
+homepage: http://www.forward.co.uk
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.7.2
+signing_key: 
+specification_version: 3
+summary: Grabs roles from ec2's tags and autogenerates capistrano tasks
+test_files: []
+