mirror of https://github.com/mastodon/mastodon
Add coverage for `URLValidator` (#25591)
parent
6602edf064
commit
660993b415
@ -1,16 +1,31 @@
|
|||||||
# frozen_string_literal: true
|
# frozen_string_literal: true
|
||||||
|
|
||||||
class URLValidator < ActiveModel::EachValidator
|
class URLValidator < ActiveModel::EachValidator
|
||||||
|
VALID_SCHEMES = %w(http https).freeze
|
||||||
|
|
||||||
def validate_each(record, attribute, value)
|
def validate_each(record, attribute, value)
|
||||||
record.errors.add(attribute, :invalid) unless compliant?(value)
|
@value = value
|
||||||
|
|
||||||
|
record.errors.add(attribute, :invalid) unless compliant_url?
|
||||||
end
|
end
|
||||||
|
|
||||||
private
|
private
|
||||||
|
|
||||||
def compliant?(url)
|
def compliant_url?
|
||||||
parsed_url = Addressable::URI.parse(url)
|
parsed_url.present? && valid_url_scheme? && valid_url_host?
|
||||||
parsed_url && %w(http https).include?(parsed_url.scheme) && parsed_url.host
|
end
|
||||||
|
|
||||||
|
def parsed_url
|
||||||
|
Addressable::URI.parse(@value)
|
||||||
rescue Addressable::URI::InvalidURIError
|
rescue Addressable::URI::InvalidURIError
|
||||||
false
|
false
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def valid_url_scheme?
|
||||||
|
VALID_SCHEMES.include?(parsed_url.scheme)
|
||||||
|
end
|
||||||
|
|
||||||
|
def valid_url_host?
|
||||||
|
parsed_url.host.present?
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue