aws_recon 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a8cd90cd73ab80c766882e88a2bbd5f03c106a7365805b7e50847997decd98de
-  data.tar.gz: 1a605e550317fdbf2295f72c0df9df6bdb02b1cb70e940d83db3c779d165d7d3
+  metadata.gz: 233f4b10a4360186449d4046d61bb0ae0e78511483e4dac27ae0dfee89a8ff04
+  data.tar.gz: 1cde6970d7f06cdfa0d52cefd9d931794c832cceeb98c7b8fe5e2f7aa5a447ab
 SHA512:
-  metadata.gz: f3f8600403036263543e0be3d9fee779d40a9cedbed38a982f8144e645c1184d0e4c135691d87d0e7809033c55e371dab4eaaba8ff80b6ba27baab7663d75ec0
-  data.tar.gz: 7e4af190e9451570c40e90fdd222899bc11c072da15f496d7739cf057da5ee04615d6a6b0cfda375308fda88a9dd91796bb0ab8e31e6d3289f7cf4daf7286e3c
+  metadata.gz: 38ee33272bf5980f4c4e5a764790e29222febaf435bab4c7a7b478a688b07211432a281b762ec94565d052be2f7496a45b028b501f1c741446b06a87da75611f
+  data.tar.gz: 5691d204c31c2423e92c86bcb05b27406c95285ef5777d1f5225496750aae35a6642c52f3b4b92ac36859c05692bced6c0968f11d41ee12a43ed628a98f74afc

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    aws_recon (0.2.1)
+    aws_recon (0.2.2)
       aws-sdk (~> 3.0)
       parallel (~> 1.19)
 

data/lib/aws_recon.rb

@@ -8,7 +8,6 @@ require 'ostruct'
 require 'optparse'
 require 'yaml'
 require 'csv'
-require 'pry'
 require 'aws-sdk'
 require 'aws_recon/options.rb'
 require 'aws_recon/lib/mapper.rb'

data/lib/aws_recon/aws_recon.rb

@@ -4,37 +4,39 @@ SERVICES_CONFIG_FILE = File.join(File.dirname(__FILE__), 'services.yaml').freeze
 
 module AwsRecon
   class CLI
-    # parse options
-    @options = Parser.parse ARGV.length < 1 ? %w[-h] : ARGV
+    def initialize
+      # parse options
+      @options = Parser.parse ARGV.length < 1 ? %w[-h] : ARGV
 
-    # timing
-    starting = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+      # timing
+      @starting = Process.clock_gettime(Process::CLOCK_MONOTONIC)
 
-    # AWS account id
-    @account_id = Aws::STS::Client.new.get_caller_identity.account
+      # AWS account id
+      @account_id = Aws::STS::Client.new.get_caller_identity.account
 
-    # AWS services
-    aws_services = YAML.load(File.read(SERVICES_CONFIG_FILE), symbolize_names: true)
+      # AWS services
+      @aws_services = YAML.load(File.read(SERVICES_CONFIG_FILE), symbolize_names: true)
 
-    # User config services
-    if @options.config_file
-      user_config = YAML.load(File.read(@options.config_file), symbolize_names: true)
+      # User config services
+      if @options.config_file
+        user_config = YAML.load(File.read(@options.config_file), symbolize_names: true)
 
-      services = user_config[:services]
-      regions = user_config[:regions]
-    else
-      regions = @options.regions
-      services = @options.services
-    end
+        @services = user_config[:services]
+        @regions = user_config[:regions]
+      else
+        @regions = @options.regions
+        @services = @options.services
+      end
 
-    # collection
-    @resources = []
+      # collection
+      @resources = []
 
-    # formatter
-    @formatter = Formatter.new
+      # formatter
+      @formatter = Formatter.new
 
-    unless @options.stream_output
-      puts "\nStarting collection with #{@options.threads} threads...\n"
+      unless @options.stream_output
+        puts "\nStarting collection with #{@options.threads} threads...\n"
+      end
     end
 
     #
@@ -66,16 +68,16 @@ module AwsRecon
     #
     # main wrapper
     #
-    def start
+    def start(_args)
       #
       # global services
       #
-      aws_services.map { |x| OpenStruct.new(x) }.filter { |s| s.global }.each do |service|
+      @aws_services.map { |x| OpenStruct.new(x) }.filter { |s| s.global }.each do |service|
         # user included this service in the args
-        next unless services.include?(service.name)
+        next unless @services.include?(service.name)
 
         # user did not exclude 'global'
-        next unless regions.include?('global')
+        next unless @regions.include?('global')
 
         collect(service, 'global')
       end
@@ -83,28 +85,28 @@ module AwsRecon
       #
       # regional services
       #
-      regions.filter { |x| x != 'global' }.each do |region|
-        Parallel.map(aws_services.map { |x| OpenStruct.new(x) }.filter { |s| !s.global }.each, in_threads: @options.threads) do |service|
+      @regions.filter { |x| x != 'global' }.each do |region|
+        Parallel.map(@aws_services.map { |x| OpenStruct.new(x) }.filter { |s| !s.global }.each, in_threads: @options.threads) do |service|
           # some services aren't available in some regions
-          skip_region = service&.excluded_regions&.include?(region)
+          skip_region = @service&.excluded_regions&.include?(region)
 
           # user included this region in the args
-          next unless regions.include?(region) && !skip_region
+          next unless @regions.include?(region) && !skip_region
 
           # user included this service in the args
-          next unless services.include?(service.name) || services.include?(service.alias) # rubocop:disable Layout/LineLength
+          next unless @services.include?(service.name) || @services.include?(service.alias) # rubocop:disable Layout/LineLength
 
           collect(service, region)
         end
       end
     rescue Interrupt # ctrl-c
-      elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - starting
+      elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - @starting
 
       puts "\nStopped early after \x1b[32m#{elapsed.to_i}\x1b[0m seconds.\n"
     ensure
       # write output file
       if @options.output_file
-        elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - starting
+        elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - @starting
 
         puts "\nFinished in \x1b[32m#{elapsed.to_i}\x1b[0m seconds. Saving resources to \x1b[32m#{@options.output_file}\x1b[0m.\n\n"
 

data/lib/aws_recon/options.rb

@@ -2,7 +2,7 @@
 
 class Parser
   DEFAULT_CONFIG_FILE = nil
-  DEFAULT_OUTPUT_FILE = File.join(File.dirname(__FILE__), '../output.json').freeze
+  DEFAULT_OUTPUT_FILE = File.expand_path(File.join(Dir.pwd, 'output.json')).freeze
   SERVICES_CONFIG_FILE = File.join(File.dirname(__FILE__), 'services.yaml').freeze
   DEFAULT_FORMAT = 'aws'
   DEFAULT_THREADS = 8
@@ -86,7 +86,7 @@ class Parser
 
       # output file
       opts.on('-o', '--output [OUTPUT]', 'Specify output file (default: output.json)') do |output|
-        args.output_file = output
+        args.output_file = File.expand_path(File.join(Dir.pwd, output))
       end
 
       # output format

data/lib/aws_recon/version.rb

@@ -1,3 +1,3 @@
 module AwsRecon
-  VERSION = "0.2.1"
+  VERSION = "0.2.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws_recon
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Josh Larsen