sty 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be460e9dd95543a4956eddfaa6e45b8f1d8f9f89d74560bc769e23ad8284fd71
-  data.tar.gz: 0f6383f3d4913d6997420b78b6a30a9c266d029a45b2539c2fee6fa0c33aba77
+  metadata.gz: 7ceaafb4b92a0332aabcc49bd368a523a4be721c4bcc1bd1cb5cee40e662d2f6
+  data.tar.gz: 57e5292eeeb81d7a550c1a923493369927ad6d6a68536675b609c25cf5d3f239
 SHA512:
-  metadata.gz: 1a6cab7166156c2be6e4f4ca0e7c66fe34687ec30b738ad4b35cf229f04852540fcc29bd2488f583e09a666c3f837bad12c1eb01c7c12c065ce282dce53e4d90
-  data.tar.gz: 67468b0793e2073b989d91d20d4e0e81e067b3e7cc9429bed8cb67de829e2161ce9cc0c898a9c97141dee04d9200e910bda0e6f8aa0d27bb902e0d19e8ce4400
+  metadata.gz: 6c4af2cfa09c0149cb2d9bfb702271f1bf5d7d535c723b74da5289a4e9b8f5afc1b424b640827dab9c54112c84f6781b8851812c9a71fa7f34423bf5710bf9d8
+  data.tar.gz: d7c6e93effa758489e3577ab8de14ec33d0ffbc70ab8236eda737a755a913bd4e334e8f926800ede30e9d5da7a7af0ddc472d4c5ca78000391a7f7846a9ca94e

data/lib/sty/auth.rb

@@ -1,11 +1,10 @@
 require_relative 'functions'
 require_relative 'dig'
 
-SESSION_DURATION_SECONDS = 43_200
-DEFAULT_ROLE_NAME = 'ReadOnlyRole'
-DEFAULT_REGION = 'ap-southeast-2'
-
 class Auth
+  SESSION_DURATION_SECONDS = 43_200
+  DEFAULT_ROLE_NAME = 'ReadOnlyRole'
+  DEFAULT_REGION = 'ap-southeast-2'
 
   def logout
     current = ENV['AWS_ACTIVE_ACCOUNT']

data/lib/sty/cli.rb

@@ -4,6 +4,7 @@ require_relative 'auth'
 require_relative 'console'
 require_relative 'proxy'
 require_relative 'ssh'
+require_relative 'ssm'
 
 class Cli < Thor
   map auth: :login, acct: :account, px: :proxy
@@ -12,7 +13,12 @@ class Cli < Thor
     'sty'
   end
 
-  desc "ssh [OPTIONS] <SEARCH_TERM...>","Creates ssh connection to desired ec2 instance through existing jumphost. SEARCH_TERM to search in EC2 instance ID, name or IP address"
+  desc "ssm [OPTIONS] <SEARCH_TERMS...>","Creates ssm session to an EC2 instance. SEARCH_TERMS to search in EC2 instance ID, name or IP address"
+  def ssm(*search_term)
+    Ssm.new.connect(search_term)
+  end
+
+  desc "ssh [OPTIONS] <SEARCH_TERMS...>","Creates ssh connection to an EC2 instance through existing jumphost. SEARCH_TERMS to search in EC2 instance ID, name or IP address"
   method_option :no_jumphost, type: :boolean, default: false, aliases: "-n", desc: "Connect directly without jumphost"
   method_option :select_jumphost, type: :boolean, aliases: "-s", desc: "Select jumphost instance"
   method_option :use_key, type: :boolean, aliases: "-k", desc: "Use private key auth for target instance. Keys are searched recursively in ~/.sty/keys"
@@ -32,7 +38,7 @@ class Cli < Thor
   method_option :role, aliases: "-r", dssc: "Override role name"
   def login(path)
     source_run(__method__)
-    Auth.new.login(path, options[:browser])
+    Auth.new.login(path, options[:role])
   end
 
   desc "logout", "Forget current credentials and clear cache"

data/lib/sty/ssm.rb

@@ -1,19 +1,18 @@
 require_relative 'functions'
-require 'aws-sdk-ec2'
-require 'aws-sdk-ssm'
-require 'optparse'
-require 'ostruct'
+require 'pathname'
 require 'uri'
-require 'pry'
 require 'json'
-require 'os'
 
-Aws.config.update(:http_proxy => ENV['https_proxy'])
+class Ssm
+  MINIMAL_AGENT_VERSION = '2.3.612.0'
+  DEFAULT_REGION = 'ap-southeast-2'
+  SSM_PLUGIN = "/usr/local/bin/session-manager-plugin"
 
-MINIMAL_AGENT_VERSION = '2.3.612.0'
-DEFAULT_REGION = 'ap-southeast-2'
-
-class SSM
+  def initialize
+    require 'aws-sdk-ec2'
+    require 'aws-sdk-ssm'
+    Aws.config.update(:http_proxy => ENV['https_proxy'])
+  end
 
   def region
     ENV['AWS_REGION'] || DEFAULT_REGION
@@ -126,21 +125,21 @@ class SSM
     target
   end
 
+
   def session_manager
-    case
-    when OS.mac?
-      "#{dir}/bin/osx/session-manager-plugin"
-    when OS.linux?
-      "#{dir}/bin/linux/session-manager-plugin"
-    else
-      raise "The tool doesn't have binary for #{OS.host}"
+    unless Pathname.new(SSM_PLUGIN).exist?
+      puts red("SSM plugin is not found: #{SSM_PLUGIN}.")
+      puts white("You must have SSM plugin to continue.\nSee: https://docs.aws.amazon.com/systems-manager/latest/userguide/session-manager-working-with-install-plugin.html")
+      exit 1
     end
+    SSM_PLUGIN
   end
 
-  def connect(args)
-    found = find(all_ec2_instances, args)
+  def connect(search)
+
+    found = find(all_ec2_instances, search)
     if found.empty?
-      puts "No instances found for search terms: #{args.join(', ')}"
+      puts "No instances found for search terms: #{search.join(', ')}"
       exit 1
     end
     target = refine(found) {|found| print_refine_instances(found, 'target')}
@@ -156,9 +155,6 @@ class SSM
     mappings = {session_id: 'SessionId', token_value: 'TokenValue', stream_url: 'StreamUrl'}
     reqest_str = resp.to_h.map {|k, v| [mappings[k], v] }.to_h.to_json
     exec "#{session_manager} \'#{reqest_str}\' ap-southeast-2 StartSession"
-
   end
 
 end
-
-SSM.new.connect(ARGV)

data/sty.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift lib unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |s|
   s.name = 'sty'
-  s.version = '0.0.1'
+  s.version = '0.0.3'
   s.description = "Command line tools"
   s.authors = ["Andrew Sozonnyk"]
   s.email = ''

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sty
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Andrew Sozonnyk
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-12-07 00:00:00.000000000 Z
+date: 2019-12-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -150,7 +150,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: 1.3.5
 requirements: []
-rubygems_version: 3.0.3
+rubyforge_project: 
+rubygems_version: 2.7.7
 signing_key: 
 specification_version: 4
 summary: Command line tools