4691b0068c
* Remove the search button from UI header when logged out (#25631) * Change account search to match by text when opted-in (#25599) Co-authored-by: Eugen Rochko <eugen@zeonfederated.com> * Fix ResolveURLService not resolving local URLs for remote content (#25637) * Remove `pkg-config` gem dependency (#25615) * Update Crowdin configuration file * Fix onboarding prompt being displayed because of disconnection gaps (#25617) * Use an Immutable Record as the root state (#25584) * Add index to backups on `user_id` column (#25647) * Fix rails `rewhere` deprecation warning in directories api controller (#25625) * Remove unused routes (#25578) * Fixing an issue with a missing argument (#2261) undefined * Update uri to version 0.12.2 (CVE fix) (#25657) * Change local and federated timelines to be in a single firehose column (#25641) * Fix HTTP 500 in `/api/v1/emails/check_confirmation` (#25595) * Rails 7 update (#24241) * Change dropdown icon above compose form from ellipsis to bars in web UI (#25661) * Prevent duplicate concurrent calls of `/api/*/instance` in web UI (#25663) * Revert "Rails 7 update" (#25667) * [Glitch] Remove the search button from UI header when logged out Port285a691936
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * [Glitch] Fix onboarding prompt being displayed because of disconnection gaps Port9934949fc4
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * [Glitch] Use an Immutable Record as the root state Port78ba12f0bf
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * [Glitch] Change local and federated timelines to be in a single firehose column Portcea9db5a0b
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * [Glitch] Change dropdown icon above compose form from ellipsis to bars in web UI Port0512537eb6
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * [Glitch] Prevent duplicate concurrent calls of `/api/*/instance` in web UI Port5b46345459
to glitch-soc Signed-off-by: Claire <claire.github-309c@sitedethib.com> * Show local-only posts in “All” by default, and add back option to toggle it * Fix showing local only toots in "All" (#2265) * Fix warnings about missing dependency in hooks Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> * Add `allowLocalOnly` to timelineId Without this local-only toots will never be loaded. feedType is checked to be public to not show local-only toots in the "Remote" tab. Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> --------- Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> * Add regex filter back to firehose (#2266) * Add regex filter back to firehose The regex filter will apply to all tabs and not be automatically applied when pinned. Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> * Keep regex when pinned Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> --------- Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> --------- Signed-off-by: Claire <claire.github-309c@sitedethib.com> Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com> Co-authored-by: Claire <claire.github-309c@sitedethib.com> Co-authored-by: jsgoldstein <jakegoldstein95@gmail.com> Co-authored-by: Eugen Rochko <eugen@zeonfederated.com> Co-authored-by: Renaud Chaput <renchap@gmail.com> Co-authored-by: Matt Jankowski <matt@jankowski.online> Co-authored-by: Vivianne <puttabutta@gmail.com> Co-authored-by: Daniel M Brasil <danielmbrasil@protonmail.com> Co-authored-by: mogaminsk <mgmnjp@icloud.com> Co-authored-by: Plastikmensch <Plastikmensch@users.noreply.github.com>
215 lines
4.9 KiB
Ruby
215 lines
4.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class AccountSearchService < BaseService
|
|
attr_reader :query, :limit, :offset, :options, :account
|
|
|
|
MENTION_ONLY_RE = /\A#{Account::MENTION_RE}\z/i
|
|
|
|
# Min. number of characters to look for non-exact matches
|
|
MIN_QUERY_LENGTH = 5
|
|
|
|
def call(query, account = nil, options = {})
|
|
@query = query&.strip&.gsub(/\A@/, '')
|
|
@limit = options[:limit].to_i
|
|
@offset = options[:offset].to_i
|
|
@options = options
|
|
@account = account
|
|
|
|
search_service_results.compact.uniq
|
|
end
|
|
|
|
private
|
|
|
|
def search_service_results
|
|
return [] if query.blank? || limit < 1
|
|
|
|
[exact_match] + search_results
|
|
end
|
|
|
|
def exact_match
|
|
return unless offset.zero? && username_complete?
|
|
|
|
return @exact_match if defined?(@exact_match)
|
|
|
|
match = if options[:resolve]
|
|
ResolveAccountService.new.call(query)
|
|
elsif domain_is_local?
|
|
Account.find_local(query_username)
|
|
else
|
|
Account.find_remote(query_username, query_domain)
|
|
end
|
|
|
|
match = nil if !match.nil? && !account.nil? && options[:following] && !account.following?(match)
|
|
|
|
@exact_match = match
|
|
end
|
|
|
|
def search_results
|
|
return [] if limit_for_non_exact_results.zero?
|
|
|
|
@search_results ||= begin
|
|
results = from_elasticsearch if Chewy.enabled?
|
|
results ||= from_database
|
|
results
|
|
end
|
|
end
|
|
|
|
def from_database
|
|
if account
|
|
advanced_search_results
|
|
else
|
|
simple_search_results
|
|
end
|
|
end
|
|
|
|
def advanced_search_results
|
|
Account.advanced_search_for(terms_for_query, account, limit: limit_for_non_exact_results, following: options[:following], offset: offset)
|
|
end
|
|
|
|
def simple_search_results
|
|
Account.search_for(terms_for_query, limit: limit_for_non_exact_results, offset: offset)
|
|
end
|
|
|
|
def from_elasticsearch
|
|
must_clauses = must_clause
|
|
should_clauses = should_clause
|
|
|
|
if account
|
|
return [] if options[:following] && following_ids.empty?
|
|
|
|
if options[:following]
|
|
must_clauses << { terms: { id: following_ids } }
|
|
elsif following_ids.any?
|
|
should_clauses << { terms: { id: following_ids, boost: 100 } }
|
|
end
|
|
end
|
|
|
|
query = { bool: { must: must_clauses, should: should_clauses } }
|
|
functions = [reputation_score_function, followers_score_function, time_distance_function]
|
|
|
|
records = AccountsIndex.query(function_score: { query: query, functions: functions })
|
|
.limit(limit_for_non_exact_results)
|
|
.offset(offset)
|
|
.objects
|
|
.compact
|
|
|
|
ActiveRecord::Associations::Preloader.new.preload(records, :account_stat)
|
|
|
|
records
|
|
rescue Faraday::ConnectionFailed, Parslet::ParseFailed
|
|
nil
|
|
end
|
|
|
|
def reputation_score_function
|
|
{
|
|
script_score: {
|
|
script: {
|
|
source: "(Math.max(doc['followers_count'].value, 0) + 0.0) / (Math.max(doc['followers_count'].value, 0) + Math.max(doc['following_count'].value, 0) + 1)",
|
|
},
|
|
},
|
|
}
|
|
end
|
|
|
|
def followers_score_function
|
|
{
|
|
script_score: {
|
|
script: {
|
|
source: "Math.log10(Math.max(doc['followers_count'].value, 0) + 2)",
|
|
},
|
|
},
|
|
}
|
|
end
|
|
|
|
def time_distance_function
|
|
{
|
|
gauss: {
|
|
last_status_at: {
|
|
scale: '30d',
|
|
offset: '30d',
|
|
decay: 0.3,
|
|
},
|
|
},
|
|
}
|
|
end
|
|
|
|
def must_clause
|
|
fields = %w(username username.* display_name display_name.*)
|
|
fields << 'text' << 'text.*' if options[:use_searchable_text]
|
|
|
|
[
|
|
{
|
|
multi_match: {
|
|
query: terms_for_query,
|
|
fields: fields,
|
|
type: 'best_fields',
|
|
operator: 'or',
|
|
},
|
|
},
|
|
]
|
|
end
|
|
|
|
def should_clause
|
|
[
|
|
{
|
|
multi_match: {
|
|
query: terms_for_query,
|
|
fields: %w(username username.* display_name display_name.*),
|
|
type: 'best_fields',
|
|
operator: 'and',
|
|
boost: 10,
|
|
},
|
|
},
|
|
]
|
|
end
|
|
|
|
def following_ids
|
|
@following_ids ||= account.active_relationships.pluck(:target_account_id) + [account.id]
|
|
end
|
|
|
|
def limit_for_non_exact_results
|
|
return 0 if @account.nil? && query.size < MIN_QUERY_LENGTH
|
|
|
|
if exact_match?
|
|
limit - 1
|
|
else
|
|
limit
|
|
end
|
|
end
|
|
|
|
def terms_for_query
|
|
if domain_is_local?
|
|
query_username
|
|
else
|
|
query
|
|
end
|
|
end
|
|
|
|
def split_query_string
|
|
@split_query_string ||= query.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.present?
|
|
end
|
|
|
|
def username_complete?
|
|
query.include?('@') && "@#{query}".match?(MENTION_ONLY_RE)
|
|
end
|
|
end
|