module Twitter
  # A module provides base methods to rewrite usernames, lists, hashtags and URLs.
  module Rewriter extend self
    def rewrite_entities(text, entities)
      chars = text.to_s.to_char_a

      # sort by start index
      entities = entities.sort_by do |entity|
        indices = entity.respond_to?(:indices) ? entity.indices : entity[:indices]
        indices.first
      end

      result = []
      last_index = entities.inject(0) do |index, entity|
        indices = entity.respond_to?(:indices) ? entity.indices : entity[:indices]
        result << chars[index...indices.first]
        result << yield(entity, chars)
        indices.last
      end
      result << chars[last_index..-1]

      result.flatten.join
    end

    # These methods are deprecated, will be removed in future.
    extend Deprecation

    def rewrite(text, options = {})
      [:hashtags, :urls, :usernames_or_lists].inject(text) do |key|
        options[key] ? send(:"rewrite_#{key}", text, &options[key]) : text
      end
    end
    deprecate :rewrite, :rewrite_entities

    def rewrite_usernames_or_lists(text)
      entities = Extractor.extract_mentions_or_lists_with_indices(text)
      rewrite_entities(text, entities) do |entity, chars|
        at = chars[entity[:indices].first]
        list_slug = entity[:list_slug]
        list_slug = nil if list_slug.empty?
        yield(at, entity[:screen_name], list_slug)
      end
    end
    deprecate :rewrite_usernames_or_lists, :rewrite_entities

    def rewrite_hashtags(text)
      entities = Extractor.extract_hashtags_with_indices(text)
      rewrite_entities(text, entities) do |entity, chars|
        hash = chars[entity[:indices].first]
        yield(hash, entity[:hashtag])
      end
    end
    deprecate :rewrite_hashtags, :rewrite_entities

    def rewrite_urls(text)
      entities = Extractor.extract_urls_with_indices(text, :extract_url_without_protocol => false)
      rewrite_entities(text, entities) do |entity, chars|
        yield(entity[:url])
      end
    end
    deprecate :rewrite_urls, :rewrite_entities
  end
end