ec2-starter 0.1.0

data/.document

@@ -0,0 +1,5 @@
+lib/**/*.rb
+bin/*
+- 
+features/**/*.feature
+LICENSE.txt

data/Gemfile

@@ -0,0 +1,14 @@
+source "http://rubygems.org"
+# Add dependencies required to use your gem here.
+# Example:
+gem "amazon-ec2", ">= 0.9.17"
+gem 'net-ssh', '>= 2.1.4'
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "shoulda", ">= 0"
+  gem "bundler", "~> 1.0.0"
+  gem "jeweler", "~> 1.5.2"
+  gem "rcov", ">= 0"
+end

data/Gemfile.lock

@@ -0,0 +1,26 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    amazon-ec2 (0.9.17)
+      xml-simple (>= 1.0.12)
+    git (1.2.5)
+    jeweler (1.5.2)
+      bundler (~> 1.0.0)
+      git (>= 1.2.5)
+      rake
+    net-ssh (2.1.4)
+    rake (0.9.2)
+    rcov (0.9.9)
+    shoulda (2.11.3)
+    xml-simple (1.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  amazon-ec2 (>= 0.9.17)
+  bundler (~> 1.0.0)
+  jeweler (~> 1.5.2)
+  net-ssh (>= 2.1.4)
+  rcov
+  shoulda

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2011 Oliver Kiessler
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,12 @@
+ec2-starter
+-----------
+
+Very Ruby DSL to perform standard tasks on a single EC2 instance. With this you can start an instance, assign an IP, assign volumes and perform SSH shell commands.
+
+See example.rb
+
+Copyright
+---------
+
+Copyright (c) 2011 Oliver Kiessler. See LICENSE.txt for
+further details.

data/Rakefile

@@ -0,0 +1,50 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'rake'
+
+require 'jeweler'
+Jeweler::Tasks.new do |gem|
+  gem.name = "ec2-starter"
+  gem.homepage = "http://github.com/okiess/ec2-starter"
+  gem.license = "MIT"
+  gem.summary = %Q{Very simple EC2 instance starter}
+  gem.description = %Q{Very simple EC2 instance starter for single server deployments}
+  gem.email = "kiessler@inceedo.com"
+  gem.authors = ["Oliver Kiessler"]
+  gem.add_runtime_dependency 'amazon-ec2'
+  gem.add_development_dependency 'net-ssh'
+end
+Jeweler::RubygemsDotOrgTasks.new
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+require 'rcov/rcovtask'
+Rcov::RcovTask.new do |test|
+  test.libs << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "ec2-starter #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/example.rb

@@ -0,0 +1,26 @@
+require "lib/ec2-starter"
+
+service_options = {
+  :access_key_id => "YOUR KEY",
+  :secret_access_key => "YOUR SECRET",
+  :server => "eu-west-1.ec2.amazonaws.com" # Target Zone
+}
+
+start_options = {
+  :instance_type => "t1.micro",
+  :key_name => "YOUR KEY",
+  :availability_zone => "eu-west-1a",
+  :architecture => "x86_64",
+  :kernel_id => 'aki-4feec43b', # your kernel, remove if not needed
+  :ssh_user => 'deploy',
+  :ssh_keys => ['/Users/your_user/.ssh/your_key']
+  :sudo_password => nil # set if necessary
+}
+
+Ec2Starter.start 'YOUR AMI_ID', service_options, start_options do
+  ip 'YOUR ELASTIC IP'
+  volume :volume_id => 'YOUR VOLUME', :mount_point => '/dev/sdf'
+  
+  command :sudo => '/root/attach_volume.sh' # Shell command on the instance
+  command '/home/your_user/some_script.sh'
+end

data/lib/ec2-starter.rb

@@ -0,0 +1,8 @@
+require 'rubygems'
+gem "net-ssh"
+require 'net/ssh'
+gem 'amazon-ec2'
+require 'AWS'
+
+require File.dirname(__FILE__) + '/ec2-starter/starter'
+require File.dirname(__FILE__) + '/ec2-starter/execution'

data/lib/ec2-starter/execution.rb

@@ -0,0 +1,148 @@
+module Ec2Starter
+  class Execution
+    attr_accessor :ec2_instance, :dns_name
+
+    def initialize(ec2_instance)
+      @ec2_instance = ec2_instance
+    end
+
+    def start
+      success = false
+      result = launch_ami(@ec2_instance.ami_id)
+      instance_id = result.instancesSet.item[0].instanceId
+      puts "=> Starting Instance Id: #{instance_id.white}"
+
+      instance_state = nil
+      while(instance_state != 'running')
+        instance_state, @dns_name = get_instance_state(instance_id)
+        puts "=> Checking for running state... #{output_running_state(instance_state)}"
+        puts "=> Public DNS: #{@dns_name.white}" if instance_state == 'running'
+        sleep 10 unless instance_state == 'running'
+      end
+
+      if @ec2_instance.ips.size > 0
+        result = ec2.associate_address(:instance_id => instance_id, :public_ip => @ec2_instance.ips.first)
+        puts "=> Elastic IP: #{@ec2_instance.ips.first.white}" if result["return"] == "true"
+      else
+        skip_ip = true
+      end
+      if result["return"] == "true" or skip_ip
+        if @ec2_instance.volumes.size > 0
+          result = ec2.attach_volume(:volume_id => @ec2_instance.volumes.first[:volume_id], :instance_id => instance_id,
+           :device => @ec2_instance.volumes.first[:mount_point])
+
+          volume_state = nil
+          while(volume_state != 'attached')
+            volume_state = get_volume_state(@ec2_instance.volumes.first[:volume_id])
+            puts "=> Checking for attachment state... #{output_running_state(volume_state)}"
+            sleep 10 unless volume_state == 'attached'
+          end
+
+          success = true
+          puts "=> EBS Volume attached: #{@ec2_instance.volumes.first[:volume_id]}"
+        end
+        if @ec2_instance.commands.size > 0
+          ip = @ec2_instance.ips.first || @dns_name
+          system("ssh-keygen -R #{ip}")
+          i = 0
+          while(i < 3) do
+            begin
+              ssh
+              break
+            rescue => e
+              puts e.message
+              puts "Retrying in 10sec..."
+              sleep 10
+            end
+            i += 1
+          end
+          success = true
+          puts "=> SSH Commands were executed!"
+        end
+      else
+        puts "=> Could not assign Elastic IP!"
+      end
+
+      unless success
+        puts "Terminating instance: #{instance_id}".red
+        @ec2.terminate_instances(:instance_id => instance_id)
+      end
+    end
+
+    def ssh
+      ip = @ec2_instance.ips.first || @dns_name
+      Net::SSH.start(ip, @ec2_instance.default_options[:ssh_user], :keys => @ec2_instance.default_options[:ssh_keys]) do |ssh|
+        @ec2_instance.commands.each do |command|
+          if command.is_a?(String)
+            ssh.exec(command)
+          elsif command.is_a?(Hash) and command.include?(:sudo)
+            ssh.sudo @ec2_instance.default_options[:sudo_password], command[:sudo]
+          end
+        end
+        ssh.close
+      end
+    end
+
+    def ec2
+       @ec2 ||= AWS::EC2::Base.new(:access_key_id => @ec2_instance.service_options[:access_key_id],
+        :secret_access_key => @ec2_instance.service_options[:secret_access_key],
+        :server => @ec2_instance.service_options[:server])
+    end
+
+    def get_instance_state(instance_id)
+      result = ec2.describe_instances(:instance_id => instance_id)
+      instance_state = result["reservationSet"]["item"].first["instancesSet"]["item"].first["instanceState"]["name"]
+      dns_name = result["reservationSet"]["item"].first["instancesSet"]["item"].first["dnsName"]
+      return instance_state, dns_name
+    end
+
+    def get_volume_state(volume_id)
+      result = ec2.describe_volumes(:volume_id => volume_id)
+      volume_state = result["volumeSet"]["item"].first["attachmentSet"]["item"].first["status"]
+      return volume_state
+    end
+
+    def launch_ami(ami_id, options = {})
+      default_options = @ec2_instance.default_options.merge(:image_id => ami_id).merge(options)
+      run_options = default_options.merge(options)
+      puts "Launch Options: #{run_options.inspect}"
+      ec2.run_instances(run_options)
+    end
+
+    def output_running_state(running_state)
+      if running_state == 'running' or running_state == 'attached'
+        running_state.green
+      elsif running_state == 'terminated' or running_state == 'shutting-down'
+        running_state.red
+      elsif running_state == 'pending' or running_state == 'attaching'
+        running_state.yellow
+      else
+        running_state
+      end
+    end
+  end
+end
+
+class String
+  def green
+    "\033[32m#{self}\033[0m" 
+  end
+
+  def red
+    "\033[31m#{self}\033[0m"
+  end
+
+  def yellow
+    "\033[33m#{self}\033[0m"
+  end
+
+  def white
+    "\033[1m#{self}\033[0m"
+  end
+end
+
+class Net::SSH::Connection::Session
+  def sudo(password, command)
+    exec %Q%echo "#{password}" | sudo -S #{command}% 
+  end
+end

data/lib/ec2-starter/starter.rb

@@ -0,0 +1,34 @@
+module Ec2Starter
+  def self.start(ami_id, service_options, run_options, &block)
+    instance = Instance.new(ami_id, service_options, run_options)
+    instance.instance_eval(&block)
+    ec2_starter = Ec2Starter::Execution.new(instance)
+    ec2_starter.start
+    return instance
+  end
+
+  class Instance
+    attr_accessor :service_options, :default_options, :ami_id, :instance_id, :ips, :volumes, :commands
+
+    def initialize(ami_id, service_options, options)
+      @ami_id = ami_id.to_s
+      @ips = []
+      @volumes = []
+      @commands = []
+      @default_options = {}.merge(options)
+      @service_options = {}.merge(service_options)
+    end
+
+    def ip(ip_address)
+      @ips << ip_address
+    end
+
+    def volume(volume_hash)
+      @volumes << volume_hash
+    end
+
+    def command(command)
+      @commands << command
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'ec2-starter'
+
+class Test::Unit::TestCase
+end

data/test/test_ec2-starter.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestEc2Starter < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

metadata

@@ -0,0 +1,199 @@
+--- !ruby/object:Gem::Specification 
+name: ec2-starter
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Oliver Kiessler
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-06-05 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: amazon-ec2
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 25
+        segments: 
+        - 0
+        - 9
+        - 17
+        version: 0.9.17
+  type: :runtime
+  requirement: *id001
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 1
+        - 4
+        version: 2.1.4
+  type: :runtime
+  requirement: *id002
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: shoulda
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  requirement: *id003
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  type: :development
+  requirement: *id004
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: jeweler
+  version_requirements: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 1
+        - 5
+        - 2
+        version: 1.5.2
+  type: :development
+  requirement: *id005
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: rcov
+  version_requirements: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  requirement: *id006
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: amazon-ec2
+  version_requirements: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  requirement: *id007
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  name: net-ssh
+  version_requirements: &id008 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  requirement: *id008
+  prerelease: false
+description: Very simple EC2 instance starter for single server deployments
+email: kiessler@inceedo.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE.txt
+- README.md
+files: 
+- .document
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- example.rb
+- lib/.DS_Store
+- lib/ec2-starter.rb
+- lib/ec2-starter/execution.rb
+- lib/ec2-starter/starter.rb
+- test/helper.rb
+- test/test_ec2-starter.rb
+homepage: http://github.com/okiess/ec2-starter
+licenses: 
+- MIT
+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.8.24
+signing_key: 
+specification_version: 3
+summary: Very simple EC2 instance starter
+test_files: 
+- test/helper.rb
+- test/test_ec2-starter.rb