mirror of
https://github.com/whippyshou/mastodon
synced 2024-12-11 13:18:56 +09:00
7d00e4edbd
* Make custom emoji domains case sensitive #9351 * Fixup style in downcase_domain to comply with codeclimate. * switch if! to unless * Don't use transactions, operate in batches. Also revert spurious schema change.
83 lines
2.2 KiB
Ruby
83 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
# == Schema Information
|
|
#
|
|
# Table name: custom_emojis
|
|
#
|
|
# id :bigint(8) not null, primary key
|
|
# shortcode :string default(""), not null
|
|
# domain :string
|
|
# image_file_name :string
|
|
# image_content_type :string
|
|
# image_file_size :integer
|
|
# image_updated_at :datetime
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# disabled :boolean default(FALSE), not null
|
|
# uri :string
|
|
# image_remote_url :string
|
|
# visible_in_picker :boolean default(TRUE), not null
|
|
#
|
|
|
|
class CustomEmoji < ApplicationRecord
|
|
LIMIT = 50.kilobytes
|
|
|
|
SHORTCODE_RE_FRAGMENT = '[a-zA-Z0-9_]{2,}'
|
|
|
|
SCAN_RE = /(?<=[^[:alnum:]:]|\n|^)
|
|
:(#{SHORTCODE_RE_FRAGMENT}):
|
|
(?=[^[:alnum:]:]|$)/x
|
|
|
|
has_one :local_counterpart, -> { where(domain: nil) }, class_name: 'CustomEmoji', primary_key: :shortcode, foreign_key: :shortcode
|
|
|
|
has_attached_file :image, styles: { static: { format: 'png', convert_options: '-coalesce -strip' } }
|
|
|
|
before_validation :downcase_domain
|
|
|
|
validates_attachment :image, content_type: { content_type: 'image/png' }, presence: true, size: { less_than: LIMIT }
|
|
validates :shortcode, uniqueness: { scope: :domain }, format: { with: /\A#{SHORTCODE_RE_FRAGMENT}\z/ }, length: { minimum: 2 }
|
|
|
|
scope :local, -> { where(domain: nil) }
|
|
scope :remote, -> { where.not(domain: nil) }
|
|
scope :alphabetic, -> { order(domain: :asc, shortcode: :asc) }
|
|
|
|
remotable_attachment :image, LIMIT
|
|
|
|
include Attachmentable
|
|
|
|
after_commit :remove_entity_cache
|
|
|
|
def local?
|
|
domain.nil?
|
|
end
|
|
|
|
def object_type
|
|
:emoji
|
|
end
|
|
|
|
class << self
|
|
def from_text(text, domain)
|
|
return [] if text.blank?
|
|
|
|
shortcodes = text.scan(SCAN_RE).map(&:first).uniq
|
|
|
|
return [] if shortcodes.empty?
|
|
|
|
EntityCache.instance.emoji(shortcodes, domain)
|
|
end
|
|
|
|
def search(shortcode)
|
|
where('"custom_emojis"."shortcode" ILIKE ?', "%#{shortcode}%")
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def remove_entity_cache
|
|
Rails.cache.delete(EntityCache.instance.to_key(:emoji, shortcode, domain))
|
|
end
|
|
|
|
def downcase_domain
|
|
self.domain = domain.downcase unless domain.nil?
|
|
end
|
|
end
|