mirror of https://github.com/glitch-soc/mastodon
60 lines
1.6 KiB
Ruby
60 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class StatusLengthValidator < ActiveModel::Validator
|
|
MAX_CHARS = (ENV['MAX_TOOT_CHARS'] || 500).to_i
|
|
URL_PLACEHOLDER_CHARS = 23
|
|
URL_PLACEHOLDER = 'x' * 23
|
|
|
|
def validate(status)
|
|
return unless status.local? && !status.reblog?
|
|
|
|
status.errors.add(:text, I18n.t('statuses.over_character_limit', max: MAX_CHARS)) if too_long?(status)
|
|
end
|
|
|
|
private
|
|
|
|
def too_long?(status)
|
|
countable_length(combined_text(status)) > MAX_CHARS
|
|
end
|
|
|
|
def countable_length(str)
|
|
str.mb_chars.grapheme_length
|
|
end
|
|
|
|
def combined_text(status)
|
|
[status.spoiler_text, countable_text(status.text)].join
|
|
end
|
|
|
|
def countable_text(str)
|
|
return '' if str.blank?
|
|
|
|
# To ensure that we only give length concessions to entities that
|
|
# will be correctly parsed during formatting, we go through full
|
|
# entity extraction
|
|
|
|
entities = Extractor.remove_overlapping_entities(Extractor.extract_urls_with_indices(str, extract_url_without_protocol: false) + Extractor.extract_mentions_or_lists_with_indices(str))
|
|
|
|
rewrite_entities(str, entities) do |entity|
|
|
if entity[:url]
|
|
URL_PLACEHOLDER
|
|
elsif entity[:screen_name]
|
|
"@#{entity[:screen_name].split('@').first}"
|
|
end
|
|
end
|
|
end
|
|
|
|
def rewrite_entities(str, entities)
|
|
entities.sort_by! { |entity| entity[:indices].first }
|
|
result = +''
|
|
|
|
last_index = entities.reduce(0) do |index, entity|
|
|
result << str[index...entity[:indices].first]
|
|
result << yield(entity)
|
|
entity[:indices].last
|
|
end
|
|
|
|
result << str[last_index..]
|
|
result
|
|
end
|
|
end
|