178 lines
3.8 KiB
Ruby
178 lines
3.8 KiB
Ruby
class Dmail < ActiveRecord::Base
|
|
validates_presence_of :to_id
|
|
validates_presence_of :from_id
|
|
validates_format_of :title, :with => /\S/
|
|
validates_format_of :body, :with => /\S/
|
|
before_validation :initialize_from_id, :on => :create
|
|
belongs_to :owner, :class_name => "User"
|
|
belongs_to :to, :class_name => "User"
|
|
belongs_to :from, :class_name => "User"
|
|
after_create :update_recipient
|
|
after_create :send_dmail
|
|
attr_accessible :title, :body, :is_deleted, :to_id, :to, :to_name
|
|
|
|
module AddressMethods
|
|
def to_name
|
|
User.id_to_pretty_name(to_id)
|
|
end
|
|
|
|
def from_name
|
|
User.id_to_pretty_name(from_id)
|
|
end
|
|
|
|
def to_name=(name)
|
|
user = User.find_by_name(name)
|
|
return if user.nil?
|
|
self.to_id = user.id
|
|
end
|
|
|
|
def initialize_from_id
|
|
self.from_id = CurrentUser.id
|
|
end
|
|
end
|
|
|
|
module FactoryMethods
|
|
extend ActiveSupport::Concern
|
|
|
|
module ClassMethods
|
|
def create_split(params)
|
|
copy = nil
|
|
|
|
Dmail.transaction do
|
|
copy = Dmail.new(params)
|
|
copy.owner_id = copy.to_id
|
|
unless copy.to_id == CurrentUser.id
|
|
copy.save
|
|
end
|
|
|
|
copy = Dmail.new(params)
|
|
copy.owner_id = CurrentUser.id
|
|
copy.is_read = true
|
|
copy.save
|
|
end
|
|
|
|
copy
|
|
end
|
|
|
|
def new_blank
|
|
Dmail.new do |dmail|
|
|
dmail.from_id = CurrentUser.id
|
|
end
|
|
end
|
|
end
|
|
|
|
def build_response(options = {})
|
|
Dmail.new do |dmail|
|
|
dmail.title = "Re: #{title}"
|
|
dmail.owner_id = from_id
|
|
dmail.body = quoted_body
|
|
dmail.to_id = from_id unless options[:forward]
|
|
dmail.from_id = to_id
|
|
end
|
|
end
|
|
end
|
|
|
|
module SearchMethods
|
|
def for(user)
|
|
where("owner_id = ?", user)
|
|
end
|
|
|
|
def inbox
|
|
where("to_id = owner_id")
|
|
end
|
|
|
|
def sent
|
|
where("from_id = owner_id")
|
|
end
|
|
|
|
def active
|
|
where("is_deleted = ?", false)
|
|
end
|
|
|
|
def deleted
|
|
where("is_deleted = ?", true)
|
|
end
|
|
|
|
def search_message(query)
|
|
where("message_index @@ plainto_tsquery(?)", query)
|
|
end
|
|
|
|
def unread
|
|
where("is_read = false and is_deleted = false")
|
|
end
|
|
|
|
def visible
|
|
where("owner_id = ?", CurrentUser.id)
|
|
end
|
|
|
|
def to_name_matches(name)
|
|
where("to_id = (select _.id from users _ where lower(_.name) = ?)", name.downcase)
|
|
end
|
|
|
|
def from_name_matches(name)
|
|
where("from_id = (select _.id from users _ where lower(_.name) = ?)", name.downcase)
|
|
end
|
|
|
|
def search(params)
|
|
q = scoped
|
|
return q if params.blank?
|
|
|
|
if params[:message_matches]
|
|
q = q.search_message(params[:message_matches])
|
|
end
|
|
|
|
if params[:owner_id]
|
|
q = q.for(params[:owner_id].to_i)
|
|
end
|
|
|
|
if params[:to_name]
|
|
q = q.to_name_matches(params[:to_name])
|
|
end
|
|
|
|
if params[:to_id]
|
|
q = q.where("to_id = ?", params[:to_id].to_i)
|
|
end
|
|
|
|
if params[:from_name]
|
|
q = q.from_name_matches(params[:from_name])
|
|
end
|
|
|
|
if params[:from_id]
|
|
q = q.where("from_id = ?", params[:from_id].to_i)
|
|
end
|
|
|
|
q
|
|
end
|
|
end
|
|
|
|
include AddressMethods
|
|
include FactoryMethods
|
|
extend SearchMethods
|
|
|
|
def quoted_body
|
|
"[quote]#{body}[/quote]"
|
|
end
|
|
|
|
def send_dmail
|
|
if to.receive_email_notifications? && to.email.include?("@")
|
|
UserMailer.dmail_notice(self).deliver
|
|
end
|
|
end
|
|
|
|
def mark_as_read!
|
|
update_column(:is_read, true)
|
|
|
|
unless Dmail.exists?(["to_id = ? AND is_read = false", to_id])
|
|
to.update_column(:has_mail, false)
|
|
end
|
|
end
|
|
|
|
def update_recipient
|
|
to.update_column(:has_mail, true)
|
|
end
|
|
|
|
def visible_to?(user)
|
|
user.is_moderator? || owner_id == user.id
|
|
end
|
|
end
|