Clean up the whole loading mess and closes #247. This commit depends on latest Rails.

lib/devise.rb

@@ -230,16 +230,17 @@ module Devise
   # A method used internally to setup warden manager from the Rails initialize
   # block.
   def self.configure_warden! #:nodoc:
-    return unless warden_config
+    @@warden_configured ||= begin
+      warden_config.failure_app   = Devise::FailureApp
+      warden_config.default_scope = Devise.default_scope
 
-    warden_config.failure_app   = Devise::FailureApp
-    warden_config.default_scope = Devise.default_scope
+      Devise.mappings.each_value do |mapping|
+        warden_config.scope_defaults mapping.name, :strategies => mapping.strategies
+      end
 
-    Devise.mappings.each_value do |mapping|
-      warden_config.scope_defaults mapping.name, :strategies => mapping.strategies
+      @@warden_config_block.try :call, Devise.warden_config
+      true
     end
-
-    @@warden_config_block.try :call, Devise.warden_config
   end
 
   # Generate a friendly string randomically to be used as token.

lib/devise/failure_app.rb

@@ -9,6 +9,7 @@ module Devise
     include ActionController::RackDelegation
     include ActionController::UrlFor
     include ActionController::Redirecting
+    include Rails.application.routes.url_helpers
 
     delegate :flash, :to => :request
 

lib/devise/rails.rb

@@ -1,32 +1,32 @@
 require 'devise/rails/routes'
 require 'devise/rails/warden_compat'
 
+# Include UrlHelpers in ActionController and ActionView as soon as they are loaded.
+ActiveSupport.on_load(:action_controller) { include Devise::Controllers::UrlHelpers }
+ActiveSupport.on_load(:action_view) { include Devise::Controllers::UrlHelpers }
+
 module Devise
   class Engine < ::Rails::Engine
     config.devise = Devise
 
-    initializer "devise.ensure_routes_are_loaded", :before => :load_app_classes, :after => :load_config_initializers do |app|
-      app.reload_routes!
+    config.app_middleware.use Warden::Manager do |config|
+      Devise.warden_config = config
     end
 
-    initializer "devise.add_middleware" do |app|
-      app.config.middleware.use Warden::Manager do |config|
-        Devise.warden_config = config
-        Devise.configure_warden!
-      end
-    end
-
-    initializer "devise.add_url_helpers" do |app|
-      Devise::FailureApp.send :include, app.routes.url_helpers
-      ActionController::Base.send :include, Devise::Controllers::UrlHelpers
-      ActionView::Base.send :include, Devise::Controllers::UrlHelpers
-    end
+    # Force routes to be loaded if we are doing any eager load.
+    config.before_eager_load { |app| app.reload_routes! }
 
     config.after_initialize do
-      I18n.available_locales
       flash = [:unauthenticated, :unconfirmed, :invalid, :invalid_token, :timeout, :inactive, :locked]
 
-      I18n.backend.send(:translations).each do |locale, translations|
+      translations = begin
+        I18n.available_locales
+        I18n.backend.send(:translations)
+      rescue Exception => e # Do not care if something fails
+        {}
+      end
+
+      translations.each do |locale, translations|
         keys = flash & (translations[:devise][:sessions].keys) rescue []
 
         if keys.any?

test/devise_test.rb

@@ -2,6 +2,7 @@ require 'test_helper'
 
 module Devise
   def self.yield_and_restore
+    @@warden_configured = nil
     c, b = @@warden_config, @@warden_config_block
     yield
   ensure