forked from asonix/mastodon
* Change e-mail domain blocks to block IPs dynamically * Update app/workers/scheduler/email_domain_block_refresh_scheduler.rb Co-authored-by: Yamagishi Kazutoshi <ykzts@desire.sh> * Update app/workers/scheduler/email_domain_block_refresh_scheduler.rb Co-authored-by: Yamagishi Kazutoshi <ykzts@desire.sh> Co-authored-by: Yamagishi Kazutoshi <ykzts@desire.sh>
61 lines
1.7 KiB
Ruby
61 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
# == Schema Information
|
|
#
|
|
# Table name: email_domain_blocks
|
|
#
|
|
# id :bigint(8) not null, primary key
|
|
# domain :string default(""), not null
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# parent_id :bigint(8)
|
|
# ips :inet is an Array
|
|
# last_refresh_at :datetime
|
|
#
|
|
|
|
class EmailDomainBlock < ApplicationRecord
|
|
include DomainNormalizable
|
|
|
|
belongs_to :parent, class_name: 'EmailDomainBlock', optional: true
|
|
has_many :children, class_name: 'EmailDomainBlock', foreign_key: :parent_id, inverse_of: :parent, dependent: :destroy
|
|
|
|
validates :domain, presence: true, uniqueness: true, domain: true
|
|
|
|
# Used for adding multiple blocks at once
|
|
attr_accessor :other_domains
|
|
|
|
def history
|
|
@history ||= Trends::History.new('email_domain_blocks', id)
|
|
end
|
|
|
|
def self.block?(domain_or_domains, ips: [], attempt_ip: nil)
|
|
domains = Array(domain_or_domains).map do |str|
|
|
domain = begin
|
|
if str.include?('@')
|
|
str.split('@', 2).last
|
|
else
|
|
str
|
|
end
|
|
end
|
|
|
|
TagManager.instance.normalize_domain(domain) if domain.present?
|
|
rescue Addressable::URI::InvalidURIError
|
|
nil
|
|
end
|
|
|
|
# If some of the inputs passed in are invalid, we definitely want to
|
|
# block the attempt, but we also want to register hits against any
|
|
# other valid matches
|
|
|
|
blocked = domains.any?(&:nil?)
|
|
|
|
scope = where(domain: domains)
|
|
scope = scope.or(where('ips && ARRAY[?]::inet[]', ips)) if ips.any?
|
|
|
|
scope.find_each do |block|
|
|
blocked = true
|
|
block.history.add(attempt_ip) if attempt_ip.present?
|
|
end
|
|
|
|
blocked
|
|
end
|
|
end
|