awful 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8279d47c43bd2ad05a4091376089c74750afcc79
-  data.tar.gz: ada93e597d39fdbc85294f9eeb2ca72a11dfcf9b
+  metadata.gz: f2edeb3d038db1c9063d5808e2475b027a057dce
+  data.tar.gz: 6d04587e11d128aa9363ef8cf332886a26e80b3e
 SHA512:
-  metadata.gz: 486a2e6c76c70e4f24bfb0e02da5beadf868f84fd6d5825eec0640ceb08e077371877289ca9baec0ac8b3a85346c8046eec1a61e69e126698a788c55615b3f6a
-  data.tar.gz: 47ac174a20005af8e1028168b7fb4ac602c1ef1220e7d1610232d4357ad4f87c2e3e1e72e586c8677445a0bd4521050c2c720741b273acd68765cfaadd30959d
+  metadata.gz: dc25885681695b9eb9e15a55847ac46e978c7ba91457bc4bce2562b847d736ba9eaa4bf1d58fbc371032c0bea5f971d1c1009f2193c234a5b4f65d97989cef0b
+  data.tar.gz: af16e76f55bc4895fba349b7c551a1cd35157b1aa5b420087dfbb59734d27544b59c1392c334672aa5289b59f43938b1556c6990dae86a2b7ebd6a575cd95cdf

data/bin/sg

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

data/lib/awful/auto_scaling.rb

@@ -92,9 +92,10 @@ module Awful
     end
 
     desc 'update NAME', 'update existing auto-scaling group'
