about summary refs log tree commit diff
path: root/app/services/block_domain_service.rb
blob: 36634fdd5ed1a626312e3f367e8dacc0888738e4 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
# frozen_string_literal: true

class BlockDomainService < BaseService
  attr_reader :domain_block

  def call(domain_block)
    @domain_block = domain_block
    @affected_status_ids = []

    remove_existing_block!
    process_domain_block!
    invalidate_association_caches!

    @domain_block.update!(processing: false)
  end

  private

  def remove_existing_block!
    UnblockDomainService.new.call(@domain_block, false)
  end

  def process_domain_block!
    clear_media! if domain_block.reject_media? || domain_block.suspend?
    force_accounts_sensitive! if domain_block.force_sensitive?
    mark_unknown_accounts! if domain_block.reject_unknown?
    mark_accounts_manual_only! if domain_block.manual_only?

    if domain_block.force_unlisted?
      force_accounts_unlisted!
    elsif domain_block.silence?
      silence_accounts!
    elsif domain_block.suspend?
      suspend_accounts!
    end
  end

  def invalidate_association_caches!
    # 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
    @affected_status_ids.each { |id| Rails.cache.delete_matched("statuses/#{id}-*") }
  end

  def force_accounts_sensitive!
    ApplicationRecord.transaction do
      blocked_domain_accounts.in_batches.update_all(force_sensitive: true)
      blocked_domain_accounts.find_each do |account|
        @affected_status_ids |= account.statuses.where(sensitive: false).pluck(:id)
        account.statuses.where(sensitive: false).in_batches.update_all(sensitive: true)
      end
    end
  end

  def mark_accounts_manual_only!
    blocked_domain_accounts.in_batches.update_all(manual_only: true)
  end

  def mark_unknown_accounts!
    ApplicationRecord.transaction do
      unknown_accounts.in_batches.update_all(known: false)
      unknown_accounts.find_each do |account|
        account.avatar = nil
        account.header = nil
        account.save!
      end
    end
  end

  def force_accounts_unlisted!
    ApplicationRecord.transaction do
      blocked_domain_accounts.in_batches.update_all(force_unlisted: true)
      blocked_domain_accounts.find_each do |account|
        @affected_status_ids |= account.statuses.with_public_visibility.pluck(:id)
        account.statuses.with_public_visibility.in_batches.update_all(visibility: :unlisted)
      end
    end
  end

  def silence_accounts!
    blocked_domain_accounts.without_silenced.in_batches.update_all(silenced_at: @domain_block.created_at)
  end

  def clear_media!

    clear_account_images!
    clear_account_attachments!
    clear_emojos!

  end

  def suspend_accounts!
    blocked_domain_accounts.without_suspended.find_each do |account|
      SuspendAccountService.new.call(account, suspended_at: @domain_block.created_at)
    end
  end

  def clear_account_images!
    blocked_domain_accounts.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.find_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
  end

  def clear_emojos!
    emojis_from_blocked_domains.destroy_all
  end

  def blocked_domain
    domain_block.domain
  end

  def blocked_domain_accounts
    Account.where(domain: blocked_domain).reorder(nil)
  end

  def media_from_blocked_domain
    MediaAttachment.joins(:account).merge(blocked_domain_accounts).reorder(nil)
  end

  def emojis_from_blocked_domains
    CustomEmoji.where(domain: blocked_domain)
  end

  def unknown_accounts
    Account.where(id: blocked_domain_accounts.pluck(:id) - known_account_ids).reorder(nil)
  end

  def known_account_ids
    local_accounts | packmates | boosted_authors
  end

  def boosted_authors
    Status.where(id: Status.local.reblogs.reorder(nil).select(:reblog_of_id)).reorder(nil).pluck(:account_id)
  end

  def local_accounts
    Account.local.pluck(:id)
  end

  def packmates
    Account.local.flat_map { |account| account.following_ids | account.follower_ids }
  end
end