process_mentions_service.rb 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. # frozen_string_literal: true
  2. class ProcessMentionsService < BaseService
  3. include StreamEntryRenderer
  4. # Scan status for mentions and fetch remote mentioned users, create
  5. # local mention pointers, send Salmon notifications to mentioned
  6. # remote users
  7. # @param [Status] status
  8. def call(status)
  9. return unless status.local?
  10. @status = status
  11. mentions = []
  12. status.text = status.text.gsub(Account::MENTION_RE) do |match|
  13. username, domain = Regexp.last_match(1).split('@')
  14. mentioned_account = Account.find_remote(username, domain)
  15. if mention_undeliverable?(mentioned_account)
  16. begin
  17. mentioned_account = resolve_account_service.call(Regexp.last_match(1))
  18. rescue Goldfinger::Error, HTTP::Error, OpenSSL::SSL::SSLError, Mastodon::UnexpectedResponseError
  19. mentioned_account = nil
  20. end
  21. end
  22. next match if mention_undeliverable?(mentioned_account) || mentioned_account&.suspended
  23. mentions << mentioned_account.mentions.where(status: status).first_or_create(status: status)
  24. "@#{mentioned_account.acct}"
  25. end
  26. status.save!
  27. mentions.each { |mention| create_notification(mention) }
  28. end
  29. private
  30. def mention_undeliverable?(mentioned_account)
  31. mentioned_account.nil? || (!mentioned_account.local? && mentioned_account.ostatus? && @status.stream_entry.hidden?)
  32. end
  33. def create_notification(mention)
  34. mentioned_account = mention.account
  35. if mentioned_account.local?
  36. LocalNotificationWorker.perform_async(mentioned_account.id, mention.id, mention.class.name)
  37. elsif mentioned_account.ostatus? && !@status.stream_entry.hidden?
  38. NotificationWorker.perform_async(ostatus_xml, @status.account_id, mentioned_account.id)
  39. elsif mentioned_account.activitypub?
  40. ActivityPub::DeliveryWorker.perform_async(activitypub_json, mention.status.account_id, mentioned_account.inbox_url)
  41. end
  42. end
  43. def ostatus_xml
  44. @ostatus_xml ||= stream_entry_to_xml(@status.stream_entry)
  45. end
  46. def activitypub_json
  47. return @activitypub_json if defined?(@activitypub_json)
  48. payload = ActiveModelSerializers::SerializableResource.new(
  49. @status,
  50. serializer: ActivityPub::ActivitySerializer,
  51. adapter: ActivityPub::Adapter
  52. ).as_json
  53. @activitypub_json = Oj.dump(@status.distributable? ? ActivityPub::LinkedDataSignature.new(payload).sign!(@status.account) : payload)
  54. end
  55. def resolve_account_service
  56. ResolveAccountService.new
  57. end
  58. end