capistrano-elobuff 0.1.4 → 0.2.0

data/elobuff_capistrano.gemspec

@@ -11,7 +11,7 @@ Gem::Specification.new do |gem|
   gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
   gem.name          = "capistrano-elobuff"
   gem.require_paths = ["lib"]
-  gem.version       = "0.1.4"
+  gem.version       = "0.2.0"
 
   gem.add_dependency "capistrano"
 end

data/lib/capistrano/elobuff/ec2.rb

@@ -0,0 +1,15 @@
+require "fog"
+require "capistrano/elobuff/ec2/helpers"
+require "capistrano/elobuff/ec2/compute"
+require "capistrano/elobuff/ec2/server"
+
+Capistrano::Configuration.instance.load do
+  require_settings :ec2_ami, :ec2_instance_type, :ec2_security_group, :ec2_availability_zone
+
+  set_default(:ec2_key)       { ENV["AWS_EC2_KEY"] || ENV["AWS_S3_KEY"] || raise("Please specify an AWS_EC2_KEY") }
+  set_default(:ec2_secret)    { ENV["AWS_EC2_SECRET"] || ENV["AWS_S3_SECRET"] || raise("Please specify an AWS_EC2_SECRET") }
+
+  set_default(:ec2)           { Capistrano::Elobuff::Ec2::Compute.new application, ec2_key, ec2_secret }
+end
+
+Dir.glob(File.join(File.dirname(__FILE__), "/ec2/recipes/*.rb")).sort.each {|f| load f }

data/lib/capistrano/elobuff/ec2/compute.rb

@@ -0,0 +1,63 @@
+require "fog"
+
+module Capistrano
+  module Elobuff
+    module Ec2
+      class Compute
+        def initialize(application, access_key_id, secret_access_key)
+          @application = application
+          @access_key_id = access_key_id
+          @secret_access_key = secret_access_key
+        end
+
+        def connection
+          @connection ||= Fog::Compute.new provider: "AWS", aws_access_key_id: @access_key_id, aws_secret_access_key: @secret_access_key
+        end
+
+        def servers
+          @servers ||= connection.servers.select {|s| s.tags.keys.include? @application.to_s }.map {|s| Server.new s, @application }
+        end
+
+        def find_server(id)
+          servers.select {|s| (s.id == id) || (s.name == id) || (s.address == id) }.first || die("Could not find server with id, name or address: #{id}")
+        end
+
+        def available_servers
+          servers.select {|s| s.available? }
+        end
+
+        def roles
+          available_servers.map {|s| s.roles }.flatten.uniq
+        end
+
+        def zones
+          available_servers.map {|s| s.zone }.flatten.uniq
+        end
+
+        def available_servers_in_role(role)
+          available_servers.select {|s| s.roles.include? role }
+        end
+
+        def available_servers_in_zone(zone)
+          available_servers.select {|s| s.zone == zone }
+        end
+
+        def create_server(ami, type, security_groups, availability_zone, name, roles)
+          puts "Creating server"
+          server = connection.servers.create(
+            image_id: ami,
+            flavor_id: type,
+            groups: security_groups,
+            availability_zone: availability_zone,
+            tags: { "Name" => name, "#{@application}" => roles })
+          server.wait_for { ready? }
+          Server.new(server, @application)
+        end
+
+        def update_server_tags(server)
+          connection.create_tags server.id, server.tags
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/elobuff/ec2/helpers.rb

@@ -0,0 +1,16 @@
+def die(why)
+  $stdout.sync = true
+  puts ""
+  puts "  FATAL ERROR: #{why}"
+  exit 1
+end
+
+def find_servers
+  if ids = ENV["SERVERS"]
+    ids.split(",").map {|id| ec2.find_server id }
+  elsif id = ENV["SERVER"]
+    Array(ec2.find_server id)
+  else
+    die "Please specify SERVER or SERVERS"
+  end
+end

data/lib/capistrano/elobuff/ec2/recipes/a.rb

@@ -0,0 +1,12 @@
+Capistrano::Configuration.instance.load do
+  require_settings :ec2
+
+  namespace :a do
+    desc "Invoke Servers"
+    task :default do
+      ec2.available_servers.each do |s|
+        server s.address, *s.roles
+      end
+    end
+  end
+end

data/lib/capistrano/elobuff/ec2/recipes/r.rb

@@ -0,0 +1,30 @@
+Capistrano::Configuration.instance.load do
+  require_settings :ec2
+
+  namespace :r do
+    desc "List roles"
+    task :default do
+      puts "Listing roles:"
+      ec2.roles.each do |r|
+        puts "%-16s %s" % ["#{r}:", ec2.available_servers_in_role(r).map(&:id).join(", ")]
+      end
+      puts ""
+      puts "Commands: #{ec2.roles.map {|r| "r:#{r}" }.join(", ")}"
+    end
+  end
+end
+
+Capistrano::Configuration.instance(:must_exist).load do
+  def ec2_roles
+    namespace :r do
+      ec2.roles.each do |r|
+        desc "Invoke Role: #{r}"
+        task r do
+          ec2.available_servers_in_role(r).each do |s|
+            server s.address, *s.roles
+          end
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/elobuff/ec2/recipes/s.rb

