@@ -15,7 +15,7 @@ class AccountDomainBlock < ApplicationRecord
 | 
			
		||||
  include DomainNormalizable
 | 
			
		||||
 | 
			
		||||
  belongs_to :account
 | 
			
		||||
  validates :domain, presence: true, uniqueness: { scope: :account_id }
 | 
			
		||||
  validates :domain, presence: true, uniqueness: { scope: :account_id }, domain: true
 | 
			
		||||
 | 
			
		||||
  after_commit :remove_blocking_cache
 | 
			
		||||
  after_commit :remove_relationship_cache
 | 
			
		||||
 
 | 
			
		||||
@@ -4,7 +4,7 @@ module DomainNormalizable
 | 
			
		||||
  extend ActiveSupport::Concern
 | 
			
		||||
 | 
			
		||||
  included do
 | 
			
		||||
    before_validation :normalize_domain
 | 
			
		||||
    before_save :normalize_domain
 | 
			
		||||
  end
 | 
			
		||||
 | 
			
		||||
  private
 | 
			
		||||
 
 | 
			
		||||
@@ -13,7 +13,7 @@
 | 
			
		||||
class DomainAllow < ApplicationRecord
 | 
			
		||||
  include DomainNormalizable
 | 
			
		||||
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true, domain: true
 | 
			
		||||
 | 
			
		||||
  scope :matches_domain, ->(value) { where(arel_table[:domain].matches("%#{value}%")) }
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -19,7 +19,7 @@ class DomainBlock < ApplicationRecord
 | 
			
		||||
 | 
			
		||||
  enum severity: [:silence, :suspend, :noop]
 | 
			
		||||
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true, domain: true
 | 
			
		||||
 | 
			
		||||
  has_many :accounts, foreign_key: :domain, primary_key: :domain
 | 
			
		||||
  delegate :count, to: :accounts, prefix: true
 | 
			
		||||
 
 | 
			
		||||
@@ -12,7 +12,7 @@
 | 
			
		||||
class EmailDomainBlock < ApplicationRecord
 | 
			
		||||
  include DomainNormalizable
 | 
			
		||||
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true
 | 
			
		||||
  validates :domain, presence: true, uniqueness: true, domain: true
 | 
			
		||||
 | 
			
		||||
  def self.block?(email)
 | 
			
		||||
    _, domain = email.split('@', 2)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										17
									
								
								app/validators/domain_validator.rb
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										17
									
								
								app/validators/domain_validator.rb
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,17 @@
 | 
			
		||||
# frozen_string_literal: true
 | 
			
		||||
 | 
			
		||||
class DomainValidator < ActiveModel::EachValidator
 | 
			
		||||
  def validate_each(record, attribute, value)
 | 
			
		||||
    return if value.blank?
 | 
			
		||||
 | 
			
		||||
    record.errors.add(attribute, I18n.t('domain_validator.invalid_domain')) unless compliant?(value)
 | 
			
		||||
  end
 | 
			
		||||
 | 
			
		||||
  private
 | 
			
		||||
 | 
			
		||||
  def compliant?(value)
 | 
			
		||||
    Addressable::URI.new.tap { |uri| uri.host = value }
 | 
			
		||||
  rescue Addressable::URI::InvalidURIError
 | 
			
		||||
    false
 | 
			
		||||
  end
 | 
			
		||||
end
 | 
			
		||||
		Reference in New Issue
	
	Block a user