lib/prx_auth/rails/railtie.rb in prx_auth-rails-0.0.1 vs lib/prx_auth/rails/railtie.rb in prx_auth-rails-0.0.2

@@ -1,17 +1,17 @@
 require 'rails/railtie'
 require 'prx_auth/rails/ext/controller'
 require 'rack/prx_auth'
 
-module PrxAuth
-  module Rails
-    class Railtie < ::Rails::Railtie
-      config.to_prepare do
-        ApplicationController.send(:include, PrxAuth::Rails::Controller)
-      end
+module PrxAuth::Rails
+  class Railtie < ::Rails::Railtie
+    config.to_prepare do
+      ApplicationController.send(:include, PrxAuth::Rails::Controller)
+    end
 
-      initializer 'prx_auth.insert_middleware' do |app|
-        app.config.middleware.insert_before ActionDispatch::ParamsParser, 'Rack::PrxAuth'
+    initializer 'prx_auth.insert_middleware' do |app|
+      if PrxAuth::Rails.middleware
+        app.config.middleware.insert ActionDispatch::ParamsParser, Rack::PrxAuth
       end
     end
   end
 end