Domain block service cleanup (#2490)

* Add coverage for domain block service with silence

* Get rid of warning about find_each and order

* Move domain_block to attr_reader

* Move optional clear_media into silence_accounts method

* Use blocked_domain method to reduce passed vars

* Extract blocked_domain_accounts method to find accounts on the domain

* Extract media_from_blocked_domain method to find relevant attachments

* Separate destruction of account images and account attachments
This commit is contained in:
Matt Jankowski 2017-04-26 14:09:01 -04:00 committed by Eugen Rochko
parent affd75936e
commit 8857cabca4
2 changed files with 82 additions and 31 deletions

View File

@ -1,36 +1,62 @@
# frozen_string_literal: true # frozen_string_literal: true
class BlockDomainService < BaseService class BlockDomainService < BaseService
attr_reader :domain_block
def call(domain_block) def call(domain_block)
if domain_block.silence? @domain_block = domain_block
silence_accounts!(domain_block.domain) process_domain_block
clear_media!(domain_block.domain) if domain_block.reject_media?
else
suspend_accounts!(domain_block.domain)
end
end end
private private
def silence_accounts!(domain) def process_domain_block
Account.where(domain: domain).update_all(silenced: true) if domain_block.silence?
end silence_accounts!
else
def clear_media!(domain) suspend_accounts!
Account.where(domain: domain).find_each do |account|
account.avatar.destroy
account.header.destroy
end
MediaAttachment.where(account: Account.where(domain: domain)).find_each do |attachment|
attachment.file.destroy
end end
end end
def suspend_accounts!(domain) def silence_accounts!
Account.where(domain: domain).where(suspended: false).find_each do |account| 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? account.subscription(api_subscription_url(account.id)).unsubscribe if account.subscribed?
SuspendAccountService.new.call(account) SuspendAccountService.new.call(account)
end end
end end
def clear_account_images
blocked_domain_accounts.find_each do |account|
account.avatar.destroy
account.header.destroy
end
end
def clear_account_attachments
media_from_blocked_domain.find_each do |attachment|
attachment.file.destroy
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 end

View File

@ -13,7 +13,10 @@ RSpec.describe BlockDomainService do
bad_status1 bad_status1
bad_status2 bad_status2
bad_attachment bad_attachment
end
describe 'for a suspension' do
before do
subject.call(DomainBlock.create!(domain: 'evil.org', severity: :suspend)) subject.call(DomainBlock.create!(domain: 'evil.org', severity: :suspend))
end end
@ -31,3 +34,25 @@ RSpec.describe BlockDomainService do
expect { bad_attachment.reload }.to raise_exception ActiveRecord::RecordNotFound expect { bad_attachment.reload }.to raise_exception ActiveRecord::RecordNotFound
end end
end end
describe 'for a silence with reject media' do
before do
subject.call(DomainBlock.create!(domain: 'evil.org', severity: :silence, reject_media: true))
end
it 'does not create a domain block' do
expect(DomainBlock.blocked?('evil.org')).to be false
end
it 'silences remote accounts from that domain' do
expect(Account.find_remote('badguy666', 'evil.org').silenced?).to be true
end
it 'leaves the domains status and attachements, but clears media' do
expect { bad_status1.reload }.not_to raise_error
expect { bad_status2.reload }.not_to raise_error
expect { bad_attachment.reload }.not_to raise_error
expect(bad_attachment.file.exists?).to be false
end
end
end