lib/mystique.rb in mystique-0.7.0 vs lib/mystique.rb in mystique-0.8.0

@@ -10,25 +10,30 @@
   module_function
 
   def present(object, with: nil, context: nil, &block)
     begin
       presenter_class = presenter_class_for(object, with)
-      presenter_class.present(object, context, &block)
+      presenter_class.for(object, context, &block)
     rescue NameError
       return object
     end
   end
 
   def present_collection(collection, context: nil, &block)
     return to_enum(:present_collection, collection, context: context, &block) unless block_given?
 
     collection.each do |element|
-      block.call present(element, context: context)
+      case block.arity
+      when 2
+        block.call( present(element, context: context), element )
+      else
+        block.call(present(element, context: context))
+      end
     end
   end
 
   def presenter_class_for(object, with)
-    if with.respond_to?(:present)
+    if with.respond_to?(:for)
       with
     else
       StringPlus.constantize("#{with || object.class}Presenter")
     end
   end