awful 0.0.183 → 0.1.0.alpha.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: c1ab2fc365e4aaccd052206843ef554704ffd86d
-  data.tar.gz: b4afadc9463ef855311cdea9e3e03ce6ebb3ee7a
+SHA256:
+  metadata.gz: a6d920e5f1bbc5756ffe4eb9bccfde37f6203454ff1e8ab6f2c9fc05a464eb4c
+  data.tar.gz: 39315ee5a3569c65a329828158a8f38611db2dac2e3f62d02e2d3610a8dcf12f
 SHA512:
-  metadata.gz: f7fbe4141ddaf40a4482d9562e12fd542ac0cdd42e84093beccdbfa36fd0e9c49525566cfa030a5e561e6b94fbd674f68ba1c248e8cede2a262e870a6fb108f8
-  data.tar.gz: 89db4d459b0846fa4d7b2b4b7e1075a9daf7f538833e14139cdf937eb3b0212d59bfbea0ce31b0fe47545658d5a4f3877096945c2c881bae47a8d2098404f9a3
+  metadata.gz: ce7775521074720421fe220028e0dad577aff399b0239585ced2126056dc469f3f64e431b90f9ffbb6cc012b3bfde40c37082879296d6530179db322ccc43a5f
+  data.tar.gz: 970509127de9063ec007ca57488c933158f7a4a08f4919d7540da5538da0d544e502c3a09d95e39d26a48c1500dbd4e65226bc2127b69fdb0a30358027c0d8be

data/awful.gemspec

@@ -18,10 +18,20 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_development_dependency "bundler", "~> 1.7"
+  spec.add_development_dependency "bundler"
   spec.add_development_dependency "rake"
 
-  spec.add_dependency('aws-sdk', '~> 2')
   spec.add_dependency('thor')
   spec.add_dependency('dotenv')
-end
+
+  spec.add_dependency('aws-sdk-autoscaling')
+  spec.add_dependency('aws-sdk-cloudformation')
+  spec.add_dependency('aws-sdk-dynamodb')
+  spec.add_dependency('aws-sdk-ec2')
+  spec.add_dependency('aws-sdk-ecr')
+  spec.add_dependency('aws-sdk-elasticloadbalancingv2')
+  spec.add_dependency('aws-sdk-rds')
+  spec.add_dependency('aws-sdk-s3')
+  spec.add_dependency('aws-sdk-secretsmanager')
+  spec.add_dependency('aws-sdk-ssm')
+end

data/bin/firehose

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+#-*- mode: ruby; -*-
+
+require 'awful'
+require 'awful/firehose'
+
+Awful::Firehose.start(ARGV)

data/bin/kinesis

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+#-*- mode: ruby; -*-
+
+require 'awful'
+require 'awful/kinesis'
+
+Awful::Kinesis.start(ARGV)

data/bin/{param → secret}

@@ -2,6 +2,6 @@
 #-*- mode: ruby; -*-
 
 require 'awful'
-require 'awful/param'
+require 'awful/secret'
 
-Awful::Param.start(ARGV)
+Awful::Secret.start(ARGV)

data/lib/awful/alb.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-elasticloadbalancingv2'
+
 module Awful
   module Short
     def alb(*args)
@@ -17,10 +19,6 @@ module Awful
     }
 
     no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-
       def alb
         @alb ||= Aws::ElasticLoadBalancingV2::Client.new
       end
@@ -167,4 +165,4 @@ module Awful
     end
 
   end
-end
+end

data/lib/awful/ami.rb

@@ -28,10 +28,6 @@ module Awful
 
         ec2.describe_images(params).images
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'ls [IDS]', 'list AMIs'
@@ -106,4 +102,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/auto_scaling.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-autoscaling'
+
 module Awful
   module Short
     def asg(*args)
@@ -28,10 +30,6 @@ module Awful
     }
 
     no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-
       def all_matching_asgs(name)
         autoscaling.describe_auto_scaling_groups.map(&:auto_scaling_groups).flatten.select do |asg|
           asg.auto_scaling_group_name.match(name) or tag_name(asg, '').match(name)
@@ -416,4 +414,4 @@ module Awful
       autoscaling.exit_standby(auto_scaling_group_name: name, instance_ids: instance_ids)
     end
   end
-end
+end

data/lib/awful/changesets.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-cloudformation'
+
 module Awful
   module Short
     def changesets(*args)
@@ -29,10 +31,6 @@ module Awful
       def cf
         @cf ||= Aws::CloudFormation::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.downcase.to_sym, :blue))
