ec2-ssh 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 72f253fec1905d28a541b4a90f5539f850690ff1
+  data.tar.gz: 5497db4af38b1ca7df990f23c18adbca843578f5
+SHA512:
+  metadata.gz: d0b7adbbe570f304bed85d7ae171a1f871375061eeb64a914b590684b40b32e39f53cb6adcc4481f9414895f7aadc1e79e9bd09b12e7c21a2e366d4f95bb9935
+  data.tar.gz: 52078a2cd777a8477180f8e4a71fba99379c318b358e4a83a087fdfed5ddad0a5ffa2e2ff743d59c7529a78ccc37f6dfe4713d50e2400ba5553bb00b6e326ce1

data/.gitignore

@@ -0,0 +1,3 @@
+.DS_Store
+*.gem
+

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015-2016 Glide Talk, LTD.
+
+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/Rakefile

@@ -0,0 +1,8 @@
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+
+desc "Run tests"
+task :default => :test

data/bin/ec2-ssh

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), %w[.. lib]))
+require 'thor'
+require 'ec2-ssh'
+
+def help
+  Ec2Ssh::Cli.start(%w{help connect})
+end
+
+
+begin
+  ENV["THOR_DEBUG"] = "1" 
+  Ec2Ssh::Cli.start
+rescue Thor::RequiredArgumentMissingError => e
+  puts "\e[31mMissing Arguments: #{e}\e[0m\n\n"
+  help
+rescue Thor::InvocationError => e
+  puts "\e[31m#{e.to_s.gsub(/Usage:.+"/, '').chomp} but there's no such option\e[0m\n\n"
+  help
+end
+

data/ec2-ssh.gemspec

@@ -0,0 +1,22 @@
+$:.push File.expand_path("../lib", __FILE__)
+$:.push File.expand_path("../lib/ec2-ssh", __FILE__)
+
+require File.expand_path('../lib/ec2-ssh', __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "ec2-ssh"
+  s.version     = Ec2Ssh::VERSION
+  s.authors     = ["Ami Mahloof"]
+  s.email       = "ami.mahloof@gmail.com"
+  s.homepage    = "https://github.com/innovia/ec2-ssh"
+  s.summary     = "run commmands on multiple servers by tag or by autoscale group"
+  s.description = "EC2 SSH Utility"
+  s.required_rubygems_version = ">= 1.3.6"
+  s.files = `git ls-files`.split($\).reject{|n| n =~ %r[png|gif\z]}.reject{|n| n =~ %r[^(test|spec|features)/]}
+  s.add_runtime_dependency 'thor', '~> 0.19', '>= 0.19.1'
+  s.add_runtime_dependency 'aws-sdk', '~> 2.0.45', '>= 2.0.45'
+  s.add_runtime_dependency 'sshkit', '~> 1.7.1', '>= 1.7.1'
+  s.executables   = s.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  s.extra_rdoc_files = ['README.md', 'LICENSE']
+  s.license = 'MIT'
+end

data/lib/ec2-ssh/aws.rb

@@ -0,0 +1,61 @@
+module Ec2Ssh::Cli::Aws
+   def aws_init(profile, region)
+    ENV['AWS_PROFILE'] = profile
+    Aws.config.update({region: region})
+
+    @region = region
+    @as = Aws::AutoScaling::Client.new()
+    @ec2 = Aws::EC2::Client.new()
+
+    say "Currently running user is: #{Aws::IAM::CurrentUser.new.arn}"
+  end
+
+  def get_auto_scale_groups
+    say "Fetching AutoScale Groups - please wait..."
+      @as_groups = @as.describe_auto_scaling_groups.auto_scaling_groups
+
+      as_group_names = @as_groups.inject([]) {|acc, asg| acc << asg.auto_scaling_group_name; acc }
+
+      as_selection = {}
+      as_group_names.each_with_index.inject(as_selection) {|acc, pair|
+        element, index = pair
+        as_selection[index] = element
+        acc
+      } 
+      
+      say "AutoScale Group in #{options[:region]}:\n"
+      as_selection.each {|k,v| say "#{k}: #{v}"}
+
+      selected_as = ask("Which server group do you want to ssh to?", color = :yellow)
+      
+      get_instances('aws:autoscaling:groupName', as_selection[selected_as.to_i])
+  end
+
+  def get_instances(tag_key, tag_value)
+    @all_servers = []
+
+    response = @ec2.describe_instances({
+      filters: [
+      {
+        name: 'instance-state-code',
+        values: ['16']
+
+      },{
+        name: 'tag-key',
+        values: ["#{tag_key}"]
+      },{
+        name: 'tag-value',
+        values: ["#{tag_value}"]
+      }
+    ]
+    })
+
+    if !response.reservations.empty?
+      response.reservations.each {|r| r.instances.inject(@all_servers){|acc, k| acc << k.public_ip_address; acc}}
+    else
+      say "could not find any instances with the tag #{tag_key}: #{tag_value} on #{@region}"
+    end
+
+    say "All servers: #{@all_servers.size}"
+  end
+end

data/lib/ec2-ssh/cli.rb

@@ -0,0 +1,76 @@
+#!/usr/bin/env ruby
+# encoding: utf-8
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), %w[.. lib]))
+
+require 'aws-sdk'
+require 'thor'
+require 'pp'
+require 'sshkit'
+require 'sshkit/dsl'
+
+class Ec2Ssh::Cli  < Thor
+  include Thor::Actions
+  autoload :Aws,            'ec2-ssh/aws'
+  autoload :Ssh,            'ec2-ssh/ssh'
+  autoload :Utils,          'ec2-ssh/utils'
+
+  default_task :connect
+
+  desc "connect", "Connect to autoscale instance (random instance), Pass --cmd='whatever' to run a cmd on the server (use ; to seperate commands)"
+  method_option :cmd,                              :desc => 'commmand to run on remote servers', :required => true
+  method_option :profile,                          :desc => 'aws cli profile', :default => 'default'
+  method_option :region,                           :desc => "region", :default => 'us-east-1'
+  method_option :user,            :aliases => 'u', :desc => 'run as user', :default => 'ec2-user'
+  method_option :parallel,        :aliases => 'p', :desc => 'run in parallel'
+  method_option :sequence,        :aliases => 's', :desc => 'run in sequence'
+  method_option :groups,          :aliases => 'g', :desc => 'run in groups'
+  method_option :groups_limit,    :aliases => 'l', :desc => 'limit', :type => :numeric
+  method_option :wait,            :aliases => 'w', :desc => 'wait',  :type => :numeric
+  method_option :as,                               :desc => 'get autoscale groups'
+  method_option :tag_key,                          :desc => 'tag key to filter instances by', :default => 'Name'
+  method_option :tag_value,                        :desc => 'tag value to filter instances by'
+  method_option :terminal,        :aliases => 't', :desc => 'open terminal tabs for all servers'
+  def connect
+    extend Aws
+    extend Ssh
+    extend Utils
+    set_ssh(options[:user])
+    aws_init(options[:profile], options[:region])
+      
+    if options[:as]
+      get_auto_scale_groups
+    elsif options[:tag_value]
+      get_instances(options[:tag_key].chomp, options[:tag_value].chomp)
+    end
+    
+    if options[:terminal]
+      open_in_terminal
+    else
+
+      if options[:parallel] && options[:sequence]
+        say "You can't run both in sequence and in parallel at the same time"
+        exit 1
+      end
+
+      mode = :parallel if options[:parallel]
+      mode = :groups   if options[:groups]
+      mode = :sequence if options[:sequence]
+
+      mode = :parallel if mode.nil? 
+      dsl_options = {}
+      dsl_options[:in] = mode
+      dsl_options[:wait]  = options[:wait] if options[:wait]
+      dsl_options[:limit] = options[:groups_limit] if options[:groups_limit]
+      
+      say "dsl opts: #{dsl_options}", color = :cyan
+      ssh_to(options[:user], dsl_options, options[:cmd])
+    end
+  end
+
+  map ["-v", "--version"] => :version
+  desc "version", "version"
+  def version
+    say Ec2Ssh::ABOUT, color = :green
+  end  
+
+end

data/lib/ec2-ssh/ssh.rb

@@ -0,0 +1,21 @@
+module Ec2Ssh::Cli::Ssh
+  def set_ssh(user)
+    ENV['SSHKIT_COLOR'] = 'TRUE'
+    SSHKit.config.output_verbosity = Logger::DEBUG
+    SSHKit::Backend::Netssh.configure { |ssh|
+      ssh.ssh_options = {
+        :user => user,
+        :paranoid => false,
+        :forward_agent => true,
+        :user_known_hosts_file => '/dev/null'
+      }
+    }
+  end
+
+  def ssh_to(user, dsl_options, cmd)
+    say "Running #{cmd} via ssh in #{dsl_options}", color = :cyan
+    on @all_servers, dsl_options  do |host|
+        execute cmd
+    end
+  end
+end

data/lib/ec2-ssh/utils.rb

@@ -0,0 +1,19 @@
+module Ec2Ssh::Cli::Utils
+  def open_in_terminal
+    @all_servers.each do |server|
+      `osascript <<-eof
+         tell application "iTerm"
+          make new terminal
+          tell the current terminal
+            activate current session
+            launch session "Default Session"
+            tell the last session
+              set name to "#{server}"
+              write text "ssh ec2-user@#{server}"
+            end tell
+          end tell
+         end tell
+      eof`
+    end
+  end
+end

data/lib/ec2-ssh.rb

@@ -0,0 +1,8 @@
+module Ec2Ssh
+  VERSION   = "1.0.0"
+  ABOUT     = "ec2-ssh v#{VERSION} (c) #{Time.now.strftime("2015-%Y")} @innovia"
+
+  $:.unshift File.expand_path(File.join(File.dirname(__FILE__), %w[.. lib]))
+
+  autoload :Cli,      'ec2-ssh/cli'
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: ec2-ssh
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Ami Mahloof
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.19'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.19.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.19'
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 0.19.1
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.45
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.45
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.0.45
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.45
+- !ruby/object:Gem::Dependency
+  name: sshkit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.7.1
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.7.1
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.7.1
+description: EC2 SSH Utility
+email: ami.mahloof@gmail.com
+executables:
+- ec2-ssh
+extensions: []
+extra_rdoc_files:
+- README.md
+- LICENSE
+files:
+- .gitignore
+- LICENSE
+- README.md
+- Rakefile
+- bin/ec2-ssh
+- ec2-ssh.gemspec
+- lib/ec2-ssh.rb
+- lib/ec2-ssh/aws.rb
+- lib/ec2-ssh/cli.rb
+- lib/ec2-ssh/ssh.rb
+- lib/ec2-ssh/utils.rb
+homepage: https://github.com/innovia/ec2-ssh
+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: 1.3.6
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.2
+signing_key: 
+specification_version: 4
+summary: run commmands on multiple servers by tag or by autoscale group
+test_files: []