36 lines
843 B
Ruby
36 lines
843 B
Ruby
module Sinatra
|
|
module DevopsAuth
|
|
module Helpers
|
|
def protect!
|
|
return if auth_with_basic?
|
|
headers['WWW-Authenticate'] = 'Basic realm="Restricted Area"'
|
|
halt 401, "Not authorized\n"
|
|
end
|
|
|
|
def auth_with_basic?
|
|
@auth ||= Rack::Auth::Basic::Request.new(request.env)
|
|
if @auth.provided? and @auth.basic? and @auth.credentials
|
|
c = @auth.credentials
|
|
begin
|
|
u = Devops::Db.connector.user_auth(c[0], c[1])
|
|
request.env['REMOTE_USER'] = c[0]
|
|
request.env['USER'] = u
|
|
true
|
|
rescue RecordNotFound => e
|
|
false
|
|
end
|
|
else
|
|
false
|
|
end
|
|
end
|
|
end
|
|
|
|
def self.registered(app)
|
|
app.helpers Sinatra::DevopsAuth::Helpers
|
|
end
|
|
|
|
end
|
|
|
|
register Sinatra::DevopsAuth
|
|
end
|