mirror of
https://github.com/omniauth/omniauth.git
synced 2022-11-09 12:31:49 -05:00
34 lines
659 B
Ruby
34 lines
659 B
Ruby
|
require 'omniauth/core'
|
||
|
|
|
||
|
|
module OmniAuth
|
||
|
|
class Builder < ::Rack::Builder
|
||
|
|
def initialize(app, &block)
|
||
|
|
@app = app
|
||
|
|
super(&block)
|
||
|
|
end
|
||
|
|
|
||
|
|
def on_failure(&block)
|
||
|
|
OmniAuth.config.on_failure = block
|
||
|
|
end
|
||
|
|
|
||
|
|
def configure(&block)
|
||
|
|
OmniAuth.configure(&block)
|
||
|
|
end
|
||
|
|
|
||
|
|
def provider(klass, *args, &block)
|
||
|
|
if klass.is_a?(Class)
|
||
|
|
middleware = klass
|
||
|
|
else
|
||
|
|
middleware = OmniAuth::Strategies.const_get("#{OmniAuth::Utils.camelize(klass.to_s)}")
|
||
|
|
end
|
||
|
|
|
||
|
|
use middleware, *args, &block
|
||
|
|
end
|
||
|
|
|
||
|
|
def call(env)
|
||
|
|
@ins << @app unless @ins.include?(@app)
|
||
|
|
to_app.call(env)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|