166 lines
5 KiB
Ruby
166 lines
5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class StatusesIndex < Chewy::Index
|
|
settings index: { refresh_interval: '15m' }, analysis: {
|
|
filter: {
|
|
english_stop: {
|
|
type: 'stop',
|
|
stopwords: '_english_',
|
|
},
|
|
|
|
english_stemmer: {
|
|
type: 'stemmer',
|
|
language: 'english',
|
|
},
|
|
|
|
english_possessive_stemmer: {
|
|
type: 'stemmer',
|
|
language: 'possessive_english',
|
|
},
|
|
},
|
|
|
|
char_filter: {
|
|
tsconvert: {
|
|
type: 'stconvert',
|
|
keep_both: false,
|
|
delimiter: '#',
|
|
convert_type: 't2s',
|
|
},
|
|
},
|
|
|
|
tokenizer: {
|
|
kuromoji_user_dict: {
|
|
type: 'kuromoji_tokenizer',
|
|
user_dictionary: 'userdic.txt',
|
|
},
|
|
|
|
nori_user_dict: {
|
|
type: 'nori_tokenizer',
|
|
decompound_mode: 'mixed',
|
|
},
|
|
},
|
|
|
|
analyzer: {
|
|
en_content: {
|
|
tokenizer: 'uax_url_email',
|
|
filter: %w(
|
|
english_possessive_stemmer
|
|
lowercase
|
|
asciifolding
|
|
cjk_width
|
|
english_stop
|
|
english_stemmer
|
|
),
|
|
},
|
|
|
|
ja_content: {
|
|
type: 'custom',
|
|
char_filter: %w(
|
|
icu_normalizer
|
|
kuromoji_iteration_mark
|
|
),
|
|
tokenizer: 'kuromoji_user_dict',
|
|
filter: %w(
|
|
kuromoji_baseform
|
|
kuromoji_part_of_speech
|
|
ja_stop
|
|
kuromoji_stemmer
|
|
kuromoji_number
|
|
cjk_width
|
|
lowercase
|
|
),
|
|
},
|
|
|
|
ko_content: {
|
|
tokenizer: 'nori_user_dict',
|
|
filter: %w(
|
|
english_possessive_stemmer
|
|
lowercase
|
|
asciifolding
|
|
cjk_width
|
|
english_stop
|
|
english_stemmer
|
|
),
|
|
},
|
|
|
|
zh_content: {
|
|
tokenizer: 'ik_max_word',
|
|
filter: %w(
|
|
english_possessive_stemmer
|
|
lowercase
|
|
asciifolding
|
|
cjk_width
|
|
english_stop
|
|
english_stemmer
|
|
),
|
|
char_filter: %w(tsconvert),
|
|
},
|
|
},
|
|
}
|
|
|
|
index_scope ::Status.unscoped.kept.without_reblogs.includes(:media_attachments, :preloadable_poll)
|
|
|
|
crutch :mentions do |collection|
|
|
data = ::Mention.where(status_id: collection.map(&:id)).where(account: Account.local, silent: false).pluck(:status_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :favourites do |collection|
|
|
data = ::Favourite.where(status_id: collection.map(&:id)).where(account: Account.local).pluck(:status_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :reblogs do |collection|
|
|
data = ::Status.where(reblog_of_id: collection.map(&:id)).where(account: Account.local).pluck(:reblog_of_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :bookmarks do |collection|
|
|
data = ::Bookmark.where(status_id: collection.map(&:id)).where(account: Account.local).pluck(:status_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :votes do |collection|
|
|
data = ::PollVote.joins(:poll).where(poll: { status_id: collection.map(&:id) }).where(account: Account.local).pluck(:status_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :emoji_reactions do |collection|
|
|
data = ::EmojiReaction.where(status_id: collection.map(&:id)).where(account: Account.local).pluck(:status_id, :account_id)
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
crutch :status_references do |collection|
|
|
data = ::StatusReference.joins(:status).where(target_status_id: collection.map(&:id)).where(status: { account: Account.local }).pluck(:target_status_id, :'status.account_id')
|
|
data.each.with_object({}) { |(id, name), result| (result[id] ||= []).push(name) }
|
|
end
|
|
|
|
root date_detection: false do
|
|
field :id, type: 'long'
|
|
field :account_id, type: 'long'
|
|
field :domain, type: 'keyword', value: ->(status) { status.account_domain }
|
|
|
|
field :text, type: 'text', value: ->(status) { status.index_text } do
|
|
field :en_stemmed, type: 'text', analyzer: 'en_content'
|
|
field :ja_stemmed, type: 'text', analyzer: 'ja_content'
|
|
field :ko_stemmed, type: 'text', analyzer: 'ko_content'
|
|
field :zh_stemmed, type: 'text', analyzer: 'zh_content'
|
|
end
|
|
|
|
field :mentioned_account_id, type: 'long'
|
|
field :tag_id, type: 'long'
|
|
field :media_type, type: 'keyword'
|
|
field :reference_type, type: 'keyword'
|
|
field :language, type: 'keyword'
|
|
|
|
field :replies_count, type: 'long'
|
|
field :reblogs_count, type: 'long'
|
|
field :favourites_count, type: 'long'
|
|
field :emoji_reactions_count, type: 'long'
|
|
field :status_referred_by_count, type: 'long'
|
|
|
|
field :visibility, type: 'keyword'
|
|
field :searchable_by, type: 'long', value: ->(status, crutches) { status.searchable_by(crutches) }
|
|
field :searchability, type: 'keyword', value: ->(status) { status.compute_searchability }
|
|
end
|
|
end
|