aws-ssh 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 528f3fee7a959f32e7e5cda371d4e35fe98fe403
+  data.tar.gz: d5d3faf114cd04448bd45ea259d4b1a9e69b8428
+SHA512:
+  metadata.gz: 037d03e913644404aad34e487c3709d1a62c1b068efe018583c653066dfa7f96f3a1b1b390e6f9ceff0a2651bede193185fa95e5a9b5d2dc9f2c6e346e5c901b
+  data.tar.gz: 3560dd269f9687cd9f17a56cdd47b04d34031531887b948b527b537d4f419ab3d4d6a4f8c98aa9a8e53d04646b4b0c927356b530801f5700b6a0b10ba63a2c40

data/bin/awssh

@@ -0,0 +1,162 @@
+#!/usr/bin/env ruby
+
+require 'slop'
+require 'aws-sdk'
+require 'yaml'
+
+module AWSSH
+  class Runner
+    DEFAULT_OPTIONS = {
+      'user'    => ENV['user'],
+      'format'  => 'ssh',
+      'verbose' => false
+    }
+
+    attr_accessor :host_regex
+
+    def run
+      self.host_regex = cli.args.first
+      exit_error! 'AWS Region not set' unless cli[:region]
+      exit_error! 'AWS Profile not set' unless cli[:profile]
+
+      shout "Hostname: #{host_regex}"
+      shout "Options: #{cli.to_h.inspect}"
+      shout "Defaults: #{defaults.inspect}"
+
+      Aws.config.merge!(profile: cli[:profile], region: cli[:region])
+
+      if cli['show-only']
+        print_output
+      else
+        ssh!
+      end
+    end
+
+    def cli
+      @cli ||= Slop.parse do |o|
+        o.banner = <<-USAGE
+awssh - a tool that makes easy to ssh into AWS EC2 servers
+
+Usage: awssh [hostname regex]
+
+Examples:
+  $ awssh prod.*app2
+  ... will SSH into the instance and you will see:
+  user@prod-rails-app4~$
+
+  $ awssh --show-only --stack qa -u worker
+  ... will show all instances registered in QA stack of Opsworks
+  ssh worker@10.20.30.40        => MyApp QA - qa-rails-app
+  ssh worker@10.20.30.41        => MyApp QA - qa-sidekiq
+
+  # All hosts in one-line (suitable for CSSH):
+  worker@10.20.30.40 worker@10.20.30.41
+        USAGE
+
+        o.separator "Options:"
+        o.string '-s',  '--stack',      'stack name regex. E.g: `-s prod` will match "Production" stack name'
+        o.string '-p',  '--profile',    'AWS config profile name. Default: profile set in .awssh file', default: defaults['profile']
+        o.string '-r',  '--region',     'region. E.g: us-east-1. Default: region set in .awssh file', default: defaults['region']
+        o.string '-u',  '--user',       'SSH username to use. Default: user set in .awssh file or current machine user', default: defaults['user']
+        o.bool   '-so', '--show-only',  'Only show the matched hosts instead of ssh. Default: false', default: false
+        o.bool   '-cs', '--csensitive', 'Uses case-sensitive for regex matching. Default: false', default: false
+        o.bool   '-v',  '--verbose',    'Verbose mode. Default: false', default: false
+        o.on '--help', 'Shows help' do
+          puts o
+          exit
+        end
+      end
+    end
+
+    def ssh!
+      if instances.size > 1
+        print_output
+        exit_error! "\nMultiple hostnames matched. Please be more specific in hostnames regex or use the last output line to use CSSH.\n"
+      end
+      exec "ssh #{connection_string(instances.first)}"
+    end
+
+    def print_output
+      instances.each do |i|
+        puts "ssh #{connection_string(i)}      \t => #{i.name}"
+      end
+      puts "\n# All hosts in one-line (suitable for CSSH):"
+      puts instances.map { |i| connection_string(i) }.join(' ')
+    end
+
+    def connection_string(instance)
+      "#{cli[:user]}@#{instance.ip}"
+    end
+
+    def defaults
+      @defaults ||= begin
+        defaults = File.exist?('.awssh') ? YAML.load_file('.awssh') : {}
+        DEFAULT_OPTIONS.merge(defaults)
+      end
+    end
+
+    def instances
+      @instances ||= AWSEC2.new.instances(host_regex, cli[:stack], cli[:csensitive]).select do |i|
+        shout "Instance #{i.name} is not part of an Opsworks Stack" unless i.stack_name
+        shout "Instance #{i.name} not running. Current state: #{i.state}" unless i.running?
+        i.running?
+      end
+    end
+
+    def exit_error!(msg)
+      $stderr.puts msg
+      exit(1)
+    end
+
+    def shout(msg)
+      puts "[DEBUG] #{msg}" if cli.verbose?
+    end
+  end
+
+  EC2Instance = Struct.new(:name, :hostname, :stack_name, :state, :ip) do
+    def initialize(instance)
+      name       = extract_tag(instance, 'Name')
+      hostname   = extract_tag(instance, 'opsworks:instance')
+      stack_name = extract_tag(instance, 'opsworks:stack')
+
+      super(name, hostname, stack_name, instance.state.name, instance.public_ip_address)
+    end
+
+    def running?
+      state == 'running'
+    end
+
+    def extract_tag(instance, tag)
+      tag = instance.tags.find { |t| t.key == tag }
+      tag ? tag.value : nil
+    end
+  end
+
+  class AWSEC2
+    def instances(host_regex, stack_regex, case_sensitive)
+      ec2.describe_instances.reservations.map(&:instances).flatten.map do |instance_data|
+        instance = EC2Instance.new(instance_data)
+        if stack_regex
+          next unless word_match?(instance.stack_name, stack_regex, case_sensitive)
+        end
+        if host_regex
+          next unless word_match?(instance.hostname, host_regex, case_sensitive)
+          next unless word_match?(instance.name, host_regex, case_sensitive)
+        end
+
+        instance
+      end.compact.sort_by(&:name)
+    end
+
+    def ec2
+      @ec2 ||= Aws::EC2::Client.new
+    end
+
+    def word_match?(value, regex_str, case_sensitive)
+      regex = case_sensitive ? /#{regex_str}/ : /#{regex_str}/i
+      !!(value =~ regex)
+    end
+  end
+end
+
+AWSSH::Runner.new.run

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: aws-ssh
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Rafael Sales
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: slop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+description: A tool that makes easy to ssh into AWS EC2 servers
+email:
+- rafaelcds@gmail.com
+executables:
+- awssh
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/awssh
+homepage: https://github.com/buzzstarter/awssh
+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.4.3
+signing_key: 
+specification_version: 4
+summary: A tool that makes easy to ssh into AWS EC2 servers
+test_files: []
+has_rdoc: