104 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			104 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| class EmojiFormatter
 | |
|   include RoutingHelper
 | |
| 
 | |
|   DISALLOWED_BOUNDING_REGEX = /[[:alnum:]:]/
 | |
| 
 | |
|   attr_reader :html, :custom_emojis, :options
 | |
| 
 | |
|   # @param [ActiveSupport::SafeBuffer] html
 | |
|   # @param [Array<CustomEmoji>] custom_emojis
 | |
|   # @param [Hash] options
 | |
|   # @option options [Boolean] :animate
 | |
|   # @option options [String] :style
 | |
|   # @option options [String] :raw_shortcode
 | |
|   def initialize(html, custom_emojis, options = {})
 | |
|     raise ArgumentError unless html.html_safe?
 | |
| 
 | |
|     @html = html
 | |
|     @custom_emojis = custom_emojis
 | |
|     @options = options
 | |
|   end
 | |
| 
 | |
|   def to_s
 | |
|     return html if custom_emojis.empty? || html.blank?
 | |
| 
 | |
|     tree = Nokogiri::HTML5.fragment(html)
 | |
|     tree.xpath('./text()|.//text()[not(ancestor[@class="invisible"])]').to_a.each do |node|
 | |
|       i                     = -1
 | |
|       inside_shortname      = false
 | |
|       shortname_start_index = -1
 | |
|       last_index            = 0
 | |
|       text                  = node.content
 | |
|       result                = Nokogiri::XML::NodeSet.new(tree.document)
 | |
| 
 | |
|       while i + 1 < text.size
 | |
|         i += 1
 | |
| 
 | |
|         if inside_shortname && text[i] == ':'
 | |
|           inside_shortname = false
 | |
|           shortcode = text[shortname_start_index + 1..i - 1]
 | |
|           char_after = text[i + 1]
 | |
| 
 | |
|           next unless (char_after.nil? || !DISALLOWED_BOUNDING_REGEX.match?(char_after)) && (emoji = emoji_map[shortcode])
 | |
| 
 | |
|           result << tree.document.create_text_node(text[last_index..shortname_start_index - 1]) if shortname_start_index.positive?
 | |
|           result << tree.document.fragment(tag_for_emoji(shortcode, emoji))
 | |
| 
 | |
|           last_index = i + 1
 | |
|         elsif text[i] == ':' && (i.zero? || !DISALLOWED_BOUNDING_REGEX.match?(text[i - 1]))
 | |
|           inside_shortname = true
 | |
|           shortname_start_index = i
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       result << tree.document.create_text_node(text[last_index..])
 | |
|       node.replace(result)
 | |
|     end
 | |
| 
 | |
|     tree.to_html.html_safe # rubocop:disable Rails/OutputSafety
 | |
|   end
 | |
| 
 | |
|   private
 | |
| 
 | |
|   def emoji_map
 | |
|     @emoji_map ||= custom_emojis.each_with_object({}) { |e, h| h[e.shortcode] = [full_asset_url(e.image.url), full_asset_url(e.image.url(:static))] }
 | |
|   end
 | |
| 
 | |
|   def tag_for_emoji(shortcode, emoji)
 | |
|     return content_tag(:span, ":#{shortcode}:", translate: 'no') if raw_shortcode?
 | |
| 
 | |
|     original_url, static_url = emoji
 | |
| 
 | |
|     image_tag(
 | |
|       animate? ? original_url : static_url,
 | |
|       image_attributes.merge(alt: ":#{shortcode}:", title: ":#{shortcode}:", data: image_data_attributes(original_url, static_url))
 | |
|     )
 | |
|   end
 | |
| 
 | |
|   def image_attributes
 | |
|     { rel: 'emoji', draggable: false, width: 16, height: 16, class: image_class_names, style: image_style }
 | |
|   end
 | |
| 
 | |
|   def image_data_attributes(original_url, static_url)
 | |
|     { original: original_url, static: static_url } unless animate?
 | |
|   end
 | |
| 
 | |
|   def image_class_names
 | |
|     animate? ? 'emojione' : 'emojione custom-emoji'
 | |
|   end
 | |
| 
 | |
|   def image_style
 | |
|     @options[:style]
 | |
|   end
 | |
| 
 | |
|   def animate?
 | |
|     @options[:animate] || @options.key?(:style)
 | |
|   end
 | |
| 
 | |
|   def raw_shortcode?
 | |
|     @options[:raw_shortcode]
 | |
|   end
 | |
| end
 |