mastodon/app/models/trends/status_filter.rb

56 lines
954 B
Ruby

# frozen_string_literal: true
class Trends::StatusFilter
KEYS = %i(
trending
locale
).freeze
attr_reader :params
def initialize(params)
@params = params
end
def results
scope = initial_scope
params.each do |key, value|
next if %w(page).include?(key.to_s)
scope.merge!(scope_for(key, value.to_s.strip)) if value.present?
end
scope
end
private
def initial_scope
Status.select(Status.arel_table[Arel.star])
.joins(:trend)
.eager_load(:trend)
.reorder(score: :desc)
end
def scope_for(key, value)
case key.to_s
when 'trending'
trending_scope(value)
when 'locale'
StatusTrend.where(language: value)
else
raise Mastodon::InvalidParameterError, "Unknown filter: #{key}"
end
end
def trending_scope(value)
case value
when 'allowed'
StatusTrend.allowed
else
StatusTrend.all
end
end
end