aws-inventory 0.2.0

data/Guardfile

@@ -0,0 +1,19 @@
+guard "bundler", cmd: "bundle" do
+  watch("Gemfile")
+  watch(/^.+\.gemspec/)
+end
+
+guard :rspec, cmd: "bundle exec rspec" do
+  require "guard/rspec/dsl"
+  dsl = Guard::RSpec::Dsl.new(self)
+
+  # RSpec files
+  rspec = dsl.rspec
+  watch(rspec.spec_helper) { rspec.spec_dir }
+  watch(rspec.spec_support) { rspec.spec_dir }
+  watch(rspec.spec_files)
+
+  # Ruby files
+  ruby = dsl.ruby
+  dsl.watch_spec_files_for(ruby.lib_files)
+end

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Tung Nguyen
+
+MIT License
+
+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/README.md

@@ -0,0 +1,65 @@
+# AWS Inventory
+
+AWS Inventory tool. Useful to get summarized information on AWS account.  The tool by default produces a reoprt that can easily be read from a terminal, but it can also be used produce a tab separated output report that can be pasted into a spreadsheet and then copied to a report.  This is controlled via a `AWS_INVENTORY_FORMAT` env variable and covered in the Format Options section.
+
+## Usage
+
+```sh
+aws-inventory acm             # report acm inventory
+aws-inventory cfn             # report cfn inventory
+aws-inventory cw              # report cloudwatch inventory
+aws-inventory eb              # report eb inventory
+aws-inventory ec2             # report ec2 inventory
+aws-inventory ecs             # report ecs inventory
+aws-inventory elb             # report elb inventory
+aws-inventory iam             # report iam inventory
+aws-inventory keypair         # report keypair inventory
+aws-inventory rds             # report rds inventory
+aws-inventory route53         # report route53 inventory
+aws-inventory sg              # report security group inventory
+aws-inventory vpc             # report vpc inventory
+```
+
+## Example
+
+What the output looks something like:
+
+```sh
+$ aws-inventory ec2
++-------+-------------+---------------+----------+-----------------+
+| Name  | Instance Id | Instance Type | Platform | Security Groups |
++-------+-------------+---------------+----------+-----------------+
+| name1 | i-123       | m3.medium     | linux    | sg-123          |
+| name2 | i-456       | t2.small      | linux    | sg-456          |
++-------+-------------+---------------+----------+-----------------+
+$
+```
+
+If you want to copy this to an spreadsheet, you can use the tab format.  Here's an example with `pbcopy`:
+
+```sh
+export AWS_INVENTORY_FORMAT=tab
+aws-inventory ec2 | pbcopy
+```
+
+### To Check Every Region for EC2 Instances
+
+```sh
+GREEN='\033[0;32m'
+NC='\033[0m' # No Color
+for i in $(aws ec2 describe-regions | jq -r '.Regions[].RegionName') ; do
+  echo -e "$GREEN$i$NC"
+  AWS_REGION=$i inventory ec2
+done
+```
+
+### Format Option
+
+There are 2 supported formats: tab and table.  The default is table.  To switch between formats use the AWS_INVENTORY_FORMAT environment variable.
+
+```sh
+export AWS_INVENTORY_FORMAT=tab
+aws-inventory ec2
+export AWS_INVENTORY_FORMAT=table
+aws-inventory ec2
+```

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+task :default => :spec
+
+RSpec::Core::RakeTask.new

data/exe/aws-inventory

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+# Trap ^C
+Signal.trap("INT") {
+  puts "\nCtrl-C detected. Exiting..."
+  sleep 1
+  exit
+}
+
+$:.unshift(File.expand_path("../../lib", __FILE__))
+require "inventory"
+require "inventory/cli"
+
+Inventory::CLI.start(ARGV)

data/inventory.gemspec

@@ -0,0 +1,37 @@
+# coding: utf-8
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "inventory/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "aws-inventory"
+  spec.version       = Inventory::VERSION
+  spec.authors       = ["Tung Nguyen"]
+  spec.email         = ["tongueroo@gmail.com"]
+  spec.description   = %q{Tool to inventory AWS account}
+  spec.summary       = %q{Tool to inventory AWS account}
+  spec.homepage      = "https://github.com/tongueroo/aws-inventory"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "thor"
+  spec.add_dependency "hashie"
+  spec.add_dependency "colorize"
+  spec.add_dependency "aws-sdk"
+  spec.add_dependency "activesupport"
+  spec.add_dependency "text-table"
+  spec.add_dependency "facets"
+  spec.add_dependency "actionview"
+
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "byebug"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "guard"
+  spec.add_development_dependency "guard-bundler"
+  spec.add_development_dependency "guard-rspec"
+end

