refactored omniturize method

eLafo · Oct 26, 2011 · 5091ce7 · 5091ce7
1 parent 0e00a98
commit 5091ce7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/lib/omniturize_rails.rb b/lib/omniturize_rails.rb
@@ -8,7 +8,7 @@ def self.included(base)
     module ClassMethods
       def omniturize(options = {})
         include InstanceMethods
-        before_filter :set_reporter, options
+        before_filter{|c| c.send(:set_reporter, options.reverse_merge(:reporter => c.class.name.gsub(/Controller$/, '')))}
         attr_accessor :reporter
       end
     end
@@ -18,10 +18,8 @@ module InstanceMethods
       private
 
       def set_reporter(options = {})
-
         @reporter ||=  begin
-          options[:controller].present? ? "#{options[:controller].classify}Reporter".constantize.new(self) :
-                                          "#{self.class.name.gsub(/Controller$/, '')}Reporter".constantize.new(self)
+          "#{options[:reporter].classify.pluralize}Reporter".constantize.new(self)
         rescue NameError
           BasicReporter.new(self)
         end