aws-ssh 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 528f3fee7a959f32e7e5cda371d4e35fe98fe403
-  data.tar.gz: d5d3faf114cd04448bd45ea259d4b1a9e69b8428
+  metadata.gz: 78ca737b88d4a338e7f18dfab077b2fe7948d1ba
+  data.tar.gz: 3b14eb94c34a986a0afe5412e5167681ad599b93
 SHA512:
-  metadata.gz: 037d03e913644404aad34e487c3709d1a62c1b068efe018583c653066dfa7f96f3a1b1b390e6f9ceff0a2651bede193185fa95e5a9b5d2dc9f2c6e346e5c901b
-  data.tar.gz: 3560dd269f9687cd9f17a56cdd47b04d34031531887b948b527b537d4f419ab3d4d6a4f8c98aa9a8e53d04646b4b0c927356b530801f5700b6a0b10ba63a2c40
+  metadata.gz: 7643046e0babf3bd28ee73ebc595148929dd14524aa02992c4b1e91c05f385d31b65521c43470181af647cb0082689eb8aada2567fe21f580a364064dcfbd78b
+  data.tar.gz: fe55c550ecda8d8dfdd1577a06b1ca7a97becce04299218d6d332f95bf1708d4f251b02a2b5c686d4e6a1da3c0913ad01236672e885a58c824d9da719922d83a

data/bin/aws-ssh

@@ -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

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-ssh
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Rafael Sales
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-17 00:00:00.000000000 Z
+date: 2015-08-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: slop
@@ -43,11 +43,13 @@ email:
 - rafaelcds@gmail.com
 executables:
 - awssh
+- aws-ssh
 extensions: []
 extra_rdoc_files: []
 files:
+- bin/aws-ssh
 - bin/awssh
-homepage: https://github.com/buzzstarter/awssh
+homepage: https://github.com/buzzstarter/aws-ssh
 licenses:
 - MIT
 metadata: {}