ec2-instance-manager 0.2.1 → 0.3.0

data/Rakefile

@@ -10,6 +10,7 @@ begin
     gem.email = "kiessler@inceedo.com"
     gem.homepage = "http://github.com/okiess/ec2-instance-manager"
     gem.authors = ["Oliver Kiessler"]
+    gem.add_dependency "amazon-ec2"
     gem.add_development_dependency "thoughtbot-shoulda"
     # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
   end

data/VERSION

@@ -1 +1 @@
-0.2.1
+0.3.0

data/config.yml.sample

@@ -5,6 +5,11 @@ default:
   key: 
   availability_zone: eu-west-1a
   instance_type: m1.small
+  launch_plan:
+    group1:
+      "ami_id_xxx1": 1
+    group2:
+      "ami_id_xxx2": 1  
 
 customer1:
   amazon_access_key_id: 
@@ -21,4 +26,6 @@ customer2:
   key: 
   availability_zone: eu-west-1a
   instance_type: m1.small
-
+  launch_plan:
+    group1:
+      ami_id_xxx1: 1

data/ec2-instance-manager.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{ec2-instance-manager}
-  s.version = "0.2.1"
+  s.version = "0.3.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Oliver Kiessler"]
-  s.date = %q{2009-10-09}
+  s.date = %q{2009-10-16}
   s.default_executable = %q{ec2-instance-manager}
   s.description = %q{Launches EC2 instances for multiple AWS accounts}
   s.email = %q{kiessler@inceedo.com}
@@ -31,6 +31,7 @@ Gem::Specification.new do |s|
      "lib/ec2-instance-manager.rb",
      "lib/ec2-instance-manager/ec2_instance_manager.rb",
      "lib/ec2-instance-manager/launch.rb",
+     "lib/ec2-instance-manager/output.rb",
      "lib/ec2-instance-manager/status.rb",
      "test/ec2-instance-manager_test.rb",
      "test/test_helper.rb"

data/lib/ec2-instance-manager/ec2_instance_manager.rb

@@ -1,10 +1,10 @@
 require File.dirname(__FILE__) + '/status'
 require File.dirname(__FILE__) + '/launch'
+require File.dirname(__FILE__) + '/output'
 
 class Ec2InstanceManager
-  include Status
-  include Launch
-  VERSION = '0.2'
+  include Status, Launch, Output
+  VERSION = '0.3'
   
   attr_reader :config, :customer_key, :options
 
@@ -14,7 +14,7 @@ class Ec2InstanceManager
     options = {}
     
     optparse = OptionParser.new do |opts|
-      opts.banner = "Usage: ec2-instance-manager [options]"
+      opts.banner = "Usage: ec2-instance-manager #{VERSION} [options]"
 
       options[:status] = false
       opts.on( '-s', '--status', 'Status only' ) do
@@ -25,7 +25,12 @@ class Ec2InstanceManager
       opts.on( '-t', '--terminate-all', 'Terminates all instances running under a config key' ) do
         options[:terminate] = true
       end
-      
+
+      options[:start_launch_plan] = false
+      opts.on( '-l', '--start-launch-plan', 'Starts a launch plan under a config key' ) do
+        options[:start_launch_plan] = true
+      end
+
       options[:config] = nil
       opts.on( '-c', '--config CONFIG_KEY', 'Sets the config key' ) do |key|
         options[:config] = key
@@ -44,9 +49,11 @@ class Ec2InstanceManager
   def config; @config ||= read_config; end
   def read_config
     if File.exists?("config.yml")
+      puts "Using config in this directory"
       @config = YAML.load(File.read("config.yml"))
     else
       begin
+        puts "Using config in your home directory"
         @config = YAML.load(File.read("#{ENV['HOME']}/.ec2_instance_manager_config.yml"))
       rescue Errno::ENOENT
         raise "config.yml expected in current directory or ~/.ec2_instance_manager_config.yml"
@@ -60,7 +67,7 @@ class Ec2InstanceManager
   end
 
   def run
-    puts "EC2 Instance Manager"
+    puts "EC2 Instance Manager #{VERSION}"
     puts
     unless options[:config]
       puts "Which customer config do you want to use? (#{config.keys.join(", ")})"
@@ -71,7 +78,7 @@ class Ec2InstanceManager
       @customer_key = options[:config]
     end
 
-    puts "Configuration: #{@customer_key}"
+    puts "Configuration Key: #{@customer_key}"
     puts "AMAZON_ACCESS_KEY_ID: #{config[@customer_key]['amazon_access_key_id']}"
     puts "KEY: #{config[@customer_key]['key']}"
     puts "ZONE: #{config[@customer_key]['availability_zone']}"
@@ -81,6 +88,8 @@ class Ec2InstanceManager
 
     if options[:terminate]
       terminate
+    elsif options[:start_launch_plan]
+      start_launch_plan
     else
       launch unless options[:status]
     end

data/lib/ec2-instance-manager/launch.rb

@@ -5,12 +5,7 @@ module Launch
     ami_id = gets
     ami_id = ami_id.lstrip.rstrip
 
-    result = ec2.run_instances(
-    	:image_id => ami_id,
-    	:instance_type => config[@customer_key]['instance_type'],
-    	:key_name => config[@customer_key]['key'],
-    	:availability_zone => config[@customer_key]['availability_zone']
-    )
+    result = launch_ami(ami_id)
 
     instance_id = result.instancesSet.item[0].instanceId
     puts "=> Starting Instance Id: #{instance_id}"
@@ -54,21 +49,71 @@ module Launch
     instances = get_running_instances_list
     if instances and instances.any?
       puts
-      puts "Warning: Terminating all instances: #{instances.join(", ")}"
-      puts "Please cancel within the next 5 seconds..."
+      puts red("Warning: Terminating all instances: #{instances.join(", ")}")
+      puts red("Please cancel within the next 5 seconds...")
       sleep 5
       ec2.terminate_instances(:instance_id => instances)
       puts
       puts "All instances are going to terminate now."
+    else
+      puts
+      puts "No running instances."
+    end
+  end
+
+  def start_launch_plan
+    ami_ids_to_launch = []
+    puts
+    puts "Your Launch Plan:"
+    if config[@customer_key]["launch_plan"] and config[@customer_key]["launch_plan"].any?
+      config[@customer_key]["launch_plan"].keys.sort.each do |launch_plan_group|
+        puts
+        puts "Group: #{launch_plan_group}"
+        if config[@customer_key]["launch_plan"][launch_plan_group] and config[@customer_key]["launch_plan"][launch_plan_group].any?
+          config[@customer_key]["launch_plan"][launch_plan_group].keys.each do |ami_id|
+            puts "#{ami_id} => #{config[@customer_key]["launch_plan"][launch_plan_group][ami_id]} Instances to launch"
+            ami_ids_to_launch << [ami_id, config[@customer_key]["launch_plan"][launch_plan_group][ami_id]]
+          end
+        else
+          puts "No Ami Id's to launch defined."
+        end
+      end
+      
+      puts
+      puts red("Warning: Now launching your plan...")
+      puts red("Please cancel within the next 5 seconds...")
+      puts
+      sleep 5
+      
+      ami_ids_to_launch.each do |group_ami_id_pair|
+        group_ami_id_pair[1].times {
+          puts "Launching #{group_ami_id_pair[0]}..."
+          result = launch_ami(group_ami_id_pair[0])
+        }
+      end
+
+      puts "All instances are launching now..."
+    else
+      puts "No launch groups defined."
     end
   end
+  
+  def launch_ami(ami_id)
+    ec2.run_instances(
+    	:image_id => ami_id,
+    	:instance_type => config[@customer_key]['instance_type'],
+    	:key_name => config[@customer_key]['key'],
+    	:availability_zone => config[@customer_key]['availability_zone']
+    )
+  end
 
+  private
   def get_running_instances_list
     result = ec2.describe_instances
     if result and result["reservationSet"]
       result["reservationSet"]["item"].collect do |item|
         item["instancesSet"]["item"].first["instanceId"] if item["instancesSet"]["item"].first["instanceState"]["name"] == 'running'
-      end
+      end.compact
     end
   end
 end

data/lib/ec2-instance-manager/output.rb

@@ -0,0 +1,25 @@
+module Output
+  def output_running_state(running_state)
+    if running_state == 'running'
+      green(running_state)
+    elsif running_state == 'terminated'
+      red(running_state)
+    elsif running_state == 'pending'
+      yellow(running_state)
+    else
+      running_state
+    end
+  end
+  
+  def green(str)
+    "\033[32m#{str}\033[0m" 
+  end
+  
+  def red(str)
+    "\033[31m#{str}\033[0m"
+  end
+  
+  def yellow(str)
+    "\033[33m#{str}\033[0m"
+  end
+end

data/lib/ec2-instance-manager/status.rb

@@ -23,13 +23,13 @@ module Status
         ami_id = item["instancesSet"]["item"].first["imageId"]
         running_state = item["instancesSet"]["item"].first["instanceState"]["name"]
         dns_name = item["instancesSet"]["item"].first["dnsName"]
-        puts "Instance Id: #{instance_id} - #{running_state} (AMI Id: #{ami_id}) #{dns_name}"
+        puts "Instance Id: #{instance_id} - #{output_running_state(running_state)} (AMI Id: #{ami_id}) #{dns_name}"
       end
     else
       puts "No instances."
     end
   end
-  
+    
   def display_addresses
     result = ec2.describe_addresses
     if result and result["addressesSet"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: ec2-instance-manager
 version: !ruby/object:Gem::Version 
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors: 
 - Oliver Kiessler
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-09 00:00:00 +02:00
+date: 2009-10-16 00:00:00 +02:00
 default_executable: ec2-instance-manager
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -44,6 +44,7 @@ files:
 - lib/ec2-instance-manager.rb
 - lib/ec2-instance-manager/ec2_instance_manager.rb
 - lib/ec2-instance-manager/launch.rb
+- lib/ec2-instance-manager/output.rb
 - lib/ec2-instance-manager/status.rb
 - test/ec2-instance-manager_test.rb
 - test/test_helper.rb