openstreetmap-website/lib/rich_text.rb
2024-08-02 20:19:10 +03:00

129 lines
2.7 KiB
Ruby

module RichText
SPAMMY_PHRASES = [
"Business Description:", "Additional Keywords:"
].freeze
def self.new(format, text)
case format
when "html" then HTML.new(text || "")
when "markdown" then Markdown.new(text || "")
when "text" then Text.new(text || "")
end
end
class SimpleFormat
include ActionView::Helpers::TextHelper
include ActionView::Helpers::OutputSafetyHelper
def sanitize(text, _options = {})
Sanitize.clean(text, Sanitize::Config::OSM).html_safe
end
end
class Base < String
include ActionView::Helpers::TagHelper
def spam_score
link_count = 0
link_size = 0
doc = Nokogiri::HTML(to_html)
if doc.content.empty?
link_proportion = 0
else
doc.xpath("//a").each do |link|
link_count += 1
link_size += link.content.length
end
link_proportion = link_size.to_f / doc.content.length
end
spammy_phrases = SPAMMY_PHRASES.count do |phrase|
doc.content.include?(phrase)
end
([link_proportion - 0.2, 0.0].max * 200) +
(link_count * 40) +
(spammy_phrases * 40)
end
def image
nil
end
protected
def simple_format(text)
SimpleFormat.new.simple_format(text)
end
def sanitize(text)
Sanitize.clean(text, Sanitize::Config::OSM).html_safe
end
def linkify(text, mode = :urls)
if text.html_safe?
Rinku.auto_link(text, mode, tag_builder.tag_options(:rel => "nofollow noopener noreferrer")).html_safe
else
Rinku.auto_link(text, mode, tag_builder.tag_options(:rel => "nofollow noopener noreferrer"))
end
end
end
class HTML < Base
def to_html
linkify(sanitize(simple_format(self)))
end
def to_text
to_s
end
end
class Markdown < Base
def to_html
linkify(sanitize(document.to_html), :all)
end
def to_text
to_s
end
def image
return @image if defined? @image
@image = first_image_element(document.root)&.attr&.[]("src")
end
private
def document
@document ||= Kramdown::Document.new(self)
end
def first_image_element(element)
return element if image?(element) && element.attr["src"].present?
element.children.find do |child|
nested_image = first_image_element(child)
break nested_image if nested_image
end
end
def image?(element)
element.type == :img || (element.type == :html_element && element.value == "img")
end
end
class Text < Base
def to_html
linkify(simple_format(ERB::Util.html_escape(self)))
end
def to_text
to_s
end
end
end