108 lines
3.2 KiB
Ruby
108 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class SearchService < BaseService
|
|
def call(query, account, limit, options = {})
|
|
@query = query&.strip
|
|
@account = account
|
|
@options = options
|
|
@limit = limit.to_i
|
|
@offset = options[:type].blank? ? 0 : options[:offset].to_i
|
|
@resolve = options[:resolve] || false
|
|
@onlyVerified = options[:onlyVerified] || false
|
|
|
|
default_results.tap do |results|
|
|
next if @query.blank? || @limit.zero?
|
|
|
|
if @query.present?
|
|
results[:accounts] = perform_accounts_search! if account_searchable?
|
|
results[:statuses] = perform_statuses_search! if full_text_searchable? && !account.nil?
|
|
results[:links] = [] # perform_links_search! if !account.nil?
|
|
results[:groups] = perform_groups_search!
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def perform_accounts_search!
|
|
AccountSearchService.new.call(
|
|
@query,
|
|
@account,
|
|
limit: @limit,
|
|
resolve: @resolve,
|
|
offset: @offset,
|
|
onlyVerified: @onlyVerified
|
|
)
|
|
end
|
|
|
|
def perform_groups_search!
|
|
Group.search_for(
|
|
@query.gsub(/\A#/, ''),
|
|
@offset
|
|
)
|
|
end
|
|
|
|
def perform_links_search!
|
|
PreviewCard.search_for(
|
|
@query.gsub(/\A#/, ''),
|
|
@offset
|
|
)
|
|
end
|
|
|
|
def perform_statuses_search!
|
|
definition = StatusesIndex.filter(term: { searchable_by: @account.id })
|
|
.query(multi_match: { type: 'most_fields', query: @query, operator: 'and', fields: %w(text text.stemmed) })
|
|
|
|
if @options[:account_id].present?
|
|
definition = definition.filter(term: { account_id: @options[:account_id] })
|
|
end
|
|
|
|
if @options[:min_id].present? || @options[:max_id].present?
|
|
range = {}
|
|
range[:gt] = @options[:min_id].to_i if @options[:min_id].present?
|
|
range[:lt] = @options[:max_id].to_i if @options[:max_id].present?
|
|
definition = definition.filter(range: { id: range })
|
|
end
|
|
|
|
results = definition.limit(@limit).offset(@offset).objects.compact
|
|
account_ids = results.map(&:account_id)
|
|
account_domains = results.map(&:account_domain)
|
|
preloaded_relations = relations_map_for_account(@account, account_ids, account_domains)
|
|
|
|
results.reject { |status| StatusFilter.new(status, @account, preloaded_relations).filtered? }
|
|
rescue Faraday::ConnectionFailed
|
|
[]
|
|
end
|
|
|
|
def default_results
|
|
{ accounts: [], statuses: [], links: [], groups: [] }
|
|
end
|
|
|
|
def full_text_searchable?
|
|
return false unless Chewy.enabled?
|
|
|
|
statuses_search? && !@account.nil? && !((@query.start_with?('#') || @query.include?('@')) && !@query.include?(' '))
|
|
end
|
|
|
|
def account_searchable?
|
|
account_search? && !(@query.include?('@') && @query.include?(' '))
|
|
end
|
|
|
|
def account_search?
|
|
@options[:type].blank? || @options[:type] == 'accounts'
|
|
end
|
|
|
|
def statuses_search?
|
|
@options[:type].blank? || @options[:type] == 'statuses'
|
|
end
|
|
|
|
def relations_map_for_account(account, account_ids, domains)
|
|
{
|
|
blocking: Account.blocking_map(account_ids, account.id),
|
|
blocked_by: Account.blocked_by_map(account_ids, account.id),
|
|
muting: Account.muting_map(account_ids, account.id),
|
|
following: Account.following_map(account_ids, account.id),
|
|
}
|
|
end
|
|
end
|