aws_csshx 0.1.1

data/.gitignore

@@ -0,0 +1,7 @@
+pkg/*
+*.gem
+.bundle
+*.swo
+*.swp
+Manifest
+.rvmrc

data/Changelog.md

@@ -0,0 +1,8 @@
+### aws_csshx 0.1.1 2012-05-04
+
+ * Added command line switches (Eric Lubow)
+ * Added config file support using .csshrc (Eric Lubow)
+
+### aws_csshx 0.1.0 2012-05-01
+
+ * Initial vesion (Russell Bradberry)

data/Gemfile

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

data/Gemfile.lock

@@ -0,0 +1,18 @@
+PATH
+  remote: .
+  specs:
+    aws_csshx (0.0.1)
+      right_aws
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    right_aws (3.0.4)
+      right_http_connection (>= 1.2.5)
+    right_http_connection (1.3.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  aws_csshx!

data/README.md

@@ -0,0 +1,8 @@
+# AWS CSSHX Wrapper
+
+This is a wrapper script for ClusterSSHX (csshx).  It allows the user the ability to ssh to all machines in an AWS security group using a specific key and user for those machines.
+
+## Installation
+
+    gem install aws_csshx
+

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/TODO.md

@@ -0,0 +1,3 @@
+### TODO List for aws_csshx
+
+ * Add ability to use -H/--hosts to tack on additional hosts to the SSH cluster

data/aws_csshx.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "aws_csshx/version"
+
+Gem::Specification.new do |s|
+  s.name        = "aws_csshx"
+  s.version     = AwsCsshx::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Russell Bradberry", "Eric Lubow"]
+  s.email       = ["eric@lubow.org"]
+  s.homepage    = ""
+  s.summary     = %q{csshx wrapper that interacts with your AWS account for group ssh sessions}
+  s.description = %q{csshx wrapper that interacts with your AWS account for group ssh sessions}
+
+  s.add_dependency 'right_aws'
+
+  s.rubyforge_project = "aws_csshx"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/bin/aws_csshx

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', 'lib')
+require 'aws_csshx'
+
+exit AwsCsshx::Application.run!(*ARGV)

data/lib/aws_csshx.rb

@@ -0,0 +1,8 @@
+require 'etc'
+require 'right_aws'
+
+require 'aws_csshx/application'
+require 'aws_csshx/configuration_file'
+require 'aws_csshx/misc'
+require 'aws_csshx/options'
+require 'aws_csshx/version'

data/lib/aws_csshx/application.rb

@@ -0,0 +1,94 @@
+module AwsCsshx
+  class Application
+    attr :options
+
+    class << self
+
+      def csshx_exists?
+        `which csshx > /dev/null 2>&1`
+      end
+
+      def aws_settings_exist?
+        @options[:aws_access_key] and @options[:aws_secret_key] and @options[:aws_region] and File.exists?(@options[:ec2_private_key])
+      end
+
+      def has_servers?(list)
+        server_list.count > 0 ? true : false
+      end
+
+      def run!(*arguments)
+        @options = {}
+
+        # No need to go further unless csshX exists
+        abort('csshX file is required') unless csshx_exists?
+
+        begin
+          # First we load the config file and then process the command line
+          # options since they take precedence.
+
+          # Load the config file if it exists
+          conf_file = "#{Etc.getpwuid.dir}/.csshrc"
+          if File.exists?(conf_file)
+            config = AwsCsshx::ConfigurationFile.new(conf_file)
+            @options = config.options
+          end
+
+          command_line_options = AwsCsshx::Options.new(arguments)
+          @options.merge!(command_line_options[:options])
+          @options = AwsCsshx::Misc.symbolize_hash_keys(@options)
+
+          # Load the new config file if different
+          if conf_file != @options[:conf] and File.exists?(conf_file)
+            config = AwsCsshx::ConfigurationFile.new(@options[:conf])
+          end
+
+          # Deal with the option issues
+          if command_line_options[:invalid_argument]
+            $stderr.puts command_line_options[:invalid_argument]
+            @options[:help] = true
+          end
+
+          # Show the version screen and bounce if requested
+          if @options[:version]
+            puts "#{$0} #{AwsCsshx::VERSION}"
+            return 0
+          end
+
+          # Show the help screen and bounce if requested
+          if @options[:help]
+            puts command_line_options.opts
+            return 0
+          end
+
+          # Stop here without all our AWS settings
+          abort("Invalid AWS settings") unless aws_settings_exist?
+
+          abort("Cannot continue without AWS security group (-g)") unless @options[:group]
+          @server_list = aws_server_list_by_group @options[:group]
+
+          if has_servers? @server_list
+            `csshx --login #{@options[:login]} --ssh_args='-i #{@options[:ec2_private_key]}' #{@server_list.join(' ')}`
+            puts "Opened connections to #{@server_list.count} servers in the '#{@options[:group]}' security group."
+          else
+            puts "No servers found...bailing out!"
+          end
+        rescue Exception => e
+          puts "Error: #{e.inspect}"
+          puts "Trace: #{e.backtrace.join('\n')}"
+        end
+
+        # Exit cleanly
+        return 0
+      end
+
+      def aws_server_list_by_group(group)
+        @ec2_api ||= RightAws::Ec2.new(@options[:aws_access_key], @options[:aws_secret_key])
+
+        @ec2_api.describe_instances.reject{|i| i[:aws_state] != "running"}.map do |instance|
+          instance[:dns_name] if instance[:groups].map{|g| g[:group_name]}.include?(group)
+        end.compact
+      end
+
+    end
+  end
+end

data/lib/aws_csshx/configuration_file.rb

@@ -0,0 +1,82 @@
+# TODO Create config option errors handling class
+# TODO Ensure environment variables exist before blindly setting options to them
+module AwsCsshx
+  class ConfigurationFile
+    attr :options
+
+    CONFIG_OPTS = %q{ec2_private_key aws_access_key aws_secret_key aws_region}
+
+    def initialize(conf)
+      @config_file = conf
+      @options = {}
+      if File.exists?(conf)
+        load_config_from_file @config_file
+      else
+        raise FileDoesNotExist, "Config file #{@config_file} cannot be found."
+      end
+    end
+
+    # Iterate over the config file line by line and throw away things not in our config opts array
+    def load_config_from_file(file)
+      f = File.open(file, "r")
+      f.each_line do |line|
+        field,value = line.split('=')
+        next if field.nil? or value.nil?
+        field.strip!.downcase!
+        value.strip!
+        next unless CONFIG_OPTS.include?(field)
+        @options[field] =
+          if    field == 'ec2_private_key' then set_ec2_private_key(value)
+          elsif field == 'aws_access_key' then set_aws_access_key(value)
+          elsif field == 'aws_secret_key' then set_aws_secret_key(value)
+          elsif field == 'aws_region' then set_aws_region(value)
+          else  puts "No such option #{field} skipping..."
+          end
+      end
+    end
+
+
+    #
+    # Private(ish) methods for config settings
+    #
+
+    def set_aws_region(val)
+      if val
+        val
+      elsif ENV['AWS_REGION']
+        ENV['AWS_REGION']
+      else
+        'us-east-1'
+      end
+    end
+
+    def set_aws_secret_key(val)
+      if val
+        val
+      elsif ENV['AWS_SECRET_KEY']
+        ENV['AWS_SECRET_KEY']
+      elsif ENV['AMAZON_SECRET_ACCESS_KEY']
+        ENV['AMAZON_SECRET_ACCESS_KEY']
+      end
+    end
+
+    def set_aws_access_key(val)
+      if val
+        val
+      elsif ENV['AWS_ACCESS_KEY']
+        ENV['AWS_ACCESS_KEY']
+      elsif ENV['AMAZON_ACCESS_KEY_ID']
+        ENV['AMAZON_ACCESS_KEY_ID']
+      end
+    end
+
+    def set_ec2_private_key(key_file)
+      if File.exists?(key_file)
+        key_file
+      else
+        puts "#{key_file} does not exist, using default."
+        ENV['EC2_PRIVATE_KEY']
+      end
+    end
+  end
+end

data/lib/aws_csshx/misc.rb

@@ -0,0 +1,13 @@
+module AwsCsshx
+  class Misc
+    class << self
+
+      def symbolize_hash_keys(hash)
+        hsh = {}
+        hash.each_pair { |k,v|  hsh[k.downcase.to_sym] = v }
+        hsh
+      end
+
+    end
+  end
+end

data/lib/aws_csshx/options.rb

@@ -0,0 +1,89 @@
+module AwsCsshx
+  class Options < Hash
+    attr_reader :opts, :orig_args
+
+    def initialize(args)
+      super()
+
+      user_home = ENV['HOME']
+
+      @orig_args = args.clone
+
+      options = {}
+
+      require 'optparse'
+      @opts = OptionParser.new do |o|
+          o.banner = "Usage: #{File.basename($0)} <file1> <file2> ..."
+
+          o.separator ""
+          o.separator "AWS Options"
+
+          options[:group] = 'default'
+          o.on( '-g', '--group <group>', 'AWS security group name to use for the csshX sessions' ) do |group|
+              options[:group] = group
+          end
+
+          o.on( '-i', '--aws-identity <identity>', 'Use this keyfile as your SSH private key' ) do |identity|
+              options[:ec2_private_key] = identity
+          end
+
+          o.on( '-r', '--aws-region <region>', 'AWS region to query for the csshX sessions (default: us-east-1)' ) do |region|
+              options[:aws_region] = region
+          end
+
+          o.separator ""
+          o.separator "csshX Options"
+
+          options[:login] = 'root'
+          o.on( '-l', '--login <login>', 'User login to use for the csshX sessions (default: root)' ) do |login|
+              options[:login] = login
+          end
+
+          options[:csshx_opts] = ''
+          o.on( '-o', '--csshx-opts <csshx_opts>', 'Pass the options listed directly to csshx' ) do |csshx_opts|
+              options[:csshx_opts] = csshx_opts
+          end
+
+          options[:iterm2] = false
+          o.on( '-2', '--iterm2', 'Use csshX.iterm instead of csshx' ) do |iterm2|
+              options[:iterm2] = true
+          end
+
+          o.separator ""
+
+          options[:conf] = "#{Etc.getpwuid.dir}/.csshrc"
+          o.on( '-c', '--conf <file>', 'aws_csshX config file (default: .csshrc)' ) do |file|
+              if File.exists?(file)
+                  options[:conf] = file
+              else
+                  raise FileDoesNotExist, "Config file #{file} cannot be found."
+              end
+          end
+
+          o.on('-V', '--version', "Display version information") do
+              options[:version] = true
+          end
+
+          options[:help] = false
+          o.on( '-h', '--help', 'Display this help screen' ) do
+              options[:help] = true
+          end
+
+          o.separator ""
+          o.separator "Examples:"
+          o.separator "  Cluster SSH to all machines in the 'utility' security group:"
+          o.separator "    #{File.basename($0)} -g utility"
+          o.separator ""
+      end
+
+      begin
+          @opts.parse!(args)
+          self[:options] = options
+      rescue OptionParser::InvalidOption => e
+          self[:invalid_argument] = e.message
+          @opts.parse(args, flags={ :delete_invalid_opts => true })
+          self[:options] = options
+      end
+    end
+  end
+end

data/lib/aws_csshx/version.rb

@@ -0,0 +1,3 @@
+module AwsCsshx
+  VERSION = "0.1.1"
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: aws_csshx
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+  prerelease: 
+platform: ruby
+authors:
+- Russell Bradberry
+- Eric Lubow
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-05-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: right_aws
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: csshx wrapper that interacts with your AWS account for group ssh sessions
+email:
+- eric@lubow.org
+executables:
+- aws_csshx
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Changelog.md
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- TODO.md
+- aws_csshx.gemspec
+- bin/aws_csshx
+- lib/aws_csshx.rb
+- lib/aws_csshx/application.rb
+- lib/aws_csshx/configuration_file.rb
+- lib/aws_csshx/misc.rb
+- lib/aws_csshx/options.rb
+- lib/aws_csshx/version.rb
+homepage: ''
+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: aws_csshx
+rubygems_version: 1.8.18
+signing_key: 
+specification_version: 3
+summary: csshx wrapper that interacts with your AWS account for group ssh sessions
+test_files: []