2017-04-13 20:04:23 +09:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
class AccountFilter
|
2020-01-20 23:55:03 +09:00
|
|
|
KEYS = %i(
|
2021-12-06 05:48:39 +09:00
|
|
|
origin
|
|
|
|
status
|
2022-07-05 09:41:40 +09:00
|
|
|
role_ids
|
2020-01-20 23:55:03 +09:00
|
|
|
username
|
2021-12-06 05:48:39 +09:00
|
|
|
by_domain
|
2020-01-20 23:55:03 +09:00
|
|
|
display_name
|
|
|
|
email
|
|
|
|
ip
|
2021-12-06 05:48:39 +09:00
|
|
|
invited_by
|
2020-03-08 23:39:13 +09:00
|
|
|
order
|
2020-01-20 23:55:03 +09:00
|
|
|
).freeze
|
|
|
|
|
2017-04-13 20:04:23 +09:00
|
|
|
attr_reader :params
|
|
|
|
|
|
|
|
def initialize(params)
|
2023-03-03 00:21:04 +09:00
|
|
|
@params = params.to_h.symbolize_keys
|
2017-04-13 20:04:23 +09:00
|
|
|
end
|
|
|
|
|
|
|
|
def results
|
2022-01-16 21:23:50 +09:00
|
|
|
scope = Account.includes(:account_stat, user: [:ips, :invite_request]).without_instance_actor.reorder(nil)
|
2017-10-06 06:42:05 +09:00
|
|
|
|
2023-03-03 00:21:04 +09:00
|
|
|
relevant_params.each do |key, value|
|
2022-02-15 05:27:53 +09:00
|
|
|
next if key.to_s == 'page'
|
|
|
|
|
2022-07-05 09:41:40 +09:00
|
|
|
scope.merge!(scope_for(key, value)) if value.present?
|
2017-04-13 20:04:23 +09:00
|
|
|
end
|
2017-10-06 06:42:05 +09:00
|
|
|
|
2017-04-13 20:04:23 +09:00
|
|
|
scope
|
|
|
|
end
|
|
|
|
|
2017-05-17 10:00:34 +09:00
|
|
|
private
|
|
|
|
|
2023-03-03 00:21:04 +09:00
|
|
|
def relevant_params
|
|
|
|
params.tap do |args|
|
|
|
|
args.delete(:origin) if origin_is_remote_and_domain_present?
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def origin_is_remote_and_domain_present?
|
|
|
|
params[:origin] == 'remote' && params[:by_domain].present?
|
|
|
|
end
|
|
|
|
|
2017-04-13 20:04:23 +09:00
|
|
|
def scope_for(key, value)
|
2017-05-17 10:00:34 +09:00
|
|
|
case key.to_s
|
2021-12-06 05:48:39 +09:00
|
|
|
when 'origin'
|
|
|
|
origin_scope(value)
|
2022-07-05 09:41:40 +09:00
|
|
|
when 'role_ids'
|
|
|
|
role_scope(value)
|
2021-12-06 05:48:39 +09:00
|
|
|
when 'status'
|
|
|
|
status_scope(value)
|
2017-05-17 10:00:34 +09:00
|
|
|
when 'by_domain'
|
2022-11-28 04:47:29 +09:00
|
|
|
Account.where(domain: value.to_s.strip)
|
2017-05-17 10:00:34 +09:00
|
|
|
when 'username'
|
2023-04-17 21:16:36 +09:00
|
|
|
Account.matches_username(value.to_s.strip.delete_prefix('@'))
|
2017-05-17 10:00:34 +09:00
|
|
|
when 'display_name'
|
2022-11-28 04:47:29 +09:00
|
|
|
Account.matches_display_name(value.to_s.strip)
|
2017-05-17 10:00:34 +09:00
|
|
|
when 'email'
|
2022-11-28 04:47:29 +09:00
|
|
|
accounts_with_users.merge(User.matches_email(value.to_s.strip))
|
2017-05-17 10:00:34 +09:00
|
|
|
when 'ip'
|
2022-02-13 09:58:26 +09:00
|
|
|
valid_ip?(value) ? accounts_with_users.merge(User.matches_ip(value).group('users.id, accounts.id')) : Account.none
|
2021-12-06 05:48:39 +09:00
|
|
|
when 'invited_by'
|
|
|
|
invited_by_scope(value)
|
2020-03-08 23:39:13 +09:00
|
|
|
when 'order'
|
|
|
|
order_scope(value)
|
2017-04-13 20:04:23 +09:00
|
|
|
else
|
2022-11-14 16:06:06 +09:00
|
|
|
raise Mastodon::InvalidParameterError, "Unknown filter: #{key}"
|
2017-04-13 20:04:23 +09:00
|
|
|
end
|
|
|
|
end
|
2017-05-17 10:00:34 +09:00
|
|
|
|
2021-12-06 05:48:39 +09:00
|
|
|
def origin_scope(value)
|
|
|
|
case value.to_s
|
|
|
|
when 'local'
|
|
|
|
Account.local
|
|
|
|
when 'remote'
|
|
|
|
Account.remote
|
|
|
|
else
|
2022-11-14 16:06:06 +09:00
|
|
|
raise Mastodon::InvalidParameterError, "Unknown origin: #{value}"
|
2021-12-06 05:48:39 +09:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def status_scope(value)
|
|
|
|
case value.to_s
|
|
|
|
when 'active'
|
|
|
|
Account.without_suspended
|
|
|
|
when 'pending'
|
|
|
|
accounts_with_users.merge(User.pending)
|
|
|
|
when 'suspended'
|
|
|
|
Account.suspended
|
2022-03-29 06:57:38 +09:00
|
|
|
when 'disabled'
|
2022-12-16 01:30:47 +09:00
|
|
|
accounts_with_users.merge(User.disabled).without_suspended
|
2022-03-29 06:57:38 +09:00
|
|
|
when 'silenced'
|
|
|
|
Account.silenced
|
2022-11-14 16:06:06 +09:00
|
|
|
when 'sensitized'
|
|
|
|
Account.sensitized
|
2021-12-06 05:48:39 +09:00
|
|
|
else
|
2022-11-14 16:06:06 +09:00
|
|
|
raise Mastodon::InvalidParameterError, "Unknown status: #{value}"
|
2021-12-06 05:48:39 +09:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2020-03-08 23:39:13 +09:00
|
|
|
def order_scope(value)
|
2021-12-06 05:48:39 +09:00
|
|
|
case value.to_s
|
2020-03-08 23:39:13 +09:00
|
|
|
when 'active'
|
2023-03-03 00:21:04 +09:00
|
|
|
accounts_with_users
|
|
|
|
.left_joins(:account_stat)
|
|
|
|
.order(
|
|
|
|
Arel.sql(
|
|
|
|
<<~SQL.squish
|
|
|
|
COALESCE(users.current_sign_in_at, account_stats.last_status_at, to_timestamp(0)) DESC, accounts.id DESC
|
|
|
|
SQL
|
|
|
|
)
|
|
|
|
)
|
2020-03-08 23:39:13 +09:00
|
|
|
when 'recent'
|
|
|
|
Account.recent
|
|
|
|
else
|
2022-11-14 16:06:06 +09:00
|
|
|
raise Mastodon::InvalidParameterError, "Unknown order: #{value}"
|
2020-03-08 23:39:13 +09:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2021-12-06 05:48:39 +09:00
|
|
|
def invited_by_scope(value)
|
|
|
|
Account.left_joins(user: :invite).merge(Invite.where(user_id: value.to_s))
|
|
|
|
end
|
|
|
|
|
2022-07-05 09:41:40 +09:00
|
|
|
def role_scope(value)
|
|
|
|
accounts_with_users.merge(User.where(role_id: Array(value).map(&:to_s)))
|
2021-12-06 05:48:39 +09:00
|
|
|
end
|
|
|
|
|
2017-05-23 04:50:58 +09:00
|
|
|
def accounts_with_users
|
2021-12-06 05:48:39 +09:00
|
|
|
Account.left_joins(:user)
|
2017-05-23 04:50:58 +09:00
|
|
|
end
|
|
|
|
|
2017-05-17 10:00:34 +09:00
|
|
|
def valid_ip?(value)
|
2022-07-05 09:41:40 +09:00
|
|
|
IPAddr.new(value.to_s) && true
|
2017-05-17 10:00:34 +09:00
|
|
|
rescue IPAddr::InvalidAddressError
|
|
|
|
false
|
|
|
|
end
|
2017-04-13 20:04:23 +09:00
|
|
|
end
|