docbrown/app/models/mention.rb
Andy Zhao 2db2dd1e5a WIP Refactor/notifications (#1131)
* Add new columns to notifications table

* Remove stream from notifications and add new methods

* Turn off moderation comment notifications

* Add notifiable_type index to migration

* Stop creation of Stream-based notification model instances

* Add notification model create hooks

* Add badge achievement notification creation hook

* Fix some specs

* Add missing @ symbol

* Fix for tests and rename a few things

* forgot to uncomment code sigh
2018-11-19 16:09:02 -05:00

86 lines
2.4 KiB
Ruby

class Mention < ApplicationRecord
include StreamRails::Activity
as_activity
belongs_to :user
belongs_to :mentionable, polymorphic: true
validates :user_id, presence: true,
uniqueness: { scope: %i[mentionable_id
mentionable_type] }
validates :mentionable_id, presence: true
validates :mentionable_type, presence: true
validate :permission
after_create :send_email_notification
class << self
def create_all(notifiable)
# Only works for comments right now.
# Paired with the process that creates the "comment-mentioned-user"
@notifiable = notifiable
doc = Nokogiri::HTML(notifiable.processed_html)
usernames = []
mentions = []
doc.css(".comment-mentioned-user").each do |link|
username = link.text.gsub("@", "").downcase
if user = User.find_by_username(username)
usernames << username
mentions << create_mention(user)
end
end
delete_removed_mentions(usernames)
mentions
end
handle_asynchronously :create_all
private
def delete_removed_mentions(usernames)
user_ids = User.where(username: usernames).pluck(:id)
mentions = @notifiable.mentions.where.not(user_id: user_ids).destroy_all
Notification.remove_each(mentions) unless mentions.blank?
end
def create_mention(user)
mention = Mention.create(user_id: user.id, mentionable_id: @notifiable.id, mentionable_type: @notifiable.class.name)
# mentionable_type = model that created the mention, user = user to be mentioned
Notification.send_mention_notification(mention)
mention
end
end
def activity_actor
mentionable.user
end
def activity_notify
return if mentionable.parent_user && mentionable.parent_user.id == user.id
[StreamNotifier.new(user.id).notify]
end
def activity_object
mentionable
end
def activity_target
"mention_#{Time.current}"
end
def remove_from_feed
super
User.find_by(id: user.id)&.touch(:last_notification_activity)
end
def send_email_notification
if User.find(user_id).email.present? && User.find(user_id).email_mention_notifications
NotifyMailer.new_mention_email(self).deliver
end
end
handle_asynchronously :send_email_notification
def permission
if !mentionable.valid?
errors.add(:mentionable_id, "is not valid.")
end
end
end