X-Git-Url: https://git.openstreetmap.org./rails.git/blobdiff_plain/9c5c71f2d7413c280c046f45d9b0565981c7bbab..3458d881af4584e44d72bb682771e3c7a3cf6e68:/app/models/message.rb

diff --git a/app/models/message.rb b/app/models/message.rb
index 590e3e357..f897af3c2 100644
--- a/app/models/message.rb
+++ b/app/models/message.rb
@@ -1,2 +1,58 @@
+require 'validators'
+
 class Message < ActiveRecord::Base
+  belongs_to :sender, :class_name => "User", :foreign_key => :from_user_id
+  belongs_to :recipient, :class_name => "User", :foreign_key => :to_user_id
+
+  validates_presence_of :title, :body, :sent_on, :sender, :recipient
+  validates_length_of :title, :within => 1..255
+  validates_inclusion_of :message_read, :in => [ true, false ]
+  validates_as_utf8 :title
+
+  attr_accessible :title, :body
+
+  after_initialize :set_defaults
+
+  def self.from_mail(mail, from, to)
+    if mail.multipart?
+      if mail.text_part
+        body = mail.text_part.decoded
+      elsif mail.html_part
+        body = HTMLEntities.new.decode(Sanitize.clean(mail.html_part.decoded))
+      end
+    elsif mail.text? and mail.sub_type == "html"
+      body = HTMLEntities.new.decode(Sanitize.clean(mail.decoded))
+    else
+      body = mail.decoded
+    end
+
+    message = Message.new({
+      :sender => from,
+      :recipient => to,
+      :sent_on => mail.date.new_offset(0),
+      :title => mail.subject.sub(/\[OpenStreetMap\] */, ""),
+      :body => body,
+      :body_format => "text"
+    }, :without_protection => true)
+  end
+
+  def body
+    RichText.new(read_attribute(:body_format), read_attribute(:body))
+  end
+
+  def digest
+    md5 = Digest::MD5.new
+    md5 << from_user_id.to_s
+    md5 << to_user_id.to_s
+    md5 << sent_on.xmlschema
+    md5 << title
+    md5 << body
+    md5.hexdigest
+  end
+
+private
+
+  def set_defaults
+    self.body_format = "markdown" unless self.attribute_present?(:body_format)
+  end
 end