X-Git-Url: https://git.openstreetmap.org/rails.git/blobdiff_plain/e660e609661edadc1ed5ad49d6e83e936b2f91cd..15c156483263893d2c767030bd544d589833af93:/app/models/diary_comment.rb

diff --git a/app/models/diary_comment.rb b/app/models/diary_comment.rb
index b915e027a..9d29f52b7 100644
--- a/app/models/diary_comment.rb
+++ b/app/models/diary_comment.rb
@@ -5,7 +5,12 @@ class DiaryComment < ActiveRecord::Base
   validates_presence_of :body
   validates_associated :diary_entry
 
-  attr_accessible :body
+  after_initialize :set_defaults
+  after_save :spam_check
+
+  def body
+    RichText.new(read_attribute(:body_format), read_attribute(:body))
+  end
 
   def digest
     md5 = Digest::MD5.new
@@ -15,4 +20,14 @@ class DiaryComment < ActiveRecord::Base
     md5 << body
     md5.hexdigest
   end
+
+private
+
+  def set_defaults
+    self.body_format = "markdown" unless self.attribute_present?(:body_format)
+  end
+
+  def spam_check
+    user.spam_check
+  end
 end