Remove System namespace

lib/draper.rb

@@ -1,7 +1,6 @@
 require 'action_view'
 
 require 'draper/version'
-require 'draper/system'
 require 'draper/view_helpers'
 require 'draper/finders'
 require 'draper/decorator'
@@ -20,6 +19,29 @@
 require 'draper/test/minitest_integration' if defined?(MiniTest::Rails)
 
 module Draper
+  def self.setup_action_controller(base)
+    base.class_eval do
+      include Draper::ViewContext
+      extend  Draper::HelperSupport
+      before_filter ->(controller) {
+        Draper::ViewContext.current = nil
+        Draper::ViewContext.current_controller = controller
+      }
+    end
+  end
+
+  def self.setup_action_mailer(base)
+    base.class_eval do
+      include Draper::ViewContext
+    end
+  end
+
+  def self.setup_active_record(base)
+    base.class_eval do
+      include Draper::Decoratable
+    end
+  end
+
   class UninferrableDecoratorError < NameError
     def initialize(klass)
       super("Could not infer a decorator for #{klass}.")

lib/draper/railtie.rb

@@ -24,19 +24,19 @@ class Railtie < Rails::Railtie
 
     initializer "draper.extend_action_controller_base" do |app|
       ActiveSupport.on_load(:action_controller) do
-        Draper::System.setup_action_controller(self)
+        Draper.setup_action_controller(self)
       end
     end
 
     initializer "draper.extend_action_mailer_base" do |app|
       ActiveSupport.on_load(:action_mailer) do
-        Draper::System.setup_action_mailer(self)
+        Draper.setup_action_mailer(self)
       end
     end
 
     initializer "draper.extend_active_record_base" do |app|
       ActiveSupport.on_load(:active_record) do
-        self.send(:include, Draper::Decoratable)
+        Draper.setup_active_record(self)
       end
     end
 

spec/support/action_controller.rb

@@ -1,6 +1,6 @@
 module ActionController
   class Base
-    Draper::System.setup_action_controller(self)
+    Draper.setup_action_controller(self)
   end
 end