awsm 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e24a01866b9592580f781071d0930eb454b81a27
+  data.tar.gz: 5ae12ba51869dd5339f515f770bcf8b0b44470ef
+SHA512:
+  metadata.gz: 72f890cb1c74a2a3913c7477be147e8cf40b2cb9e20e1d2d6cc59472d0bfa84f0ec966a363df41b9f865fca88ece6f839780da063235bd7b8b00ecb8bac8a8b5
+  data.tar.gz: 3a84fe176fdfe08a7b085dbd57ddbcea3e9e857626de1906d5915f0781a457063915d032c861d53a1ad58e97d75278abaac2f100ecc4e5210230ca3cbb673868

data/.gitignore

@@ -0,0 +1,23 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log
+.*.swp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in awsome.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Daniel Kendell
+
+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,39 @@
+# Awsm
+
+Awsm is an awesome AWS querying tool. Pun most certainly intended.
+
+## Make it go!
+
+Awsm is written in Ruby 2.1, so you'll need that to start with, you'll also need the `bundler` gem.
+
+	[you@host:~]$ git clone https://github.com/mduk/awsome
+	[you@host:~]$ cd awsome
+
+Awsm requires four enviornment variables to be set in order to work. I use something like this:
+
+	#!/bin/bash
+
+	export AWS_ACCESS_KEY_ID="You do not leak keys on github."
+	export AWS_SECRET_ACCESS_KEY="You DO, NOT, LEAK, keys on github!"
+	export AWS_REGION="eu-west-1"
+	export AWSM_HOSTEDZONE="Don't leak this either."
+
+	echo -e "\033[1m\033[92mAWS Environment Set.\033[0m"
+
+Once you have that in place, install dependencies.
+
+	[you@host:~/awsome]$ bundle install
+
+Initialise environment variables.
+
+	[you@host:~/awsome]$ source ./setvars.sh
+	AWS Environment Set.
+
+And off you go.
+
+	[you@host:~/awsome]$ bundle exec awsome search -dia beta
+
+
+# Licence
+
+See `LICENCE.txt` file.

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/awsm.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'awsm/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "awsm"
+  spec.version       = Awsm::VERSION
+  spec.authors       = ["Daniel Kendell"]
+  spec.email         = ["daniel@kendell.org"]
+  spec.summary       = %q{Awsm AWS queryer}
+  spec.description   = File.open( 'README.md', 'rb' ).read
+  spec.homepage      = "http://github.com/mduk/awsm"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = [ 'awsm' ]
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency 'thor', '~> 0.19'
+  spec.add_runtime_dependency 'aws-sdk-core', '~> 2.0'
+
+  spec.add_development_dependency "bundler", "~> 1.6"
+  spec.add_development_dependency "rake", '~> 0'
+  spec.add_development_dependency "pry", '~> 0'
+end

data/bin/awsm

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'awsm'
+
+Awsm::Cli.start( ARGV )
+

data/lib/awsm.rb

@@ -0,0 +1,9 @@
+require "awsm/version"
+require "awsm/loadbalancers"
+require "awsm/autoscalinggroups"
+require "awsm/dns"
+require "awsm/instances"
+require "awsm/cli"
+
+module Awsm
+end

data/lib/awsm/autoscalinggroups.rb

@@ -0,0 +1,28 @@
+module Awsm
+  class AutoScalingGroups
+    def initialize
+      @client = Aws::AutoScaling::Client.new
+      @asg_map = {}
+      load_auto_scaling_groups
+    end
+
+    def find_for( elb_name )
+      @asg_map[ elb_name ]
+    end
+
+    def load_auto_scaling_groups
+      @client.describe_auto_scaling_groups.each_page do |p|
+        p.auto_scaling_groups.map do |asg|
+          asg.load_balancer_names.each do |elb_name|
+            @asg_map[elb_name] = [] if @asg_map[elb_name].nil?
+            @asg_map[elb_name] << {
+              asg_name: asg.auto_scaling_group_name,
+              instances: asg.instances.map{ |i| i.to_h },
+              instance_ids: asg.instances.map { |i| i.instance_id }
+            }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/awsm/cli.rb

