lib/rails_warden.rb in rails_warden-0.5.4 vs lib/rails_warden.rb in rails_warden-0.5.5

@@ -84,18 +84,26 @@
       include RailsWarden::Mixins::HelperMethods
     end
   end
 else
   class RailsWarden::Railtie < Rails::Railtie
-    initializer :warden do
+    include_block = lambda do
       ::ActionController::Base.class_eval do
         include RailsWarden::Mixins::HelperMethods
         include RailsWarden::Mixins::ControllerOnlyMethods
       end
 
       ::ActionView::Base.class_eval do
         include RailsWarden::Mixins::HelperMethods
       end
+    end
+
+    if respond_to?(:initializer)
+      initializer :warden, &include_block
+    elsif respond_to?(:config) && config.respond_to?(:before_initialize)
+      config.before_initialize(&include_block)
+    else
+      Rails.configuration.after_initialize &include_block
     end
   end
 end
 
 class Warden::SessionSerializer