103 lines
2.7 KiB
Ruby
103 lines
2.7 KiB
Ruby
require "action_controller/metal"
|
|
|
|
module Devise
|
|
# Failure application that will be called every time :warden is thrown from
|
|
# any strategy or hook. Responsible for redirect the user to the sign in
|
|
# page based on current scope and mapping. If no scope is given, redirect
|
|
# to the default_url.
|
|
class FailureApp < ActionController::Metal
|
|
include ActionController::RackDelegation
|
|
include ActionController::UrlFor
|
|
include ActionController::Redirecting
|
|
|
|
delegate :flash, :to => :request
|
|
|
|
def self.call(env)
|
|
action(:respond).call(env)
|
|
end
|
|
|
|
def self.default_url_options(*args)
|
|
ApplicationController.default_url_options(*args)
|
|
end
|
|
|
|
def respond
|
|
if http_auth?
|
|
http_auth
|
|
elsif warden_options[:recall]
|
|
recall
|
|
else
|
|
redirect
|
|
end
|
|
end
|
|
|
|
def http_auth
|
|
self.status = 401
|
|
self.headers["WWW-Authenticate"] = %(Basic realm=#{Devise.http_authentication_realm.inspect})
|
|
self.content_type = request.format.to_s
|
|
self.response_body = http_auth_body
|
|
end
|
|
|
|
def recall
|
|
env["PATH_INFO"] = attempted_path
|
|
flash.now[:alert] = i18n_message(:invalid)
|
|
self.response = recall_controller.action(warden_options[:recall]).call(env)
|
|
end
|
|
|
|
def redirect
|
|
store_location!
|
|
flash[:alert] = i18n_message unless flash[:notice]
|
|
redirect_to send(:"new_#{scope}_session_path")
|
|
end
|
|
|
|
protected
|
|
|
|
def i18n_message(default = nil)
|
|
message = warden.message || warden_options[:message] || default || :unauthenticated
|
|
|
|
if message.is_a?(Symbol)
|
|
I18n.t(:"#{scope}.#{message}", :resource_name => scope,
|
|
:scope => "devise.failure", :default => [message, message.to_s])
|
|
else
|
|
message.to_s
|
|
end
|
|
end
|
|
|
|
def http_auth?
|
|
request.authorization
|
|
end
|
|
|
|
def http_auth_body
|
|
method = :"to_#{request.format.to_sym}"
|
|
{}.respond_to?(method) ? { :error => i18n_message }.send(method) : i18n_message
|
|
end
|
|
|
|
def recall_controller
|
|
"#{params[:controller].camelize}Controller".constantize
|
|
end
|
|
|
|
def warden
|
|
env['warden']
|
|
end
|
|
|
|
def warden_options
|
|
env['warden.options']
|
|
end
|
|
|
|
def scope
|
|
@scope ||= warden_options[:scope]
|
|
end
|
|
|
|
def attempted_path
|
|
warden_options[:attempted_path]
|
|
end
|
|
|
|
# Stores requested uri to redirect the user after signing in. We cannot use
|
|
# scoped session provided by warden here, since the user is not authenticated
|
|
# yet, but we still need to store the uri based on scope, so different scopes
|
|
# would never use the same uri to redirect.
|
|
def store_location!
|
|
session[:"#{scope}_return_to"] = attempted_path if request && request.get?
|
|
end
|
|
end
|
|
end
|