# frozen_string_literal: true require 'singleton' require_relative './sanitize_config' class Formatter include Singleton include RoutingHelper include ActionView::Helpers::TextHelper def format(status) return reformat(status.content) unless status.local? html = status.text html = encode_and_link_urls(html) html = simple_format(html, {}, sanitize: false) html = html.delete("\n") html = link_mentions(html, status.mentions) html = link_hashtags(html) html.html_safe # rubocop:disable Rails/OutputSafety end def reformat(html) sanitize(html, Sanitize::Config::MASTODON_STRICT).html_safe # rubocop:disable Rails/OutputSafety end def plaintext(status) return status.text if status.local? strip_tags(status.text) end def simplified_format(account) return reformat(account.note) unless account.local? html = encode_and_link_urls(account.note) html = simple_format(html, {}, sanitize: false) html = html.delete("\n") html = link_accounts(html) html = link_hashtags(html) html.html_safe # rubocop:disable Rails/OutputSafety end def sanitize(html, config) Sanitize.fragment(html, config) end private def encode(html) HTMLEntities.new.encode(html) end def encode_and_link_urls(html) entities = Twitter::Extractor.extract_urls_with_indices(html, extract_url_without_protocol: false) entities = entities.sort_by { |entity| entity[:indices].first } chars = html.to_s.to_char_a html_attrs = { target: '_blank', rel: 'nofollow noopener', } result = '' last_index = entities.reduce(0) do |index, entity| normalized_url = Addressable::URI.parse(entity[:url]).normalize indices = entity[:indices] result += encode(chars[index...indices.first].join) result += Twitter::Autolink.send(:link_to_text, entity, link_html(entity[:url]), normalized_url, html_attrs) indices.last end result += encode(chars[last_index..-1].join) end def link_mentions(html, mentions) html.gsub(Account::MENTION_RE) do |match| acct = Account::MENTION_RE.match(match)[1] mention = mentions.find { |item| TagManager.instance.same_acct?(item.account.acct, acct) } mention.nil? ? match : mention_html(match, mention.account) end end def link_accounts(html) html.gsub(Account::MENTION_RE) do |match| acct = Account::MENTION_RE.match(match)[1] username, domain = acct.split('@') domain = nil if TagManager.instance.local_domain?(domain) account = Account.find_remote(username, domain) account.nil? ? match : mention_html(match, account) end end def link_hashtags(html) html.gsub(Tag::HASHTAG_RE) do |match| hashtag_html(match) end end def link_html(url) url = Addressable::URI.parse(url).display_uri.to_s prefix = url.match(/\Ahttps?:\/\/(www\.)?/).to_s text = url[prefix.length, 30] suffix = url[prefix.length + 30..-1] cutoff = url[prefix.length..-1].length > 30 "#{prefix}#{text}#{suffix}" end def hashtag_html(match) prefix, _, affix = match.rpartition('#') "#{prefix}##{affix}" end def mention_html(match, account) "#{match.split('@').first}@#{account.username}" end end