Merge pull request drapergem#25 from edtsech/block

Ability to pass block to decorator method

lib/draper/model_support.rb

@@ -1,5 +1,6 @@
 module Draper::ModelSupport
   def decorator
     @decorator ||= "#{self.class.name}Decorator".constantize.decorate(self)
+    block_given? ? yield(@decorator) : @decorator
   end
 end

spec/draper/model_support_spec.rb

@@ -6,5 +6,10 @@
   describe '#decorator' do
     its(:decorator) { should be_kind_of(ProductDecorator) }
     its(:decorator) { should be(subject.decorator) }
+
+    it 'should have abillity to pass block' do
+      a = Product.new.decorator { |d| d.awesome_title }
+      a.should eql "Awesome Title"
+    end
   end
 end

spec/samples/active_record.rb

@@ -1,4 +1,4 @@
 module ActiveRecord
   class Base
   end
-end
+end

spec/samples/product_decorator.rb

@@ -1,3 +1,7 @@
 class ProductDecorator < Draper::Base
   decorates :product
+
+  def awesome_title
+    "Awesome Title"
+  end
 end