discourse/lib/admin_user_index_query.rb

149 lines
4.1 KiB
Ruby
Raw Normal View History

require_dependency 'trust_level'
class AdminUserIndexQuery
def initialize(params = {}, klass = User, trust_levels = TrustLevel.levels)
@params = params
@query = initialize_query_with_order(klass.joins(:primary_email))
@trust_levels = trust_levels
end
attr_reader :params, :trust_levels
SORTABLE_MAPPING = {
'created' => 'created_at',
'last_emailed' => "COALESCE(last_emailed_at, to_date('1970-01-01', 'YYYY-MM-DD'))",
'seen' => "COALESCE(last_seen_at, to_date('1970-01-01', 'YYYY-MM-DD'))",
'username' => 'username',
'email' => 'email',
'trust_level' => 'trust_level',
'days_visited' => 'user_stats.days_visited',
'posts_read' => 'user_stats.posts_read_count',
'topics_viewed' => 'user_stats.topics_entered',
'posts' => 'user_stats.post_count',
'read_time' => 'user_stats.time_read'
}
2017-07-27 21:20:09 -04:00
def find_users(limit = 100)
page = params[:page].to_i - 1
if page < 0
2017-07-27 21:20:09 -04:00
page = 0
end
find_users_query.limit(limit).offset(page * limit)
end
def count_users
find_users_query.count
end
def custom_direction
asc = params[:ascending]
asc.present? && asc ? "ASC" : "DESC"
end
2014-11-24 13:48:08 -05:00
def initialize_query_with_order(klass)
order = []
custom_order = params[:order]
if custom_order.present? &&
without_dir = SORTABLE_MAPPING[custom_order.downcase.sub(/ (asc|desc)$/, '')]
order << "#{without_dir} #{custom_direction}"
end
if !custom_order.present?
if params[:query] == "active"
order << "COALESCE(users.last_seen_at, to_date('1970-01-01', 'YYYY-MM-DD')) DESC"
else
order << "users.created_at DESC"
end
order << "users.username"
2014-11-24 13:48:08 -05:00
end
if params[:stats].present? && params[:stats] == false
klass.order(order.reject(&:blank?).join(","))
else
2018-06-28 04:12:32 -04:00
klass.includes(:user_stat, :user_second_factors)
.order(order.reject(&:blank?).join(","))
end
2014-11-24 13:48:08 -05:00
end
2014-11-24 13:48:08 -05:00
def filter_by_trust
levels = trust_levels.map { |key, _| key.to_s }
if levels.include?(params[:query])
@query.where('trust_level = ?', trust_levels[params[:query].to_sym])
end
2014-11-24 13:48:08 -05:00
end
def suspect_users
where_conds = []
# One signal: no reading yet the user has bio text
where_conds << "user_stats.posts_read_count <= 1 AND user_stats.topics_entered <= 1"
@query.activated
.human_users
2017-07-27 21:20:09 -04:00
.references(:user_stats)
.includes(:user_profile)
.where("COALESCE(user_profiles.bio_raw, '') != ''")
.where('users.created_at <= ?', 1.day.ago)
.where(where_conds.map { |c| "(#{c})" }.join(" OR "))
end
2014-11-24 13:48:08 -05:00
def filter_by_query_classification
case params[:query]
2017-07-27 21:20:09 -04:00
when 'staff' then @query.where("admin or moderator")
when 'admins' then @query.where(admin: true)
when 'moderators' then @query.where(moderator: true)
2017-11-10 12:18:08 -05:00
when 'silenced' then @query.silenced
2017-07-27 21:20:09 -04:00
when 'suspended' then @query.suspended
when 'pending' then @query.not_suspended.where(approved: false, active: true)
2017-07-27 21:20:09 -04:00
when 'suspect' then suspect_users
when 'staged' then @query.where(staged: true)
end
2014-11-24 13:48:08 -05:00
end
2014-11-24 13:48:08 -05:00
def filter_by_search
if params[:email].present?
return @query.where('user_emails.email = ?', params[:email].downcase)
end
filter = params[:filter]
if filter.present?
filter.strip!
if ip = IPAddr.new(filter) rescue nil
2014-11-24 13:48:08 -05:00
@query.where('ip_address <<= :ip OR registration_ip_address <<= :ip', ip: ip.to_cidr_s)
else
@query.filter_by_username_or_email(filter)
end
end
2014-11-24 13:48:08 -05:00
end
2014-11-24 13:48:08 -05:00
def filter_by_ip
if params[:ip].present?
@query.where('ip_address = :ip OR registration_ip_address = :ip', ip: params[:ip].strip)
end
2014-11-24 13:48:08 -05:00
end
2014-11-24 13:48:08 -05:00
def filter_exclude
if params[:exclude].present?
@query.where('users.id != ?', params[:exclude])
end
2014-11-24 13:48:08 -05:00
end
2014-11-24 13:48:08 -05:00
# this might not be needed in rails 4 ?
def append(active_relation)
@query = active_relation if active_relation
end
2014-11-24 13:48:08 -05:00
def find_users_query
append filter_by_trust
append filter_by_query_classification
append filter_by_ip
append filter_exclude
append filter_by_search
@query
2014-11-24 13:48:08 -05:00
end
end