FIX: make get_hostname more lenient to user input

This commit is contained in:
Régis Hanol 2018-04-12 17:09:09 +02:00
parent f042a9529b
commit a0a06492d8
2 changed files with 30 additions and 10 deletions

View File

@ -252,14 +252,18 @@ module SiteSettingExtension
refresh_settings.include?(name.to_sym)
end
HOSTNAME_SETTINGS ||= %w{
disabled_image_download_domains onebox_domains_blacklist exclude_rel_nofollow_domains
email_domains_blacklist email_domains_whitelist white_listed_spam_host_domains
}
def filter_value(name, value)
if %w[disabled_image_download_domains onebox_domains_blacklist exclude_rel_nofollow_domains email_domains_blacklist email_domains_whitelist white_listed_spam_host_domains].include? name
domain_array = []
value.split('|').each { |url| domain_array << get_hostname(url) }
value = domain_array.join("|")
end
if HOSTNAME_SETTINGS.include?(name)
value.split("|").map { |url| get_hostname(url) }.compact.uniq.join("|")
else
value
end
end
def set(name, value)
if has_setting?(name)
@ -339,16 +343,21 @@ module SiteSettingExtension
end
def get_hostname(url)
uri = begin
URI.parse(url)
url.strip!
host = begin
URI.parse(url)&.host
rescue URI::InvalidURIError
nil
end
unless uri.scheme.nil?
url = uri.host
host ||= begin
URI.parse("http://#{url}")&.host
rescue URI::InvalidURIError
nil
end
url
host.presence || url
end
private

View File

@ -586,4 +586,15 @@ describe SiteSettingExtension do
end
end
describe "get_hostname" do
it "properly extracts the hostname" do
expect(settings.send(:get_hostname, "discourse.org")).to eq("discourse.org")
expect(settings.send(:get_hostname, " discourse.org ")).to eq("discourse.org")
expect(settings.send(:get_hostname, "@discourse.org")).to eq("discourse.org")
expect(settings.send(:get_hostname, "https://discourse.org")).to eq("discourse.org")
end
end
end