Module: Twitter::TwitterText::Rewriter

Extended by:
Deprecation, Rewriter
Included in:
Rewriter
Defined in:
lib/twitter-text/rewriter.rb

Overview

A module provides base methods to rewrite usernames, lists, hashtags and URLs.

Instance Method Summary collapse

Methods included from Deprecation

deprecate

Instance Method Details

#rewrite(text, options = {}) ⇒ Object 



33
34
35
36
37
 
# File 'lib/twitter-text/rewriter.rb', line 33

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

#rewrite_entities(text, entities) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/twitter-text/rewriter.rb', line 9

def rewrite_entities(text, entities)
  codepoints = text.to_s.to_codepoint_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 << codepoints[index...indices.first]
    result << yield(entity, codepoints)
    indices.last
  end
  result << codepoints[last_index..-1]

  result.flatten.join
end

#rewrite_hashtags(text) ⇒ Object 



51
52
53
54
55
56
57
 
# File 'lib/twitter-text/rewriter.rb', line 51

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

#rewrite_urls(text) ⇒ Object 



60
61
62
63
64
65
 
# File 'lib/twitter-text/rewriter.rb', line 60

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

#rewrite_usernames_or_lists(text) ⇒ Object 



40
41
42
43
44
45
46
47
48
 
# File 'lib/twitter-text/rewriter.rb', line 40

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