68 lines
1.5 KiB
Ruby
68 lines
1.5 KiB
Ruby
module OmniAuth
|
|
class Builder < ::Rack::Builder
|
|
def initialize(app, &block)
|
|
@options = nil
|
|
if rack14? || rack2?
|
|
super
|
|
else
|
|
@app = app
|
|
super(&block)
|
|
@ins << @app
|
|
end
|
|
end
|
|
|
|
def rack14?
|
|
Rack.release.start_with?('1.') && (Rack.release.split('.')[1].to_i >= 4)
|
|
end
|
|
|
|
def rack2?
|
|
Rack.release.start_with? '2.'
|
|
end
|
|
|
|
def on_failure(&block)
|
|
OmniAuth.config.on_failure = block
|
|
end
|
|
|
|
def before_options_phase(&block)
|
|
OmniAuth.config.before_options_phase = block
|
|
end
|
|
|
|
def before_request_phase(&block)
|
|
OmniAuth.config.before_request_phase = block
|
|
end
|
|
|
|
def before_callback_phase(&block)
|
|
OmniAuth.config.before_callback_phase = block
|
|
end
|
|
|
|
def configure(&block)
|
|
OmniAuth.configure(&block)
|
|
end
|
|
|
|
def options(options = false)
|
|
return @options || {} if options == false
|
|
|
|
@options = options
|
|
end
|
|
|
|
def provider(klass, *args, &block)
|
|
if klass.is_a?(Class)
|
|
middleware = klass
|
|
else
|
|
begin
|
|
middleware = OmniAuth::Strategies.const_get(OmniAuth::Utils.camelize(klass.to_s).to_s)
|
|
rescue NameError
|
|
raise(LoadError.new("Could not find matching strategy for #{klass.inspect}. You may need to install an additional gem (such as omniauth-#{klass})."))
|
|
end
|
|
end
|
|
|
|
args.last.is_a?(Hash) ? args.push(options.merge(args.pop)) : args.push(options)
|
|
use middleware, *args, &block
|
|
end
|
|
|
|
def call(env)
|
|
to_app.call(env)
|
|
end
|
|
end
|
|
end
|