Skip to content

Commit

Permalink
Minor performance improvements and cleanup in formatter (mastodon#10765)
Browse files Browse the repository at this point in the history
  • Loading branch information
ClearlyClaire authored and hiyuki2578 committed Oct 2, 2019
1 parent 980f72f commit 8064958
Showing 1 changed file with 5 additions and 14 deletions.
19 changes: 5 additions & 14 deletions app/lib/formatter.rb
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,7 @@ def nyaize(html)
end

def rewrite(text, entities)
chars = text.to_s.to_char_a
text = text.to_s

# Sort by start index
entities = entities.sort_by do |entity|
Expand All @@ -205,12 +205,12 @@ def rewrite(text, entities)

last_index = entities.reduce(0) do |index, entity|
indices = entity.respond_to?(:indices) ? entity.indices : entity[:indices]
result << encode(chars[index...indices.first].join)
result << encode(text[index...indices.first])
result << yield(entity)
indices.last
end

result << encode(chars[last_index..-1].join)
result << encode(text[last_index..-1])

result.flatten.join
end
Expand All @@ -237,23 +237,14 @@ def utf8_friendly_extractor(text, options = {})
# Note: I couldn't obtain list_slug with @user/list-name format
# for mention so this requires additional check
special = Extractor.extract_urls_with_indices(escaped, options).map do |extract|
# exactly one of :url, :hashtag, :screen_name, :cashtag keys is present
key = (extract.keys & [:url, :hashtag, :screen_name, :cashtag]).first

new_indices = [
old_to_new_index.find_index(extract[:indices].first),
old_to_new_index.find_index(extract[:indices].last),
]

has_prefix_char = [:hashtag, :screen_name, :cashtag].include?(key)
value_indices = [
new_indices.first + (has_prefix_char ? 1 : 0), # account for #, @ or $
new_indices.last - 1,
]

next extract.merge(
:indices => new_indices,
key => text[value_indices.first..value_indices.last]
indices: new_indices,
url: text[new_indices.first..new_indices.last - 1]
)
end

Expand Down

0 comments on commit 8064958

Please sign in to comment.