speculate 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 34c067c32dad1620b96437cf3daceb9161be8c16
-  data.tar.gz: 89bba6d1e2a5e77dcfbeef174ddca5ebc427ebec
+  metadata.gz: 4c4246f0306b5f4623c563247067fa0a81314d19
+  data.tar.gz: 3224578dc4726ac6658de9484838b0b9716126c7
 SHA512:
-  metadata.gz: d0450bb9b0c9ab16eb8075adbb508b6424eb6a1412c835df57f30103fc5ee1258dca7ca4ce3f84b7619b672d0410116aaa1e83057efb1dacd1ecd8786ba1b59c
-  data.tar.gz: 42cda06e782a40816c533c5eb032ca19a87d7df0026b139a344fc8cb60515bcd08fe7443e14560662b1b36b039b26ea58e1bc16e611312d3dfc795f49984fb35
+  metadata.gz: 145a15fb717abbf5be7b0642ff18447fe32a80875fd9aa022c39f01341e060de69eed5a509cf4ff23757140cccacc26df058c94e41038c10198d8872071a5bef
+  data.tar.gz: 8faaceb19e5d1e64121a3a3daf5e0b055910c55f77cb1959670c3d7d0f95682e15fa61de4f52b38c711a340c1d653b3b52e7036cfbc4e59ab48fd2bf72810e87

data/.circle-ruby

@@ -1,3 +1,4 @@
-2.3.1
-2.2.5
+2.4.0
+2.3.3
+2.2.6
 2.1.10

data/README.md

