post_status_service.rb 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  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. if options[:idempotency].present?
  38. redis.setex("idempotency:status:#{account.id}:#{options[:idempotency]}", 3_600, status.id)
  39. end
  40. status
  41. end
  42. private
  43. def validate_media!(media_ids)
  44. return if media_ids.blank? || !media_ids.is_a?(Enumerable)
  45. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.too_many') if media_ids.size > 4
  46. media = MediaAttachment.where(status_id: nil).where(id: media_ids.take(4).map(&:to_i))
  47. raise Mastodon::ValidationError, I18n.t('media_attachments.validations.images_and_video') if media.size > 1 && media.find(&:video?)
  48. media
  49. end
  50. def attach_media(status, media)
  51. return if media.nil?
  52. media.update(status_id: status.id)
  53. end
  54. def detect_language_for(text, account)
  55. LanguageDetector.new(text, account).to_iso_s
  56. end
  57. def process_mentions_service
  58. @process_mentions_service ||= ProcessMentionsService.new
  59. end
  60. def process_hashtags_service
  61. @process_hashtags_service ||= ProcessHashtagsService.new
  62. end
  63. def redis
  64. Redis.current
  65. end
  66. end