1 files changed, 40 insertions, 14 deletions
diff --git a/app/services/block_domain_service.rb b/app/services/block_domain_service.rb
index 97d2ebcd7..5d13cfc0e 100644
--- a/app/services/block_domain_service.rb
+++ b/app/services/block_domain_service.rb
@@ -1,36 +1,62 @@
 # 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!(domain_block.domain)
-      clear_media!(domain_block.domain) if domain_block.reject_media?
+      silence_accounts!
     else
-      suspend_accounts!(domain_block.domain)
+      suspend_accounts!
     end
   end
 
-  private
+  def silence_accounts!
+    blocked_domain_accounts.update_all(silenced: true)
+    clear_media! if domain_block.reject_media?
+  end
 
-  def silence_accounts!(domain)
-    Account.where(domain: domain).update_all(silenced: true)
+  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_media!(domain)
-    Account.where(domain: domain).find_each do |account|
+  def clear_account_images
+    blocked_domain_accounts.find_each do |account|
       account.avatar.destroy
       account.header.destroy
     end
+  end
 
-    MediaAttachment.where(account: Account.where(domain: domain)).find_each do |attachment|
+  def clear_account_attachments
+    media_from_blocked_domain.find_each do |attachment|
       attachment.file.destroy
     end
   end
 
-  def suspend_accounts!(domain)
-    Account.where(domain: domain).where(suspended: false).find_each do |account|
-      account.subscription(api_subscription_url(account.id)).unsubscribe if account.subscribed?
-      SuspendAccountService.new.call(account)
-    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