2017-03-22 03:32:27 +02:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
class AccountSearchService < BaseService
|
Add type, limit, offset, min_id, max_id, account_id to search API (#10091)
* Add type, limit, offset, min_id, max_id, account_id to search API
Fix #8939
* Make the offset work on accounts and hashtags search as well
* Assure brakeman we are not doing mass assignment here
* Do not allow paginating unless a type is chosen
* Fix search query and index id field on statuses instead of created_at
2019-02-26 16:21:36 +02:00
|
|
|
attr_reader :query, :limit, :offset, :options, :account
|
2017-04-15 04:17:07 +03:00
|
|
|
|
Add type, limit, offset, min_id, max_id, account_id to search API (#10091)
* Add type, limit, offset, min_id, max_id, account_id to search API
Fix #8939
* Make the offset work on accounts and hashtags search as well
* Assure brakeman we are not doing mass assignment here
* Do not allow paginating unless a type is chosen
* Fix search query and index id field on statuses instead of created_at
2019-02-26 16:21:36 +02:00
|
|
|
def call(query, account = nil, options = {})
|
2017-12-10 20:35:46 +02:00
|
|
|
@query = query.strip
|
Add type, limit, offset, min_id, max_id, account_id to search API (#10091)
* Add type, limit, offset, min_id, max_id, account_id to search API
Fix #8939
* Make the offset work on accounts and hashtags search as well
* Assure brakeman we are not doing mass assignment here
* Do not allow paginating unless a type is chosen
* Fix search query and index id field on statuses instead of created_at
2019-02-26 16:21:36 +02:00
|
|
|
@limit = options[:limit].to_i
|
|
|
|
@offset = options[:offset].to_i
|
2017-12-06 00:02:27 +02:00
|
|
|
@options = options
|
2017-04-15 04:17:07 +03:00
|
|
|
@account = account
|
|
|
|
|
2019-04-07 05:59:13 +03:00
|
|
|
search_service_results
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
private
|
2017-03-22 03:32:27 +02:00
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
def search_service_results
|
2017-04-25 05:44:43 +03:00
|
|
|
return [] if query_blank_or_hashtag? || limit < 1
|
2017-03-22 03:32:27 +02:00
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
if resolving_non_matching_remote_account?
|
2018-01-22 15:25:09 +02:00
|
|
|
[ResolveAccountService.new.call("#{query_username}@#{query_domain}")].compact
|
2017-03-22 03:32:27 +02:00
|
|
|
else
|
2019-08-11 14:14:51 +03:00
|
|
|
search_results_and_exact_match.compact.uniq
|
2017-03-22 03:32:27 +02:00
|
|
|
end
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def resolving_non_matching_remote_account?
|
2019-08-11 14:14:51 +03:00
|
|
|
offset.zero? && options[:resolve] && !exact_match? && !domain_is_local?
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def search_results_and_exact_match
|
2019-08-11 14:14:51 +03:00
|
|
|
return search_results.to_a unless offset.zero?
|
|
|
|
|
|
|
|
results = [exact_match]
|
|
|
|
|
|
|
|
return results if exact_match? && limit == 1
|
|
|
|
|
|
|
|
results + search_results.to_a
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def query_blank_or_hashtag?
|
|
|
|
query.blank? || query.start_with?('#')
|
|
|
|
end
|
|
|
|
|
|
|
|
def split_query_string
|
2019-08-11 14:14:51 +03:00
|
|
|
@split_query_string ||= query.gsub(/\A@/, '').split('@')
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def query_username
|
2019-08-11 14:14:51 +03:00
|
|
|
@query_username ||= split_query_string.first || ''
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
2017-03-22 03:32:27 +02:00
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
def query_domain
|
2019-08-11 14:14:51 +03:00
|
|
|
@query_domain ||= query_without_split? ? nil : split_query_string.last
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def query_without_split?
|
|
|
|
split_query_string.size == 1
|
|
|
|
end
|
|
|
|
|
|
|
|
def domain_is_local?
|
2019-08-11 14:14:51 +03:00
|
|
|
@domain_is_local ||= TagManager.instance.local_domain?(query_domain)
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
2017-12-06 12:44:23 +02:00
|
|
|
def search_from
|
|
|
|
options[:following] && account ? account.following : Account
|
|
|
|
end
|
|
|
|
|
2019-08-11 14:14:51 +03:00
|
|
|
def exact_match?
|
|
|
|
exact_match.present?
|
|
|
|
end
|
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
def exact_match
|
2019-08-11 14:14:51 +03:00
|
|
|
return @exact_match if defined?(@exact_match)
|
|
|
|
|
|
|
|
@exact_match = begin
|
2017-05-27 01:55:08 +03:00
|
|
|
if domain_is_local?
|
2018-04-23 22:27:18 +03:00
|
|
|
search_from.without_suspended.find_local(query_username)
|
2017-05-27 01:55:08 +03:00
|
|
|
else
|
2018-04-23 22:27:18 +03:00
|
|
|
search_from.without_suspended.find_remote(query_username, query_domain)
|
2017-05-27 01:55:08 +03:00
|
|
|
end
|
|
|
|
end
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
2017-03-22 03:32:27 +02:00
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
def search_results
|
2019-08-11 14:14:51 +03:00
|
|
|
@search_results ||= begin
|
2017-04-30 01:23:45 +03:00
|
|
|
if account
|
|
|
|
advanced_search_results
|
|
|
|
else
|
|
|
|
simple_search_results
|
|
|
|
end
|
2017-03-22 03:32:27 +02:00
|
|
|
end
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def advanced_search_results
|
2019-08-11 14:14:51 +03:00
|
|
|
Account.advanced_search_for(terms_for_query, account, limit_for_non_exact_results, options[:following], offset)
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
2017-03-22 03:32:27 +02:00
|
|
|
|
2017-04-15 04:17:07 +03:00
|
|
|
def simple_search_results
|
2019-08-11 14:14:51 +03:00
|
|
|
Account.search_for(terms_for_query, limit_for_non_exact_results, offset)
|
|
|
|
end
|
|
|
|
|
|
|
|
def limit_for_non_exact_results
|
|
|
|
if offset.zero? && exact_match?
|
|
|
|
limit - 1
|
|
|
|
else
|
|
|
|
limit
|
|
|
|
end
|
2017-04-15 04:17:07 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
def terms_for_query
|
|
|
|
if domain_is_local?
|
|
|
|
query_username
|
|
|
|
else
|
|
|
|
"#{query_username} #{query_domain}"
|
|
|
|
end
|
2017-03-22 03:32:27 +02:00
|
|
|
end
|
|
|
|
end
|