mirror of
https://github.com/glitch-soc/mastodon.git
synced 2024-11-25 01:24:02 -05:00
dbd529109e
slightly optimized FanOutOnWriteService again
87 lines
2.3 KiB
Ruby
87 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class RemoveStatusService < BaseService
|
|
include StreamEntryRenderer
|
|
|
|
def call(status)
|
|
@payload = Oj.dump(event: :delete, payload: status.id)
|
|
|
|
remove_from_self(status) if status.account.local?
|
|
remove_from_followers(status)
|
|
remove_from_mentioned(status)
|
|
remove_reblogs(status)
|
|
remove_from_hashtags(status)
|
|
remove_from_public(status)
|
|
|
|
status.destroy!
|
|
|
|
return unless status.account.local?
|
|
|
|
Pubsubhubbub::DistributionWorker.perform_async(status.stream_entry.id)
|
|
end
|
|
|
|
private
|
|
|
|
def remove_from_self(status)
|
|
unpush(:home, status.account, status)
|
|
end
|
|
|
|
def remove_from_followers(status)
|
|
status.account.followers.where(domain: nil).each do |follower|
|
|
unpush(:home, follower, status)
|
|
end
|
|
end
|
|
|
|
def remove_from_mentioned(status)
|
|
return unless status.local?
|
|
notified_domains = []
|
|
|
|
status.mentions.each do |mention|
|
|
mentioned_account = mention.account
|
|
|
|
next if mentioned_account.local?
|
|
next if notified_domains.include?(mentioned_account.domain)
|
|
|
|
notified_domains << mentioned_account.domain
|
|
send_delete_salmon(mentioned_account, status)
|
|
end
|
|
end
|
|
|
|
def send_delete_salmon(account, status)
|
|
return unless status.local?
|
|
NotificationWorker.perform_async(stream_entry_to_xml(status.stream_entry), status.account_id, account.id)
|
|
end
|
|
|
|
def remove_reblogs(status)
|
|
status.reblogs.each do |reblog|
|
|
RemoveStatusService.new.call(reblog)
|
|
end
|
|
end
|
|
|
|
def unpush(type, receiver, status)
|
|
if status.reblog? && !redis.zscore(FeedManager.instance.key(type, receiver.id), status.reblog_of_id).nil?
|
|
redis.zadd(FeedManager.instance.key(type, receiver.id), status.reblog_of_id, status.reblog_of_id)
|
|
else
|
|
redis.zremrangebyscore(FeedManager.instance.key(type, receiver.id), status.id, status.id)
|
|
end
|
|
|
|
Redis.current.publish("timeline:#{receiver.id}", @payload)
|
|
end
|
|
|
|
def remove_from_hashtags(status)
|
|
status.tags.pluck(:name) do |hashtag|
|
|
Redis.current.publish("hashtag:#{hashtag}", @payload)
|
|
Redis.current.publish("hashtag:#{hashtag}:local", @payload) if status.local?
|
|
end
|
|
end
|
|
|
|
def remove_from_public(status)
|
|
Redis.current.publish('public', @payload)
|
|
Redis.current.publish('public:local', @payload) if status.local?
|
|
end
|
|
|
|
def redis
|
|
Redis.current
|
|
end
|
|
end
|