elb 0.0.4

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.ruby-version

@@ -0,0 +1 @@
+ree-1.8.7-2012.02

data/.travis.yml

@@ -0,0 +1,9 @@
+language: ruby
+cache: bundler
+rvm:
+- 1.8.7
+- 2.0.0
+addons:
+  code_climate:
+    repo_token:
+      secure: AK6jL+T50l6SbGw/+xo6dpLUz++EQJl56DW+CDOp9+/SYZNL3PqtNGSApMEzXiAf1C+1TAhXRNkH8uyV7x8w6/SGFe+ZzxW1hupqJocDOaFL6sRHBXT9Fg9JNn3FULUgUjZDQn7t7u7QGBpykcGW3JNbKuyiQgfiSYSLx5Ee6OE=

data/Gemfile

@@ -0,0 +1,6 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in elb.gemspec
+gemspec
+
+gem "codeclimate-test-reporter", :group => :test, :require => nil

data/Guardfile

@@ -0,0 +1,12 @@
+guard 'rspec', :version => 2 do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})      { "spec/elb_spec.rb" }
+  watch(%r{^lib/elb/(.+)\.rb$})  { "spec/elb_spec.rb" }
+  watch('spec/spec_helper.rb')   { "spec/elb_spec.rb" }
+  watch(%r{^lib/elb/(.+)\.rb$})   { |m| "spec/lib/#{m[1]}_spec.rb" }
+end
+
+guard 'bundler' do
+  watch('Gemfile')
+  watch(/^.+\.gemspec/)
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Tung Nguyen
+
+MIT License
+
+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,57 @@
+# Elb
+
+[![Build Status](https://travis-ci.org/tongueroo/elb.svg?branch=master)](https://travis-ci.org/tongueroo/elb)
+[![Code Climate](https://codeclimate.com/github/tongueroo/elb.png)](https://codeclimate.com/github/tongueroo/elb)
+[![Code Climate](https://codeclimate.com/github/tongueroo/elb/coverage.png)](https://codeclimate.com/github/tongueroo/elb)
+
+Tool gracefully restarts app server without affecting users by deregistering the instance from the elb, restarting it and registering it back to the elb.  
+
+It makes the assumption that the instance belongs to an autoscaling group that was created with cloudformation.  Cloudformation automatically tags these instances with a `aws:autoscaling:groupName` tag, which is required.
+
+## Installation
+
+```
+$ gem install elb
+```
+
+## Setup
+
+The tool requires access to a limited number of AWS resources.  This tool is meant to be ran on an ec2 instance, so using an IAM role is ideal.  An [example IAM policy](https://github.com/tongueroo/elb/wiki/IAM-Role-Example) is on the wiki.
+
+If you cannot use IAM roles, you can use a credentials file instead.  The location of the file should be at ~/.br/aws.yml.  An example of the format of the file is provided [here](https://github.com/tongueroo/elb/blob/master/spec/fixtures/aws.yml).
+
+## Usage
+
+```
+$ elb restart
+```
+The `elb restart` command is meant to be ran on an ec2 instance and does the following:
+
+1. deregister the instance from the autoscaling group it's associated with
+2. restart the app
+3. warm up the app by hitting it with a local curl request
+4. register the instance back to the elb
+
+Options can be used to override default behaviors.
+
+```
+$ elb restart --wait 30 # wait 30 seconds after deregistering
+
+$ elb restart --warm-command "curl -s -v -o /dev/null localhost/custom-warm-url 2>&1 | grep '< HTTP'"
+
+$ elb restart --restart-command "restart nginx"
+
+$ elb help
+```
+
+### Local Usage
+
+You can also use the tool locally and test on remote ec2 instances by specifying the `--instance-id` option.  
+
+```bash
+$ elb restart --instance-id i-xxxxx
+
+$ elb deregister --instance-id i-xxxxx # only runs deregister part
+
+$ elb register --instance-id i-xxxxx # only runs register part
+```

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+task :default => :spec
+
+RSpec::Core::RakeTask.new

data/bin/elb

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+
+$:.unshift(File.expand_path('../../lib', __FILE__))
+require 'elb'
+require 'elb/cli'
+
+Elb::CLI.start(ARGV)

data/elb.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'elb/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "elb"
+  spec.version       = Elb::VERSION
+  spec.authors       = ["Tung Nguyen"]
+  spec.email         = ["tongueroo@gmail.com"]
+  spec.description   = %q{Tool to gracefully restart app servers without affecting users}
+  spec.summary       = %q{Tool to gracefully restart app servers without affecting users.}
+  spec.homepage      = "https://github.com/br/elb"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thor"
+  spec.add_dependency "aws-sdk"
+  spec.add_dependency "hashie"
+  spec.add_dependency "colorize"
+  spec.add_dependency "nokogiri", "1.5.11" # fix for 1.8.7
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+  if RUBY_VERSION != "1.8.7"
+    spec.add_development_dependency "guard"
+    spec.add_development_dependency "guard-bundler"
+    spec.add_development_dependency "guard-rspec"
+  end
+end

data/lib/elb.rb

@@ -0,0 +1,10 @@
+$:.unshift(File.expand_path("../", __FILE__))
+require "elb/version"
+
+module Elb
+  autoload :CLI, 'elb/cli'
+  autoload :Deploy, 'elb/deploy'
+  autoload :UI, 'elb/ui'
+  autoload :Helper, 'elb/helper'
+  autoload :Settings, 'elb/settings'
+end

data/lib/elb/cli.rb

@@ -0,0 +1,30 @@
+require 'thor'
+require 'elb/cli/help'
+
+module Elb
+
+  class CLI < Thor
+    class_option :noop, :type => :boolean
+    class_option :mute, :default => false, :type => :boolean
+    class_option :instance_id, :desc => 'instance id to use, defaults to instance command is running on'
+
+    desc "restart", "restart server"
+    long_desc Help.restart
+    option :wait, :type => :numeric, :default => 20, :desc => 'period to wait after deregistering'
+    option :warm_command, :desc => 'command use to warm app, defaults to curling homepage', :default => "curl -s -v -o /dev/null localhost/ 2>&1 | grep '< HTTP'"
+    option :restart_command, :desc => 'command use to restart app, defaults to touch tmp/restart.txt', :default => "touch tmp/restart.txt"
+    def restart
+      Deploy.new(options).run
+    end
+
+    desc "deregister", "deregister server from elb"
+    def deregister
+      Deploy.new(options.merge(:wait => 0)).deregister
+    end
+
+    desc "register", "register server with elb"
+    def register
+      Deploy.new(options).register
+    end
+  end
+end

data/lib/elb/cli/help.rb

@@ -0,0 +1,29 @@
+module Elb
+  class CLI < Thor
+    class Help
+      class << self
+        def restart
+<<-EOL
+The command is meant to be ran on the instance itself and will basically:
+
+1. deregister the instance from the autoscaling group it's associated with
+
+2. restart the app
+
+3. warm up the app by hitting it with a local curl request
+
+4. register the instance back to the elb
+
+Examples:
+
+$ elb restart --wait 30 # wait 30 seconds after deregistering
+
+$ elb restart --warm-command "curl -s -v -o /dev/null localhost/custom-warm-url 2>&1 | grep '< HTTP'"
+
+$ elb restart --restart-command "/etc/init.d/nginx restart"
+EOL
+        end
+      end
+    end
+  end
+end

data/lib/elb/deploy.rb

@@ -0,0 +1,65 @@
+module Elb
+  class Deploy
+    include Helper
+
+    attr_reader :instance_id
+    def initialize(options={})
+      @options = options
+      @instance_id = options[:instance_id] || %x[curl -s http://instance-data.ec2.internal/latest/meta-data/instance-id].strip
+      setup_aws
+    end
+
+    def run
+      UI.say("Restarting server started")
+      deregister
+      restart
+      warm
+      register
+      UI.say("Restarting server completed")
+    end
+
+    def asg
+      return @asg if @asg
+      name = ec2.instances[instance_id].tags["aws:autoscaling:groupName"]
+      @asg = as.groups[name]
+    end
+
+    def elb
+      return @elb if @elb
+      @elb = asg.load_balancers.first
+    end
+
+    def deregister
+      UI.say("Deregistering server from ELB")
+      return true if @options[:noop]
+      elb.client.deregister_instances_from_load_balancer(
+        :load_balancer_name => elb.name,
+        :instances => [{:instance_id => @instance_id}]
+      )
+      wait(@options[:wait]) # takes a while for the elb to deregister
+    end
+
+    def restart
+      UI.say("Restarting server")
+      system(@options[:restart_command])
+    end
+
+    def warm
+      UI.say("Warming up server")
+      system(@options[:warm_command])
+    end
+
+    def register
+      UI.say("Registering server with ELB")
+      return true if @options[:noop]
+      elb.client.register_instances_with_load_balancer(
+        :load_balancer_name => elb.name,
+        :instances => [{:instance_id => @instance_id}]
+      )
+    end
+
+    def wait(n)
+      sleep n
+    end
+  end
+end

data/lib/elb/helper.rb

@@ -0,0 +1,41 @@
+require 'aws-sdk'
+
+module Elb
+  module Helper
+    @@setup_aws = nil
+    def setup_aws(region='us-east-1', options={})
+      return if @@setup_aws
+      path = Elb::Settings.new.aws_path
+      if File.exist?(path)
+        @config = YAML.load(IO.read(path))
+        AWS.config(
+          :http_wire_trace => options[:debug],
+          :access_key_id => @config[:aws_access_key_id], 
+          :secret_access_key => @config[:aws_secret_access_key]
+        )
+      end
+      @@setup_aws = true
+    end
+
+    def ec2
+      @ec2 ||= AWS::EC2.new
+    end
+
+    def cfn
+      @cfn ||= AWS::CloudFormation.new
+    end
+
+    def rds
+      @rds ||= AWS::RDS.new
+    end
+
+    def as
+      @as ||= AWS::AutoScaling.new
+    end
+
+    def r53
+      @r53 ||= AWS::Route53.new
+    end
+
+  end
+end

data/lib/elb/settings.rb

@@ -0,0 +1,26 @@
+require 'yaml'
+require 'hashie'
+
+module Elb
+  class Settings
+    def aws_path
+      path = ENV['TEST'] ? "spec/fixtures/aws.yml" : "#{br_path}/aws.yml"
+      path = ENV['AWS'] if ENV['AWS']
+      path
+    end
+
+    def br_path
+      self.class.br_path
+    end
+
+    def self.br_path
+      ENV['BR_PATH'] || "#{ENV['HOME']}/.br"
+    end
+
+    def self.setup_br_path
+      FileUtils.mkdir(br_path) unless File.exist?(br_path)
+    end
+  end
+end
+
+Elb::Settings.setup_br_path

data/lib/elb/ui.rb

@@ -0,0 +1,16 @@
+module Elb
+  class UI
+    class << self
+      @@mute = false
+      def mute
+        @@mute
+      end
+      def mute=(v)
+        @@mute=v
+      end
+      def say(msg='')
+        puts msg unless mute
+      end
+    end
+  end
+end

data/lib/elb/version.rb

@@ -0,0 +1,3 @@
+module Elb
+  VERSION = "0.0.4"
+end

data/spec/fixtures/aws.yml

@@ -0,0 +1,3 @@
+---
+:aws_access_key_id: xxx
+:aws_secret_access_key: xxx

data/spec/lib/cli_spec.rb

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+describe Elb::CLI do
+  before(:all) do
+    ensure_tmp
+    @args = "--noop --instance-id #{instance_id} --mute"
+  end
+
+  describe "elb" do
+    it "should restart" do
+      out = execute("bin/elb restart #{@args}")
+      expect(out).to include("Restarting server started")
+      expect(out).to include("Restarting server completed")
+    end
+
+    it "should deregister" do
+      out = execute("bin/elb deregister #{@args}")
+      expect(out).to include("Deregistering server from ELB")
+    end
+
+    it "should register" do
+      out = execute("bin/elb register #{@args}")
+      expect(out).to include("Registering server with ELB")
+    end
+  end
+end

data/spec/lib/deploy_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+
+describe Elb::Deploy do
+  before(:all) do
+    AWS.stub!
+    ensure_tmp
+    @options = {
+      :instance_id => instance_id,
+      :noop => true,
+      :restart_command => "touch tmp/restart.txt"
+    }
+    Elb::UI.mute = true
+  end
+  let(:lb) { double("Aws::Elb") }
+  let(:deploy) { 
+    deploy = Elb::Deploy.new(@options)
+    allow(deploy.asg).to receive(:load_balancers).and_return([lb])
+    allow(deploy).to receive(:wait).and_return(true)
+    deploy
+  }
+
+  describe "deploy" do
+    it "should run" do
+      expect(deploy).to receive(:deregister)
+      expect(deploy).to receive(:restart)
+      expect(deploy).to receive(:warm)
+      expect(deploy).to receive(:register)
+      deploy.run
+    end
+
+    it "should have asg" do
+      expect(deploy.asg).to_not be_nil
+    end
+
+    it "should have elb" do
+      expect(deploy.elb).to_not be_nil
+    end
+
+    it "should deregister" do
+      expect(deploy.deregister).to be true
+    end
+
+    it "should register" do
+      expect(deploy.register).to be true
+    end
+
+    it "should restart" do
+      expect(deploy.restart).to be true
+    end
+
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,32 @@
+ENV['TEST'] = '1'
+
+if RUBY_VERSION != "1.8.7"
+  require "codeclimate-test-reporter"
+  CodeClimate::TestReporter.start
+end
+
+require "pp"
+ 
+root = File.expand_path('../../', __FILE__)
+require "#{root}/lib/elb"
+
+module Helpers
+  def execute(cmd)
+    puts "Running: #{cmd}" if ENV['DEBUG']
+    out = `#{cmd}`
+    puts out if ENV['DEBUG']
+    out
+  end
+
+  def instance_id
+    ENV['INSTANCE_ID']
+  end
+
+  def ensure_tmp
+    FileUtils.mkdir('tmp') unless File.exist?('tmp')
+  end
+end
+
+RSpec.configure do |c|
+  c.include Helpers
+end

metadata

@@ -0,0 +1,204 @@
+--- !ruby/object:Gem::Specification 
+name: elb
+version: !ruby/object:Gem::Version 
+  hash: 23
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 4
+  version: 0.0.4
+platform: ruby
+authors: 
+- Tung Nguyen
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2014-06-08 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: aws-sdk
+  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
+- !ruby/object:Gem::Dependency 
+  name: hashie
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: colorize
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: nokogiri
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 1
+        - 5
+        - 11
+        version: 1.5.11
+  type: :runtime
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 1
+        - 3
+        version: "1.3"
+  type: :development
+  version_requirements: *id006
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id007 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id007
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id008 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id008
+description: Tool to gracefully restart app servers without affecting users
+email: 
+- tongueroo@gmail.com
+executables: 
+- elb
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .ruby-version
+- .travis.yml
+- Gemfile
+- Guardfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/elb
+- elb.gemspec
+- lib/elb.rb
+- lib/elb/cli.rb
+- lib/elb/cli/help.rb
+- lib/elb/deploy.rb
+- lib/elb/helper.rb
+- lib/elb/settings.rb
+- lib/elb/ui.rb
+- lib/elb/version.rb
+- spec/fixtures/aws.yml
+- spec/lib/cli_spec.rb
+- spec/lib/deploy_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/br/elb
+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.15
+signing_key: 
+specification_version: 3
+summary: Tool to gracefully restart app servers without affecting users.
+test_files: 
+- spec/fixtures/aws.yml
+- spec/lib/cli_spec.rb
+- spec/lib/deploy_spec.rb
+- spec/spec_helper.rb