statuses.rb

# frozen_string_literal: true

class Trends::Statuses < Trends::Base
  PREFIX = 'trending_statuses'

  self.default_options = {
    threshold: 5,
    review_threshold: 3,
    score_halflife: 2.hours.freeze,
  }

  class Query < Trends::Query
    def filtered_for!(account)
      @account = account
      self
    end

    def filtered_for(account)
      clone.filtered_for!(account)
    end

    private

    def apply_scopes(scope)
      scope.includes(:account)
    end

    def perform_queries
      return super if @account.nil?

      statuses        = super
      account_ids     = statuses.map(&:account_id)
      account_domains = statuses.map(&:account_domain)

      preloaded_relations = {
        blocking: Account.blocking_map(account_ids, @account.id),
        blocked_by: Account.blocked_by_map(account_ids, @account.id),
        muting: Account.muting_map(account_ids, @account.id),
        following: Account.following_map(account_ids, @account.id),
        domain_blocking_by_domain: Account.domain_blocking_map_by_domain(account_domains, @account.id),
      }

      statuses.reject { |status| StatusFilter.new(status, @account, preloaded_relations).filtered? }
    end
  end

  def register(status, at_time = Time.now.utc)
    add(status.proper, status.account_id, at_time) if eligible?(status)
  end

  def add(status, _account_id, at_time = Time.now.utc)
    # We rely on the total reblogs and favourites count, so we
    # don't record which account did the what and when here

    record_used_id(status.id, at_time)
  end

  def query
    Query.new(key_prefix, klass)
  end

  def refresh(at_time = Time.now.utc)
    statuses = Status.where(id: (recently_used_ids(at_time) + currently_trending_ids(false, -1)).uniq).includes(:account, :media_attachments)
    calculate_scores(statuses, at_time)
    trim_older_items
  end

  def request_review
    statuses = Status.where(id: currently_trending_ids(false, -1)).includes(:account)