@@ -0,0 +1,147 @@
+require 'thor'
+
+module Awsm 
+  class Cli < Thor
+    class_option :instances, :type => :boolean, :default => false, :aliases => "-i",
+      :desc => "Also show instance information. Including ID, Private IP, and current health."
+    class_option :dns, :type => :boolean, :default => false, :aliases => "-d",
+      :desc => "Check if there are any Route53 records pointing at this Load Balancer"
+    class_option :asg, :type => :boolean, :default => false, :aliases => "-a",
+      :desc => "Show Auto Scaling Groups that are receiving traffic from this Load Balancer"
+
+    desc "specific <comma-separated-elb-names>",
+      "Only find specific ELBs named in a comma-separated list."
+    def specific( elb_names )
+      results = load_balancers.get(elb_names)
+      results = do_options( results )
+      results.each { |elb| print_result(elb) }
+    end
+
+    desc "search <search-terms>",
+      "Show ELBs where the name matches the search terms."
+    def search( search_terms )
+      results = load_balancers.getAll.select { |combined| combined[:elb][:name] =~ /#{search_terms}/ }
+      results = do_options( results )
+      results.each { |elb| print_result(elb) }
+    end
+
+    no_commands do
+      def load_balancers
+        Awsm::LoadBalancers.new
+      end
+
+      def instances
+        Awsm::Instances.new
+      end
+
+      def dns
+        Awsm::Dns.new
+      end
+
+      def asg
+        Awsm::AutoScalingGroups.new
+      end
+
+      def do_options( results )
+        if options[:instances]
+          results.map do |combined|
+            instance_data = Awsm::Instances.new.get_instance_data( combined[:elb_instance_ids] )
+            combined[:instances] = instance_data
+          end
+          say "Added instance data", :white
+        end
+
+        if options[:dns]
+          results.map do |combined|
+            elb_dns = combined[:elb][:dns]
+            r53_dns = dns.find_for( elb_dns )
+            combined[:r53_dns] = r53_dns
+          end
+          say "Added dns data", :white
+        end
+
+        if options[:asg]
+          results.map do |combined|
+            elb_name = combined[:elb][:name]
+            asgs = asg.find_for( elb_name )
+            combined[:asgs] = asgs
+          end
+        end
+
+        results
+      end
+
+      def print_result(result)
+        say result[:elb][:name], :green, false
+        print ' <= '
+
+	# DNS
+        if result[:r53_dns].nil?
+          say result[:elb][:dns], :white, false
+        else
+          say result[:r53_dns], :bold, false
+        end
+
+        puts
+
+	# Load Balancer Tags
+        say "  NO TAGS!", :red  unless result[:elb][:tags].length > 0
+        say "  Tags:" if result[:elb][:tags].length > 0
+        result[:elb][:tags].each do |k, v|
+          print '       '
+          say k, :cyan, false
+          print ' => '
+          say v, :yellow
+        end
+
+        if options[:instances]
+          say "  Instances:"
+          result[:instances].each do |instance_and_id|
+            instance = instance_and_id[1]
+            print_instance( 12, instance )
+          end
+        end
+
+	# Auto Scaling Groups
+	if options[:asg]
+	  say "  Auto Scaling Groups:"
+	end
+        if options[:asg] and result[:asgs].nil?
+	  say "    No Auto Scaling Groups!", :red
+	end
+
+        if options[:asg] and !result[:asgs].nil?
+          result[:asgs].each do |asg|
+          enabled = result[:elb_instance_ids] & asg[:instance_ids]
+
+          if enabled.length > 0
+            say "    >> #{asg[:asg_name]} (#{asg[:instances].length})", :magenta
+          else
+            say "       #{asg[:asg_name]} (#{asg[:instances].length})", :magenta
+          end
+
+          if options[:instances]
+            asg[:instances].each do |asg_instance|
+              instance_id = asg_instance[:instance_id]
+              instance_health = asg_instance[:health_status]
+              instance_health_colour = ( instance_health == "Healthy" ? :green : :red )
+              instance_description = result[:instances][ instance_id ]
+              instance_ip = instance_description[:private_ip_address]
+              unless result[:instances][ instance_id ].nil?
+                say "         #{instance_id} : #{instance_ip}", instance_health_colour
+              end
+            end
+          end
+        end
+      end
+    end
+
+      def print_instance( pad_length, instance )
+        padding = " " * pad_length
+        instance_id = instance[:instance_id]
+        instance_ip = instance[:private_ip_address]
+        say "#{padding}#{instance_id} : #{instance_ip}"
+      end
+    end
+  end
+end

data/lib/awsm/dns.rb

@@ -0,0 +1,37 @@
+require 'aws-sdk-core'
+
+module Awsm
+  class Dns
+    def initialize
+      @client = Aws::Route53::Client.new
+      @dns_map = {}
+      load_dns_records
+    end
+
+    def find_for( target )
+      @dns_map[ target ]
+    end
+
+    def load_dns_records
+      paged_response = @client.list_resource_record_sets(
+        hosted_zone_id: ENV['AWSM_HOSTEDZONE']
+      )
+
+      paged_response.each_page do |p|
+        records = p.resource_record_sets.select do |r|
+          r.type == 'A' || r.type == 'CNAME'
+        end
+
+        records.each do |r|
+          if r.resource_records == []
+            target = r.alias_target.dns_name
+          else
+            target = r.resource_records[0].value
+          end
+
+          @dns_map[target] = r.name
+        end
+      end
+    end
+  end
+end

data/lib/awsm/instances.rb

@@ -0,0 +1,21 @@
+module Awsm
+  class Instances
+    def initialize
+      @client = Aws::EC2::Client.new
+    end
+
+    def get_instance_data( instance_ids )
+      descriptions = @client.describe_instances( {
+        filters: [
+          { name: "instance-id", values: instance_ids }
+        ]
+      } ).reservations.first.instances
+
+      instance_hash = {}
+      descriptions.each do |description|
+        instance_hash[ description.instance_id ] = description.to_h
+      end
+      instance_hash
+    end
+  end
+end

data/lib/awsm/loadbalancers.rb

@@ -0,0 +1,60 @@
+module Awsm
+  class LoadBalancers
+    def initialize
+      @elb_client = Aws::ElasticLoadBalancing::Client.new
+      @asg_client = Aws::AutoScaling::Client.new
+    end
+
+    def get(elb_names)
+      processResponse(@elb_client.describe_load_balancers({
+        load_balancer_names: elb_names.split(',')
+      }))
+    end
+
+    def getAll
+      processResponse(@elb_client.describe_load_balancers)
+    end
+
+    private
+
+    def processResponse(pagableResponse)
+      elbs = []
+
+      pagableResponse.each_page do |p|
+        p.load_balancer_descriptions.each do |elb|
+          elbs << {
+            load_balancer_name: elb.load_balancer_name,
+            dns_name: elb.dns_name,
+            instance_ids: elb.instances.map { |i| i.instance_id }
+          }
+        end
+      end
+
+      combined = []
+      elbs.each do |elb|
+        result = {
+          elb: {
+            dns: elb[:dns_name],
+            name: elb[:load_balancer_name],
+            tags: getLoadBalancerTags( elb[:load_balancer_name] )
+          },
+          elb_instance_ids: elb[:instance_ids],
+        }
+
+        combined << result
+      end
+      combined
+    end
+
+    def getLoadBalancerTags(elb_name)
+      tags = {}
+      @elb_client.describe_tags( load_balancer_names: [ elb_name ] ).each_page do |p|
+        p.tag_descriptions.first.tags.each do |t|
+          tags[t.key] = t.value
+        end
+      end
+      tags
+    end
+  end
+end
+

data/lib/awsm/version.rb

@@ -0,0 +1,3 @@
+module Awsm
+  VERSION = "0.0.1"
+end

metadata

@@ -0,0 +1,141 @@
+--- !ruby/object:Gem::Specification
+name: awsm
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Daniel Kendell
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-03-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+description: "# Awsm\n\nAwsm is an awesome AWS querying tool. Pun most certainly intended.\n\n##
+  Make it go!\n\nAwsm is written in Ruby 2.1, so you'll need that to start with, you'll
+  also need the `bundler` gem.\n\n\t[you@host:~]$ git clone https://github.com/mduk/awsome\n\t[you@host:~]$
+  cd awsome\n\nAwsm requires four enviornment variables to be set in order to work.
+  I use something like this:\n\n\t#!/bin/bash\n\n\texport AWS_ACCESS_KEY_ID=\"You
+  do not leak keys on github.\"\n\texport AWS_SECRET_ACCESS_KEY=\"You DO, NOT, LEAK,
+  keys on github!\"\n\texport AWS_REGION=\"eu-west-1\"\n\texport AWSM_HOSTEDZONE=\"Don't
+  leak this either.\"\n\n\techo -e \"\\033[1m\\033[92mAWS Environment Set.\\033[0m\"\n\nOnce
+  you have that in place, install dependencies.\n\n\t[you@host:~/awsome]$ bundle install\n\nInitialise
+  environment variables.\n\n\t[you@host:~/awsome]$ source ./setvars.sh\n\tAWS Environment
+  Set.\n\nAnd off you go.\n\n\t[you@host:~/awsome]$ bundle exec awsome search -dia
+  beta\n\n\n# Licence\n\nSee `LICENCE.txt` file.\n"
+email:
+- daniel@kendell.org
+executables:
+- awsm
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- awsm.gemspec
+- bin/awsm
+- lib/awsm.rb
+- lib/awsm/autoscalinggroups.rb
+- lib/awsm/cli.rb
+- lib/awsm/dns.rb
+- lib/awsm/instances.rb
+- lib/awsm/loadbalancers.rb
+- lib/awsm/version.rb
+homepage: http://github.com/mduk/awsm
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Awsm AWS queryer
+test_files: []
+has_rdoc: