Preserve background colour when marking messages read/unread

[rails.git] / app / models / diary_entry.rb

diff --git a/app/models/diary_entry.rb b/app/models/diary_entry.rb
index 1d836353fee8b30d9f4f1a56396a068966748a49..64a412d28c771a55da936381c04f37737371962a 100644 (file)
--- a/app/models/diary_entry.rb
+++ b/app/models/diary_entry.rb
@@ -25,4 +25,16 @@ class DiaryEntry < ActiveRecord::Base
   validates_associated :language
 
   attr_accessible :title, :body, :language_code, :latitude, :longitude
+
+  after_initialize :set_defaults
+
+  def body
+    RichText.new(read_attribute(:body_format), read_attribute(:body))
+  end
+
+private
+
+  def set_defaults
+    self.body_format = "markdown" unless self.attribute_present?(:body_format)
+  end
 end