-      end
     end
 
     ## if no task given , run this
@@ -87,4 +85,4 @@ module Awful
     desc 'changesets', 'control changesets'
     subcommand 'changesets', Changesets
   end
-end
+end

data/lib/awful/cloudformation.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-cloudformation'
+
 module Awful
   module Short
     def cf(*args)
@@ -8,30 +10,35 @@ module Awful
   class CloudFormation < Cli
 
     COLORS = {
-      create_in_progress:                  :yellow,
-      delete_in_progress:                  :yellow,
-      update_in_progress:                  :yellow,
-      update_complete_cleanup_in_progress: :yellow,
-      create_failed:                       :red,
-      delete_failed:                       :red,
-      update_failed:                       :red,
-      create_complete:                     :green,
-      delete_complete:                     :green,
-      update_complete:                     :green,
-      delete_skipped:                      :yellow,
-      rollback_in_progress:                :red,
-      rollback_complete:                   :red,
+      CREATE_IN_PROGRESS:                  :yellow,
+      DELETE_IN_PROGRESS:                  :yellow,
+      UPDATE_IN_PROGRESS:                  :yellow,
+      UPDATE_COMPLETE_CLEANUP_IN_PROGRESS: :yellow,
+      CREATE_FAILED:                       :red,
+      DELETE_FAILED:                       :red,
+      UPDATE_FAILED:                       :red,
+      CREATE_COMPLETE:                     :green,
+      DELETE_COMPLETE:                     :green,
+      UPDATE_COMPLETE:                     :green,
+      DELETE_SKIPPED:                      :yellow,
+      ROLLBACK_IN_PROGRESS:                :red,
+      ROLLBACK_COMPLETE:                   :red,
+      ROLLBACK_FAILED:                     :red,
+      ACTIVE:                              :green,
+      DELETED:                             :red,
     }
 
-    no_commands do
-      def cf
-        @cf ||= Aws::CloudFormation::Client.new
-      end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.downcase.to_sym, :blue))
-      end
+    ## stack statuses that are not DELETE_COMPLETE
+    STATUSES = %i[
+      CREATE_IN_PROGRESS CREATE_FAILED CREATE_COMPLETE
+      ROLLBACK_IN_PROGRESS ROLLBACK_FAILED ROLLBACK_COMPLETE
+      DELETE_IN_PROGRESS DELETE_FAILED
+      UPDATE_IN_PROGRESS UPDATE_COMPLETE_CLEANUP_IN_PROGRESS UPDATE_COMPLETE
+      UPDATE_ROLLBACK_IN_PROGRESS UPDATE_ROLLBACK_FAILED UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS UPDATE_ROLLBACK_COMPLETE
+      REVIEW_IN_PROGRESS
+    ]
 
+    no_commands do
       ## get list of stacks
       def stack_summaries(next_token = nil)
         response = cf.list_stacks(next_token: next_token)
@@ -45,28 +52,16 @@ module Awful
     end
 
     desc 'ls [PATTERN]', 'list cloudformation stacks matching PATTERN'
-    method_option :long, aliases: '-l', type: :boolean, default: false, desc: 'Long listing'
-    method_option :all,  aliases: '-a', type: :boolean, default: false, desc: 'Show all, including stacks in DELETE_COMPLETE'
-    def ls(name = /./)
-      stacks = stack_summaries
-
-      ## skip deleted stacks unless -a given
-      unless options[:all]
-        stacks = stacks.select { |stack| stack.stack_status != 'DELETE_COMPLETE' }
-      end
-
-      ## match on given arg
-      stacks.select do |stack|
-        stack.stack_name.match(name)
+    method_option :long, aliases: '-l', type: :boolean, default: false, desc: 'long listing'
+    def ls(name = nil)
+      paginate(:stack_summaries) do |next_token|
+        cf.list_stacks(stack_status_filter: STATUSES, next_token: next_token)
+      end.tap do |stacks|
+        stacks.select! { |s| s.stack_name.match(name) } if name
       end.output do |list|
         if options[:long]
           print_table list.map { |s|
-            [
-              s.stack_name,
-              s.creation_time,
-              color(s.stack_status),
-              s.template_description,
-            ]
+            [s.stack_name, s.creation_time, color(s.stack_status), s.template_description]
           }.sort
         else
           puts list.map(&:stack_name).sort
@@ -279,5 +274,23 @@ module Awful
       end
       puts cf.estimate_template_cost(template_body: template, parameters: parameters).url
     end
