image_builder 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4dad5330712e1e493380c06ce30c27c631bc8668
-  data.tar.gz: 7cc941b95547d231ae5d0f01ded310a8efa65eda
+  metadata.gz: f0421ca79ce0ab1bc492fb62634cd3299266ccf2
+  data.tar.gz: 024b9010de1ba6bca39d627f5a5599dc03111c47
 SHA512:
-  metadata.gz: 5a456a9d62cf45f077491b14500c7fc02989b6dc64c59ea89b84a73e39ed4c02366086f29cbee57b21a09d243ef3df7d902ea6f3599eb01d44c5414a42bd6060
-  data.tar.gz: 1f078bc505be1671a1f324d8ce094bbddeadce32dd179bda7b8effb0f5ccac09946144c95b6ece6aabe097fa3dae0eb75a7af8b3c7d588927151cc017ae0a0f6
+  metadata.gz: de380363d056d213b946d2435e78d0acdd747c7086aaa072f0024e76b4e47b303029a416f9c14b04a0ac243f4abd5a985163ac5cc7b1fb38d990a461d9aab929
+  data.tar.gz: b190d74ea7d04a35895951d57bbf1c7d6f57346b75e92bfa382a6aee373dde92ae4e93dc22dffc2166f9d4cae52d34932e2a76bf658f68ad0692c953563c4157

data/bin/image_builder

@@ -141,6 +141,14 @@ def convert_tags(tag_opt)
   hash
 end
 
+def find_ami_snapshot(ami_obj)
+  root_dev = ami_obj.root_device_name
+
+  ami_obj.block_device_mappings.each do |k, v|
+    return v[:snapshot_id] if root_dev.start_with? k
+  end
+end
+
 # NOTE: Array options are in format '--opt arg1,arg2' (comma-seperated, no space!)
 Commander.configure do
   program :name, prog_name
@@ -233,4 +241,75 @@ Commander.configure do
       run_packer(builders, provisioners, [vagrant].compact, options.p_packer_path, options.debug)
     end
   end
+
+  command :tag_snapshot do |c|
+    c.syntax = "#{prog_name} tag_snapshot -a ami1[,ami2,...] [-r aws_region1,aws_region2,...]"
+    c.description = 'A command to set the name of an EBS-backed AMI snapshot to match its AMI.'
+
+    c.option '-a, --amis AMIS', Array, 'An array of AMI ID or names to operate on (REQUIRED)'
+    c.option '-r, --regions REGIONS', Array, 'An array of AWS regions to do the tagging on, defaults to us-east-1'
+
+    c.action do |_args, options|
+      options.default r_regions: ['us-east-1']
+
+      fail ArgumentError, 'Missing required -a option' if options.a_amis.nil? || options.a_amis.empty?
+
+      options.r_regions.each do |r|
+        cu  = Cloudutil::AWS::EC2.new(region: r)
+        ec2 = ::AWS::EC2.new(config: cu.config(true))
+
+        options.a_amis.each do |a|
+          ami_obj = cu.lookup_ami(a).first # Get AWS SDK AMI objects instead of just the id
+
+          if ami_obj.nil?
+            puts "WARNING: AMI #{a} not found in region #{r}, skipping"
+          else
+            if ami_obj.root_device_type.eql? :ebs
+              # AWS SDK only allows setting a snapshot description at create time, so we can't
+              # add a custom description here :(
+              snap_id  = find_ami_snapshot(ami_obj)
+              ami_name = ami_obj.name
+              ec2.snapshots[snap_id].add_tag('Name', value: ami_name)
+              puts "Set name for #{snap_id} (#{a}, #{r}) to #{ami_name}"
+            end
+          end
+        end
+      end
+    end
+  end
+
+  command :list_ami do |c|
+    c.syntax = "#{prog_name} list_ami -a ami1[,ami2,...] [-r aws_region1,aws_region2,...]"
+    c.description = 'A command to generate a JSON mapping of AMI names to AMI ids across AWS regions.'
+
+    c.option '-a, --amis AMIS', Array, 'An array of AMI ID or names to operate on (REQUIRED)'
+    c.option '-r, --regions REGIONS', Array, 'An array of AWS regions to do the tagging on, defaults to us-east-1'
+
+    c.action do |_args, options|
+      options.default r_regions: ['us-east-1']
+
+      fail ArgumentError, 'Missing required -a option' if options.a_amis.nil? || options.a_amis.empty?
+
+      ami_data = {}
+
+      options.r_regions.each do |r|
+        cu  = Cloudutil::AWS::EC2.new(region: r)
+
+        options.a_amis.each do |a|
+          ami_obj = cu.lookup_ami(a).first # Get AWS SDK AMI objects instead of just the id
+
+          if ami_obj.nil?
+            puts "WARNING: AMI #{a} not found in region #{r}, skipping"
+          else
+            ami_name = ami_obj.name
+
+            ami_data[ami_name] = {} unless ami_data.key? ami_name
+            ami_data[ami_name][r] = ami_obj.id
+          end
+        end
+      end
+
+      puts ami_data.to_json
+    end
+  end
 end

data/lib/image_builder/version.rb

@@ -1,4 +1,4 @@
 # Generic top-level module comment
 module ImageBuilder
-  VERSION = '0.0.6'
+  VERSION = '0.0.7'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: image_builder
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Mike Morris
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-11-25 00:00:00.000000000 Z
+date: 2014-11-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler