Implementing application-wide 'decorate' view helper

lib/draper.rb

@@ -3,6 +3,7 @@ require 'draper/system'
 require 'draper/base'
 require 'draper/lazy_helpers'
 require 'draper/model_support'
+require 'draper/helper_support'
 require 'draper/view_context'
 require 'draper/decorated_enumerable_proxy'
 

lib/draper/system.rb

@@ -3,6 +3,7 @@ module Draper
     def self.setup
       ActionController::Base.send(:include, Draper::ViewContextFilter) if defined?(ActionController::Base)
       ActionMailer::Base.send(:include, Draper::ViewContextFilter) if defined?(ActionMailer::Base)
+      ActionController::Base.send(:helper, Draper::HelperSupport) if defined?(ActionController::Base)
     end
   end
 end

spec/support/samples/application_controller.rb

@@ -9,6 +9,9 @@ module ActionController
     def self.before_filter(name)
       @@before_filters << name
     end
+    def self.helper(mod)
+      extend mod
+    end
   end
 end
 
@@ -29,6 +32,15 @@ class ApplicationController < ActionController::Base
   def self.hello
     "Hello!"
   end
+  
+  def self.capture(&block)
+    @@capture = true
+    block.call
+  end
+  
+  def self.capture_triggered
+    @@capture ||= false
+  end
 end
 
 Draper::System.setup