+
+    desc 'sets', 'list stack sets'
+    method_option :long,   aliases: '-l', type: :boolean, default: false,    desc: 'long listing'
+    method_option :status, aliases: '-s', type: :string,  default: 'ACTIVE', desc: 'ACTIVE or DELETED'
+    def sets
+      paginate(:summaries) do |next_token|
+        cf.list_stack_sets(status: options[:status].upcase, next_token: next_token)
+      end.output do |sets|
+        if options[:long]
+          print_table sets.map { |s|
+            [s.stack_set_name, s.stack_set_id, color(s.status), s.description]
+          }
+        else
+          puts sets.map(&:stack_set_name)
+        end
+      end
+    end
+
   end
-end
+end

data/lib/awful/cloudfront.rb

@@ -18,10 +18,6 @@ module Awful
       def cloudfront
         @cloudfront ||= Aws::CloudFront::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.downcase.to_sym, :blue))
-      end
     end
 
     desc 'ls', 'list distributions'
@@ -101,4 +97,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/cloudwatch_events.rb

@@ -15,10 +15,6 @@ module Awful
       def events
         @events ||= Aws::CloudWatchEvents::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'ls [PREFIX]', 'list events'
@@ -77,4 +73,4 @@ module Awful
     end
 
   end
-end
+end

data/lib/awful/config.rb

@@ -21,10 +21,6 @@ module Awful
       def config
         @_config ||= Aws::ConfigService::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'recorders', 'show delivery recorders'
@@ -85,4 +81,4 @@ module Awful
     end
 
   end
-end
+end

data/lib/awful/dynamodb.rb

@@ -1,3 +1,4 @@
+require 'aws-sdk-dynamodb'
 require 'json'
 require 'awful/dynamodb_streams'
 
@@ -17,10 +18,6 @@ module Awful
     }
 
     no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-
       ## return array of tables names matching name
       def all_matching_tables(name)
         tables = []
@@ -428,4 +425,4 @@ module Awful
     desc 'streams SUBCOMMANDS', 'subcommands for dynamodb streams'
     subcommand 'streams', Streams
   end
-end
+end

data/lib/awful/ec2.rb

@@ -1,26 +1,13 @@
 require 'base64'
 
 module Awful
-  module Short
-    def ec2(*args)
-      Awful::Ec2.new.invoke(*args)
-    end
-  end
-
   class Ec2 < Cli
-
     COLORS = {
       running:    :green,
       stopped:    :yellow,
       terminated: :red,
     }
 
-    no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-    end
-
     desc 'ls [NAME]', 'get instances with given name regex'
     method_option :long,        aliases: '-l', type: :boolean, default: false, desc: 'Long listing'
     method_option :ids,         aliases: '-i', type: :array,   default: [],    desc: 'List of instance ids to retrieve'

data/lib/awful/ecr.rb

@@ -1,3 +1,4 @@
+require 'aws-sdk-ecr'
 require 'json'
 require 'base64'
 
@@ -191,4 +192,4 @@ module Awful
       puts "deleted: #{deleted}, failures: #{failures}"
     end
   end
-end
+end

data/lib/awful/ecs.rb

@@ -21,10 +21,6 @@ module Awful
       def ecs
         @ecs ||= Aws::ECS::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'ls NAME', 'list ECS clusters'
@@ -253,4 +249,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/elasticache.rb

@@ -17,10 +17,6 @@ module Awful
       def elasticache
         @elasticache ||= Aws::ElastiCache::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'ls [ID]', 'list clusters'
@@ -71,4 +67,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/elb.rb

@@ -12,10 +12,6 @@ module Awful
     }
 
     no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-
       ## cannot search ELBs by tag, so just name here
       def all_matching_elbs(name)
         elb.describe_load_balancers.map(&:load_balancer_descriptions).flatten.select do |elb|
@@ -158,4 +154,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/emr.rb

@@ -17,10 +17,6 @@ module Awful
       def emr
         @emr ||= Aws::EMR::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'ls', 'list clusters'
@@ -61,4 +57,4 @@ module Awful
     end
 
   end
-end
+end

data/lib/awful/firehose.rb

