set LAZAR_ENV to RACK_ENV

1 files changed, 12 insertions, 2 deletions

diff --git a/lib/lazar-rest.rb b/lib/lazar-rest.rb
index 2de3998..77483f2 100644
--- a/lib/lazar-rest.rb
+++ b/lib/lazar-rest.rb
@@ -1,6 +1,16 @@
 require "sinatra"
-require "../lazar/lib/lazar.rb"
-require "../qsar-report/lib/qsar-report.rb"
+
+# Environment setup from unicorn -E param
+ENV["LAZAR_ENV"] = ENV["RACK_ENV"]
+
+if ENV["LAZAR_ENV"] == "development"
+  require "../lazar/lib/lazar.rb"
+  require "../qsar-report/lib/qsar-report.rb"
+else
+  require "lazar"
+  require "qsar-report"
+end
+
 include OpenTox
 
 require 'rack/cors'