33 lines
657 B
Ruby
33 lines
657 B
Ruby
|
module Sinatra
|
||
|
|
module Devops
|
||
|
|
module Helpers
|
||
|
|
def protect!
|
||
|
|
return if auth_with_basic?
|
||
|
|
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
|
||
|
|
Devops::Db.connector.user_auth(c[0], c[1])
|
||
|
|
true
|
||
|
|
rescue RecordNotFound => e
|
||
|
|
false
|
||
|
|
end
|
||
|
|
else
|
||
|
|
false
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.registered(app)
|
||
|
|
app.helpers Devops::Helpers
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
register Devops
|
||
|
|
end
|