tootlab-mastodon/app/services/block_domain_service.rb
Patrick Figel 197af5de70 Set unknown attachment type when adding domain blocks (#2605)
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`.
2017-04-29 02:44:03 +02:00

66 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