lib/jets/resource/permission.rb in jets-0.8.8 vs lib/jets/resource/permission.rb in jets-0.8.9

@@ -1,60 +1,39 @@
 class Jets::Resource
-  class Permission
-    extend Memoist
-
+  class Permission < Jets::Resource::Base
     def initialize(replacements, associated_resource)
       @replacements = replacements
-      # puts caller
-      # puts "replacements #{replacements.inspect}"
       @associated_resource = associated_resource
     end
 
-    def logical_id
-      logical_id = "{namespace}Permission"
-      md = @associated_resource.logical_id.match(/(\d+)/)
-      counter = md[1] if md
-      logical_id = [logical_id, counter].compact.join('')
-      # replace possible {namespace} in the logical id
-      logical_id = replacer.replace_value(logical_id)
-      Jets::Camelizer.camelize(logical_id)
-    end
-
-    def type
-      attributes['Type']
-    end
-
-    def properties
-      attributes['Properties']
-    end
-
-    def attributes
-      attributes = {
-        type: "AWS::Lambda::Permission",
-        properties: {
-          function_name: "!GetAtt {namespace}LambdaFunction.Arn",
-          action: "lambda:InvokeFunction",
-          principal: principal,
-          source_arn: source_arn,
+    def definition
+      {
+        permission_logical_id => {
+          type: "AWS::Lambda::Permission",
+          properties: {
+            function_name: "!GetAtt {namespace}LambdaFunction.Arn",
+            action: "lambda:InvokeFunction",
+            principal: principal,
+            source_arn: source_arn,
+          }
         }
       }
-      attributes = replacer.replace_placeholders(attributes)
-      Jets::Camelizer.transform(attributes)
     end
-    memoize :attributes
 
+    def permission_logical_id
+      logical_id = "{namespace}_permission"
+      md = @associated_resource.logical_id.match(/(\d+)/)
+      counter = md[1] if md
+      [logical_id, counter].compact.join('').underscore
+    end
+
     # Auto-detect principal from the associated resources.
     def principal
       Replacer.principal_map(@associated_resource.type)
     end
 
     def source_arn
       default_arn = "!GetAtt #{@associated_resource.logical_id}.Arn"
       Replacer.source_arn_map(@associated_resource.type) || default_arn
     end
-
-    def replacer
-      Replacer.new(@replacements)
-    end
-    memoize :replacer
   end
 end