Rails 3 compatibility

ayb · Mar 9, 2010 · b585179 · b585179
1 parent 259733e
commit b585179
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/init.rb b/init.rb
@@ -1,3 +1,7 @@
-if RAILS_ENV == 'production'
-	ActionController::Dispatcher.middleware.use Hassle
-end
+if (defined? Rails.env ? Rails.env : RAILS_ENV) == "production"
+  if defined? Rails.configuration && defined? Rails.configuration.middleware
+    Rails.configuration.middleware.use(Hassle)
+  elsif defined? ActionController::Dispatcher && defined? ActionController::Dispatcher.middleware
+    ActionController::Dispatcher.middleware.use Hassle
+  end  
+end