forked from fedi/mastodon
40fd1de488
* Begin coverage for account search service * Coverage for hashtag query * Coverage for calling local vs remote find based on domain presence * Spec to check that exact matches are not duped * Coverage of resolve option * Coverage for account being provided * Start to refactor account search service * Isolate query username and domain methods * Isolate exact_match method * Extract methods for local and remote results * Simplify local vs remote and account isoliation * Extract methods for local and remote results * Simplify de-dupe of exact match * Simplify logic to check for non exact remotes * Cache some methods * Remove nil from exact_match from results array * Return exact matches first * Use find_remote even with no domain Account.find_local is just an alias for Account.find_remote(user, nil) - so we can not bother with the conditional here, and call find_remote directly.
87 lines
1.8 KiB
Ruby
87 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class AccountSearchService < BaseService
|
|
attr_reader :query, :limit, :resolve, :account
|
|
|
|
def call(query, limit, resolve = false, account = nil)
|
|
@query = query
|
|
@limit = limit
|
|
@resolve = resolve
|
|
@account = account
|
|
|
|
search_service_results
|
|
end
|
|
|
|
private
|
|
|
|
def search_service_results
|
|
return [] if query_blank_or_hashtag?
|
|
|
|
if resolving_non_matching_remote_account?
|
|
[FollowRemoteAccountService.new.call("#{query_username}@#{query_domain}")]
|
|
else
|
|
search_results_and_exact_match.compact.uniq
|
|
end
|
|
end
|
|
|
|
def resolving_non_matching_remote_account?
|
|
resolve && !exact_match && !domain_is_local?
|
|
end
|
|
|
|
def search_results_and_exact_match
|
|
[exact_match] + search_results.to_a
|
|
end
|
|
|
|
def query_blank_or_hashtag?
|
|
query.blank? || query.start_with?('#')
|
|
end
|
|
|
|
def split_query_string
|
|
@_split_query_string ||= query.gsub(/\A@/, '').split('@')
|
|
end
|
|
|
|
def query_username
|
|
@_query_username ||= split_query_string.first
|
|
end
|
|
|
|
def query_domain
|
|
@_query_domain ||= query_without_split? ? nil : split_query_string.last
|
|
end
|
|
|
|
def query_without_split?
|
|
split_query_string.size == 1
|
|
end
|
|
|
|
def domain_is_local?
|
|
@_domain_is_local ||= TagManager.instance.local_domain?(query_domain)
|
|
end
|
|
|
|
def exact_match
|
|
@_exact_match ||= Account.find_remote(query_username, query_domain)
|
|
end
|
|
|
|
def search_results
|
|
@_search_results ||= if account
|
|
advanced_search_results
|
|
else
|
|
simple_search_results
|
|
end
|
|
end
|
|
|
|
def advanced_search_results
|
|
Account.advanced_search_for(terms_for_query, account, limit)
|
|
end
|
|
|
|
def simple_search_results
|
|
Account.search_for(terms_for_query, limit)
|
|
end
|
|
|
|
def terms_for_query
|
|
if domain_is_local?
|
|
query_username
|
|
else
|
|
"#{query_username} #{query_domain}"
|
|
end
|
|
end
|
|
end
|