You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.7 KiB
98 lines
3.7 KiB
# frozen_string_literal: true |
|
|
|
class ProcessMentionsService < BaseService |
|
include Payloadable |
|
|
|
# Scan status for mentions and fetch remote mentioned users, |
|
# and create local mention pointers |
|
# @param [Status] status |
|
# @param [Boolean] save_records Whether to save records in database |
|
def call(status, save_records: true) |
|
@status = status |
|
@save_records = save_records |
|
|
|
return unless @status.local? |
|
|
|
@previous_mentions = @status.mentions.includes(:account).to_a |
|
@current_mentions = [] |
|
|
|
Status.transaction do |
|
scan_text! |
|
assign_mentions! |
|
end |
|
end |
|
|
|
private |
|
|
|
def scan_text! |
|
@status.text = @status.text.gsub(Account::MENTION_RE) do |match| |
|
username, domain = Regexp.last_match(1).split('@') |
|
|
|
domain = if TagManager.instance.local_domain?(domain) |
|
nil |
|
else |
|
TagManager.instance.normalize_domain(domain) |
|
end |
|
|
|
mentioned_account = Account.find_remote(username, domain) |
|
|
|
# Unapproved and unconfirmed accounts should not be mentionable |
|
next match if mentioned_account&.local? && !(mentioned_account.user_confirmed? && mentioned_account.user_approved?) |
|
|
|
# If the account cannot be found or isn't the right protocol, |
|
# first try to resolve it |
|
if mention_undeliverable?(mentioned_account) |
|
begin |
|
mentioned_account = ResolveAccountService.new.call(Regexp.last_match(1)) |
|
rescue Webfinger::Error, *Mastodon::HTTP_CONNECTION_ERRORS, Mastodon::UnexpectedResponseError |
|
mentioned_account = nil |
|
end |
|
end |
|
|
|
# If after resolving it still isn't found or isn't the right |
|
# protocol, or it's a federated account and the status is local-only, |
|
# then give up |
|
next match if mention_undeliverable?(mentioned_account) || mentioned_account&.unavailable? || (!mentioned_account.local? && @status.local_only?) |
|
|
|
mention = @previous_mentions.find { |x| x.account_id == mentioned_account.id } |
|
mention ||= @current_mentions.find { |x| x.account_id == mentioned_account.id } |
|
mention ||= @status.mentions.new(account: mentioned_account) |
|
|
|
mention.silent = false |
|
|
|
@current_mentions << mention |
|
|
|
"@#{mentioned_account.acct}" |
|
end |
|
|
|
@status.save! if @save_records |
|
end |
|
|
|
def assign_mentions! |
|
# Make sure we never mention blocked accounts |
|
unless @current_mentions.empty? |
|
mentioned_domains = @current_mentions.filter_map { |m| m.account.domain }.uniq |
|
blocked_domains = Set.new(mentioned_domains.empty? ? [] : AccountDomainBlock.where(account_id: @status.account_id, domain: mentioned_domains)) |
|
mentioned_account_ids = @current_mentions.map(&:account_id) |
|
blocked_account_ids = Set.new(@status.account.block_relationships.where(target_account_id: mentioned_account_ids).pluck(:target_account_id)) |
|
|
|
dropped_mentions, @current_mentions = @current_mentions.partition { |mention| blocked_account_ids.include?(mention.account_id) || blocked_domains.include?(mention.account.domain) } |
|
dropped_mentions.each(&:destroy) |
|
end |
|
|
|
@current_mentions.each do |mention| |
|
mention.save if (mention.new_record? || mention.silent_changed?) && @save_records |
|
end |
|
|
|
# If previous mentions are no longer contained in the text, convert them |
|
# to silent mentions, since withdrawing access from someone who already |
|
# received a notification might be more confusing |
|
removed_mentions = @previous_mentions - @current_mentions |
|
|
|
Mention.where(id: removed_mentions.map(&:id), silent: false).update_all(silent: true) unless removed_mentions.empty? |
|
end |
|
|
|
def mention_undeliverable?(mentioned_account) |
|
mentioned_account.nil? || (!mentioned_account.local? && !mentioned_account.activitypub?) |
|
end |
|
end
|
|
|