Move view_context definition back into System.setup

lib/draper.rb

@@ -10,7 +10,6 @@ require 'draper/helper_support'
 require 'draper/view_context'
 require 'draper/decorated_enumerable_proxy'
 require 'draper/railtie' if defined?(Rails)
-require 'draper/action_controller/base'
 
 # Test Support
 require 'draper/test/rspec_integration'    if defined?(RSpec) and RSpec.respond_to?(:configure)

lib/draper/action_controller/base.rb

@@ -1,9 +0,0 @@
-module DraperViewContext
-  def view_context
-    super.tap do |context|
-      Draper::ViewContext.current = context
-    end
-  end
-end
-
-ActionController::Base.send(:include, DraperViewContext)

lib/draper/system.rb

@@ -2,6 +2,7 @@ module Draper
   class System
     def self.setup(component)
       component.class_eval do
+        include Draper::ViewContext
         extend  Draper::HelperSupport unless defined?(::ActionMailer) && self.is_a?(::ActionMailer::Base)
       end
     end

lib/draper/view_context.rb

@@ -7,5 +7,11 @@ module Draper
     def self.current=(input)
       Thread.current[:current_view_context] = input
     end
+
+    def view_context
+      super.tap do |context|
+        Draper::ViewContext.current = context
+      end
+    end
   end
 end