Change fetch link card service to parse as HTML5 (#31814)

This commit is contained in:
Mike Dalessio 2024-09-09 06:59:42 -04:00 committed by GitHub
parent 9d9901cc5b
commit a0ea2fa3b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 22 additions and 22 deletions

View file

@ -157,11 +157,11 @@ class LinkDetailsExtractor
end end
def title def title
html_entities_decode(structured_data&.headline || opengraph_tag('og:title') || document.xpath('//title').map(&:content).first)&.strip html_entities.decode(structured_data&.headline || opengraph_tag('og:title') || document.xpath('//title').map(&:content).first)&.strip
end end
def description def description
html_entities_decode(structured_data&.description || opengraph_tag('og:description') || meta_tag('description')) html_entities.decode(structured_data&.description || opengraph_tag('og:description') || meta_tag('description'))
end end
def published_at def published_at
@ -181,7 +181,7 @@ class LinkDetailsExtractor
end end
def provider_name def provider_name
html_entities_decode(structured_data&.publisher_name || opengraph_tag('og:site_name')) html_entities.decode(structured_data&.publisher_name || opengraph_tag('og:site_name'))
end end
def provider_url def provider_url
@ -189,7 +189,7 @@ class LinkDetailsExtractor
end end
def author_name def author_name
html_entities_decode(structured_data&.author_name || opengraph_tag('og:author') || opengraph_tag('og:author:username')) html_entities.decode(structured_data&.author_name || opengraph_tag('og:author') || opengraph_tag('og:author:username'))
end end
def author_url def author_url
@ -258,7 +258,7 @@ class LinkDetailsExtractor
next if json_ld.blank? next if json_ld.blank?
structured_data = StructuredData.new(html_entities_decode(json_ld)) structured_data = StructuredData.new(html_entities.decode(json_ld))
next unless structured_data.valid? next unless structured_data.valid?
@ -274,11 +274,20 @@ class LinkDetailsExtractor
end end
def detect_encoding_and_parse_document def detect_encoding_and_parse_document
[detect_encoding, nil, header_encoding].uniq.each do |encoding| html = nil
document = Nokogiri::HTML(@html, nil, encoding) encoding = nil
return document if document.to_s.valid_encoding?
[detect_encoding, header_encoding].compact.each do |enc|
html = @html.dup.force_encoding(enc)
if html.valid_encoding?
encoding = enc
break
end
end end
Nokogiri::HTML(@html, nil, 'UTF-8')
html = @html unless encoding
Nokogiri::HTML5(html, nil, encoding)
end end
def detect_encoding def detect_encoding
@ -299,15 +308,6 @@ class LinkDetailsExtractor
end end
end end
def html_entities_decode(string)
return if string.nil?
unicode_string = string.to_s.encode('UTF-8')
raise EncodingError, 'cannot convert string to valid UTF-8' unless unicode_string.valid_encoding?
html_entities.decode(unicode_string)
end
def html_entities def html_entities
@html_entities ||= HTMLEntities.new(:expanded) @html_entities ||= HTMLEntities.new(:expanded)
end end

View file

@ -29,7 +29,7 @@ class FetchLinkCardService < BaseService
end end
attach_card if @card&.persisted? attach_card if @card&.persisted?
rescue HTTP::Error, OpenSSL::SSL::SSLError, Addressable::URI::InvalidURIError, Mastodon::HostValidationError, Mastodon::LengthValidationError, EncodingError, ActiveRecord::RecordInvalid => e rescue HTTP::Error, OpenSSL::SSL::SSLError, Addressable::URI::InvalidURIError, Mastodon::HostValidationError, Mastodon::LengthValidationError, Encoding::UndefinedConversionError, ActiveRecord::RecordInvalid => e
Rails.logger.debug { "Error fetching link #{@original_url}: #{e}" } Rails.logger.debug { "Error fetching link #{@original_url}: #{e}" }
nil nil
end end
@ -80,7 +80,7 @@ class FetchLinkCardService < BaseService
urls = if @status.local? urls = if @status.local?
@status.text.scan(URL_PATTERN).map { |array| Addressable::URI.parse(array[1]).normalize } @status.text.scan(URL_PATTERN).map { |array| Addressable::URI.parse(array[1]).normalize }
else else
document = Nokogiri::HTML(@status.text) document = Nokogiri::HTML5(@status.text)
links = document.css('a') links = document.css('a')
links.filter_map { |a| Addressable::URI.parse(a['href']) unless skip_link?(a) }.filter_map(&:normalize) links.filter_map { |a| Addressable::URI.parse(a['href']) unless skip_link?(a) }.filter_map(&:normalize)

View file

@ -192,8 +192,8 @@ RSpec.describe FetchLinkCardService do
context 'when encoding problems appear in title tag' do context 'when encoding problems appear in title tag' do
let(:status) { Fabricate(:status, text: 'Check out http://example.com/latin1_posing_as_utf8_broken') } let(:status) { Fabricate(:status, text: 'Check out http://example.com/latin1_posing_as_utf8_broken') }
it 'does not create a preview card' do it 'creates a preview card anyway that replaces invalid bytes with U+FFFD (replacement char)' do
expect(status.preview_card).to be_nil expect(status.preview_card.title).to eq("Tofu <20> l'orange")
end end
end end
end end