@@ -0,0 +1,83 @@
+Capistrano::Configuration.instance.load do
+  require_settings :ec2
+
+  namespace :s do
+    desc "List servers"
+    task :default do
+      puts "Listing servers:"
+      puts ec2.servers.map(&:to_s)
+      puts ""
+      puts "SERVERS: #{ec2.available_servers.map(&:id).join(", ")}"
+    end
+
+    desc "Add Server"
+    task :add do
+      ami = ENV["EC2_AMI"] || ec2_ami || die("Please specify an EC2_AMI")
+      type = ENV["EC2_INSTANCE_TYPE"] || ec2_instance_type || die("Please specify an EC2_INSTANCE_TYPE")
+      group = ENV["EC2_SECURITY_GROUP"] || ec2_security_group || die("Please specify an EC2_SECURITY_GROUP")
+      zone = ENV["EC2_AVAILABILITY_ZONE"] || ec2_availability_zone || die("Please specify an EC2_AVAILABILITY_ZONE")
+      name = ENV["NAME"] || die("Please specify a NAME")
+      roles = ENV["ROLES"] || die("Please specify ROLES")
+
+      s = ec2.create_server(ami, type, group, zone, name, roles)
+      server s.address, *s.roles
+    end
+
+    desc "Update Server Roles"
+    task :rerole do
+      find_servers.each do |s|
+        if roles = ENV["ROLES"]
+          s.roles = roles
+          ec2.update_server_tags s
+        end
+
+        puts s.to_s
+      end
+    end
+
+    desc "Rename Servers"
+    task :rename do
+      find_servers.each do |s|
+        if name = ENV["NAME"]
+          s.name = name
+          ec2.update_server_tags s
+        end
+
+        puts s.to_s
+      end
+    end
+
+    desc "Console into Server"
+    task :console do
+      s = find_servers.first
+      exec "ssh #{user}@#{s.address}"
+    end
+
+    desc "Terminate Servers"
+    task :terminate do
+      puts "Preparing to terminate the following servers:"
+      find_servers.each do |s|
+        puts s.to_s
+      end
+
+      ENV["CONFIRM"] || die("Please set CONFIRM in order to terminate")
+
+      find_servers.each do |s|
+        s.terminate!
+      end
+    end
+  end
+end
+
+Capistrano::Configuration.instance(:must_exist).load do
+  def ec2_servers
+    namespace :s do
+      ec2.servers.each do |s|
+        desc "Invoke Server: #{s.id}"
+        task s.slug do
+          define_server s.address, *s.roles
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/elobuff/ec2/recipes/z.rb

@@ -0,0 +1,30 @@
+Capistrano::Configuration.instance.load do
+  require_settings :ec2
+
+  namespace :z do
+    desc "List zones"
+    task :default do
+      puts "Listing availability zones:"
+      ec2.zones.each do |z|
+        puts "%-16s %s" % ["#{z}:", ec2.available_servers_in_zone(z).map(&:id).join(", ")]
+      end
+      puts ""
+      puts "Commands: #{ec2.zones.map {|z| "z:#{z}" }.join(", ")}"
+    end
+  end
+end
+
+Capistrano::Configuration.instance(:must_exist).load do
+  def ec2_zones
+    namespace :z do
+      ec2.zones.each do |z|
+        desc "Invoke Zone: #{z}"
+        task z do
+          ec2.available_servers_in_zone(z).each do |s|
+            server s.address, *s.roles
+          end
+        end
+      end
+    end
+  end
+end

data/lib/capistrano/elobuff/ec2/server.rb

@@ -0,0 +1,62 @@
+require "fog"
+
+module Capistrano
+  module Elobuff
+    module Ec2
+      class Server
+        def initialize(fog_server, application)
+          @server = fog_server
+          @application = application
+        end
+
+        def to_s
+          "%-24s%-16s%-48s%-16s%-16s%-16s%-24s%s" % [name, id, address, @server.image_id, @server.availability_zone, @server.state, @server.groups.join(","), roles.join(",")]
+        end
+
+        def available?
+          @server.state == "running"
+        end
+
+        def id
+          @server.id
+        end
+
+        def slug
+          @server.id.gsub(/i-/, "")
+        end
+
+        def name
+          @server.tags["Name"]
+        end
+
+        def name=(string)
+          @server.tags["Name"] = string
+        end
+
+        def address
+          @server.dns_name
+        end
+
+        def tags
+          @server.tags
+        end
+
+        def zone
+          @server.availability_zone
+        end
+
+        def roles
+          @server.tags[@application].split(",").map(&:to_sym)
+        end
+
+        def roles=(string)
+          @server.tags[@application] = string
+        end
+
+        def terminate!
+          @server.destroy
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-elobuff
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-17 00:00:00.000000000 Z
+date: 2012-08-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: capistrano
@@ -43,6 +43,14 @@ files:
 - lib/capistrano/elobuff.rb
 - lib/capistrano/elobuff/base.rb
 - lib/capistrano/elobuff/defaults.rb
+- lib/capistrano/elobuff/ec2.rb
+- lib/capistrano/elobuff/ec2/compute.rb
+- lib/capistrano/elobuff/ec2/helpers.rb
+- lib/capistrano/elobuff/ec2/recipes/a.rb
+- lib/capistrano/elobuff/ec2/recipes/r.rb
+- lib/capistrano/elobuff/ec2/recipes/s.rb
+- lib/capistrano/elobuff/ec2/recipes/z.rb
+- lib/capistrano/elobuff/ec2/server.rb
 - lib/capistrano/elobuff/helpers.rb
 - lib/capistrano/elobuff/recipes/bundler.rb
 - lib/capistrano/elobuff/recipes/code.rb