lib/ahoy_matey.rb in ahoy_matey-0.1.4 vs lib/ahoy_matey.rb in ahoy_matey-0.1.5

@@ -22,9 +22,22 @@
   # performance hack for referer-parser
   def self.referrer_parser
     @referrer_parser ||= RefererParser::Referer.new("https://github.com/ankane/ahoy")
   end
 
+  def self.fetch_user(controller)
+    if user_method.respond_to?(:call)
+      user_method.call(controller)
+    else
+      controller.send(user_method)
+    end
+  end
+
+  mattr_accessor :user_method
+  self.user_method = proc do |controller|
+    (controller.respond_to?(:current_user) && controller.current_user) || (controller.respond_to?(:current_resource_owner, true) and controller.send(:current_resource_owner))
+  end
+
 end
 
 ActionController::Base.send :include, Ahoy::Controller
 ActiveRecord::Base.send(:extend, Ahoy::Model) if defined?(ActiveRecord)