diff --git a/report_generator.rb b/report_generator.rb index 3dfb146..c1b434f 100644 --- a/report_generator.rb +++ b/report_generator.rb @@ -6,6 +6,19 @@ class ReportGenerator < Sinatra::Base + + use Rack::Auth::Basic, "Restricted Area" do |username, password| + + if ENV['RACK_ENV'] == 'production' + @username = ENV["SINATRA_USERNAME"] + @password = ENV["SINATRA_PASSWORD"] + else + @username = "admin" + @password = "admin" + end + + username == @username and password == @password + end post '/kif/?' do @kif_info = JSON.parse(params["data"])