awsm 0.0.1 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e24a01866b9592580f781071d0930eb454b81a27
-  data.tar.gz: 5ae12ba51869dd5339f515f770bcf8b0b44470ef
+  metadata.gz: abfb98b976f8940d34a6cea53ed10ec196a25d9b
+  data.tar.gz: 01446fea54dc664592c5b3af40115cb5f737a5d1
 SHA512:
-  metadata.gz: 72f890cb1c74a2a3913c7477be147e8cf40b2cb9e20e1d2d6cc59472d0bfa84f0ec966a363df41b9f865fca88ece6f839780da063235bd7b8b00ecb8bac8a8b5
-  data.tar.gz: 3a84fe176fdfe08a7b085dbd57ddbcea3e9e857626de1906d5915f0781a457063915d032c861d53a1ad58e97d75278abaac2f100ecc4e5210230ca3cbb673868
+  metadata.gz: f411dac4b99cd4d8d5df477d984f8031edc41975096c8cd2f0605d9fc86b9909d427cc55d067b81eb79faeb4f8fcd80cf6e8020ebd99844e74729f0a0e367584
+  data.tar.gz: 9e66e090cde43ce40a47d8647b04c6087686989e8f3b360cc674caab0cc4de552d83d41fa32e6be937eecfa5caf4be80ef1cf7d51b1d625aa4ae2b7e7114a16e

data/README.md

@@ -2,12 +2,28 @@
 
 Awsm is an awesome AWS querying tool. Pun most certainly intended.
 
-## Make it go!
+## Install from RubyGems
+
+Awsm is available on RubyGems, to install it, all you need to do is:
+
+```
+gem install awsm
+```
+
+## Install from Source
 
 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
+```
+[you@host:~]$ git clone https://github.com/mduk/awsm
+[you@host:~]$ cd awsm
+[you@host:~/awsm]$ bundle install
+[you@host:~/awsm]$ gem build awsm.gemspec
+[you@host:~/awsm]$ gem install awsm-x.x.x.gem
+[you@host:~/awsm]$ awsm
+```
+
+## Running Awsm
 
 Awsm requires four enviornment variables to be set in order to work. I use something like this:
 
@@ -20,19 +36,22 @@ Awsm requires four enviornment variables to be set in order to work. I use somet
 
 	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
-
+## Configuring Awsm
+
+Awsm will read configuration from `~/.awsm`. This is just used for the `spin` subcommand for now.
+An example `~/.awsm` file is shown below:
+
+```
+Spin:
+  InstanceType: t2.micro
+  KeyName: my-aws-keypair
+  Subnet: subnet-a0b1c2d3
+  SecurityGroups:
+    - sg-a0b1c2d3
+  Tags:
+    "h2g2:contact": "ford.prefect@megadodo.h2g2"
+    "h2g2:planet": kakrafoon
+```
 
 # Licence
 

data/awsm.gemspec

@@ -9,8 +9,8 @@ Gem::Specification.new do |spec|
   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.description   = %q{Awsm AWS queryer}
+  spec.homepage      = ""
   spec.license       = "MIT"
 
   spec.files         = `git ls-files -z`.split("\x0")
@@ -18,10 +18,11 @@ Gem::Specification.new do |spec|
   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_dependency 'thor'
+  spec.add_dependency 'aws-sdk', '~> 2'
+  spec.add_dependency 'terminal-table', '~> 1.4.5'
 
   spec.add_development_dependency "bundler", "~> 1.6"
-  spec.add_development_dependency "rake", '~> 0'
-  spec.add_development_dependency "pry", '~> 0'
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "pry"
 end

data/bin/awsm

@@ -2,5 +2,5 @@
 
 require 'awsm'
 
-Awsm::Cli.start( ARGV )
+Awsm::CLI::Main.start( ARGV )
 

data/lib/awsm/{cli.rb → cli/main.rb}

@@ -1,29 +1,57 @@
-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
+  module CLI
+    class Main < Clibase
+      class_option :instances, :type => :boolean, :default => false, :aliases => "-i",
+        :desc => "Also show instance information. Including ID, Private IP, and current health."
 
