aws-ssh 1.0.1 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 78ca737b88d4a338e7f18dfab077b2fe7948d1ba
-  data.tar.gz: 3b14eb94c34a986a0afe5412e5167681ad599b93
+SHA256:
+  metadata.gz: 95f7d7c849326eabeacf2f9e3010424fdae6d1791983070726fd371926e9a284
+  data.tar.gz: 87e154a952299dc9f7eb7cb46bb69114ed5e684643f1ad4e33cdf45e9fe3d015
 SHA512:
-  metadata.gz: 7643046e0babf3bd28ee73ebc595148929dd14524aa02992c4b1e91c05f385d31b65521c43470181af647cb0082689eb8aada2567fe21f580a364064dcfbd78b
-  data.tar.gz: fe55c550ecda8d8dfdd1577a06b1ca7a97becce04299218d6d332f95bf1708d4f251b02a2b5c686d4e6a1da3c0913ad01236672e885a58c824d9da719922d83a
+  metadata.gz: e0bc9261566eb918f77afab42fdf859b9a61261de226c913ab161dcf78f89213d83b01dc971770146280c2dc1f5354ffb4f00f032139f2b32234a9cedd4065bf
+  data.tar.gz: 03220f46695d3ee3c98781d046906532c0acc5e29f1b843dab914f9d392d78156975d0cc18eed34c07098cf5d22089ba71dd37b988f1244253f0bb92fa1bbe11

data/bin/aws-ssh

@@ -1,16 +1,17 @@
 #!/usr/bin/env ruby
 
 require 'slop'
-require 'aws-sdk'
+require 'aws-sdk-ec2'
 require 'yaml'
 
-module AWSSH
+module AWSSSH
   class Runner
     DEFAULT_OPTIONS = {
       'user'    => ENV['user'],
       'format'  => 'ssh',
       'verbose' => false
     }
+    AWS_CONFIG_KEYS = [:profile, :region]
 
     attr_accessor :host_regex
 
@@ -23,43 +24,43 @@ module AWSSH
       shout "Options: #{cli.to_h.inspect}"
       shout "Defaults: #{defaults.inspect}"
 
-      Aws.config.merge!(profile: cli[:profile], region: cli[:region])
+      Aws.config.merge!(cli.to_h.select { |key| AWS_CONFIG_KEYS.include?(key) })
 
-      if cli['show-only']
-        print_output
-      else
-        ssh!
-      end
+      cli['show-only'] ? print_output : ssh!
     end
 
     def cli
       @cli ||= Slop.parse do |o|
         o.banner = <<-USAGE
-awssh - a tool that makes easy to ssh into AWS EC2 servers
+aws-ssh - a tool that makes easy to ssh into AWS EC2 servers
 
-Usage: awssh [hostname regex]
+Usage: aws-ssh [hostname regex]
 
 Examples:
-  $ awssh prod.*app2
+  $ aws-ssh prod.*app2
   ... will SSH into the instance and you will see:
   user@prod-rails-app4~$
 
-  $ awssh --show-only --stack qa -u worker
+  $ aws-ssh --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
+
+The config file .aws-ssh is looked up in following directories:
+  ./ (current directory)
+  ~/ ($HOME directory)
         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.string '-s',  '--stack',      'AWS OpsWorks Stack name regex. E.g: `-s prod` will match "Production" stack name'
+        o.string '-p',  '--profile',    'AWS config profile name. Default: profile set in .aws-ssh file', default: defaults['profile']
+        o.string '-r',  '--region',     'AWS region. E.g: us-east-1. Default: region set in .aws-ssh file', default: defaults['region']
+        o.string '-u',  '--user',       'SSH username to use. Default: user set in .aws-ssh 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   '-cs', '--csensitive', 'Use 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
@@ -69,16 +70,18 @@ Examples:
     end
 
     def ssh!
-      if instances.size > 1
+      if instances.empty?
+        exit_error! "No hostname matched."
+      elsif 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"
+        exit_error! "\nMultiple hostnames matched. Please be more specific in hostnames regex or use the last output line to use CSSH."
       end
       exec "ssh #{connection_string(instances.first)}"
     end
 
     def print_output
       instances.each do |i|
-        puts "ssh #{connection_string(i)}      \t => #{i.name}"
+        puts "ssh #{connection_string(i).ljust(30, ' ')} => #{i.name}"
       end
       puts "\n# All hosts in one-line (suitable for CSSH):"
       puts instances.map { |i| connection_string(i) }.join(' ')
@@ -90,8 +93,13 @@ Examples:
 
     def defaults
       @defaults ||= begin
-        defaults = File.exist?('.awssh') ? YAML.load_file('.awssh') : {}
-        DEFAULT_OPTIONS.merge(defaults)
+        if File.exist?('.awssh') || File.exist?("#{ENV['HOME']}/.awssh")
+          puts 'WARNING: please rename the config file .awssh to .aws-ssh'
+          exit(1)
+        end
+        defaults = YAML.load_file('.aws-ssh') if File.exist?('.aws-ssh')
+        defaults ||= YAML.load_file("#{ENV['HOME']}/.aws-ssh") if File.exist?("#{ENV['HOME']}/.aws-ssh")
+        DEFAULT_OPTIONS.merge(defaults) if defaults
       end
     end
 
@@ -105,6 +113,7 @@ Examples:
 
     def exit_error!(msg)
       $stderr.puts msg
+      $stderr.puts
       exit(1)
     end
 
@@ -140,8 +149,8 @@ Examples:
           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)
+          next unless word_match?(instance.hostname, host_regex, case_sensitive) ||
+                      word_match?(instance.name, host_regex, case_sensitive)
         end
 
         instance
@@ -159,4 +168,4 @@ Examples:
   end
 end
 
-AWSSH::Runner.new.run
+AWSSSH::Runner.new.run

data/bin/awssh

@@ -0,0 +1 @@
+bin/aws-ssh

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aws-ssh
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.3.0
 platform: ruby
 authors:
 - Rafael Sales
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-18 00:00:00.000000000 Z
+date: 2022-01-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: slop
@@ -25,31 +25,45 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '4'
 - !ruby/object:Gem::Dependency
-  name: aws-sdk
+  name: aws-sdk-ec2
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.11.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.11.0
 description: A tool that makes easy to ssh into AWS EC2 servers
 email:
 - rafaelcds@gmail.com
 executables:
-- awssh
 - aws-ssh
+- awssh
 extensions: []
 extra_rdoc_files: []
 files:
 - bin/aws-ssh
 - bin/awssh
-homepage: https://github.com/buzzstarter/aws-ssh
+homepage: https://github.com/rafaelsales/aws-ssh
 licenses:
 - MIT
 metadata: {}
@@ -68,10 +82,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.4.3
+rubygems_version: 3.0.3.1
 signing_key: 
 specification_version: 4
 summary: A tool that makes easy to ssh into AWS EC2 servers
 test_files: []
-has_rdoc: 

data/bin/awssh

@@ -1,162 +0,0 @@
-#!/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