post_status_service.rb 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193
  1. # frozen_string_literal: true
  2. class PostStatusService < BaseService
  3. include Redisable
  4. MIN_SCHEDULE_OFFSET = 5.minutes.freeze
  5. # Post a text status update, fetch and notify remote users mentioned
  6. # @param [Account] account Account from which to post
  7. # @param [Hash] options
  8. # @option [String] :text Message
  9. # @option [Status] :thread Optional status to reply to
  10. # @option [Boolean] :sensitive
  11. # @option [String] :visibility
  12. # @option [String] :spoiler_text
  13. # @option [String] :language
  14. # @option [String] :scheduled_at
  15. # @option [Hash] :poll Optional poll to attach
  16. # @option [Enumerable] :media_ids Optional array of media IDs to attach
  17. # @option [Doorkeeper::Application] :application
  18. # @option [String] :idempotency Optional idempotency key
  19. # @option [Boolean] :with_rate_limit
  20. # @return [Status]
  21. def call(account, options = {})
  22. @account = account
  23. @options = options
  24. @text = @options[:text] || ''
  25. @in_reply_to = @options[:thread]
  26. return idempotency_duplicate if idempotency_given? && idempotency_duplicate?
  27. validate_media!
  28. preprocess_attributes!
  29. if scheduled?
  30. schedule_status!
  31. else
  32. process_status!
  33. postprocess_status!
  34. bump_potential_friendship!
  35. end
  36. redis.setex(idempotency_key, 3_600, @status.id) if idempotency_given?
  37. @status
  38. end
  39. private
  40. def preprocess_attributes!
  41. @sensitive = (@options[:sensitive].nil? ? @account.user&.setting_default_sensitive : @options[:sensitive]) || @options[:spoiler_text].present?
  42. @text = @options.delete(:spoiler_text) if @text.blank? && @options[:spoiler_text].present?
  43. @visibility = @options[:visibility] || @account.user&.setting_default_privacy
  44. @visibility = :unlisted if @visibility&.to_sym == :public && @account.silenced?
  45. @scheduled_at = @options[:scheduled_at]&.to_datetime
  46. @scheduled_at = nil if scheduled_in_the_past?
  47. rescue ArgumentError
  48. raise ActiveRecord::RecordInvalid
  49. end
  50. def process_status!
  51. # The following transaction block is needed to wrap the UPDATEs to
  52. # the media attachments when the status is created
  53. ApplicationRecord.transaction do
  54. @status = @account.statuses.create!(status_attributes)
  55. end
  56. process_hashtags_service.call(@status)
  57. process_mentions_service.call(@status)
  58. end
  59. def schedule_status!
  60. status_for_validation = @account.statuses.build(status_attributes)
  61. if status_for_validation.valid?
  62. status_for_validation.destroy
  63. # The following transaction block is needed to wrap the UPDATEs to
  64. # the media attachments when the scheduled status is created
  65. ApplicationRecord.transaction do
  66. @status = @account.scheduled_statuses.create!(scheduled_status_attributes)
  67. end
  68. else
  69. raise ActiveRecord::RecordInvalid
  70. end
  71. end
  72. def postprocess_status!
  73. LinkCrawlWorker.perform_async(@status.id) unless @status.spoiler_text?
  74. DistributionWorker.perform_async(@status.id)
  75. ActivityPub::DistributionWorker.perform_async(@status.id)
  76. PollExpirationNotifyWorker.perform_at(@status.poll.expires_at, @status.poll.id) if @status.poll
  77. end
  78. def validate_media!
  79. return if @options[:media_ids].blank? || !@options[:media_ids].is_a?(Enumerable)
  80. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.too_many') if @options[:media_ids].size > 4 || @options[:poll].present?
  81. @media = @account.media_attachments.where(status_id: nil).where(id: @options[:media_ids].take(4).map(&:to_i))
  82. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.images_and_video') if @media.size > 1 && @media.find(&:audio_or_video?)
  83. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.not_ready') if @media.any?(&:not_processed?)
  84. end
  85. def language_from_option(str)
  86. ISO_639.find(str)&.alpha2
  87. end
  88. def process_mentions_service
  89. ProcessMentionsService.new
  90. end
  91. def process_hashtags_service
  92. ProcessHashtagsService.new
  93. end
  94. def scheduled?
  95. @scheduled_at.present?
  96. end
  97. def idempotency_key
  98. "idempotency:status:#{@account.id}:#{@options[:idempotency]}"
  99. end
  100. def idempotency_given?
  101. @options[:idempotency].present?
  102. end
  103. def idempotency_duplicate
  104. if scheduled?
  105. @account.schedule_statuses.find(@idempotency_duplicate)
  106. else
  107. @account.statuses.find(@idempotency_duplicate)
  108. end
  109. end
  110. def idempotency_duplicate?
  111. @idempotency_duplicate = redis.get(idempotency_key)
  112. end
  113. def scheduled_in_the_past?
  114. @scheduled_at.present? && @scheduled_at <= Time.now.utc + MIN_SCHEDULE_OFFSET
  115. end
  116. def bump_potential_friendship!
  117. return if !@status.reply? || @account.id == @status.in_reply_to_account_id
  118. ActivityTracker.increment('activity:interactions')
  119. return if @account.following?(@status.in_reply_to_account_id)
  120. PotentialFriendshipTracker.record(@account.id, @status.in_reply_to_account_id, :reply)
  121. end
  122. def status_attributes
  123. {
  124. text: @text,
  125. media_attachments: @media || [],
  126. thread: @in_reply_to,
  127. poll_attributes: poll_attributes,
  128. sensitive: @sensitive,
  129. spoiler_text: @options[:spoiler_text] || '',
  130. visibility: @visibility,
  131. language: language_from_option(@options[:language]) || @account.user&.setting_default_language&.presence || LanguageDetector.instance.detect(@text, @account),
  132. application: @options[:application],
  133. rate_limit: @options[:with_rate_limit],
  134. }.compact
  135. end
  136. def scheduled_status_attributes
  137. {
  138. scheduled_at: @scheduled_at,
  139. media_attachments: @media || [],
  140. params: scheduled_options,
  141. }
  142. end
  143. def poll_attributes
  144. return if @options[:poll].blank?
  145. @options[:poll].merge(account: @account, voters_count: 0)
  146. end
  147. def scheduled_options
  148. @options.tap do |options_hash|
  149. options_hash[:in_reply_to_id] = options_hash.delete(:thread)&.id
  150. options_hash[:application_id] = options_hash.delete(:application)&.id
  151. options_hash[:scheduled_at] = nil
  152. options_hash[:idempotency] = nil
  153. options_hash[:with_rate_limit] = false
  154. end
  155. end
  156. end