-    method_option :desired_capacity, aliases: '-d', default: nil, desc: 'Set desired capacity'
-    method_option :min_size,         aliases: '-m', default: nil, desc: 'Set minimum capacity'
-    method_option :max_size,         aliases: '-M', default: nil, desc: 'Set maximum capacity'
+    method_option :desired_capacity,          aliases: '-d', default: nil, desc: 'Set desired capacity'
+    method_option :min_size,                  aliases: '-m', default: nil, desc: 'Set minimum capacity'
+    method_option :max_size,                  aliases: '-M', default: nil, desc: 'Set maximum capacity'
+    method_option :launch_configuration_name, aliases: '-l', default: nil, desc: 'Launch config name'
     def update(name)
       opt = load_cfg(options)
       whitelist = %i[auto_scaling_group_name launch_configuration_name min_size max_size desired_capacity default_cooldown availability_zones
@@ -114,16 +115,24 @@ module Awful
       end
     end
 
+    no_commands do
+      def oldest(name)
+        instance_ids = autoscaling.describe_auto_scaling_instances.map(&:auto_scaling_instances).flatten.select do |instance|
+          instance.auto_scaling_group_name == name
+        end.map(&:instance_id)
+        ec2.describe_instances(instance_ids: instance_ids).map(&:reservations).flatten.map(&:instances).flatten.sort_by(&:launch_time)
+      end
+
+      def newest(name)
+        oldest(name).reverse
+      end
+    end
+
     desc 'terminate NAME [NUMBER]', 'terminate NUMBER instances in group NAME'
     method_option :decrement, aliases: '-d', default: false, type: :boolean, desc: 'Decrement desired capacity for each terminated instance'
     method_option :newest,    aliases: '-n', default: false, type: :boolean, desc: 'Delete newest instances instead of oldest'
     def terminate(name, num = 1)
-      instance_ids = autoscaling.describe_auto_scaling_instances.map(&:auto_scaling_instances).flatten.select do |instance|
-        instance.auto_scaling_group_name == name
-      end.map(&:instance_id)
-
-      ins = ec2.describe_instances(instance_ids: instance_ids).map(&:reservations).flatten.map(&:instances).flatten.sort_by(&:launch_time)
-      ins = ins.reverse if options[:newest]
+      ins = options[:newest] ? newest(name) : oldest(name)
 
       ins.first(num.to_i).map(&:instance_id).tap do |ids|
         if yes? "Really terminate #{num} instances: #{ids.join(',')}?", :yellow
@@ -132,7 +141,17 @@ module Awful
           end
         end
       end
+    end
 
+    desc 'stop NAME [NUMBER]', 'stop NUMBER instances in group NAME'
+    method_option :newest, aliases: '-n', default: false, type: :boolean, desc: 'Stop newest instances instead of oldest'
+    def stop(name, num = 1)
+      ins = options[:newest] ? newest(name) : oldest(name)
+      ins.first(num.to_i).map(&:instance_id).tap do |ids|
+        if yes? "Really stop #{num} instances: #{ids.join(',')}?", :yellow
+          ec2.stop_instances(instance_ids: ids)
+        end
+      end
     end
 
   end

data/lib/awful/launch_config.rb

@@ -44,13 +44,36 @@ module Awful
       puts YAML.dump(stringify_keys(lc))
     end
 
+    desc 'latest', 'latest'
+    def latest(name)
+      autoscaling.describe_launch_configurations.map(&:launch_configurations).flatten.select do |lc|
+        lc.launch_configuration_name.match(/^#{name}/)
+      end.sort_by(&:created_time).last.launch_configuration_name.tap do |latest|
+        puts latest
+      end
+    end
+
     desc 'create [NAME]', 'create a new launch configuration'
+    method_option :timestamp, aliases: '-t', default: false, desc: 'Add timestamp to launch config name'
+    method_option :patch,     aliases: '-p', default: false, desc: 'Add release to launch config name'
     def create(name)
-      opt = load_cfg
+      opt = load_cfg(options)
+
       whitelist = %i[launch_configuration_name image_id key_name security_groups classic_link_vpc_id classic_link_vpc_security_groups user_data
                      instance_id instance_type kernel_id ramdisk_id block_device_mappings instance_monitoring spot_price iam_instance_profile
                      ebs_optimized associate_public_ip_address placement_tenancy]
-      opt[:launch_configuration_name] = "#{name}-#{Time.now.utc.strftime('%Y%m%d%H%M%S')}"
+
+      if options[:timestamp]
+        opt[:launch_configuration_name] = "#{name}-#{Time.now.utc.strftime('%Y%m%d%H%M%S')}"
+      elsif opt[:patch]
+        print "bumping version from latest: "
+        m = latest(name).match(/^#{name}-(\w+)\.(\w+)\.(\w+)$/)
+        raise "latest launch config does not match format #{name}-x.x.x" unless m
+        opt[:launch_configuration_name] = "#{name}-#{[m[1], m[2], m[3].to_i + 1].join('.')}"
+      else
+        opt[:launch_configuration_name] = name
+      end
+
       opt[:user_data] = Base64.encode64(opt[:user_data]) # encode user data
       opt = remove_empty_strings(opt)
       opt = only_keys_matching(opt, whitelist)

data/lib/awful/security_group.rb

@@ -0,0 +1,33 @@
+module Awful
+
+  class SecurityGroup < Thor
+    include Awful
+
+    desc 'ls [NAME]', 'list security groups [matching NAME]'
+    method_option :long, aliases: '-l', default: false, desc: 'Long listing'
+    def ls(name = /./)
+      fields = options[:long] ?
+        ->(s) { [tag_name(s), s.group_id, s.group_name, s.vpc_id, s.description] } :
+        ->(s) { [s.group_name] }
+
+      ec2.describe_security_groups.map(&:security_groups).flatten.select do |sg|
+        sg.group_name.match(name) or sg.group_id.match(name)
+      end.map do |sg|
+        fields.call(sg)
+      end.tap do |list|
+        print_table list
+      end
+    end
+
+    desc 'dump NAME', 'dump security group with NAME as yaml'
+    def dump(name)
+      ec2.describe_security_groups.map(&:security_groups).flatten.find do |sg|
+        sg.group_name == name
+      end.tap do |sg|
+        puts YAML.dump(stringify_keys(sg.to_hash))
+      end
+    end
+
+  end
+
+end

data/lib/awful/version.rb

@@ -1,3 +1,3 @@
 module Awful
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awful
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Ric Lister
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-06 00:00:00.000000000 Z
+date: 2015-03-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -75,6 +75,7 @@ executables:
 - elb
 - lc
 - rds
+- sg
 - subnet
 - vpc
 extensions: []
@@ -91,6 +92,7 @@ files:
 - bin/elb
 - bin/lc
 - bin/rds
+- bin/sg
 - bin/subnet
 - bin/vpc
 - lib/awful.rb
@@ -99,6 +101,7 @@ files:
 - lib/awful/elb.rb
 - lib/awful/launch_config.rb
 - lib/awful/rds.rb
+- lib/awful/security_group.rb
 - lib/awful/subnet.rb
 - lib/awful/version.rb
 - lib/awful/vpc.rb