lib/ahoy_email.rb in ahoy_email-0.2.2 vs lib/ahoy_email.rb in ahoy_email-0.2.3

@@ -14,32 +14,32 @@
   self.options = {
     message: true,
     open: true,
     click: true,
     utm_params: true,
-    utm_source: proc {|message, mailer| mailer.mailer_name },
+    utm_source: proc { |message, mailer| mailer.mailer_name },
     utm_medium: "email",
     utm_term: nil,
     utm_content: nil,
-    utm_campaign: proc {|message, mailer| mailer.action_name },
-    user: proc{|message, mailer| (message.to.size == 1 ? User.where(email: message.to.first).first : nil) rescue nil },
-    mailer: proc{|message, mailer| "#{mailer.class.name}##{mailer.action_name}" }
+    utm_campaign: proc { |message, mailer| mailer.action_name },
+    user: proc { |message, mailer| (message.to.size == 1 ? User.where(email: message.to.first).first : nil) rescue nil },
+    mailer: proc { |message, mailer| "#{mailer.class.name}##{mailer.action_name}" },
+    url_options: {}
   }
 
   self.subscribers = []
 
   def self.track(options)
     self.options = self.options.merge(options)
   end
 
-  def self.message_model=(message_model)
-    @message_model = message_model
+  class << self
+    attr_writer :message_model
   end
 
   def self.message_model
     @message_model || Ahoy::Message
   end
-
 end
 
 ActionMailer::Base.send :include, AhoyEmail::Mailer
 ActionMailer::Base.register_interceptor AhoyEmail::Interceptor