data/lib/inventory.rb

@@ -0,0 +1,29 @@
+$:.unshift(File.expand_path("../", __FILE__))
+require "inventory/version"
+require "text-table"
+require "active_support/all"
+require "pp"
+require "byebug"
+
+module Inventory
+  autoload :Base, "inventory/base"
+  autoload :Help, "inventory/help"
+  autoload :Command, "inventory/command"
+  autoload :CLI, "inventory/cli"
+  autoload :AwsServices, "inventory/aws_services"
+  autoload :Shared, "inventory/shared"
+  autoload :Presenter, "inventory/presenter"
+  autoload :Cfn, "inventory/cfn"
+  autoload :Ec2, "inventory/ec2"
+  autoload :Vpc, "inventory/vpc"
+  autoload :SecurityGroup, "inventory/security_group"
+  autoload :Rds, "inventory/rds"
+  autoload :Route53, "inventory/route53"
+  autoload :Acm, "inventory/acm"
+  autoload :Elb, "inventory/elb"
+  autoload :Eb, "inventory/eb"
+  autoload :Ecs, "inventory/ecs"
+  autoload :Keypair, "inventory/keypair"
+  autoload :Iam, "inventory/iam"
+  autoload :Cloudwatch, "inventory/cloudwatch"
+end

data/lib/inventory/acm.rb

@@ -0,0 +1,15 @@
+class Inventory::Acm < Inventory::Base
+  def header
+    ["Domain", "Cert Arn"]
+  end
+
+  def data
+    certificate_summary_list.map do |cert|
+      [cert.domain_name, cert.certificate_arn]
+    end
+  end
+
+  def certificate_summary_list
+    @summary ||= acm.list_certificates.certificate_summary_list
+  end
+end

data/lib/inventory/aws_services.rb

@@ -0,0 +1,53 @@
+require 'aws-sdk'
+
+module Inventory::AwsServices
+  include Inventory::Shared
+
+  def cfn
+    @cfn ||= Aws::CloudFormation::Client.new
+  end
+
+  def ec2
+    @ec2 ||= Aws::EC2::Client.new
+  end
+
+  def pricing
+    @pricing ||= Aws::Pricing::Client.new
+  end
+
+  def rds
+    @rds ||= Aws::RDS::Client.new
+  end
+
+  def route53
+    @route53 ||= Aws::Route53::Client.new
+  end
+
+  def acm
+    @acm ||= Aws::ACM::Client.new
+  end
+
+  def elbv1
+    @elbv1 ||= Aws::ElasticLoadBalancing::Client.new
+  end
+
+  def elbv2
+    @elbv2 ||= Aws::ElasticLoadBalancingV2::Client.new
+  end
+
+  def eb
+    @eb ||= Aws::ElasticBeanstalk::Client.new
+  end
+
+  def ecs
+    @ecs ||= Aws::ECS::Client.new
+  end
+
+  def iam
+    @iam ||= Aws::IAM::Client.new
+  end
+
+  def cw
+    @cw ||= Aws::CloudWatch::Client.new
+  end
+end

data/lib/inventory/base.rb

@@ -0,0 +1,74 @@
+# Child class must implement this interface.
+# Methods:
+#   header - header row to display.  Array of strings.
+#   data - data to display under header. 2D Array.
+#     Each item in the array represents a row of data.
+class Inventory::Base
+  include Inventory::AwsServices
+
+  def initialize(options)
+    @options = options
+  end
+
+  def report
+    return if test_mode
+
+    results = sort(data)
+    results.unshift(header) if header
+    presenter = Inventory::Presenter.new(results)
+    presenter.display
+  end
+
+  def sort(data)
+    data.sort_by {|a| a[0]}
+  end
+
+  def test_mode
+    if ENV['TEST']
+      puts "Testing #{self.class} report" # specs tests against this
+      true
+    end
+  end
+
+  def show(report_type)
+    ["all", report_type.to_s].include?(@options[:report_type])
+  end
+
+  class << self
+    # Track all command subclasses.
+    def subclasses
+      @subclasses ||= []
+    end
+
+    def inherited(base)
+      super
+
+      if base.name
+        self.subclasses << base
+      end
+    end
+
+    # Thought this might be useful for
+    # specs. Eager load all classes so then we can loop thorugh the
+    # methods and run specs on any new cli commands.
+    # The rspec code turn out a too ugly to follow though. Leaving this
+    # around in case eager_laod is useful for other purposes
+    def eager_load!
+      path = File.expand_path("../", __FILE__)
+
+      Dir.glob("#{path}/**/*.rb").select do |path|
+        next if !File.file?(path) or path =~ /version/
+
+        class_name = path
+                      .sub('.rb','')
+                      .sub(%r{.*/inventory}, 'inventory')
+                      .camelize
+        # special rules
+        class_name.sub!("Cli", "CLI")
+
+        class_name.constantize # use constantize instead of require
+          # so we dont have to worry about require order.
+      end
+    end
+  end
+end

