Merge pull request #6 from Besermenji/vb/sinatra-auth

Sinatra authentication added
Besermenji · Jul 4, 2016 · c8c5f1c · c8c5f1c
2 parents 960814a + 20f7fb5
commit c8c5f1c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git 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"])