86 lines
		
	
	
	
		
			2.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			86 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("timeline:hashtag:#{hashtag}", @payload)
 | |
|       Redis.current.publish("timeline:hashtag:#{hashtag}:local", @payload) if status.local?
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def remove_from_public(status)
 | |
|     Redis.current.publish('timeline:public', @payload)
 | |
|     Redis.current.publish('timeline:public:local', @payload) if status.local?
 | |
|   end
 | |
| 
 | |
|   def redis
 | |
|     Redis.current
 | |
|   end
 | |
| end
 |