rails_view.rb in rails_view-1.2.0

- old
+ new
@@ -1,47 +1,64 @@
 require 'rails' unless defined?(::Rails)
 require 'action_controller' unless defined?(::ActionController)
 
 class View
-  VERSION = '1.0.1'
+  VERSION = '1.2.0'
 
   def View.version
     View::VERSION
   end
 
-  class Controller < ::ActionController::Base
-    layout false
-    helper :all
+  if defined?(Rails::Engine)
+    class Engine < Rails::Engine
+      config.after_initialize do
+        ::View.generate_controller!
+      end
+    end
+  else
+    ::View.generate_controller!
+  end
 
-    def Controller.context(*args, &block)
-      require 'rails'
-      require 'action_controller'
-      require 'action_dispatch/testing/test_request.rb' 
-      require 'action_dispatch/testing/test_response.rb' 
+  def View.generate_controller!
+    #base = (defined?(ApplicationController) ? ApplicationController : ::ActionController::Base)
+    base = ::ActionController::Base
 
-      default_url_options =
-        begin
-          require 'rails_default_url_options'
-          DefaultUrlOptions
-        rescue LoadError
-          options[:default_url_options] || {}
-        end
+    c = Class.new(base) do
+      layout false
+      helper :all
 
-      store = ActiveSupport::Cache::MemoryStore.new 
-      request = ActionDispatch::TestRequest.new 
-      response = ActionDispatch::TestResponse.new 
+      def self.context(*args, &block)
+        require 'rails'
+        require 'action_controller'
+        require 'action_dispatch/testing/test_request.rb' 
+        require 'action_dispatch/testing/test_response.rb' 
 
-      controller = new()
+        default_url_options =
+          begin
+            require 'rails_default_url_options'
+            DefaultUrlOptions
+          rescue LoadError
+            options[:default_url_options] || {}
+          end
 
-      controller.perform_caching = false
-      controller.cache_store = store 
-      controller.request = request 
-      controller.response = response 
-      #controller.send(:initialize_template_class, response) 
-      #controller.send(:assign_shortcuts, request, response) 
-      controller.send(:default_url_options).merge!(default_url_options)
-      block ? controller.instance_eval(&block) : controller
-    end 
+        store = ActiveSupport::Cache::MemoryStore.new 
+        request = ActionDispatch::TestRequest.new 
+        response = ActionDispatch::TestResponse.new 
+
+        controller = new()
+
+        controller.perform_caching = false
+        controller.cache_store = store 
+        controller.request = request 
+        controller.response = response 
+        #controller.send(:initialize_template_class, response) 
+        #controller.send(:assign_shortcuts, request, response) 
+        controller.send(:default_url_options).merge!(default_url_options)
+        block ? controller.instance_eval(&block) : controller
+      end 
+    end
+
+    const_set(:Controller, c)
   end
 
 
   def View.render(*args)
     Array(Controller.context{ render(*args) }).join.html_safe