mirror of
https://github.com/mastodon/mastodon.git
synced 2024-12-30 04:14:36 +00:00
31 lines
934 B
Ruby
31 lines
934 B
Ruby
# frozen_string_literal: true
|
|
|
|
class DomainValidator < ActiveModel::EachValidator
|
|
MAX_DOMAIN_LENGTH = 256
|
|
MIN_LABEL_LENGTH = 1
|
|
MAX_LABEL_LENGTH = 63
|
|
ALLOWED_CHARACTERS_RE = /^[a-z0-9-]+$/i
|
|
|
|
def validate_each(record, attribute, value)
|
|
return if value.blank?
|
|
|
|
(options[:multiline] ? value.split : [value]).each do |domain|
|
|
_, domain = domain.split('@') if options[:acct]
|
|
|
|
next if domain.blank?
|
|
|
|
record.errors.add(attribute, options[:multiline] ? :invalid_domain_on_line : :invalid, value: domain) unless compliant?(domain)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def compliant?(value)
|
|
uri = Addressable::URI.new
|
|
uri.host = value
|
|
uri.normalized_host.size < MAX_DOMAIN_LENGTH && uri.normalized_host.split('.').all? { |label| label.size.between?(MIN_LABEL_LENGTH, MAX_LABEL_LENGTH) && label =~ ALLOWED_CHARACTERS_RE }
|
|
rescue Addressable::URI::InvalidURIError, IDN::Idna::IdnaError
|
|
false
|
|
end
|
|
end
|