lib/loaf/railtie.rb in loaf-0.1.1 vs lib/loaf/railtie.rb in loaf-0.2.0

@@ -1,20 +1,29 @@
-module Loaf
+# encoding: utf-8
 
+require 'action_controller'
+require 'action_view'
+
+module Loaf
   if defined? Rails::Railtie
     class Railtie < Rails::Railtie
       initializer "loaf.extend_action_controller_base" do |app|
-        ActiveSupport.on_load(:action_controller) do
-          Loaf::Railtie.insert
+        ActiveSupport.on_load :action_controller do
+          Loaf::Railtie.insert_controller
+          Loaf::Railtie.insert_view
         end
       end
     end
   end
 
   class Railtie
-    def self.insert
-      ActionController::Base.send :include, Loaf::Filters 
-      ActionController::Base.helper Loaf::Helpers
+    class << self
+      def insert_view
+        ActionController::Base.helper Loaf::ViewExtensions
+      end
+      def insert_controller
+        ActionController::Base.send :include, Loaf::ControllerExtensions
+      end
     end
   end # Railtie
 
 end # Loaf