123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- # frozen_string_literal: true
- class AccountSearchService < BaseService
- attr_reader :query, :limit, :offset, :options, :account
- def call(query, account = nil, options = {})
- @query = query.strip
- @limit = options[:limit].to_i
- @offset = options[:offset].to_i
- @options = options
- @account = account
- search_service_results
- end
- private
- def search_service_results
- return [] if query_blank_or_hashtag? || limit < 1
- if resolving_non_matching_remote_account?
- [ResolveAccountService.new.call("#{query_username}@#{query_domain}")].compact
- else
- search_results_and_exact_match.compact.uniq.slice(0, limit)
- end
- end
- def resolving_non_matching_remote_account?
- options[:resolve] && !exact_match && !domain_is_local?
- end
- def search_results_and_exact_match
- exact = [exact_match]
- return exact if !exact[0].nil? && limit == 1
- exact + 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 search_from
- options[:following] && account ? account.following : Account
- end
- def exact_match
- @_exact_match ||= begin
- if domain_is_local?
- search_from.without_suspended.find_local(query_username)
- else
- search_from.without_suspended.find_remote(query_username, query_domain)
- end
- end
- end
- def search_results
- @_search_results ||= begin
- if account
- advanced_search_results
- else
- simple_search_results
- end
- end
- end
- def advanced_search_results
- Account.advanced_search_for(terms_for_query, account, limit, options[:following], offset)
- end
- def simple_search_results
- Account.search_for(terms_for_query, limit, offset)
- end
- def terms_for_query
- if domain_is_local?
- query_username
- else
- "#{query_username} #{query_domain}"
- end
- end
- end
|