soffes/quesadilla

# encoding: UTF-8

module Quesadilla
  class Extractor
    # Convert entites and entire string to HTML.
    #
    # This module has no public methods.
    module HTML
    private

      HTML_ESCAPE_MAP = [
        {
           pattern: '&',
           text: '&',
           placeholder: "\uf050",
        },
        {
           pattern: '<',
           text: '&lt;',
           placeholder: "\uf051",
        },
        {
           pattern: '>',
           text: '&gt;',
           placeholder: "\uf052",
        },
        {
           pattern: '"',
           text: '&quot;',
           placeholder: "\uf053",
        },
        {
           pattern: '\'',
           text: '&#x27;',
           placeholder: "\uf054",
        },
        {
           pattern: '/',
           text: '&#x2F;',
           placeholder: "\uf055",
        }
      ].freeze

      def display_html(display_text, entities)
         return html_escape(display_text) unless entities and entities.length > 0

         # Replace entities
         html = sub_entities(display_text, entities, true) do |entity|
           html_entity(entity)
        end

         # Return
         @renderer.post_process(html_un_pre_escape(html))
      end

      def html_entity(entity)
        display_text = html_pre_escape(entity[:display_text])
        case entity[:type]
        when ENTITY_TYPE_EMPHASIS
          @renderer.emphasis(display_text)
        when ENTITY_TYPE_DOUBLE_EMPHASIS
          @renderer.double_emphasis(display_text)
        when ENTITY_TYPE_TRIPLE_EMPHASIS
          @renderer.triple_emphasis(display_text)
        when ENTITY_TYPE_STRIKETHROUGH
          @renderer.strikethrough(display_text)
        when ENTITY_TYPE_CODE
          @renderer.code(display_text)
        when ENTITY_TYPE_HASHTAG
          @renderer.hashtag(display_text, html_pre_escape(entity[:hashtag]))
        when ENTITY_TYPE_USER
          @renderer.user(display_text, html_pre_escape(entity[:username]), html_pre_escape(entity[:user_id]))
        when ENTITY_TYPE_LINK
          @renderer.link(display_text, entity[:url], html_pre_escape(entity[:title]))
        else
          # Catchall
          html_pre_escape(entity[:text])
        end
      end

      # Pre-escape. Convert bad characters to high UTF-8 characters
      # We do this dance so we don't throw off the indexes so the entities get inserted correctly.
      def html_pre_escape(string)
         return '' unless string
         HTML_ESCAPE_MAP.each do |escape|
           string = string.to_s.gsub(escape[:pattern], escape[:placeholder])
         end
         string
      end

      # Convert bad characters (now, high UTF-8 characters) to HTML escaped ones
      def html_un_pre_escape(string)
         HTML_ESCAPE_MAP.each do |escape|
           string = string.gsub(escape[:placeholder], escape[:text])
        end
        string
      end

      def html_escape(string)
        return '' unless string
        string.to_s.gsub(/&/, '&amp;').gsub(/</, '&lt;').gsub(/>/, '&gt;').gsub(/"/, '&quot;').gsub(/'/, '&#x27;').gsub(/\//, '&#x2F;')
      end
    end
  end
end