jackal-cfn 0.1.2 → 0.1.4

data/CHANGELOG.md

@@ -1,3 +1,6 @@
+# v0.1.4
+* Include ami manager resource (handle deletion of amis only for now)
+
 # v0.1.2
 * Include request id in response content
 * Allow matching on class name to custom resource type

data/lib/jackal-cfn/resource/ami_manager.rb

@@ -0,0 +1,109 @@
+require 'jackal-cfn'
+
+module Jackal
+  module Cfn
+    class Resource
+      # Manage AMI Resources
+      #
+      # Expected resource:
+      #   {
+      #     "Type": "Custom::AmiManager",
+      #     "Properties": {
+      #       "Parameters": {
+      #         "InstanceId": "",
+      #         "AmiId": "",
+      #         "Action": "",
+      #         "Region": ""
+      #       }
+      #     }
+      #   }
+      #
+      # Required configuration:
+      #   {
+      #     "config": {
+      #       "credentials": {
+      #         "compute": {
+      #           FOG_CREDENTIALS
+      #         }
+      #       }
+      #     }
+      #   }
+      class AmiManager < Resource
+
+        PHYSICAL_ID_JOINER = '__-__'
+
+        # Process message, send value back to CFN
+        #
+        # @param message [Carnivore::Message]
+        def execute(message)
+          payload = unpack(message)
+          parameters = transform_parameters(payload[:resource_properties]['Parameters'])
+          debug "Processing payload: #{payload.inspect}"
+          response = build_response(payload)
+          case payload[:request_type]
+          when :create
+            ami_response_update(response, parameters)
+          when :delete
+            destroy_ami(response, payload, parameters)
+          when :update
+          else
+            error "Unknown request type received: #{payload[:request_type].inspect}"
+            response['Status'] = 'FAILED'
+            response['Reason'] = 'Unknown request type received'
+          end
+          respond_to_stack(response, payload[:response_url])
+          completed(
+            new_payload(
+              config.fetch(:name, :ami_manager),
+              :cfn_resource => payload
+            ),
+            message
+          )
+        end
+
+        # Update the physical resource id to include the ami id
+        #
+        # @param response [Hash] cfn response
+        # @param parameters [Hash] resource parameters
+        # @return [Hash] updated response hash
+        def ami_response_update(response, parameters)
+          response['PhysicalResourceId'] = [
+            response['PhysicalResourceId'],
+            parameters[:ami_id]
+          ].join(PHYSICAL_ID_JOINER)
+          response
+        end
+
+        # Destroy the AMI referenced by the resource
+        #
+        # @param response [Hash] cfn response
+        # @param payload [Hash] message payload
+        # @param parameters [Hash] resource parameters
+        # @return [TrueClass]
+        def destroy_ami(response, payload, parameters)
+          ami_id = payload[:physical_resource_id].split(PHYSICAL_ID_JOINER).last
+          begin
+            compute_api(parameters[:region]).deregister_image(ami_id)
+          rescue Fog::Compute::AWS::Error => e
+            warn "Non-fatal error encountered on AMI removal: #{e.class}: #{e}"
+            response['Reason'] = e.message
+          rescue => e
+            error "Failed to remove AMI: #{e.class}: #{e}"
+            response['Status'] = 'FAILED'
+            response['Reason'] = e.message
+          end
+          true
+        end
+
+        # Build new compute api connection
+        #
+        # @param region [String] AWS region ami exists
+        # @return [Fog::Compute]
+        def compute_api(region)
+          Fog::Compute.new({:provider => :aws}.merge(config.fetch(:credentials, :compute, {}).merge(:region => region)))
+        end
+
+      end
+    end
+  end
+end

data/lib/jackal-cfn/resource/hash_extractor.rb

@@ -5,8 +5,9 @@ module Jackal
     class Resource
       # Extract value from hash
       #
-      # Expected resource properties:
+      # Expected resource:
       #   {
+      #     "Type": "Custom::HashExtractor",
       #     "Properties": {
       #       "Parameters": {
       #         "Key": "path.to.value.in.hash",
@@ -16,21 +17,6 @@ module Jackal
       #   }
       class HashExtractor < Resource
 
-        # Setup the dependency requirements for the callback
-        def setup(*_)
-          require 'patron'
-        end
-
-        # Validity of message
-        #
-        # @param message [Carnivore::Message]
-        # @return [Truthy, Falsey]
-#        def valid?(message)
-#          super do |payload|
-#            payload.get('ResourceProperties', 'Action') == 'hash_extractor'
-#          end
-#        end
-
         # Process message, send value back to CFN
         #
         # @param message [Carnivore::Message]

data/lib/jackal-cfn/resource.rb

@@ -8,6 +8,12 @@ module Jackal
       VALID_RESOURCE_STATUS = ['SUCCESS', 'FAILED']
 
       autoload :HashExtractor, 'jackal-cfn/resource/hash_extractor'
+      autoload :AmiManager, 'jackal-cfn/resource/ami_manager'
+
+      # Setup the dependency requirements for the callback
+      def setup(*_)
+        require 'patron'
+      end
 
       # Physical ID of the resource created
       #
@@ -92,8 +98,10 @@ module Jackal
             payload.fetch('Body', 'Message', payload)
           )
         )
+        payload = transform_parameters(payload)
         payload[:origin_type] = message[:message].get('Body', 'Type')
         payload[:origin_subject] = message[:message].get('Body', 'Subject')
+        payload[:request_type] = snakecase(payload[:request_type])
         payload
       end
 
@@ -103,10 +111,10 @@ module Jackal
       # @return [TrueClass, FalseClass]
       def valid?(message)
         super do |payload|
-          resource_type = payload['ResourceType'].split('::').last
+          resource_type = payload[:resource_type].split('::').last
           result = payload[:origin_type] == 'Notification' &&
             payload[:origin_subject].downcase.include?('cloudformation custom resource') &&
-            (resource_type == 'CustomResource' || resource_type == self.class.name.split('::').last)
+            resource_type == self.class.name.split('::').last
           if(result && block_given?)
             yield payload
           else
@@ -136,12 +144,19 @@ module Jackal
       def transform_parameters(params)
         Smash.new.tap do |new_hash|
           params.each do |key, value|
-            key = key.gsub(/(?<![A-Z])([A-Z])/, '_\1').sub(/^_/, '').downcase.to_sym
-            new_hash[key] = value
+            new_hash[snakecase(key)] = value
           end
         end
       end
 
+      # Snake case string
+      #
+      # @param v [String]
+      # @return [Symbol]
+      def snakecase(v)
+        v.to_s.gsub(/(?<![A-Z])([A-Z])/, '_\1').sub(/^_/, '').downcase.to_sym
+      end
+
     end
   end
 end

data/lib/jackal-cfn/version.rb

@@ -1,6 +1,6 @@
 module Jackal
   module Cfn
     # Current version
-    VERSION = Gem::Version.new('0.1.2')
+    VERSION = Gem::Version.new('0.1.4')
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jackal-cfn
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-07-04 00:00:00.000000000 Z
+date: 2014-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: jackal
@@ -54,6 +54,7 @@ files:
 - lib/jackal-cfn/version.rb
 - lib/jackal-cfn/resource.rb
 - lib/jackal-cfn/resource/hash_extractor.rb
+- lib/jackal-cfn/resource/ami_manager.rb
 - jackal-cfn.gemspec
 - README.md
 - CHANGELOG.md