ec2ctl 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: bea0a185379f61297a427c2c7efc2af67df6b763
+  data.tar.gz: 6135a1d00da6cbb6df57a8f184d24dc8d87104d3
+SHA512:
+  metadata.gz: 8729284c92f26b5220aab5925715b3056696a3a03fd116254dd8920e2d3d548baa0f1b5595b185fc3b3b4b11c63ef04979f044171c3bff3c6a97e3313e839949
+  data.tar.gz: e4451be5017a641a415e2f5da2e8463ac2daf598c32d0d4025b1683d4dae3895e16308de8353148b5cfc738a82c43e80233c7f5a93285e7e638ee39dded91c63

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ec2ctl.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 y13i
+
+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,52 @@
+# Ec2ctl
+
+**!!! WORK IN PROGRESS !!!**
+
+TODO: Add spec. etc...
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'ec2ctl'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ec2ctl
+
+## Usage
+
+TODO: Write more usage instructions here
+
+```
+Commands:
+  ec2ctl add NAME ELB_NAME     # Add instance to ELB. Instance will be searched by Name tag.
+  ec2ctl help [COMMAND]        # Describe available commands or one specific command
+  ec2ctl lb ELB_NAME           # Show load balancer registered instance statuses.
+  ec2ctl list PATTERN          # List all instances in region. Specify PATTERN to filter results by name.
+  ec2ctl remove NAME ELB_NAME  # Remove instance from ELB. Instance will be searched by Name tag.
+  ec2ctl ssh NAME COMMAND      # Execute remote command to specified instance by given name.
+  ec2ctl start NAME            # Start instance. Instance will be searched by Name tag.
+  ec2ctl status NAME           # Show instance status. Instance will be searched by Name tag.
+  ec2ctl stop NAME             # Stop instance. Instance will be searched by Name tag.
+
+Options:
+  k, [--access-key-id=ACCESS_KEY_ID]
+  s, [--secret-access-key=SECRET_ACCESS_KEY]
+  r, [--region=REGION]
+      [--profile=PROFILE]
+      [--color], [--no-color]
+                                              # Default: true
+```
+
+## Contributing
+
+1. Fork it ( http://github.com/y13i/ec2ctl/fork )
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/ec2ctl

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require "ec2ctl"
+
+Ec2ctl::CLI.start ARGV

data/ec2ctl.gemspec

@@ -0,0 +1,30 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ec2ctl/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "ec2ctl"
+  spec.version       = Ec2ctl::VERSION
+  spec.authors       = ["y13i"]
+  spec.email         = ["email@y13i.com"]
+  spec.summary       = %q{Yet another handy EC2 tools.}
+  spec.description   = %q{Can start/stop instances, add/remove to load balancer, remote command via ssh.}
+  spec.homepage      = "https://github.com/y13i/ec2ctl"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  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 "net-ssh"
+  spec.add_dependency "colorize"
+  spec.add_dependency "terminal-table"
+  spec.add_dependency "unindent"
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+end

data/lib/ec2ctl/version.rb

@@ -0,0 +1,3 @@
+module Ec2ctl
+  VERSION = "0.1.0"
+end

data/lib/ec2ctl.rb

@@ -0,0 +1,198 @@
+require "ec2ctl/version"
+
+require "thor"
+require "aws-sdk"
+require "net/ssh"
+require "net/http"
+require "colorize"
+require "terminal-table"
+require "timeout"
+require "unindent"
+
+module Ec2ctl
+  class CLI < Thor
+    own_region = begin
+      timeout 3 do
+        Net::HTTP.get("169.254.169.254", "/latest/meta-data/placement/availability-zone")[0..-2]
+      end
+    rescue
+      nil
+    end
+
+    class_option :access_key_id,     default: nil,        aliases: [:k]
+    class_option :secret_access_key, default: nil,        aliases: [:s]
+    class_option :region,            default: own_region, aliases: [:r]
+    class_option :profile
+    class_option :color,             default: true,       type: :boolean
+
+    desc "list PATTERN", "List all instances in region. Specify PATTERN to filter results by name."
+    def list pattern = ""
+      rows = []
+
+      ec2.instances.each do |instance|
+        name = instance.tags["Name"]
+        next if !pattern.empty? and !name.match pattern
+
+        rows << [
+          instance.id,
+          name,
+          instance.instance_type,
+          instance.private_ip_address,
+          instance.public_ip_address,
+          status_colorize(instance.status),
+        ]
+      end
+
+      puts Terminal::Table.new headings: ["ID", "Name", "Type", "Private IP", "Public IP", "Status"], rows: rows
+    end
+
+    desc "status NAME", "Show instance status. Instance will be searched by Name tag."
+    def status name
+      instance = find_instance_by_name name
+      tags     = instance.tags.to_h
+
+      puts <<-EOS.unindent
+        * Name:              #{tags["Name"]}
+        * Status:            #{status_colorize instance.status}
+        * Instance Type:     #{instance.instance_type}
+        * Private IP:        #{instance.private_ip_address}
+        * Public IP:         #{instance.public_ip_address}
+        * Availability Zone: #{instance.availability_zone}
+        * Other Tags:        #{tags.reject {|k, v| k == "Name"}.inspect}
+      EOS
+    end
+
+    desc "start NAME", "Start instance. Instance will be searched by Name tag."
+    def start name
+      instance = find_instance_by_name name
+      abort "Instance is not stopped!" unless instance.status == :stopped
+      instance.start
+      sleep 2
+      puts "Instance is now #{status_colorize instance.status}."
+    end
+
+    desc "stop NAME", "Stop instance. Instance will be searched by Name tag."
+    def stop name
+      instance = find_instance_by_name name
+      abort "Instance is not running!" unless instance.status == :running
+      instance.stop
+      sleep 2
+      puts "Instance is now #{status_colorize instance.status}."
+    end
+
+    option :user,          default: "ec2-user",      aliases: [:u]
+    option :port,          default: 22,              aliases: [:p]
+    option :identity_file, default: "~/.ssh/id_rsa", aliases: [:i]
+    option :passphrase,    default: nil,             aliases: [:P]
+    option :via_public_ip, default: false,           type: :boolean
+    desc "ssh NAME COMMAND", "Execute remote command to specified instance by given name."
+    def ssh name, command
+      instance   = find_instance_by_name name
+      ip_address = options[:via_public_ip] ? instance.public_ip_address : instance.private_ip_address
+
+      ssh_options = {
+        keys:       [options[:identity_file]],
+        passphrase: options[:passphrase],
+      }
+
+      Net::SSH.start ip_address, options[:user], ssh_options do |ssh|
+        puts ssh.exec!(command)
+      end
+    end
+
+    desc "add NAME ELB_NAME", "Add instance to ELB. Instance will be searched by Name tag."
+    def add name, elb_name
+      instance      = find_instance_by_name name
+      load_balancer = find_load_balancer_by_name elb_name
+
+      abort "Instance is already registered to the load balancer!" if load_balancer.instances.include? instance
+
+      puts "Adding instance to the load balancer..."
+      load_balancer.instances.add instance
+    end
+
+    desc "remove NAME ELB_NAME", "Remove instance from ELB. Instance will be searched by Name tag."
+    def remove name, elb_name
+      instance      = find_instance_by_name name
+      load_balancer = find_load_balancer_by_name elb_name
+
+      abort "Instance is not registered to the load balancer!" unless load_balancer.instances.include? instance
+
+      puts "Removing instance from the load balancer..."
+      load_balancer.instances.remove instance
+    end
+
+    desc "lb ELB_NAME", "Show load balancer registered instance statuses."
+    def lb elb_name
+      load_balancer = find_load_balancer_by_name elb_name
+      instances     = load_balancer.instances
+      rows          = []
+
+      load_balancer.instances.health.each do |instance_health|
+        next unless instances.any? {|instance| instance.id == instance_health[:instance].id}
+
+        rows << [
+          instance_health[:instance].tags["Name"],
+          instance_health[:instance].id,
+          instance_health[:description],
+          instance_health[:state],
+        ]
+      end
+
+      puts Terminal::Table.new headings: ["Name", "Instance ID", "Description", "State"], rows: rows
+    end
+
+    private
+
+    def aws_config
+      hash = {}
+
+      if options[:profile]
+        provider = AWS::Core::CredentialProviders::SharedCredentialFileProvider.new profile_name: options[:profile]
+        hash.update credential_provider: provider
+      else
+        hash.update access_key_id: options[:access_key_id], secret_access_key: options[:secret_access_key] if options[:access_key_id] && options[:secret_access_key]
+      end
+
+      hash.update region: options[:region] if options[:region]
+      hash
+    end
+
+    def ec2
+      AWS::EC2.new aws_config
+    end
+
+    def elb
+      AWS::ELB.new aws_config
+    end
+
+    def status_colorize status
+      return status unless options[:color]
+
+      color = case status
+      when :pending, :stopping then :magenta
+      when :running            then :green
+      when :stopped            then :light_blue
+      when :shutting_down      then :yellow
+      when :terminated         then :red
+      else                     :default
+      end
+
+      status.to_s.colorize color
+    end
+
+    def find_instance_by_name name
+      instance = ec2.instances.with_tag("Name", name).first
+      abort "No instance found!" unless instance
+      puts "Instance found: " + instance.inspect
+      instance
+    end
+
+    def find_load_balancer_by_name name
+      load_balancer = elb.load_balancers[name]
+      abort "No load balancer found!" unless load_balancer
+      puts "Load balancer found: " + load_balancer.inspect
+      load_balancer
+    end
+  end
+end

metadata

@@ -0,0 +1,167 @@
+--- !ruby/object:Gem::Specification
+name: ec2ctl
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- y13i
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-08-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: unindent
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Can start/stop instances, add/remove to load balancer, remote command
+  via ssh.
+email:
+- email@y13i.com
+executables:
+- ec2ctl
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/ec2ctl
+- ec2ctl.gemspec
+- lib/ec2ctl.rb
+- lib/ec2ctl/version.rb
+homepage: https://github.com/y13i/ec2ctl
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Yet another handy EC2 tools.
+test_files: []