* Change domain blocks to automatically support subdomains If a more authoritative domain is blocked (example.com), then the same block will be applied to a subdomain (foo.example.com) * Match subdomains of existing accounts when blocking/unblocking domains * Improve code style
		
			
				
	
	
		
			83 lines
		
	
	
	
		
			2.3 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
	
		
			2.3 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) }
 | 
						|
  scope :by_domain_and_subdomains, ->(domain) { where(domain: domain).or(where(arel_table[:domain].matches('%.' + domain))) }
 | 
						|
 | 
						|
  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
 |