gitlab-org--gitlab-foss/lib/banzai/filter/external_link_filter.rb

49 lines
1.1 KiB
Ruby

module Banzai
module Filter
# HTML Filter to modify the attributes of external links
class ExternalLinkFilter < HTML::Pipeline::Filter
def call
links.each do |node|
href = href_to_lowercase_scheme(node["href"].to_s)
unless node["href"].to_s == href
node.set_attribute('href', href)
end
if href =~ %r{\A(https?:)?//[^/]} && external_url?(href)
node.set_attribute('rel', 'nofollow noreferrer')
node.set_attribute('target', '_blank')
end
end
doc
end
private
def links
query = 'descendant-or-self::a[@href and not(@href = "")]'
doc.xpath(query)
end
def href_to_lowercase_scheme(href)
scheme_match = href.match(/\A(\w+):\/\//)
if scheme_match
scheme_match.to_s.downcase + scheme_match.post_match
else
href
end
end
def external_url?(url)
!url.start_with?(internal_url)
end
def internal_url
@internal_url ||= Gitlab.config.gitlab.url
end
end
end
end