-    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
+      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 'spin', 'Ad-hoc instances'
+      subcommand 'spin', Awsm::CLI::Spin
+
+      desc 'unused', 'Find unused resources'
+      subcommand 'unused', Awsm::CLI::Unused
+
+      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 do |combined|
+          combined[:elb][:name] =~ /#{search_terms}/
+        end
+        results = do_options( results )
+        results.each { |elb| print_result(elb) }
+      end
+
+      desc "r53 [dns-name]",
+        "Show me what <dns-name> points at."
+      def r53( dns_name )
+        dns_len = dns_name.length
+
+        say "#{dns_name} ", :yellow
+        say "=> ", :bold
+        dns.get_by_record( dns_name ).each_with_index do |r, i|
+        if i > 0
+          say " " * ( dns_len + 4 )
+        end
+        say "(#{r.type}) ", :green
+        case r.type
+          when "A"
+            say "#{r.alias_target.dns_name}", :cyan
+          end
+        end
+      end
 
     no_commands do
       def load_balancers
@@ -75,7 +103,7 @@ module Awsm
         say result[:elb][:name], :green, false
         print ' <= '
 
-	# DNS
+        # DNS
         if result[:r53_dns].nil?
           say result[:elb][:dns], :white, false
         else
@@ -84,7 +112,7 @@ module Awsm
 
         puts
 
-	# Load Balancer Tags
+        # 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|
@@ -95,10 +123,14 @@ module Awsm
         end
 
         if options[:instances]
-          say "  Instances:"
-          result[:instances].each do |instance_and_id|
-            instance = instance_and_id[1]
-            print_instance( 12, instance )
+          if result[:instances] != []
+            say "  Instances:"
+            result[:instances].each do |instance_and_id|
+              instance = instance_and_id[1]
+              print_instance( 12, instance )
+            end
+          else
+            say "  No Instances!", :red
           end
         end
 
@@ -145,3 +177,4 @@ module Awsm
     end
   end
 end
+end

data/lib/awsm/cli/spin.rb

@@ -0,0 +1,153 @@
+module Awsm
+  module CLI
+  class Spin < Clibase
+
+    class_option :tables, :type => :boolean, :lazy_default => true, :default => true,
+      :desc => "Whether or not to draw ASCII tables."
+
+    desc 'up [AMI_ID]',
+      "Spin up an instance of the specified AMI"
+    def up( ami_id )
+      say "Spinning up #{ami_id}..."
+      response = ec2.run_instances(
+        image_id: ami_id,
+        min_count: 1,
+        max_count: 1,
+        key_name: configured_key_name,
+        instance_type: configured_instance_type,
+        security_group_ids: configured_security_groups,
+        subnet_id: configured_subnet
+      )
+
+      instance_id = response.instances.first.instance_id
+      say "Instance #{instance_id} is spinning up...", :green
+
+      while instance_extant?( instance_id ) == false
+        say '.', :green, false
+        sleep(3)
+      end
+
+      tags = [
+        { key: 'Name', value: "Temporary instance of #{ami_id} for #{whoami}" },
+        { key: 'awsm:owner', value: whoami }
+      ]
+
+      configured_tags.each do |k, v|
+        tags << { key: k, value: v }
+      end
+
+      ec2.create_tags(
+        resources: [ instance_id ],
+        tags: tags
+      )
+
+      say "Tagged #{instance_id}:"
+      tags.each do |tag|
+        say "    #{tag[:key]} ", :cyan
+        say '=> '
+        say "#{tag[:value]}", :yellow
+      end
+    end
+
+    desc 'down [INSTANCE_ID]',
+      "Spin down the specified instance"
+    def down( instance_id )
+      say "Spinning down (terminating) #{instance_id}...", :red
+      ec2.terminate_instances(
+        instance_ids: [ instance_id ]
+      )
+    end
+
+    desc 'list',
+      "List all spinning instances"
+    option :simple, :type => :boolean, :default => false, :aliases => '-s',
+      :desc => "Display list without prettiness - good for sedding"
+    def list
+      response = ec2.describe_instances(
+        filters: [
+          { name: 'tag-key', values: [ "awsm:owner" ] }
+        ]
+      )
+      spinning = []
+      response.reservations.each do |r|
+        r.instances.each do |i|
+          owner = i.tags.select { |t| t.key == 'awsm:owner' }.first.value
+
+          if owner == whoami
+            fields = [ i.instance_id, i.state.name, i.image_id, owner, i.launch_time ]
+
+            if i.state.name == 'running'
+              fields << i.private_ip_address
+            else
+              fields << 'N/A'
+            end
+
+            spinning << fields
+          end
+        end
+      end
+
+      if options[:simple]
+        spinning.each do |row|
+          say row.join(' ')
+        end
+      else
+        puts_table(
+          headings: [ 'Instance ID', 'State', 'AMI ID', 'Owner', 'Launched Time', 'Private IP' ],
+          rows: spinning
+        )
+      end
+    end
+
+    no_commands do
+      def ec2
+        Aws::EC2::Client.new
+      end
+
+      def whoami
+        me_host = `hostname -f`.strip
+        me_user = `whoami`.strip
+        "#{me_user}@#{me_host}"
+      end
+
+      def instance_extant?( instance_id )
+        description = ec2.describe_instances(
+          instance_ids: [ instance_id ]
+        )
+
+        num_found = description.reservations.first.instances.length
+
+        if num_found == 0
+          return false
+        end
+
+        true
+      end
+
+      def config
+        Config.new.get('Spin')
+      end
+
+      def configured_security_groups
+        config['SecurityGroups']
+      end
+
+      def configured_subnet
+        config['Subnet']
+      end
+
+      def configured_instance_type
+        config['InstanceType']
+      end
+
+      def configured_key_name
+        config['KeyName']
+      end
+
+      def configured_tags
+        config['Tags']
+      end
+    end
+  end
+end
+end

