sty 0.0.1

data/lib/sty/ssm.rb

@@ -0,0 +1,164 @@
+require_relative 'functions'
+require 'aws-sdk-ec2'
+require 'aws-sdk-ssm'
+require 'optparse'
+require 'ostruct'
+require 'uri'
+require 'pry'
+require 'json'
+require 'os'
+
+Aws.config.update(:http_proxy => ENV['https_proxy'])
+
+MINIMAL_AGENT_VERSION = '2.3.612.0'
+DEFAULT_REGION = 'ap-southeast-2'
+
+class SSM
+
+  def region
+    ENV['AWS_REGION'] || DEFAULT_REGION
+  end
+
+  def ec2
+    @ec2 = @ec2 || Aws::EC2::Client.new(region: region)
+  end
+
+  def ssm
+    @ssm = @ssm || Aws::SSM::Client.new(region: region)
+  end
+
+  def all_ec2_instances
+    ec2.describe_instances.map do |page|
+      page.reservations.map { |r| r.instances }
+    end.flatten
+  end
+
+  def all_ssm_details
+    ssm.describe_instance_information.map do |page|
+      page.instance_information_list
+    end.flatten
+  end
+
+  def ssm_details
+    @ssm_details = @ssm_details || all_ssm_details.map do |i|
+      [i.instance_id, i]
+    end.to_h
+  end
+
+  def instance_id(instance)
+    instance.instance_id
+  end
+
+  def instance_name(instance)
+    name_tag = instance.tags.select { |t| t.key == 'Name' }.first
+    name_tag ? name_tag.value : "None"
+  end
+
+  def instance_ips(instance)
+    instance.network_interfaces.map { |n| n.private_ip_address }
+  end
+
+  def extract_fields(inst)
+    [instance_name(inst), instance_id(inst)] + instance_ips(inst)
+  end
+
+  def prepare_regex(names)
+    names = ['.'] if names.empty?
+    names.map { |n| Regexp.new(Regexp.quote(n), Regexp::IGNORECASE) }
+  end
+
+  def win?(instance)
+    instance.platform =~ /windows/i
+  end
+
+  def platform(instance)
+    win?(instance) ? magenta("[Win] ") : ""
+  end
+
+  def ping(instance)
+    #binding.pry
+    status = ssm_details[instance_id(instance)]&.ping_status || 'Unavailable'
+    version_str = ssm_details[instance_id(instance)]&.agent_version
+    version = Gem::Version.new(version_str || '0')
+    reference_version = Gem::Version.new(MINIMAL_AGENT_VERSION)
+    old_version = version < reference_version
+
+    result = "[#{status}#{old_version ? ', old agent' : ''}]"
+
+    case
+      when status == 'Online' && !old_version
+        green(result)
+      when status == 'Online' && old_version
+        yellow(result)
+      else
+        red(result)
+    end
+
+  end
+
+  def find(instances, names)
+    re = prepare_regex(names)
+    instances.select do |i|
+      i.state.name == 'running' &&
+          extract_fields(i).any? { |f| re.all? { |r| f =~ r } }
+    end.sort{|a,b| instance_name(a) <=> instance_name(b)}
+  end
+
+  def print_refine_instances(instances, type)
+    puts "Please refine #{type} instance:"
+    instances.each_with_index do |inst, idx|
+      puts "#{idx}: #{instance_id(inst)} #{ping(inst)} [#{instance_name(inst)}] #{platform(inst)}(#{instance_ips(inst).join(', ')}) "
+    end
+  end
+
+  def refine(found)
+    if found.size > 1
+      refine = nil
+      loop do
+        yield(found)
+        refine = Integer(STDIN.gets.chomp) rescue false
+        break if refine && refine < found.size
+      end
+      target = found[refine]
+    else
+      target = found.first
+    end
+    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}"
+    end
+  end
+
+  def connect(args)
+    found = find(all_ec2_instances, args)
+    if found.empty?
+      puts "No instances found for search terms: #{args.join(', ')}"
+      exit 1
+    end
+    target = refine(found) {|found| print_refine_instances(found, 'target')}
+
+    begin
+      resp = ssm.start_session(target: instance_id(target))
+    rescue Exception => e
+      puts red("ERROR! Unable to start session")
+      puts white(e.message)
+      exit 1
+    end
+
+    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/lib/sty/store.rb

@@ -0,0 +1,3 @@
+class Store
+
+end

data/proxy.yaml

@@ -0,0 +1,7 @@
+lo:
+  proxy: http://localhost:3128
+  no-proxy: .internal
+
+ext:
+  proxy: http://proxy:8080
+  no-proxy: .internal

data/sty.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path("../lib/", __FILE__)
+$LOAD_PATH.unshift lib unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |s|
+  s.name = 'sty'
+  s.version = '0.0.1'
+  s.description = "Command line tools"
+  s.authors = ["Andrew Sozonnyk"]
+  s.email = ''
+  s.homepage = 'https://github.com/sozonnyk/sty'
+  s.license = 'MIT'
+  s.require_paths = %w(lib)
+  s.required_ruby_version = ">= 2.0.0"
+  s.required_rubygems_version = ">= 1.3.5"
+  s.summary = s.description
+
+  s.files = %w(.document sty.gemspec) + Dir["*.md", "bin/*", "lib/**/*.rb", "*.yaml"]
+  s.extensions = ['ext/install/Rakefile']
+
+  s.add_development_dependency "bundler", ">= 1.0", "< 3"
+  s.add_runtime_dependency 'aws-sdk-core', '~> 3'
+  s.add_runtime_dependency 'aws-sdk-ec2', '~> 1'
+  s.add_runtime_dependency 'aws-sdk-ssm', '~> 1'
+  s.add_runtime_dependency 'thor', '~> 0.20'
+  s.add_runtime_dependency 'ruby-keychain', '~> 0.3'
+end

metadata

@@ -0,0 +1,157 @@
+--- !ruby/object:Gem::Specification
+name: sty
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Andrew Sozonnyk
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-12-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.0'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ssm
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.20'
+- !ruby/object:Gem::Dependency
+  name: ruby-keychain
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+description: Command line tools
+email: ''
+executables: []
+extensions:
+- ext/install/Rakefile
+extra_rdoc_files: []
+files:
+- ".document"
+- README.md
+- auth-keys.yaml
+- auth.yaml
+- bin/sty
+- ec2.yaml
+- ext/install/Rakefile
+- lib/sty.rb
+- lib/sty/account.rb
+- lib/sty/auth-rotate.rb
+- lib/sty/auth.rb
+- lib/sty/cli.rb
+- lib/sty/config.rb
+- lib/sty/console.rb
+- lib/sty/dig.rb
+- lib/sty/functions.rb
+- lib/sty/info.rb
+- lib/sty/keychain.rb
+- lib/sty/proxy.rb
+- lib/sty/ssh.rb
+- lib/sty/ssm.rb
+- lib/sty/store.rb
+- proxy.yaml
+- sty.gemspec
+homepage: https://github.com/sozonnyk/sty
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.3.5
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Command line tools
+test_files: []