mirror of
https://github.com/whippyshou/mastodon
synced 2024-12-18 00:29:06 +09:00
76 lines
1.8 KiB
Ruby
76 lines
1.8 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
class StatusesSearchService < BaseService
|
||
|
def call(query, account = nil, options = {})
|
||
|
@query = query&.strip
|
||
|
@account = account
|
||
|
@options = options
|
||
|
@limit = options[:limit].to_i
|
||
|
@offset = options[:offset].to_i
|
||
|
|
||
|
status_search_results
|
||
|
end
|
||
|
|
||
|
private
|
||
|
|
||
|
def status_search_results
|
||
|
definition = parsed_query.apply(
|
||
|
StatusesIndex.filter(
|
||
|
bool: {
|
||
|
should: [
|
||
|
publicly_searchable,
|
||
|
non_publicly_searchable,
|
||
|
],
|
||
|
|
||
|
minimum_should_match: 1,
|
||
|
}
|
||
|
)
|
||
|
)
|
||
|
|
||
|
# This is the best way to submit identical queries to multi-indexes though chewy
|
||
|
definition.instance_variable_get(:@parameters)[:indices].value[:indices] << PublicStatusesIndex
|
||
|
|
||
|
results = definition.collapse(field: :id).order(_id: { order: :desc }).limit(@limit).offset(@offset).objects.compact
|
||
|
account_ids = results.map(&:account_id)
|
||
|
account_domains = results.map(&:account_domain)
|
||
|
preloaded_relations = @account.relations_map(account_ids, account_domains)
|
||
|
|
||
|
results.reject { |status| StatusFilter.new(status, @account, preloaded_relations).filtered? }
|
||
|
rescue Faraday::ConnectionFailed, Parslet::ParseFailed
|
||
|
[]
|
||
|
end
|
||
|
|
||
|
def publicly_searchable
|
||
|
{
|
||
|
bool: {
|
||
|
must_not: {
|
||
|
exists: {
|
||
|
field: 'searchable_by',
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def non_publicly_searchable
|
||
|
{
|
||
|
bool: {
|
||
|
must: [
|
||
|
{
|
||
|
exists: {
|
||
|
field: 'searchable_by',
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
term: { searchable_by: @account.id },
|
||
|
},
|
||
|
],
|
||
|
},
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def parsed_query
|
||
|
SearchQueryTransformer.new.apply(SearchQueryParser.new.parse(@query))
|
||
|
end
|
||
|
end
|