data/lib/awsm/cli/unused.rb

@@ -0,0 +1,88 @@
+module Awsm
+  module CLI
+
+    class Unused < Clibase
+
+      class_option :tables, :type => :boolean, :lazy_default => true, :default => true,
+        :desc => "Whether or not to draw ASCII tables."
+
+      desc "elasticloadbalancers",
+        "Find and prune elastic load balancers that aren't connected to anything"
+      def elasticloadbalancers
+        puts_table(
+          headings: [ "ELB Name", "DNS", "Created Time" ],
+          rows: find_instanceless_elbs.map do |elb|
+            [ elb.load_balancer_name, elb.dns_name, elb.created_time ]
+          end
+        )
+      end
+
+      desc 'launchconfigurations',
+        "Find unused launch configurations"
+      def launchconfigurations
+        puts_table(
+          headings: [ "Launch Configuration Name", "Created Time" ],
+          rows: find_unused_launchconfigurations.map do |lc|
+            [ lc.launch_configuration_name, lc.created_time ]
+          end
+        )
+      end
+
+			desc 'ami',
+				"Find unused AMIs"
+			def ami
+				puts_table(
+					headings: [ "AMI ID", "Created Time" ],
+					rows: find_unused_amis do |ami|
+						[  ]
+					end
+				)
+			end
+
+      no_commands do
+        def elb
+          Aws::ElasticLoadBalancing::Client.new
+        end
+
+				def ec2
+					Aws::EC2::Client.new
+				end
+
+        def as
+          Aws::AutoScaling::Client.new
+        end
+
+        def find_instanceless_elbs
+          elbs = []
+          elb.describe_load_balancers.load_balancer_descriptions.each do |elb|
+            if elb.instances.length == 0
+              elbs << elb
+            end
+          end
+          elbs
+        end
+
+        def find_unused_launchconfigurations
+          used_launchconfiguration_names = []
+          as.describe_auto_scaling_groups.auto_scaling_groups.each do |asg|
+            used_launchconfiguration_names << asg.launch_configuration_name
+          end
+
+          launchconfigurations = []
+          as.describe_launch_configurations.launch_configurations.each do |lc|
+            unless used_launchconfiguration_names.include?( lc.launch_configuration_name )
+              launchconfigurations << lc
+            end
+          end
+          launchconfigurations
+        end
+
+				def find_unused_amis
+binding.pry
+				end
+      end
+    
+    end
+
+  end
+end

data/lib/awsm/clibase.rb

@@ -0,0 +1,15 @@
+module Awsm
+  class Clibase < Thor
+    no_commands do
+      def puts_table( table )
+        if options[:tables] == true
+          puts Terminal::Table.new( table )
+        else
+          table[:rows].each do |r|
+            puts r.join( ' ' )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/awsm/config.rb

@@ -0,0 +1,13 @@
+module Awsm
+  class Config
+    def initialize
+      if File.exist?( ENV['HOME'] + '/.awsm' )
+        file = File.open( ENV['HOME'] + '/.awsm' ).read
+        @config = YAML.load( file )
+      end
+    end
+    def get( key )
+      @config[ key ]
+    end
+  end
+end

data/lib/awsm/dns.rb

@@ -4,12 +4,19 @@ module Awsm
   class Dns
     def initialize
       @client = Aws::Route53::Client.new
-      @dns_map = {}
+      @dns_map_by_target = {}
+      @records = []
       load_dns_records
     end
 
     def find_for( target )
-      @dns_map[ target ]
+      @dns_map_by_target[ target ]
+    end
+
+    def get_by_record( record )
+      @records.select do |r|
+        r.name =~ /^#{record}/
+      end
     end
 
     def load_dns_records
@@ -29,7 +36,8 @@ module Awsm
             target = r.resource_records[0].value
           end
 
-          @dns_map[target] = r.name
+          @dns_map_by_target[target] = r.name
+          @records << r
         end
       end
     end

data/lib/awsm/instances.rb

@@ -5,11 +5,17 @@ module Awsm
     end
 
     def get_instance_data( instance_ids )
-      descriptions = @client.describe_instances( {
+      first_reservation = @client.describe_instances( {
         filters: [
           { name: "instance-id", values: instance_ids }
         ]
-      } ).reservations.first.instances
+      } ).reservations.first
+
+      if first_reservation.nil?
+        return []
+      end
+
+	  descriptions = first_reservation.instances
 
       instance_hash = {}
       descriptions.each do |description|

data/lib/awsm/version.rb

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

data/lib/awsm.rb

@@ -1,9 +1,19 @@
-require "awsm/version"
-require "awsm/loadbalancers"
-require "awsm/autoscalinggroups"
-require "awsm/dns"
-require "awsm/instances"
-require "awsm/cli"
-
-module Awsm
-end
+require 'yaml'
+require 'thor'
+require 'terminal-table'
+
+require 'awsm/config'
+require 'awsm/version'
+require 'awsm/loadbalancers'
+require 'awsm/autoscalinggroups'
+require 'awsm/dns'
+require 'awsm/instances'
+
+require 'awsm/clibase'
+require 'awsm/cli/unused'
+require 'awsm/cli/spin'
+require 'awsm/cli/main'
+
+
+
+require 'pry'

metadata

@@ -1,43 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: awsm
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.4
 platform: ruby
 authors:
 - Daniel Kendell
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-02 00:00:00.000000000 Z
+date: 2015-04-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'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.19'
+        version: '2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.19'
+        version: '2'
 - !ruby/object:Gem::Dependency
-  name: aws-sdk-core
+  name: terminal-table
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: 1.4.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: 1.4.5
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -56,42 +70,31 @@ dependencies:
   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"
+description: Awsm AWS queryer
 email:
 - daniel@kendell.org
 executables:
@@ -108,12 +111,16 @@ files:
 - bin/awsm
 - lib/awsm.rb
 - lib/awsm/autoscalinggroups.rb
-- lib/awsm/cli.rb
+- lib/awsm/cli/main.rb
+- lib/awsm/cli/spin.rb
+- lib/awsm/cli/unused.rb
+- lib/awsm/clibase.rb
+- lib/awsm/config.rb
 - lib/awsm/dns.rb
 - lib/awsm/instances.rb
 - lib/awsm/loadbalancers.rb
 - lib/awsm/version.rb
-homepage: http://github.com/mduk/awsm
+homepage: ''
 licenses:
 - MIT
 metadata: {}