848cb2a697
* Add more accurate hashtag search Using ElasticSearch to index hashtags with edge n-grams and score them by usage within the last 7 days since last activity. Only hashtags that have been reviewed and are listable can appear in searches, unless they match the query exactly * Fix search analyzer dropping non-ascii characters
172 lines
4.6 KiB
Ruby
172 lines
4.6 KiB
Ruby
# frozen_string_literal: true
|
|
# == Schema Information
|
|
#
|
|
# Table name: tags
|
|
#
|
|
# id :bigint(8) not null, primary key
|
|
# name :string default(""), not null
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# score :integer
|
|
# usable :boolean
|
|
# trendable :boolean
|
|
# listable :boolean
|
|
# reviewed_at :datetime
|
|
# requested_review_at :datetime
|
|
# last_status_at :datetime
|
|
# last_trend_at :datetime
|
|
#
|
|
|
|
class Tag < ApplicationRecord
|
|
has_and_belongs_to_many :statuses
|
|
has_and_belongs_to_many :accounts
|
|
has_and_belongs_to_many :sample_accounts, -> { searchable.discoverable.popular.limit(3) }, class_name: 'Account'
|
|
|
|
has_many :featured_tags, dependent: :destroy, inverse_of: :tag
|
|
has_one :account_tag_stat, dependent: :destroy
|
|
|
|
HASHTAG_NAME_RE = '([[:word:]_][[:word:]_·]*[[:alpha:]_·][[:word:]_·]*[[:word:]_])|([[:word:]_]*[[:alpha:]][[:word:]_]*)'
|
|
HASHTAG_RE = /(?:^|[^\/\)\w])#(#{HASHTAG_NAME_RE})/i
|
|
|
|
validates :name, presence: true, format: { with: /\A(#{HASHTAG_NAME_RE})\z/i }
|
|
validate :validate_name_change, if: -> { !new_record? && name_changed? }
|
|
|
|
scope :reviewed, -> { where.not(reviewed_at: nil) }
|
|
scope :unreviewed, -> { where(reviewed_at: nil) }
|
|
scope :pending_review, -> { unreviewed.where.not(requested_review_at: nil) }
|
|
scope :usable, -> { where(usable: [true, nil]) }
|
|
scope :listable, -> { where(listable: [true, nil]) }
|
|
scope :discoverable, -> { listable.joins(:account_tag_stat).where(AccountTagStat.arel_table[:accounts_count].gt(0)).order(Arel.sql('account_tag_stats.accounts_count desc')) }
|
|
scope :most_used, ->(account) { joins(:statuses).where(statuses: { account: account }).group(:id).order(Arel.sql('count(*) desc')) }
|
|
|
|
delegate :accounts_count,
|
|
:accounts_count=,
|
|
:increment_count!,
|
|
:decrement_count!,
|
|
to: :account_tag_stat
|
|
|
|
after_save :save_account_tag_stat
|
|
|
|
update_index('tags#tag', :self) if Chewy.enabled?
|
|
|
|
def account_tag_stat
|
|
super || build_account_tag_stat
|
|
end
|
|
|
|
def cached_sample_accounts
|
|
Rails.cache.fetch("#{cache_key}/sample_accounts", expires_in: 12.hours) { sample_accounts }
|
|
end
|
|
|
|
def to_param
|
|
name
|
|
end
|
|
|
|
def usable
|
|
boolean_with_default('usable', true)
|
|
end
|
|
|
|
alias usable? usable
|
|
|
|
def listable
|
|
boolean_with_default('listable', true)
|
|
end
|
|
|
|
alias listable? listable
|
|
|
|
def trendable
|
|
boolean_with_default('trendable', false)
|
|
end
|
|
|
|
alias trendable? trendable
|
|
|
|
def requires_review?
|
|
reviewed_at.nil?
|
|
end
|
|
|
|
def reviewed?
|
|
reviewed_at.present?
|
|
end
|
|
|
|
def requested_review?
|
|
requested_review_at.present?
|
|
end
|
|
|
|
def trending?
|
|
TrendingTags.trending?(self)
|
|
end
|
|
|
|
def history
|
|
days = []
|
|
|
|
7.times do |i|
|
|
day = i.days.ago.beginning_of_day.to_i
|
|
|
|
days << {
|
|
day: day.to_s,
|
|
uses: Redis.current.get("activity:tags:#{id}:#{day}") || '0',
|
|
accounts: Redis.current.pfcount("activity:tags:#{id}:#{day}:accounts").to_s,
|
|
}
|
|
end
|
|
|
|
days
|
|
end
|
|
|
|
class << self
|
|
def find_or_create_by_names(name_or_names)
|
|
Array(name_or_names).map(&method(:normalize)).uniq { |str| str.mb_chars.downcase.to_s }.map do |normalized_name|
|
|
tag = matching_name(normalized_name).first || create(name: normalized_name)
|
|
|
|
yield tag if block_given?
|
|
|
|
tag
|
|
end
|
|
end
|
|
|
|
def search_for(term, limit = 5, offset = 0)
|
|
normalized_term = normalize(term.strip).mb_chars.downcase.to_s
|
|
pattern = sanitize_sql_like(normalized_term) + '%'
|
|
|
|
Tag.listable
|
|
.where(arel_table[:name].lower.matches(pattern))
|
|
.where(arel_table[:name].lower.eq(normalized_term).or(arel_table[:reviewed_at].not_eq(nil)))
|
|
.order(Arel.sql('length(name) ASC, name ASC'))
|
|
.limit(limit)
|
|
.offset(offset)
|
|
end
|
|
|
|
def find_normalized(name)
|
|
matching_name(name).first
|
|
end
|
|
|
|
def find_normalized!(name)
|
|
find_normalized(name) || raise(ActiveRecord::RecordNotFound)
|
|
end
|
|
|
|
def matching_name(name_or_names)
|
|
names = Array(name_or_names).map { |name| normalize(name).mb_chars.downcase.to_s }
|
|
|
|
if names.size == 1
|
|
where(arel_table[:name].lower.eq(names.first))
|
|
else
|
|
where(arel_table[:name].lower.in(names))
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def normalize(str)
|
|
str.gsub(/\A#/, '')
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def save_account_tag_stat
|
|
return unless account_tag_stat&.changed?
|
|
account_tag_stat.save
|
|
end
|
|
|
|
def validate_name_change
|
|
errors.add(:name, I18n.t('tags.does_not_match_previous_name')) unless name_was.mb_chars.casecmp(name.mb_chars).zero?
|
|
end
|
|
end
|