282 lines
8.3 KiB
Ruby
282 lines
8.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class PostStatusService < BaseService
|
|
include Redisable
|
|
|
|
MIN_SCHEDULE_OFFSET = 5.minutes.freeze
|
|
VISIBILITY_RANK = {
|
|
'public' => 0,
|
|
'unlisted' => 1,
|
|
'local' => 1,
|
|
'private' => 2,
|
|
'direct' => 3,
|
|
'limited' => 3,
|
|
'chat' => 4
|
|
}
|
|
|
|
# Post a text status update, fetch and notify remote users mentioned
|
|
# @param [Account] account Account from which to post
|
|
# @param [Hash] options
|
|
# @option [String] :text Message
|
|
# @option [Status] :thread Optional status to reply to
|
|
# @option [Tag] :tags Optional tags to include
|
|
# @option [Time] :created_at Optional time which status was originally posted
|
|
# @option [Boolean] :sensitive
|
|
# @option [String] :visibility
|
|
# @option [Boolean] :local_only
|
|
# @option [String] :sharekey
|
|
# @option [String] :spoiler_text
|
|
# @option [String] :language
|
|
# @option [String] :scheduled_at
|
|
# @option [Hash] :poll Optional poll to attach
|
|
# @option [Enumerable] :media_ids Optional array of media IDs to attach
|
|
# @option [Doorkeeper::Application] :application
|
|
# @option [String] :idempotency Optional idempotency key
|
|
# @return [Status]
|
|
def call(account, options = {})
|
|
@account = account
|
|
@options = options
|
|
@text = @options[:text] || ''
|
|
@footer = @options[:footer]
|
|
@in_reply_to = @options[:thread]
|
|
@tags = @options[:tags] || []
|
|
@local_only = @options[:local_only]
|
|
@sensitive = (@account.force_sensitive? ? true : @options[:sensitive])
|
|
@preloaded_tags = @options[:preloaded_tags] || []
|
|
|
|
return idempotency_duplicate if idempotency_given? && idempotency_duplicate?
|
|
|
|
validate_media!
|
|
preprocess_attributes!
|
|
|
|
if scheduled?
|
|
schedule_status!
|
|
else
|
|
process_status!
|
|
postprocess_status!
|
|
bump_potential_friendship!
|
|
end
|
|
|
|
redis.setex(idempotency_key, 3_600, @status.id) if idempotency_given?
|
|
|
|
@status
|
|
end
|
|
|
|
private
|
|
|
|
def set_footer_from_i_am
|
|
return if @footer.nil? || @options[:no_footer]
|
|
name = @account.vars['_they:are']
|
|
return if name.blank?
|
|
@footer = @account.vars["_they:are:#{name}"]
|
|
end
|
|
|
|
def set_initial_visibility
|
|
@visibility = @options[:visibility] || @account.user_default_visibility
|
|
end
|
|
|
|
def limit_visibility_if_silenced
|
|
@visibility = :unlisted if @visibility.in?([nil, 'public']) && @account.silenced? || @account.force_unlisted
|
|
end
|
|
|
|
def limit_visibility_to_reply
|
|
return if @in_reply_to.nil?
|
|
@visibility = @in_reply_to.visibility if @visibility.nil? ||
|
|
VISIBILITY_RANK[@visibility] < VISIBILITY_RANK[@in_reply_to.visibility]
|
|
end
|
|
|
|
def set_local_only
|
|
@local_only = true if @account.user_always_local_only? || @in_reply_to&.local_only
|
|
end
|
|
|
|
def set_chat
|
|
if @in_reply_to.present?
|
|
unless @in_reply_to.chat_tags.blank?
|
|
@preloaded_tags |= @in_reply_to.chat_tags
|
|
@visibility = :chat
|
|
@in_reply_to = nil
|
|
end
|
|
elsif @tags.present? && @tags.any? { |t| t.start_with?('chat.', '.chat.') }
|
|
@visibility = :chat
|
|
@local_only = true if @tags.any? { |t| t.in?(%w(chat.local .chat.local)) || t.start_with?('chat.local.', '.chat.local.') }
|
|
end
|
|
end
|
|
|
|
# move tags out of body so we can format them later
|
|
def extract_tags
|
|
chunks = []
|
|
@text.split(/^((?:#[\w:._·\-]*\s*)+)/).each do |chunk|
|
|
if chunk.start_with?('#')
|
|
@tags |= chunk[1..-1].split(/\s+/)
|
|
else
|
|
chunks << chunk
|
|
end
|
|
end
|
|
@text = chunks.join
|
|
end
|
|
|
|
def preprocess_attributes!
|
|
if @text.blank? && @options[:spoiler_text].present?
|
|
@text = '.'
|
|
@text = @media.find(&:video?) ? '📹' : '🖼' if @media.size > 0
|
|
end
|
|
|
|
set_footer_from_i_am
|
|
extract_tags
|
|
set_chat
|
|
set_local_only
|
|
|
|
unless @visibility == :chat
|
|
set_initial_visibility
|
|
limit_visibility_if_silenced
|
|
limit_visibility_to_reply
|
|
end
|
|
|
|
@sensitive = (@account.user_defaults_to_sensitive? || @options[:spoiler_text].present?) if @sensitive.nil?
|
|
|
|
@scheduled_at = @options[:scheduled_at]&.to_datetime
|
|
@scheduled_at = nil if scheduled_in_the_past?
|
|
rescue ArgumentError
|
|
raise ActiveRecord::RecordInvalid
|
|
end
|
|
|
|
def process_status!
|
|
# The following transaction block is needed to wrap the UPDATEs to
|
|
# the media attachments when the status is created
|
|
|
|
ApplicationRecord.transaction do
|
|
@status = @account.statuses.create!(status_attributes)
|
|
end
|
|
|
|
process_hashtags_service.call(@status, @tags, @preloaded_tags)
|
|
process_mentions_service.call(@status)
|
|
end
|
|
|
|
def schedule_status!
|
|
status_for_validation = @account.statuses.build(status_attributes)
|
|
|
|
if status_for_validation.valid?
|
|
status_for_validation.destroy
|
|
|
|
# The following transaction block is needed to wrap the UPDATEs to
|
|
# the media attachments when the scheduled status is created
|
|
|
|
ApplicationRecord.transaction do
|
|
@status = @account.scheduled_statuses.create!(scheduled_status_attributes)
|
|
end
|
|
else
|
|
raise ActiveRecord::RecordInvalid
|
|
end
|
|
end
|
|
|
|
def postprocess_status!
|
|
LinkCrawlWorker.perform_async(@status.id) unless @status.spoiler_text?
|
|
DistributionWorker.perform_async(@status.id) unless @options[:distribute] == false
|
|
|
|
unless @status.local_only? || @options[:distribute] == false || @options[:federate] == false
|
|
ActivityPub::DistributionWorker.perform_async(@status.id)
|
|
end
|
|
|
|
PollExpirationNotifyWorker.perform_at(@status.poll.expires_at, @status.poll.id) if @status.poll
|
|
end
|
|
|
|
def validate_media!
|
|
return if @options[:media_ids].blank? || !@options[:media_ids].is_a?(Enumerable)
|
|
|
|
raise Mastodon::ValidationError, I18n.t('media_attachments.validations.too_many') if @options[:media_ids].size > 6 || @options[:poll].present?
|
|
|
|
@media = @account.media_attachments.where(status_id: nil).where(id: @options[:media_ids].take(6).map(&:to_i))
|
|
|
|
raise Mastodon::ValidationError, I18n.t('media_attachments.validations.images_and_video') if @media.size > 1 && @media.find(&:video?)
|
|
end
|
|
|
|
def language_from_option(str)
|
|
ISO_639.find(str)&.alpha2
|
|
end
|
|
|
|
def process_mentions_service
|
|
ProcessMentionsService.new
|
|
end
|
|
|
|
def process_hashtags_service
|
|
ProcessHashtagsService.new
|
|
end
|
|
|
|
def scheduled?
|
|
@scheduled_at.present?
|
|
end
|
|
|
|
def idempotency_key
|
|
"idempotency:status:#{@account.id}:#{@options[:idempotency]}"
|
|
end
|
|
|
|
def idempotency_given?
|
|
@options[:idempotency].present?
|
|
end
|
|
|
|
def idempotency_duplicate
|
|
if scheduled?
|
|
@account.schedule_statuses.find(@idempotency_duplicate)
|
|
else
|
|
@account.statuses.find(@idempotency_duplicate)
|
|
end
|
|
end
|
|
|
|
def idempotency_duplicate?
|
|
@idempotency_duplicate = redis.get(idempotency_key)
|
|
end
|
|
|
|
def scheduled_in_the_past?
|
|
@scheduled_at.present? && @scheduled_at <= Time.now.utc + MIN_SCHEDULE_OFFSET
|
|
end
|
|
|
|
def bump_potential_friendship!
|
|
return if !@status.reply? || @account.id == @status.in_reply_to_account_id
|
|
ActivityTracker.increment('activity:interactions')
|
|
return if @account.following?(@status.in_reply_to_account_id)
|
|
PotentialFriendshipTracker.record(@account.id, @status.in_reply_to_account_id, :reply)
|
|
end
|
|
|
|
def status_attributes
|
|
{
|
|
created_at: @options[:created_at] ? @options[:created_at].to_datetime.utc : Time.now.utc,
|
|
text: @text,
|
|
footer: @footer,
|
|
media_attachments: @media || [],
|
|
thread: @in_reply_to,
|
|
poll_attributes: poll_attributes,
|
|
sensitive: @sensitive,
|
|
spoiler_text: @options[:spoiler_text] || '',
|
|
visibility: @visibility,
|
|
local_only: @local_only,
|
|
sharekey: @sharekey,
|
|
language: language_from_option(@options[:language]) || @account.user_default_language&.presence || 'en',
|
|
application: @options[:application],
|
|
content_type: @options[:content_type] || @account.user&.setting_default_content_type,
|
|
}.compact
|
|
end
|
|
|
|
def scheduled_status_attributes
|
|
{
|
|
scheduled_at: @scheduled_at,
|
|
media_attachments: @media || [],
|
|
params: scheduled_options,
|
|
}
|
|
end
|
|
|
|
def poll_attributes
|
|
return if @options[:poll].blank?
|
|
|
|
@options[:poll].merge(account: @account)
|
|
end
|
|
|
|
def scheduled_options
|
|
@options.tap do |options_hash|
|
|
options_hash[:in_reply_to_id] = options_hash.delete(:thread)&.id
|
|
options_hash[:application_id] = options_hash.delete(:application)&.id
|
|
options_hash[:scheduled_at] = nil
|
|
options_hash[:idempotency] = nil
|
|
end
|
|
end
|
|
end
|