121 lines
2.8 KiB
Ruby
121 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class RelationshipFilter
|
|
KEYS = %i(
|
|
relationship
|
|
status
|
|
by_domain
|
|
activity
|
|
order
|
|
location
|
|
).freeze
|
|
|
|
attr_reader :params, :account
|
|
|
|
def initialize(account, params)
|
|
@account = account
|
|
@params = params
|
|
|
|
set_defaults!
|
|
end
|
|
|
|
def results
|
|
scope = scope_for('relationship', params['relationship'].to_s.strip)
|
|
|
|
params.each do |key, value|
|
|
next if %w(relationship page).include?(key)
|
|
|
|
scope.merge!(scope_for(key.to_s, value.to_s.strip)) if value.present?
|
|
end
|
|
|
|
scope
|
|
end
|
|
|
|
private
|
|
|
|
def set_defaults!
|
|
params['relationship'] = 'following' if params['relationship'].blank?
|
|
params['order'] = 'recent' if params['order'].blank?
|
|
end
|
|
|
|
def scope_for(key, value)
|
|
case key
|
|
when 'relationship'
|
|
relationship_scope(value)
|
|
when 'by_domain'
|
|
by_domain_scope(value)
|
|
when 'location'
|
|
location_scope(value)
|
|
when 'status'
|
|
status_scope(value)
|
|
when 'order'
|
|
order_scope(value)
|
|
when 'activity'
|
|
activity_scope(value)
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown filter: #{key}"
|
|
end
|
|
end
|
|
|
|
def relationship_scope(value)
|
|
case value
|
|
when 'following'
|
|
account.following.includes(:account_stat).reorder(nil)
|
|
when 'followed_by'
|
|
account.followers.includes(:account_stat).reorder(nil)
|
|
when 'mutual'
|
|
account.followers.includes(:account_stat).reorder(nil).merge(Account.where(id: account.following))
|
|
when 'invited'
|
|
Account.joins(user: :invite).merge(Invite.where(user: account.user)).includes(:account_stat).reorder(nil)
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown relationship: #{value}"
|
|
end
|
|
end
|
|
|
|
def by_domain_scope(value)
|
|
Account.where(domain: value)
|
|
end
|
|
|
|
def location_scope(value)
|
|
case value
|
|
when 'local'
|
|
Account.local
|
|
when 'remote'
|
|
Account.remote
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown location: #{value}"
|
|
end
|
|
end
|
|
|
|
def status_scope(value)
|
|
case value
|
|
when 'moved'
|
|
Account.where.not(moved_to_account_id: nil)
|
|
when 'primary'
|
|
Account.where(moved_to_account_id: nil)
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown status: #{value}"
|
|
end
|
|
end
|
|
|
|
def order_scope(value)
|
|
case value
|
|
when 'active'
|
|
Account.by_recent_status
|
|
when 'recent'
|
|
params[:relationship] == 'invited' ? Account.recent : Follow.recent
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown order: #{value}"
|
|
end
|
|
end
|
|
|
|
def activity_scope(value)
|
|
case value
|
|
when 'dormant'
|
|
Account.joins(:account_stat).where(account_stat: { last_status_at: [nil, ...1.month.ago] })
|
|
else
|
|
raise Mastodon::InvalidParameterError, "Unknown activity: #{value}"
|
|
end
|
|
end
|
|
end
|