70 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			70 lines
		
	
	
	
		
			1.9 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class ClearDomainMediaService < BaseService
 | |
|   attr_reader :domain_block
 | |
| 
 | |
|   def call(domain_block)
 | |
|     @domain_block = domain_block
 | |
|     clear_media! if domain_block.reject_media?
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def invalidate_association_caches!(status_ids)
 | |
|     # Normally, associated models of a status are immutable (except for accounts)
 | |
|     # so they are aggressively cached. After updating the media attachments to no
 | |
|     # longer point to a local file, we need to clear the cache to make those
 | |
|     # changes appear in the API and UI
 | |
|     Rails.cache.delete_multi(status_ids.map { |id| "statuses/#{id}" })
 | |
|   end
 | |
| 
 | |
|   def clear_media!
 | |
|     clear_account_images!
 | |
|     clear_account_attachments!
 | |
|     clear_emojos!
 | |
|   end
 | |
| 
 | |
|   def clear_account_images!
 | |
|     blocked_domain_accounts.reorder(nil).find_each do |account|
 | |
|       account.avatar.destroy if account.avatar&.exists?
 | |
|       account.header.destroy if account.header&.exists?
 | |
|       account.save
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def clear_account_attachments!
 | |
|     media_from_blocked_domain.reorder(nil).find_in_batches do |attachments|
 | |
|       affected_status_ids = []
 | |
| 
 | |
|       attachments.each do |attachment|
 | |
|         affected_status_ids << attachment.status_id if attachment.status_id.present?
 | |
| 
 | |
|         attachment.file.destroy if attachment.file&.exists?
 | |
|         attachment.type = :unknown
 | |
|         attachment.save
 | |
|       end
 | |
| 
 | |
|       invalidate_association_caches!(affected_status_ids) unless affected_status_ids.empty?
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def clear_emojos!
 | |
|     emojis_from_blocked_domains.destroy_all
 | |
|   end
 | |
| 
 | |
|   def blocked_domain
 | |
|     domain_block.domain
 | |
|   end
 | |
| 
 | |
|   def blocked_domain_accounts
 | |
|     Account.by_domain_and_subdomains(blocked_domain)
 | |
|   end
 | |
| 
 | |
|   def media_from_blocked_domain
 | |
|     MediaAttachment.joins(:account).merge(blocked_domain_accounts).reorder(nil)
 | |
|   end
 | |
| 
 | |
|   def emojis_from_blocked_domains
 | |
|     CustomEmoji.by_domain_and_subdomains(blocked_domain)
 | |
|   end
 | |
| end
 |