Unfloyded/Legamunt
0
0
Fork 0
Code Activity

Improve notification model

Browse source
This commit is contained in:
Eugen Rochko 2016-12-03 20:04:19 +01:00
parent 5abf64d647
commit b14b5e3b44
5 changed files with 27 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
app/models/notification.rb
View file

@ -17,10 +17,12 @@ class Notification < ApplicationRecord
STATUS_INCLUDES = [:account, :stream_entry, :media_attachments, :tags, mentions: :account, reblog: [:stream_entry, :account, :media_attachments, :tags, mentions: :account]].freeze
scope :cache_ids, -> { select(:id, :updated_at, :activity_type, :activity_id) }
cache_associated :from_account, status: STATUS_INCLUDES, mention: [status: STATUS_INCLUDES], favourite: [:account, status: STATUS_INCLUDES], follow: :account
def activity
send(activity_type.downcase)
def activity(eager_loaded = true)
eager_loaded ? send(activity_type.downcase) : super
end
def type
@ -51,4 +53,18 @@ class Notification < ApplicationRecord
end
end
end
after_initialize :set_from_account
before_validation :set_from_account
private
def set_from_account
case activity_type
when 'Status', 'Follow', 'Favourite'
self.from_account_id = activity(false)&.account_id
when 'Mention'
self.from_account_id = activity(false)&.status&.account_id
end
end
end