lib/pakyow/presenter/processor.rb in pakyow-presenter-1.0.0.rc1 vs lib/pakyow/presenter/processor.rb in pakyow-presenter-1.0.0.rc2

@@ -1,22 +1,24 @@
 # frozen_string_literal: true
 
+require "pakyow/support/class_state"
+
 module Pakyow
   module Presenter
     class ProcessorCaller
       def initialize(instances)
         @processors = normalize(instances)
       end
 
       def process(content, extension)
         processors_for_extension(extension).each do |processor|
-          content = processor.process(content)
+          content = processor.call(content)
         end
 
         unless extension == :html
           processors_for_extension(:html).each do |processor|
-            content = processor.process(content)
+            content = processor.call(content)
           end
         end
 
         content
       end
@@ -31,30 +33,52 @@
         @processors[extension] || []
       end
 
       def normalize(instances)
         instances.each_with_object({}) { |instance, processors|
-          instance.extensions.each do |extension|
+          instance.class.extensions.each do |extension|
             (processors[extension] ||= []) << instance
           end
         }
       end
     end
 
     class Processor
+      extend Support::ClassState
+      class_state :name
+      class_state :block
+      class_state :extensions, default: [], getter: false
+
       extend Support::Makeable
 
-      class << self
-        attr_reader :name, :extensions, :block
+      def initialize(app)
+        @app = app
+      end
 
+      def call(content)
+        self.class.process(content)
+      end
+
+      class << self
         def make(name, *extensions, **kwargs, &block)
-          # name is expected to also be an extension
+          # Name is expected to also be an extension.
+          #
           extensions.unshift(name).map!(&:to_sym)
+
           super(name, extensions: extensions, block: block, **kwargs) {}
         end
 
         def process(content)
           block.call(content)
+        end
+
+        def extensions(*extensions)
+          if extensions.any?
+            @extensions ||= []
+            @extensions.concat(extensions.map(&:to_sym)).uniq
+          else
+            @extensions
+          end
         end
       end
     end
   end
 end