@@ -0,0 +1,63 @@
+require 'yaml'
+
+module Awful
+  module Short
+    def firehose(*args)
+      Awful::Firehose.new.invoke(*args)
+    end
+  end
+
+  class Firehose < Cli
+    COLORS = {
+      ACTIVE:   :green,
+      DELETING: :red,
+    }
+
+    no_commands do
+      def firehose
+        @_firehose ||= Aws::Firehose::Client.new
+      end
+
+      ## special-case paginator for delivery streams
+      def paginate_delivery_streams(thing)
+        token = nil
+        things = []
+        loop do
+          resp = yield(token)
+          items = resp.send(thing)
+          things += items
+          token = items.last
+          break unless resp.has_more_delivery_streams
+        end
+        things
+      end
+    end
+
+    desc 'ls', 'list firehose streams'
+    method_option :long, aliases: '-l', type: :boolean, default: false, desc: 'long listing'
+    def ls
+      paginate_delivery_streams(:delivery_stream_names) do |start|
+        firehose.list_delivery_streams(exclusive_start_delivery_stream_name: start)
+      end.output do |streams|
+        if options[:long]
+          print_table streams.map { |name|
+
+            s = firehose.describe_delivery_stream(delivery_stream_name: name).delivery_stream_description
+            op = s.has_more_destinations ? '>' : ''
+            [s.delivery_stream_name, op + s.destinations.count.to_s, color(s.delivery_stream_status), s.delivery_stream_type, 'v' + s.version_id.to_s, s.create_timestamp]
+          }
+        else
+          puts streams
+        end
+      end
+    end
+
+    desc 'dump NAME', 'describe firehose stream'
+    def dump(name)
+      firehose.describe_delivery_stream(delivery_stream_name: name).delivery_stream_description.output do |stream|
+        puts YAML.dump(stringify_keys(stream.to_hash))
+      end
+    end
+
+  end
+end

data/lib/awful/iam.rb

@@ -11,10 +11,6 @@ module Awful
       def iam
         @iam ||= Aws::IAM::Client.new
       end
-
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
     end
 
     desc 'users', 'list users'
@@ -159,4 +155,4 @@ module Awful
 
   end
 
-end
+end

data/lib/awful/kinesis.rb

@@ -0,0 +1,62 @@
+require 'yaml'
+
+module Awful
+  module Short
+    def kinesis(*args)
+      Awful::Kinesis.new.invoke(*args)
+    end
+  end
+
+  class Kinesis < Cli
+    COLORS = {
+      ACTIVE:   :green,
+      DELETING: :red,
+    }
+
+    no_commands do
+      def kinesis
+        @_kinesis ||= Aws::Kinesis::Client.new
+      end
+
+      ## special-case paginator for streams
+      def paginate_streams(thing)
+        token = nil
+        things = []
+        loop do
+          resp = yield(token)
+          items = resp.send(thing)
+          things += items
+          token = items.last
+          break unless resp.has_more_streams
+        end
+        things
+      end
+    end
+
+    desc 'ls', 'list streams'
+    method_option :long, aliases: '-l', type: :boolean, default: false, desc: 'long listing'
+    def ls
+      paginate_streams(:stream_names) do |start|
+        kinesis.list_streams(exclusive_start_stream_name: start)
+      end.output do |streams|
+        if options[:long]
+          print_table streams.map { |name|
+            s = kinesis.describe_stream(stream_name: name).stream_description
+            op = s.has_more_shards ? '>' : ''
+            [s.stream_name, op + s.shards.count.to_s, color(s.stream_status), s.encryption_type, s.retention_period_hours.to_s + 'h', s.stream_creation_timestamp]
+          }
+        else
+          puts streams
+        end
+      end
+    end
+
+    desc 'dump [NAME]', 'describe a stream by name'
+    def dump(name)
+      kinesis.describe_stream(stream_name: name).stream_description.output do |stream|
+        puts YAML.dump(stringify_keys(stream.to_hash))
+      end
+    end
+
+  end
+end

data/lib/awful/kms.rb

@@ -18,10 +18,6 @@ module Awful
         @_kms ||= Aws::KMS::Client.new
       end
 
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-
       def list_aliases
         paginate(:aliases) do |marker|
           kms.list_aliases(marker: marker)
@@ -135,4 +131,4 @@ module Awful
     end
 
   end
-end
+end

data/lib/awful/lambda_events.rb

@@ -13,12 +13,6 @@ module Awful
       Disabled: :red
     }
 
-    no_commands do
-      def color(string)
-        set_color(string, COLORS.fetch(string.to_sym, :yellow))
-      end
-    end
-
     desc 'ls [FUNCTION_NAME]', 'list event source mappings'
     method_option :long, aliases: '-l', type: :boolean, default: false, desc: 'Long listing'
     def ls(name = nil)
@@ -67,4 +61,4 @@ module Awful
       end
     end
   end
-end
+end

data/lib/awful/rds.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-rds'
+
 module Awful
 
   class Rds < Cli