data/lib/inventory/cfn.rb

@@ -0,0 +1,50 @@
+class Inventory::Cfn < Inventory::Base
+  ALL_STATUSES = %w[
+    REVIEW_IN_PROGRESS
+    CREATE_FAILED
+    UPDATE_ROLLBACK_FAILED
+    UPDATE_ROLLBACK_IN_PROGRESS
+    CREATE_IN_PROGRESS
+    UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS
+    ROLLBACK_IN_PROGRESS
+    DELETE_COMPLETE
+    UPDATE_COMPLETE
+    UPDATE_IN_PROGRESS
+    DELETE_FAILED
+    DELETE_IN_PROGRESS
+    ROLLBACK_COMPLETE
+    ROLLBACK_FAILED
+    UPDATE_COMPLETE_CLEANUP_IN_PROGRESS
+    CREATE_COMPLETE
+    UPDATE_ROLLBACK_COMPLETE
+  ]
+  ACTIVE_STATUSES = ALL_STATUSES - %w[DELETE_COMPLETE]
+
+  def header
+    ["Stack Name", "Description"]
+  end
+
+  def data
+    stack_summaries.map do |summary|
+      [summary.stack_name, summary.template_description]
+    end
+  end
+
+  def stack_summaries
+    @stack_summaries ||= cfn.list_stacks(stack_status_filter: ACTIVE_STATUSES).stack_summaries
+  end
+
+  # unused right now but leaving around to later figure out how to integrate
+  def text_table
+    stack_summaries.each do |summary|
+      table.rows << [summary.stack_name, summary.template_description]
+    end
+
+    table = Text::Table.new
+    table.head = %w[Name Description]
+    stack_summaries.each do |summary|
+      table.rows << [summary.stack_name, summary.template_description]
+    end
+    puts table
+  end
+end

data/lib/inventory/cli.rb

@@ -0,0 +1,86 @@
+module Inventory
+  class CLI < Command
+    class_option :verbose, type: :boolean
+
+    desc "cfn", "report cfn inventory"
+    long_desc Help.text(:cfn)
+    def cfn
+      Cfn.new(options).report
+    end
+
+    desc "ec2", "report ec2 inventory"
+    long_desc Help.text(:ec2)
+    def ec2
+      Ec2.new(options).report
+    end
+
+    desc "vpc", "report vpc inventory"
+    long_desc Help.text(:vpc)
+    def vpc
+      Vpc.new(options).report
+    end
+
+    desc "sg", "report security group inventory"
+    long_desc Help.text(:sg)
+    option :report_type, default: "open", desc: "all, summary, or open"
+    def sg
+      SecurityGroup.new(options).report
+    end
+
+    desc "rds", "report rds inventory"
+    long_desc Help.text(:rds)
+    option :report_type, default: "summary", desc: "all, summary, or port"
+    def rds
+      Rds.new(options).report
+    end
+
+    desc "route53", "report route53 inventory"
+    long_desc Help.text(:route53)
+    def route53
+      Route53.new(options).report
+    end
+
+    desc "acm", "report acm inventory"
+    long_desc Help.text(:acm)
+    def acm
+      Acm.new(options).report
+    end
+
+    desc "elb", "report elb inventory"
+    long_desc Help.text(:elb)
+    def elb
+      Elb.new(options).report
+    end
+
+    desc "eb", "report eb inventory"
+    long_desc Help.text(:eb)
+    def eb
+      Eb.new(options).report
+    end
+
+    desc "ecs", "report ecs inventory"
+    long_desc Help.text(:ecs)
+    def ecs
+      Ecs.new(options).report
+    end
+
+    desc "keypair", "report keypair inventory"
+    long_desc Help.text(:keypair)
+    def keypair
+      Keypair.new(options).report
+    end
+
+    desc "iam", "report iam inventory"
+    long_desc Help.text(:iam)
+    option :report_type, default: "groups", desc: "all, groups, users, or summary"
+    def iam
+      Iam.new(options).report
+    end
+
+    desc "cw", "report cloudwatch inventory"
+    long_desc Help.text(:cw)
+    def cw
+      Cloudwatch.new(options).report
+    end
+  end
+end