Follow-up to #2599. When a domain block with `reject_media` is added or `rake mastodon:media:remove_remote` is invoked, mastodon deletes the locally cached attachments and avatars but does not reflect that change in the database, causing the `file` fields to still have values. This change persists the deletion in the database and sets the attachment type to unknown. This also introduces a one-off rake task that sets all attachments without a local file to the "unknown" type. The upgrade notes for the next release should contain a post-upgrade step with `rake mastodon:media:set_unknown`.
		
			
				
	
	
		
			65 lines
		
	
	
	
		
			1.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			65 lines
		
	
	
	
		
			1.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class BlockDomainService < BaseService
 | |
|   attr_reader :domain_block
 | |
| 
 | |
|   def call(domain_block)
 | |
|     @domain_block = domain_block
 | |
|     process_domain_block
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def process_domain_block
 | |
|     if domain_block.silence?
 | |
|       silence_accounts!
 | |
|     else
 | |
|       suspend_accounts!
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def silence_accounts!
 | |
|     blocked_domain_accounts.update_all(silenced: true)
 | |
|     clear_media! if domain_block.reject_media?
 | |
|   end
 | |
| 
 | |
|   def clear_media!
 | |
|     clear_account_images
 | |
|     clear_account_attachments
 | |
|   end
 | |
| 
 | |
|   def suspend_accounts!
 | |
|     blocked_domain_accounts.where(suspended: false).find_each do |account|
 | |
|       account.subscription(api_subscription_url(account.id)).unsubscribe if account.subscribed?
 | |
|       SuspendAccountService.new.call(account)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def clear_account_images
 | |
|     blocked_domain_accounts.find_each do |account|
 | |
|       account.avatar.destroy
 | |
|       account.header.destroy
 | |
|       account.save
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def clear_account_attachments
 | |
|     media_from_blocked_domain.find_each do |attachment|
 | |
|       attachment.file.destroy
 | |
|       attachment.type = :unknown
 | |
|       attachment.save
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def blocked_domain
 | |
|     domain_block.domain
 | |
|   end
 | |
| 
 | |
|   def blocked_domain_accounts
 | |
|     Account.where(domain: blocked_domain)
 | |
|   end
 | |
| 
 | |
|   def media_from_blocked_domain
 | |
|     MediaAttachment.where(account: blocked_domain_accounts).reorder(nil)
 | |
|   end
 | |
| end
 |