56 lines
1.5 KiB
Ruby
56 lines
1.5 KiB
Ruby
module OmniAuth
|
|
# This simple Rack endpoint that serves as the default
|
|
# 'failure' mechanism for OmniAuth. If a strategy fails for
|
|
# any reason this endpoint will be invoked. The default behavior
|
|
# is to redirect to `/auth/failure` except in the case of
|
|
# a development `RACK_ENV`, in which case an exception will
|
|
# be raised.
|
|
class FailureEndpoint
|
|
attr_reader :env
|
|
|
|
def self.call(env)
|
|
new(env).call
|
|
end
|
|
|
|
def initialize(env)
|
|
@env = env
|
|
end
|
|
|
|
def call
|
|
raise_out! if OmniAuth.config.failure_raise_out_environments.include?(ENV['RACK_ENV'].to_s)
|
|
redirect_to_failure
|
|
end
|
|
|
|
def raise_out!
|
|
raise(env['omniauth.error'] || OmniAuth::Error.new(env['omniauth.error.type']))
|
|
end
|
|
|
|
def redirect_to_failure
|
|
message_key = env['omniauth.error.type']
|
|
|
|
new_path = "#{env['SCRIPT_NAME']}#{strategy_path_prefix}/failure?message=#{Rack::Utils.escape(message_key)}#{origin_query_param}#{strategy_name_query_param}"
|
|
Rack::Response.new(['302 Moved'], 302, 'Location' => new_path).finish
|
|
end
|
|
|
|
def strategy_path_prefix
|
|
if env['omniauth.error.strategy']
|
|
env['omniauth.error.strategy'].path_prefix
|
|
else
|
|
OmniAuth.config.path_prefix
|
|
end
|
|
end
|
|
|
|
def strategy_name_query_param
|
|
return '' unless env['omniauth.error.strategy']
|
|
|
|
"&strategy=#{env['omniauth.error.strategy'].name}"
|
|
end
|
|
|
|
def origin_query_param
|
|
return '' unless env['omniauth.origin']
|
|
|
|
"&origin=#{Rack::Utils.escape(env['omniauth.origin'])}"
|
|
end
|
|
end
|
|
end
|