From b58517976acad3f2f30214fba6165f8a8e90e540 Mon Sep 17 00:00:00 2001 From: Trym Date: Tue, 9 Mar 2010 01:35:37 +0100 Subject: [PATCH] Rails 3 compatibility --- init.rb | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/init.rb b/init.rb index 86bd533..d3fc931 100644 --- 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 \ No newline at end of file