claws 1.0.0

data/README.md

@@ -0,0 +1,80 @@
+## Command Line AWS (CLAWS) [![Build Status](https://secure.travis-ci.org/wbailey/claws.png?branch=master)](http://travis-ci.org/wbailey/claws) 
+
+This tool provides a simple and powerful way to interact with the hosts in your Amazon Web Services
+account via the command line.
+
+It provides a configurable text based interface of the status of your hosts similar to the AWS web
+console.  An example of its usage appears below:
+
+![Instances](http://i.imgur.com/VEC8V.png)
+
+It gives you a choice of which host to connect to and invokes the proper ssh command:
+
+![Connecting](http://i.imgur.com/b6ieS.png)
+
+### Installation
+
+It is up on rubygems.org so add it to your bundle in the Gemfile
+
+```bash
+gem 'claws'
+```
+
+or do it the old fashioned way:
+
+```bash
+gem install claws
+```
+
+After you have the gem installed you will need to _initialize_ it to install a configuation file:
+
+```bash
+claws --init
+```
+
+Read more about the configuration file in this [wiki post](https://github.com/wbailey/claws/wiki/Configuration-file)
+
+### Usage
+
+Once the initial configuration is completed you can simply run claws from the command line:
+
+```bash
+claws
+```
+
+The full list of options includes:
+
+```bash
+Usage: claws [options]
+    -s, --status-only                Display host status only and exit
+    -c, --choice N                   Enter the number of the host to automatically connect to
+    -i, --init                       Install the default configuration file for the application
+```
+
+### To Do
+
+* Add filtering by regular expression for any field
+
+* Integrate with your projects Capistrano definitions so that one can filter by environment and
+  roles.
+
+* Add RDS and ELB support
+
+### License
+
+Copyright (c) 2011-2012 Wes Bailey
+
+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/bin/claws

@@ -0,0 +1,36 @@
+#!/bin/env ruby
+require 'claws'
+require 'optparse'
+require 'ostruct'
+require 'yaml'
+
+options = OpenStruct.new(
+  {
+    :config_file => nil,
+    :connect => true,
+    :initialize => false,
+    :selection => nil,
+  }
+)
+
+OptionParser.new do |opts|
+  opts.banner = 'Usage: claws [options]'
+
+  opts.on('-s', '--status-only', 'Display host status only and exit') do
+    options.connect = false
+  end
+
+  opts.on('-c', '--choice N', Float, 'Enter the number of the host to automatically connect to') do |n|
+    options.selection = n.to_i
+  end
+
+  opts.on('-i', '--init', 'Install the default configuration file for the application') do
+    options.initialize = true
+  end
+end.parse!
+
+if options.initialize
+  Claws::Command::Initialize.exec
+else
+  Claws::Command::EC2.exec options
+end

data/lib/claws.rb

@@ -0,0 +1,6 @@
+require 'claws/options'
+require 'claws/configuration'
+require 'claws/collection/ec2'
+require 'claws/report/ec2'
+require 'claws/command/initialize'
+require 'claws/command/ec2'

data/lib/claws/capistrano.rb

@@ -0,0 +1,63 @@
+module Claws
+  class Capistrano
+    attr_accessor :home
+
+    def initialize(home = nil)
+      self.home = home || File.join('config', 'deploy')
+    end
+
+    def all_host_roles
+      @all_roles ||= begin
+        roles = Hash.new
+
+        Dir.glob(File.join(self.home, '**/*.rb')).each do |f|
+          environment = File.basename(f)[0..-4]
+          roles[environment.to_sym] = get_roles(environment)
+        end
+
+        roles
+      end
+    end
+
+    def roles(host)
+      self.all_host_roles.each do |env, hh|
+        hh.each do |k,v|
+          return v if k == host
+        end
+      end
+    end
+
+    private
+
+    def get_roles(environment)
+      role_records = File.readlines(File.join(self.home, "#{environment}.rb")).select {|r| r.match(/^role/)}
+
+      # At this point we have an array of strings with:
+      #
+      # [
+      #   "role :app, \"ec2-263-56-231-91.compute-1.amazonaws.com\", 'ec2-263-23-118-57.compute-1.amazonaws.com'",
+      #   "role :web, \"ec2-263-56-231-91.compute-1.amazonaws.com\", 'ec2-263-23-118-57.compute-1.amazonaws.com', \"ec2-23-20-43-198.compute-1.amazonaws.com\"",
+      # ]
+      #
+      # and we want to convert that to:
+      #
+      # {
+      #   'ec2-263-56-231-91.compute-1.amazonaws.com' => ["app", "web"],
+      #   'ec2-263-23-118-57.compute-1.amazonaws.com' => ["app", "web"],
+      #   'ec2-23-20-43-198.compute-1.amazonaws.com' => ["web"],
+      # }
+
+      roles = Hash.new {|h,k| h[k] = Array.new}
+
+      role_records.each do |record|
+        role, *hosts = record.split(',').map {|v| v.strip.chomp.gsub(/"|'/, '')}
+
+        hosts.each do |host|
+          roles[host] << role.split(':')[1]
+        end
+      end
+
+      roles
+    end
+  end
+end

data/lib/claws/collection/base.rb

@@ -0,0 +1,19 @@
+require 'aws-sdk'
+
+module Claws
+  module Collection
+    class Base
+      def self.connect(credentials)
+        AWS.config(credentials)
+        AWS.start_memoizing
+      end
+
+      # Seems unnecessary
+      def self.build
+        collection = []
+        yield(collection)
+        collection
+      end
+    end
+  end
+end

data/lib/claws/collection/ec2.rb

@@ -0,0 +1,17 @@
+require 'aws-sdk'
+require 'claws/collection/base'
+require 'claws/presenter/ec2'
+
+module Claws
+  module Collection
+    class EC2 < Claws::Collection::Base
+      def self.get(filters = {})
+        collection = []
+        AWS::EC2.new.instances.each do |instance|
+          collection << Claws::EC2::Presenter.new(instance)
+        end
+        collection
+      end
+    end
+  end
+end

data/lib/claws/command/ec2.rb

@@ -0,0 +1,43 @@
+module Claws
+  module Command
+    class EC2
+      def self.exec(options)
+        begin
+          config = Claws::Configuration.new( options.config_file )
+        rescue Claws::ConfigurationError => e
+          puts e.message
+          puts 'Use the --init option to create a configuration file.'
+          exit 1
+        end
+
+        Claws::Collection::EC2.connect( config.aws )
+
+        begin
+          instances = Claws::Collection::EC2.get
+        rescue Exception => e
+          puts e.message
+        end
+
+        Claws::Report::EC2.new( config, instances ).run
+
+        if options.connect
+          if instances.size == 1
+            puts
+            selection = 0
+          elsif options.selection
+            puts
+            selection = options.selection
+          else
+            print "Select server (enter q to quit): "
+            selection = gets.chomp
+            exit 0 if selection.match(/^q.*/i)
+          end
+
+          puts 'connecting to server...'
+
+          system "ssh #{config.ssh.user}@#{instances[selection.to_i].dns_name}"
+        end
+      end
+    end
+  end
+end

data/lib/claws/command/initialize.rb

@@ -0,0 +1,63 @@
+require 'yaml'
+
+module Claws
+  module Command
+    class Initialize
+      def self.exec
+        h = {
+          'capistrano' => {
+            'home' => nil,
+          },
+          'ssh' => {
+            'user' => nil,
+          },
+          'aws' => {
+            'access_key_id' => nil,
+            'secret_access_key' => nil,
+          },
+          'ec2' => {
+            'fields' => {
+              'id' => {
+                'width' => 10,
+                'title' => 'ID',
+              },
+              'name' => {
+                'width' => 20,
+                'title' => 'Name',
+              },
+              'status' => {
+                'width' => 8,
+                'title' => 'Status',
+              },
+              'dns_name' => {
+                'width' => 42,
+                'title' => 'DNS Name',
+              },
+              'instance_type' => {
+                'width' => 13,
+                'title' => 'Instance Type',
+              },
+              'public_ip_address' => {
+                'width' => 16,
+                'title' => 'Public IP',
+              },
+              'private_ip_address' => {
+                'width' => 16,
+                'title' => 'Private IP',
+              },
+              'tags' => {
+                'width' => 30,
+                'title' => 'tags',
+              },
+            },
+          }
+        }
+
+        conf = File.join(ENV['HOME'], '.claws.yml')
+        puts "Creating configuration file: #{conf}\n..."
+        File.open(conf, 'w').write(h.to_yaml)
+        puts 'Complete!'
+      end
+    end
+  end
+end

data/lib/claws/configuration.rb

@@ -0,0 +1,25 @@
+require 'yaml'
+require 'ostruct'
+
+module Claws
+  class ConfigurationError < StandardError; end
+
+  class Configuration
+    attr_accessor :path, :capistrano, :ssh, :aws, :ec2
+
+    def initialize(use_path = nil)
+      self.path = use_path || File.join(ENV['HOME'], '.claws.yml')
+
+      begin
+        yaml = YAML.load_file(path)
+      rescue Exception
+        raise ConfigurationError, "Unable to locate configuration file: #{self.path}"
+      end
+
+      self.capistrano = OpenStruct.new( yaml['capistrano'] )
+      self.ssh = OpenStruct.new( yaml['ssh'] )
+      self.aws = yaml['aws']
+      self.ec2 = OpenStruct.new( { :fields => yaml['ec2']['fields'] } )
+    end
+  end
+end

data/lib/claws/options.rb

@@ -0,0 +1,37 @@
+require 'ostruct'
+
+module Claws
+  class Options
+    def self.parse
+      options = OpenStruct.new(
+        {
+          :connect => true,
+          :source => 'ec2',
+          :choice => nil,
+        }
+      )
+
+      OptionParser.new do |opts|
+        opts.banner = "Usage: script/aws [options] [environment] [role]"
+
+        opts.on('-d', '--display-only', 'display host information only and exit') do
+          options.connect = false
+        end
+
+        opts.on('-c', '--choice N', Float, 'enter the identity number of the host to automatically connect to') do |n|
+          options.choice = n.to_i
+        end
+
+        opts.on('-s', '--source', 'define the AWS source - default is ec2') do
+          options.source = 'elb'
+          options.connect = false
+        end
+      end.parse!
+
+      options.environment = ARGV.shift
+      options.role = ARGV.shift
+
+      options
+    end
+  end
+end

data/lib/claws/presenter/ec2.rb

@@ -0,0 +1,43 @@
+require 'claws/capistrano'
+require 'claws/support'
+
+module Claws
+  module EC2
+    class Presenter
+      attr_writer :roles
+
+      def initialize(instance, has_roles = [])
+        @ec2 = instance.extend(Claws::Support)
+        @roles = has_roles
+        freeze
+      end
+
+      def roles
+        @roles.empty? ? 'N/A' : @roles.join(', ')
+      end
+
+      def tags
+        @ec2.try(:tags) ? @ec2.tags.select {|k,v| [k,v] unless k.downcase == 'name'}.map{|k,v| "#{k}: #{v}"}.join(', ') : 'N/A'
+      end
+
+      def security_groups
+        @ec2.try(:security_groups) ? @ec2.security_groups.map {|sg| "#{sg.id}: #{sg.name}"}.join(', ') : 'N/A'
+      end
+
+      def method_missing(meth)
+        case meth
+        when :name
+          @ec2.send(:tags)['Name'] || 'N/A'
+        when @ec2.try(:tags) && @ec2.tags.has_key?(meth)
+          @ec2.tags[meth] || 'N/A'
+        else
+          begin
+            @ec2.send(meth)
+          rescue Exception
+            'N/A'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/claws/report/ec2.rb

@@ -0,0 +1,54 @@
+require 'command_line_reporter'
+
+module Claws
+  module Report
+    class EC2
+      attr_accessor :config, :instances
+
+      include CommandLineReporter
+
+      def initialize config, instances
+        self.config = config
+        self.instances = instances
+      end
+
+      def run
+        table :border => true do
+          row :header => true do
+            column 'Choice', :width => 6, :color => 'blue', :bold => true, :align => 'right'
+
+            self.config.ec2.fields.each do |field, properties|
+              text = properties['title'] || field
+              width = properties['width'] || nil
+              column text, :width => width, :color => 'blue', :bold => true
+            end
+          end
+
+          choice = 0
+
+          self.instances.each do |i|
+            color = case i.status
+                    when :running
+                      'green'
+                    when :stopped
+                      'red'
+                    else
+                      'white'
+                    end
+
+            row do
+              column choice
+
+              self.config.ec2.fields.each do |field, properties|
+                props = ( field == 'status' ) ? {:color => color} : {}
+                column i.send( field ), props
+              end
+            end
+
+            choice += 1
+          end
+        end
+      end
+    end
+  end
+end

data/lib/claws/support.rb

@@ -0,0 +1,12 @@
+# Wish this was part of the ruby core
+module Claws
+  module Support
+    def try(meth, *args, &block)
+      self.respond_to?(meth) ? self.send(meth, *args, &block) : nil
+    end
+  end
+end
+
+class Array
+  include Claws::Support
+end

data/lib/claws/version.rb

@@ -0,0 +1,3 @@
+module Claws
+  VERSION = '1.0.0'
+end

data/spec/base_spec.rb

@@ -0,0 +1,6 @@
+require 'spec_helper'
+require 'claws'
+
+describe Claws do
+
+end

data/spec/capistrano_spec.rb

@@ -0,0 +1,74 @@
+require 'spec_helper'
+require 'claws/capistrano'
+
+describe Claws::Capistrano do
+  before :each do
+    @roles = {
+      :staging => [
+        %q{role :app, 'ec2-175-65-213-31.compute-1.amazonaws.com'},
+        %q{role :web, 'ec2-175-65-213-31.compute-1.amazonaws.com'},
+        %q{role :batch, "ec2-175-65-213-31.compute-1.amazonaws.com"},
+        %q{role :redis, "ec2-223-40-143-23.compute-1.amazonaws.com"},
+        %q{role :search, "ec2-147-32-151-54.compute-1.amazonaws.com"},
+        %q{role :search_slave, "ec2-147-32-151-54.compute-1.amazonaws.com"},
+      ],
+
+      :production => [
+        %q{role :app, "ec2-263-56-231-91.compute-1.amazonaws.com", 'ec2-263-23-118-57.compute-1.amazonaws.com'},
+        %q{role :web, "ec2-263-56-231-91.compute-1.amazonaws.com", 'ec2-263-23-118-57.compute-1.amazonaws.com', "ec2-23-20-43-198.compute-1.amazonaws.com"},
+        %q{role :batch, "ec2-23-20-43-198.compute-1.amazonaws.com"},
+        %q{role :redis, "ec2-23-20-43-198.compute-1.amazonaws.com"},
+        %q{role :search, "ec2-107-21-131-545.compute-1.amazonaws.com"},
+        %q{role :search_slave, "ec2-263-23-144-120.compute-1.amazonaws.com"},
+      ]
+    }
+    @mappings = {
+      :staging => {
+        'ec2-175-65-213-31.compute-1.amazonaws.com' => %w{app web batch},
+        'ec2-223-40-143-23.compute-1.amazonaws.com' => %w{redis},
+        'ec2-147-32-151-54.compute-1.amazonaws.com' => %w{search search_slave},
+      },
+      :production => {
+        'ec2-263-56-231-91.compute-1.amazonaws.com' => %w{app web},
+        'ec2-263-23-118-57.compute-1.amazonaws.com' => %w{app web},
+        'ec2-23-20-43-198.compute-1.amazonaws.com' => %w{web batch redis},
+        'ec2-107-21-131-545.compute-1.amazonaws.com' => %w{search},
+        'ec2-263-23-144-120.compute-1.amazonaws.com' => %w{search_slave},
+      },
+    }
+
+    @default_path = 'config/deploy'
+  end
+
+  context 'defines roles hash per environment per host' do
+    it 'from default path' do
+      Dir.should_receive(:glob).and_return(%W{#{@default_path}/staging.rb #{@default_path}/production.rb})
+      File.should_receive(:readlines).with("#{@default_path}/staging.rb").and_return(@roles[:staging])
+      File.should_receive(:readlines).with("#{@default_path}/production.rb").and_return(@roles[:production])
+
+      cap = Claws::Capistrano.new
+      cap.home.should == @default_path
+      cap.all_host_roles.should == @mappings
+    end
+
+    it 'from custom path' do
+      custom_path = '/home/app/config/deploy'
+      Dir.should_receive(:glob).and_return(%W{#{custom_path}/staging.rb #{custom_path}/production.rb})
+      File.should_receive(:readlines).with("#{custom_path}/staging.rb").and_return(@roles[:staging])
+      File.should_receive(:readlines).with("#{custom_path}/production.rb").and_return(@roles[:production])
+
+      cap = Claws::Capistrano.new(custom_path)
+      cap.home.should == custom_path
+      cap.all_host_roles.should == @mappings
+    end
+  end
+
+  it 'returns roles for host' do
+    Dir.should_receive(:glob).and_return(%W{#{@default_path}/staging.rb #{@default_path}/production.rb})
+    File.should_receive(:readlines).with("#{@default_path}/staging.rb").and_return(@roles[:staging])
+    File.should_receive(:readlines).with("#{@default_path}/production.rb").and_return(@roles[:production])
+
+    cap = Claws::Capistrano.new
+    cap.roles('ec2-263-56-231-91.compute-1.amazonaws.com').should == %w{app web}
+  end
+end

data/spec/collection/base_spec.rb

@@ -0,0 +1,30 @@
+require 'spec_helper'
+require 'claws/collection/base'
+require 'claws/configuration'
+
+describe Claws::Collection::Base do
+  subject { Claws::Collection::Base }
+
+  before :each do
+    @yaml = {"capistrano_home"=>"test", "access_key_id"=>"asdf", "secret_access_key"=>"qwer"}
+    @credentials = {:access_key_id => 'asdf', :secret_access_key => 'qwer'}
+
+    @config = double('Claws::Configuration', :aws_credentials => @credentials)
+    AWS.should_receive(:config).with(@credentials).and_return(true)
+    AWS.should_receive(:start_memoizing).and_return(nil)
+  end
+
+  it 'establishes a connection to the mothership' do
+    expect {
+      subject.connect(@config.aws_credentials)
+    }.to_not raise_exception
+  end
+
+  it 'builds a collection' do
+    subject.connect(@config.aws_credentials)
+
+    subject.build do |collection|
+      10.times {|i| collection << i}
+    end.should == (0..9).to_a
+  end
+end

data/spec/collection/ec2_spec.rb

@@ -0,0 +1,18 @@
+require 'spec_helper'
+require 'aws-sdk'
+require 'claws/collection/ec2'
+
+describe Claws::Collection::EC2 do
+  subject { Claws::Collection::EC2 }
+
+  it 'gets all instances' do
+    subject.should_receive(:get).with(no_args).and_return(
+      [
+        double('AWS::EC2::Instance'),
+        double('AWS::EC2::Instance'),
+      ]
+    )
+
+    subject.get.size.should == 2
+  end
+end

data/spec/command/ec2_spec.rb

@@ -0,0 +1,179 @@
+require 'spec_helper'
+
+describe Claws::Command::EC2 do
+  subject { Claws::Command::EC2 }
+
+  describe '#exec' do
+    context 'configuration files' do
+      let(:options) { OpenStruct.new( { :config_file => '/doesnotexist', } ) }
+
+      it 'missing files' do
+        subject.should_receive(:puts).twice
+
+        expect {
+          subject.exec options
+        }.should raise_exception SystemExit
+      end
+
+      it 'invalid file' do
+        YAML.should_receive(:load_file).and_raise(Exception)
+
+        subject.should_receive(:puts).twice
+
+        expect {
+          subject.exec options
+        }.should raise_exception SystemExit
+      end
+    end
+
+    context 'valid config file' do
+      before :each do
+        Claws::Configuration.stub(:new).and_return(
+          OpenStruct.new(
+            {
+              :ssh => OpenStruct.new(
+                { :user => 'test' }
+              ),
+              :ec2 => OpenStruct.new(
+                :fields => {
+                  :id => {
+                    :width => 10,
+                    :title => 'ID',
+                  }
+                }
+              )
+            }
+          )
+        )
+      end
+
+      let(:options) { OpenStruct.new( { :config_file => nil, } ) }
+
+      context 'instance collections' do
+
+        it 'retrieves' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_return([])
+
+          capture_stdout {
+            subject.exec options
+          }
+        end
+
+        it 'handles errors retrieving' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_raise(Exception)
+          subject.should_receive(:puts).once
+
+          expect {
+            subject.exec options
+          }.should raise_exception
+        end
+      end
+
+      it 'performs report' do
+        Claws::Collection::EC2.should_receive(:connect).and_return(true)
+        Claws::Collection::EC2.should_receive(:get).and_return([])
+
+        expect {
+          capture_stdout {
+            subject.exec options
+          }
+        }.should_not raise_exception
+      end
+    end
+
+    context 'connect options' do
+      let(:options) { OpenStruct.new( { :config_file => nil, :connect => true } ) }
+
+      before :each do
+        Claws::Configuration.stub(:new).and_return(
+          OpenStruct.new(
+            {
+              :ssh => OpenStruct.new(
+                { :user => 'test' }
+              ),
+              :ec2 => OpenStruct.new(
+                :fields => {
+                  :id => {
+                    :width => 10,
+                    :title => 'ID',
+                  }
+                }
+              )
+            }
+          )
+        )
+      end
+
+      context 'single instance' do
+        let(:instances) do
+          [
+            double(AWS::EC2::Instance, :id => 'test', :status => 'running', :dns_name => 'test.com')
+          ]
+        end
+
+        it 'automatically connects to the server' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_return(instances)
+
+          subject.should_receive(:puts).twice
+          subject.should_receive(:system).with('ssh test@test.com').and_return(0)
+
+          capture_stdout {
+            subject.exec options
+          }
+        end
+      end
+
+      context 'multiple instances' do
+        let(:instances) do
+          [
+            double(AWS::EC2::Instance, :id => 'test1', :status => 'running', :dns_name => 'test1.com'),
+            double(AWS::EC2::Instance, :id => 'test2', :status => 'running', :dns_name => 'test2.com'),
+            double(AWS::EC2::Instance, :id => 'test3', :status => 'running', :dns_name => 'test3.com'),
+          ]
+        end
+
+        it 'handles user inputed selection from the command line' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_return(instances)
+
+          subject.should_receive(:puts).twice
+          subject.should_receive(:system).with('ssh test@test2.com').and_return(0)
+
+          capture_stdout {
+            subject.exec OpenStruct.new( {:selection => 1, :config_file => nil, :connect => true} )
+          }
+
+        end
+
+        it 'presents a selection and connects to the server' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_return(instances)
+
+          subject.should_receive(:gets).and_return('1\n')
+          subject.should_receive(:puts).once
+          subject.should_receive(:system).with('ssh test@test2.com').and_return(0)
+
+          capture_stdout {
+            subject.exec options
+          }
+        end
+
+        it 'presents a selection and allows a user to quit' do
+          Claws::Collection::EC2.should_receive(:connect).and_return(true)
+          Claws::Collection::EC2.should_receive(:get).and_return(instances)
+
+          subject.should_receive(:gets).and_return('q\n')
+
+          expect {
+            capture_stdout {
+              subject.exec options
+            }
+          }.should raise_exception SystemExit
+        end
+      end
+    end
+  end
+end

data/spec/command/initialize_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+
+describe Claws::Command::Initialize do
+  subject { Claws::Command::Initialize }
+
+  let(:config) { 'test/.test.yml' }
+
+  it 'works' do
+    File.should_receive(:join).and_return(config)
+    subject.should_receive(:puts)
+
+    fh = double( File, :write => true )
+
+    File.should_receive(:open).with(config, 'w').and_return(fh)
+    subject.should_receive(:puts)
+
+    subject.exec
+  end
+end

data/spec/configuration_spec.rb

@@ -0,0 +1,98 @@
+require 'spec_helper'
+require 'claws/configuration'
+
+describe Claws::Configuration do
+  let (:yaml) do
+    {
+      'capistrano' => {
+        'home' => 'test',
+      },
+      'aws' => {
+        'access_key_id' => 'asdf',
+        'secret_access_key' => 'qwer',
+        'aws_user' => 'test',
+      },
+      'ec2' => {
+        'fields' => {
+          'id' => {
+            'width' => 10,
+            'title' => 'ID'
+          },
+          'name' => {
+            'width' => 20,
+            'title' => 'Name'
+          },
+        }
+      }
+    }
+  end
+
+  let (:config) { Claws::Configuration.new }
+
+  describe '#initialize' do
+    it 'defines default path' do
+      YAML.should_receive(:load_file).and_return( yaml )
+      c = Claws::Configuration.new
+      c.path.should == File.join(ENV['HOME'], '.claws.yml')
+    end
+
+    it 'defines a custom path' do
+      YAML.should_receive(:load_file).and_return( yaml )
+      c = Claws::Configuration.new '/home/test'
+      c.path.should == '/home/test'
+    end
+
+    it 'raises ConfigurationError' do
+      YAML.should_receive(:load_file).and_raise( Exception.new )
+
+      expect {
+        Claws::Configuration.new
+      }.should raise_exception Claws::ConfigurationError
+    end
+
+    context 'Capistrano' do
+      it 'defines home' do
+        YAML.should_receive(:load_file).and_return(yaml)
+        config.capistrano.home.should == 'test'
+      end
+    end
+
+    context 'AWS' do
+      before :each do
+        YAML.should_receive(:load_file).and_return(yaml)
+      end
+
+      it 'defines user' do
+        config.aws['aws_user'].should == 'test'
+      end
+
+      it 'defines secret access key' do
+        config.aws['secret_access_key'].should == 'qwer'
+      end
+
+      it 'defines access key id' do
+        config.aws['access_key_id'].should == 'asdf'
+      end
+    end
+
+    context 'EC2' do
+      before :each do
+        YAML.should_receive(:load_file).and_return(yaml)
+      end
+
+      context 'fields' do
+        it 'defines id hash' do
+          id = config.ec2.fields['id']
+          id['width'].should == 10
+          id['title'].should == 'ID'
+        end
+
+        it 'defines name hash' do
+          name = config.ec2.fields['name']
+          name['width'].should == 20
+          name['title'].should == 'Name'
+        end
+      end
+    end
+  end
+end

data/spec/options_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+require 'claws/options'
+
+ARGV.clear
+
+def cli(args)
+  ARGV.push(*args)
+  yield
+  ARGV.clear
+end
+
+describe Claws::Options do
+  it 'defines default options' do
+    cli %w{production app} do
+      options = Claws::Options.parse
+      options.connect.should be_true
+      options.source.should == 'ec2'
+      options.choice.should be_nil
+    end
+  end
+
+  it 'accepts display only flag' do
+    # By default we want to connect to the instance
+    Claws::Options.parse.connect.should be_true
+
+    # Allow the user to override and just display information
+    cli %w{-d} do
+      Claws::Options.parse.connect.should be_false
+    end
+
+    cli %w{--display-only} do
+      Claws::Options.parse.connect.should be_false
+    end
+  end
+
+  it 'accepts a choice flag' do
+    cli %w{-c 10} do
+      Claws::Options.parse.choice.should == 10
+    end
+
+    cli %w{--choice 10} do
+      Claws::Options.parse.choice.should == 10
+    end
+  end
+
+  it 'accepts a source flag' do
+    cli %w{-s elb} do
+      options = Claws::Options.parse
+      options.source.should == 'elb'
+      options.connect.should be_false
+    end
+
+    cli %w{--source elb} do
+      options = Claws::Options.parse
+      options.source.should == 'elb'
+      options.connect.should be_false
+    end
+  end
+
+  context 'capistrano' do
+    it 'defines the environment' do
+      cli %w{-s production app} do
+        Claws::Options.parse.environment.should == 'production'
+      end
+    end
+
+    it 'defines the role' do
+      cli %w{-s production app} do
+        Claws::Options.parse.role.should == 'app'
+      end
+    end
+  end
+end

data/spec/presenter/ec2_spec.rb

@@ -0,0 +1,94 @@
+require 'spec_helper'
+require 'aws-sdk'
+require 'claws/presenter/ec2'
+require 'claws/capistrano'
+
+describe Claws::EC2::Presenter do
+  subject { Claws::EC2::Presenter }
+
+  before :each do
+    host = 'ec2-263-56-231-91.compute-1.amazonaws.com'
+
+    full_instance = double('AWS::EC2',
+      :public_dns => host,
+      :security_groups => [
+        double(AWS::EC2::SecurityGroup, :name => 'search', :id => 'sg-0f0f0f0f'),
+        double(AWS::EC2::SecurityGroup, :name => 'mongo', :id => 'sg-0d0d0d0d'),
+        double(AWS::EC2::SecurityGroup, :name => 'app', :id => 'sg-0c0c0c0c'),
+      ],
+      :tags => double(AWS::EC2::ResourceTagCollection, :select => [
+          ['environment', 'production'],
+          ['function', 'master'],
+        ],
+        'has_key?'.to_sym => true
+      ),
+      :elastic_ip => '11.111.111.111'
+    )
+
+    cap = double('Claws::Capistrano')
+    cap.stub(:roles).with(host).and_return(%w{app web})
+
+    @full_presenter = subject.new(full_instance, cap.roles(full_instance.public_dns))
+
+    less_instance = double(AWS::EC2, :tags => nil)
+    @less_presenter = subject.new(less_instance)
+  end
+
+  describe '#initialize' do
+    it 'requires a valid ec2 instance' do
+      expect {
+        subject.new
+      }.to raise_error =~ /ArgumentError/
+    end
+  end
+
+  describe '#roles' do
+    it 'can be defined' do
+      @full_presenter.roles.should == 'app, web'
+    end
+
+    it 'are not required' do
+      @less_presenter.roles.should == 'N/A'
+    end
+  end
+
+  describe '#tags' do
+    it 'present a string summary' do
+      @full_presenter.tags.should == 'environment: production, function: master'
+    end
+
+    it 'are not required' do
+      @less_presenter.tags.should == 'N/A'
+    end
+  end
+
+  describe '#security_groups' do
+    it 'presents summary of names' do
+      @full_presenter.security_groups.should == 'sg-0f0f0f0f: search, sg-0d0d0d0d: mongo, sg-0c0c0c0c: app'
+    end
+
+    it 'are not required' do
+      @less_presenter.security_groups.should == 'N/A'
+    end
+  end
+
+  describe '#public_dns' do
+    it 'displays when available' do
+      @full_presenter.public_dns.should == 'ec2-263-56-231-91.compute-1.amazonaws.com'
+    end
+
+    it 'is not required' do
+      @less_presenter.public_dns.should == 'N/A'
+    end
+  end
+
+  describe '#elastic_ip' do
+    it 'displays when available' do
+      @full_presenter.elastic_ip.should == '11.111.111.111'
+    end
+
+    it 'is not required' do
+      @less_presenter.elastic_ip.should == 'N/A'
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,4 @@
+$: << File.join(File.dirname(__FILE__), '..', 'lib')
+
+Dir[File.dirname(__FILE__) + "/../lib/**/*.rb"].each {|f| require f}
+Dir[File.dirname(__FILE__) + "/support/**/*.rb"].each {|f| require f}

data/spec/support/helpers/stdout_helper.rb

@@ -0,0 +1,11 @@
+require 'stringio'
+
+def capture_stdout
+  $stdout = StringIO.new
+  $stdin = StringIO.new("y\n")
+  yield
+  $stdout.string.strip
+ensure
+  $stdout = STDOUT
+  $stdin = STDIN
+end

data/spec/support_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+require 'claws/support'
+
+describe Claws::Support do
+  subject { Array.new }
+
+  describe '#try' do
+    it 'handles undefined methods sanely' do
+      subject.try('asdf').should be_nil
+    end
+
+    it 'performs defined methods' do
+      subject.try('push', 2)
+      subject.should == [2]
+    end
+  end
+end

metadata

@@ -0,0 +1,120 @@
+--- !ruby/object:Gem::Specification
+name: claws
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Wes
+- Bailey
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: &2156422520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: *2156422520
+- !ruby/object:Gem::Dependency
+  name: command_line_reporter
+  requirement: &2156421940 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156421940
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: &2156420960 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2156420960
+description: A command line tool that provides a configurable report on the status
+  of all of your EC2 hosts and provides trivial ssh access for connectivity.  Never
+  copy and paste the public dns for a host again!
+email: baywes@gmail.com
+executables:
+- claws
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/claws/capistrano.rb
+- lib/claws/collection/base.rb
+- lib/claws/collection/ec2.rb
+- lib/claws/command/ec2.rb
+- lib/claws/command/initialize.rb
+- lib/claws/configuration.rb
+- lib/claws/options.rb
+- lib/claws/presenter/ec2.rb
+- lib/claws/report/ec2.rb
+- lib/claws/support.rb
+- lib/claws/version.rb
+- lib/claws.rb
+- README.md
+- spec/base_spec.rb
+- spec/capistrano_spec.rb
+- spec/collection/base_spec.rb
+- spec/collection/ec2_spec.rb
+- spec/command/ec2_spec.rb
+- spec/command/initialize_spec.rb
+- spec/configuration_spec.rb
+- spec/options_spec.rb
+- spec/presenter/ec2_spec.rb
+- spec/spec_helper.rb
+- spec/support/helpers/stdout_helper.rb
+- spec/support_spec.rb
+- !binary |-
+  YmluL2NsYXdz
+homepage: http://github.com/wbailey/claws
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: A Command Line Tool For Amazon Web Services
+test_files:
+- spec/base_spec.rb
+- spec/capistrano_spec.rb
+- spec/collection/base_spec.rb
+- spec/collection/ec2_spec.rb
+- spec/command/ec2_spec.rb
+- spec/command/initialize_spec.rb
+- spec/configuration_spec.rb
+- spec/options_spec.rb
+- spec/presenter/ec2_spec.rb
+- spec/spec_helper.rb
+- spec/support/helpers/stdout_helper.rb
+- spec/support_spec.rb