post_status_service.rb 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. # frozen_string_literal: true
  2. class PostStatusService < BaseService
  3. # Post a text status update, fetch and notify remote users mentioned
  4. # @param [Account] account Account from which to post
  5. # @param [String] text Message
  6. # @param [Status] in_reply_to Optional status to reply to
  7. # @param [Hash] options
  8. # @option [Boolean] :sensitive
  9. # @option [String] :visibility
  10. # @option [String] :spoiler_text
  11. # @option [Enumerable] :media_ids Optional array of media IDs to attach
  12. # @option [Doorkeeper::Application] :application
  13. # @option [String] :idempotency Optional idempotency key
  14. # @return [Status]
  15. def call(account, text, in_reply_to = nil, options = {})
  16. if options[:idempotency].present?
  17. existing_id = redis.get("idempotency:status:#{account.id}:#{options[:idempotency]}")
  18. return Status.find(existing_id) if existing_id
  19. end
  20. media = validate_media!(options[:media_ids])
  21. status = nil
  22. ApplicationRecord.transaction do
  23. status = account.statuses.create!(text: text,
  24. thread: in_reply_to,
  25. sensitive: options[:sensitive],
  26. spoiler_text: options[:spoiler_text] || '',
  27. visibility: options[:visibility],
  28. language: detect_language_for(text, account),
  29. application: options[:application])
  30. attach_media(status, media)
  31. end
  32. process_mentions_service.call(status)
  33. process_hashtags_service.call(status)
  34. LinkCrawlWorker.perform_async(status.id) unless status.spoiler_text?
  35. DistributionWorker.perform_async(status.id)
  36. Pubsubhubbub::DistributionWorker.perform_async(status.stream_entry.id)
  37. ActivityPub::DistributionWorker.perform_async(status.id)
  38. ActivityPub::ReplyDistributionWorker.perform_async(status.id) if status.reply? && status.thread.account.local?
  39. if options[:idempotency].present?
  40. redis.setex("idempotency:status:#{account.id}:#{options[:idempotency]}", 3_600, status.id)
  41. end
  42. status
  43. end
  44. private
  45. def validate_media!(media_ids)
  46. return if media_ids.blank? || !media_ids.is_a?(Enumerable)
  47. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.too_many') if media_ids.size > 4
  48. media = MediaAttachment.where(status_id: nil).where(id: media_ids.take(4).map(&:to_i))
  49. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.images_and_video') if media.size > 1 && media.find(&:video?)
  50. media
  51. end
  52. def attach_media(status, media)
  53. return if media.nil?
  54. media.update(status_id: status.id)
  55. end
  56. def detect_language_for(text, account)
  57. LanguageDetector.new(text, account).to_iso_s
  58. end
  59. def process_mentions_service
  60. @process_mentions_service ||= ProcessMentionsService.new
  61. end
  62. def process_hashtags_service
  63. @process_hashtags_service ||= ProcessHashtagsService.new
  64. end
  65. def redis
  66. Redis.current
  67. end
  68. end