@@ -5,7 +5,7 @@ speculate
 [![Dependency Status](https://img.shields.io/gemnasium/akerl/speculate.svg)](https://gemnasium.com/akerl/speculate)
 [![Build Status](https://img.shields.io/circleci/project/akerl/speculate.svg)](https://circleci.com/gh/akerl/speculate)
 [![Coverage Status](https://img.shields.io/codecov/c/github/akerl/speculate.svg)](https://codecov.io/github/akerl/speculate)
-[![Code Quality](https://img.shields.io/codacy/.svg)](https://www.codacy.com/app/akerl/speculate)
+[![Code Quality](https://img.shields.io/codacy/1a0888d1510f4e39b62db9c8699c4946.svg)](https://www.codacy.com/app/akerl/speculate)
 [![MIT Licensed](https://img.shields.io/badge/license-MIT-green.svg)](https://tldrlegal.com/license/mit-license)
 
 Tool for assuming roles in AWS accounts

data/Rakefile

@@ -7,7 +7,7 @@ RSpec::Core::RakeTask.new(:spec)
 
 desc 'Run Rubocop on the gem'
 RuboCop::RakeTask.new(:rubocop) do |task|
-  task.patterns = ['lib/**/*.rb', 'spec/**/*.rb']
+  task.patterns = ['lib/**/*.rb', 'spec/**/*.rb', 'bin/*']
   task.fail_on_error = true
 end
 

data/bin/speculate

@@ -1,20 +1,38 @@
 #!/usr/bin/env ruby
 
 require 'speculate'
+require 'userinput'
 require 'mercenary'
 
-# rubocop:disable Metrics/BlockLength
+MFA_ASKER = UserInput.new(validation: /\d{6}/, message: 'MFA code')
+
 Mercenary.program(:speculate) do |p|
   p.version Speculate::VERSION
   p.description 'Tool for assuming roles in AWS accounts'
-  p.syntax 'speculate [options] account-id role-name'
+  p.syntax 'speculate [options] role-name'
+
+  # rubocop:disable Metrics/LineLength
+  p.option :account_id, '-a ACCOUNT_ID', '--accountid ACCOUNT_ID', 'Account ID to assume role on'
+  p.option :browser, '-b', '--browser', 'Open console URL in browser'
+  p.option :logout, '-l', '--logout', 'Log out of existing console browser session first'
+  p.option :mfa, '-m [TOKEN]', '--mfa [TOKEN]', 'Use MFA when assuming role'
+  # rubocop:enable Metrics/LineLength
 
   p.action do |_, options|
-    account_id, role_name = ARGV.shift 2
-    if !account_id || !role_name
+    options[:role_name] = ARGV.shift
+    unless options[:role_name]
       puts p
       exit 1
     end
-    Speculate.new(account_id: account_id, role_name: role_name).print!
+    options[:mfa] = MFA_ASKER.ask if options.fetch(:mfa, 0).nil?
+    session = Speculate.new(options)
+    puts session.to_envvars, "\n# #{session.url}"
+    if options[:browser]
+      if options[:logout]
+        system('open', session.logout_url)
+        sleep 1
+      end
+      system('open', session.url)
+    end
   end
 end

data/circle.yml

@@ -1,5 +1,6 @@
 dependencies:
   override:
+  - 'for i in $(cat .circle-ruby) ; do rvm-exec $i gem update --system --no-doc || exit 1 ; done'
   - 'for i in $(cat .circle-ruby) ; do rvm-exec $i bundle install || exit 1 ; done'
 test:
   override:

data/lib/speculate.rb

@@ -1,5 +1,3 @@
-require 'aws-sdk'
-
 ##
 # Base module for Speculate
 module Speculate
@@ -14,4 +12,6 @@ module Speculate
 end
 
 require 'speculate/version'
+require 'speculate/formatters'
+require 'speculate/console'
 require 'speculate/assumption'

data/lib/speculate/assumption.rb

@@ -1,6 +1,88 @@
-##
-# Base class for assumptions
+require 'aws-sdk'
+
 module Speculate
+  ##
+  # Base class for assumptions
   class Assumption
+    include Speculate::Formatters
+
+    def initialize(params = {})
+      @options = params
+    end
+
+    def creds
+      @creds ||= assumed_role.credentials
+    end
+
+    def url
+      @url ||= console.url
+    end
+
+    def logout_url
+      @logout_url ||= console.logout_url
+    end
+
+    private
+
+    def assumed_role
+      @assumed_role ||= client.assume_role(
+        role_arn: role_arn,
+        role_session_name: role_session_name,
+        **mfa_args
+      )
+    end
+
+    def client
+      @client ||= Aws::STS::Client.new(local_config)
+    end
+
+    def console
+      @console ||= Speculate::Console.new(creds: creds)
+    end
+
+    def mfa
+      @mfa ||= @options[:mfa]
+    end
+
+    def mfa_device
+      @mfa_device ||= local_identity.arn.sub('user/', 'mfa/')
+    end
+
+    def mfa_args
+      return {} unless mfa
+      { serial_number: mfa_device, token_code: mfa }
+    end
+
+    def role_name
+      @role ||= @options[:role_name] || raise('No role_name supplied')
+    end
+
+    def role_arn
+      @role_arn ||= "arn:aws:iam::#{target_account_id}:role/#{role_name}"
+    end
+
+    def role_session_name
+      @role_session_name ||= local_user_name
+    end
+
+    def target_account_id
+      @target_account_id ||= @options[:account_id] || local_account_id
+    end
+
+    def local_account_id
+      @local_account_id ||= local_identity.account
+    end
+
+    def local_user_name
+      @local_user_name ||= local_identity.arn.split(':').last.tr('/', '_')
+    end
+
+    def local_identity
+      @local_identity ||= client.get_caller_identity
+    end
+
+    def local_config
+      @local_config ||= @options[:creds] || {}
+    end
   end
 end

data/lib/speculate/console.rb

@@ -0,0 +1,52 @@
+require 'cgi'
+require 'open-uri'
+require 'json'
+
+module Speculate
+  ##
+  # Define console URL class
+  class Console
+    def initialize(params = {})
+      @options = params
+    end
+
+    def url
+      @url ||= [
+        'https://signin.aws.amazon.com/federation',
+        '?Action=login',
+        '&Issuer=',
+        '&Destination=',
+        CGI.escape('https://console.aws.amazon.com/'),
+        '&SigninToken=',
+        signin_token
+      ].join
+    end
+
+    def logout_url
+      @logout_url ||= 'https://console.aws.amazon.com/ec2/logout!doLogout'.freeze
+    end
+
+    private
+
+    def signin_token
+      @signin_token ||= JSON.parse(open(signin_token_url).read)['SigninToken']
+    end
+
+    def signin_token_url
+      @signin_token_url ||= [
+        'https://signin.aws.amazon.com/federation',
+        '?Action=getSigninToken',
+        '&Session=',
+        CGI.escape(signin_token_creds.to_json)
+      ].join
+    end
+
+    def signin_token_creds
+      @signin_token_creds ||= {
+        'sessionId' => @options[:creds].access_key_id,
+        'sessionKey' => @options[:creds].secret_access_key,
+        'sessionToken' => @options[:creds].session_token
+      }
+    end
+  end
+end

data/lib/speculate/formatters.rb

@@ -0,0 +1,14 @@
+module Speculate
+  ##
+  # Formatter module for printing creds to terminal
+  module Formatters
+    def to_envvars
+      @to_envvars ||= {
+        AWS_ACCESS_KEY_ID: creds.access_key_id,
+        AWS_SECRET_ACCESS_KEY: creds.secret_access_key,
+        AWS_SECURITY_TOKEN: creds.session_token,
+        AWS_SESSION_TOKEN: creds.session_token
+      }.map { |k, v| "export #{k}=#{v}" }.join("\n")
+    end
+  end
+end

data/lib/speculate/version.rb

@@ -1,5 +1,5 @@
 ##
 # Set version for module
 module Speculate
-  VERSION = '0.0.0'.freeze
+  VERSION = '0.0.3'.freeze
 end

data/speculate.gemspec

@@ -1,6 +1,9 @@
+$:.unshift File.expand_path('../lib/', __FILE__)
+require 'speculate/version'
+
 Gem::Specification.new do |s|
   s.name        = 'speculate'
-  s.version     = '0.0.1'
+  s.version     = Speculate::VERSION
   s.date        = Time.now.strftime('%Y-%m-%d')
 
   s.summary     = 'Tool for assuming roles in AWS accounts'
@@ -12,12 +15,15 @@ Gem::Specification.new do |s|
 
   s.files       = `git ls-files`.split
   s.test_files  = `git ls-files spec/*`.split
+  s.executables = ['speculate']
 
-  s.add_dependency 'aws-sdk', '~> 2.6.0'
+  s.add_dependency 'aws-sdk', '~> 2.8.0'
+  s.add_dependency 'mercenary', '~> 0.3.4'
+  s.add_dependency 'userinput', '~> 1.0.0'
 
-  s.add_development_dependency 'rubocop', '~> 0.42.0'
-  s.add_development_dependency 'rake', '~> 11.2.0'
+  s.add_development_dependency 'rubocop', '~> 0.47.0'
+  s.add_development_dependency 'rake', '~> 12.0.0'
   s.add_development_dependency 'codecov', '~> 0.1.1'
   s.add_development_dependency 'rspec', '~> 3.5.0'
-  s.add_development_dependency 'fuubar', '~> 2.1.0'
+  s.add_development_dependency 'fuubar', '~> 2.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: speculate
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Les Aker
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-12-01 00:00:00.000000000 Z
+date: 2017-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -16,42 +16,70 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.6.0
+        version: 2.8.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.6.0
+        version: 2.8.0
+- !ruby/object:Gem::Dependency
+  name: mercenary
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.4
+- !ruby/object:Gem::Dependency
+  name: userinput
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.42.0
+        version: 0.47.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.42.0
+        version: 0.47.0
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 11.2.0
+        version: 12.0.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 11.2.0
+        version: 12.0.0
 - !ruby/object:Gem::Dependency
   name: codecov
   requirement: !ruby/object:Gem::Requirement
@@ -86,17 +114,18 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.1.0
+        version: 2.2.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.1.0
+        version: 2.2.0
 description: Tool for assuming roles in AWS accounts
 email: me@lesaker.org
-executables: []
+executables:
+- speculate
 extensions: []
 extra_rdoc_files: []
 files:
@@ -113,6 +142,8 @@ files:
 - circle.yml
 - lib/speculate.rb
 - lib/speculate/assumption.rb
+- lib/speculate/console.rb
+- lib/speculate/formatters.rb
 - lib/speculate/version.rb
 - spec/spec_helper.rb
 - spec/speculate_spec.rb