demarches-normaliennes/app/validators/url_validator.rb
2024-08-22 09:26:48 +02:00

75 lines
2.5 KiB
Ruby

# frozen_string_literal: true
require 'active_model'
require 'active_support/i18n'
require 'public_suffix'
require 'addressable/uri'
# Most of this code is borowed from https://github.com/perfectline/validates_url
class URLValidator < ActiveModel::EachValidator
RESERVED_OPTIONS = [:schemes, :no_local]
def initialize(options)
options.reverse_merge!(schemes: ['http', 'https'])
options.reverse_merge!(message: :url)
options.reverse_merge!(no_local: false)
options.reverse_merge!(public_suffix: false)
options.reverse_merge!(accept_array: false)
options.reverse_merge!(accept_email: false)
super(options)
end
def validate_each(record, attribute, value)
message = options.fetch(:message)
schemes = [*options.fetch(:schemes)].map(&:to_s)
if value.respond_to?(:each)
# Error out if we're not allowing arrays
if !options.include?(:accept_array) || !options.fetch(:accept_array)
record.errors.add(attribute, message, **filtered_options(value))
end
# We have to manually handle `:allow_nil` and `:allow_blank` since it's not caught by
# ActiveRecord's own validators. We do that by just removing all the nil's if we want to
# allow them so it's not passed on later.
value = value.compact if options.include?(:allow_nil) && options.fetch(:allow_nil)
value = value.compact_blank if options.include?(:allow_blank) && options.fetch(:allow_blank)
result = value.flat_map { validate_url(record, attribute, _1, message, schemes) }
errors = result.compact
return errors.any? ? errors.first : true
end
validate_url(record, attribute, value, message, schemes)
end
protected
def filtered_options(value)
filtered = options.except(*RESERVED_OPTIONS)
filtered[:value] = value
filtered
end
def validate_url(record, attribute, value, message, schemes)
uri = Addressable::URI.parse(value)
unless options.fetch(:accept_email) && uri.path.match?(/^(.+)@(.+)$/)
host = uri && uri.host
scheme = uri && uri.scheme
valid_scheme = host && scheme && schemes.include?(scheme)
valid_no_local = !options.fetch(:no_local) || (host && host.include?('.'))
valid_suffix = !options.fetch(:public_suffix) || (host && PublicSuffix.valid?(host, default_rule: nil))
unless valid_scheme && valid_no_local && valid_suffix
record.errors.add(attribute, message, **filtered_options(value))
end
end
rescue Addressable::URI::InvalidURIError
record.errors.add(attribute, message, **filtered_options(value))
end
end