lib/debug_extras/ext/action_controller.rb in debug-extras-0.4.1 vs lib/debug_extras/ext/action_controller.rb in debug-extras-0.4.2

@@ -1,8 +1,15 @@
 module DebugExtras
   module ActionController
-    def render(*args)
-      super
+    extend ActiveSupport::Concern
+
+    included do
+      after_action :debug_extras_inject
+    end
+
+    private
+
+    def debug_extras_inject
       injector = Injector.new(response)
       response.body = injector.process_response
     end
   end
 end