discourse/app/models/stat.rb

# frozen_string_literal: true

class Stat
  def initialize(name, expose_via_api: false, &block)
    @name = name
    @expose_via_api = expose_via_api
    @block = block
  end

  attr_reader :name, :expose_via_api

  def calculate
    @block.call.transform_keys { |key| build_key(key) }
  rescue StandardError => err
    Discourse.warn_exception(err, message: "Unexpected error when collecting #{@name} About stats.")
    {}
  end

  def self.all_stats
    calculate(_all_stats)
  end

  def self.api_stats
    calculate(_api_stats)
  end

  private

  def build_key(key)
    "#{@name}_#{key}".to_sym
  end

  def self._all_stats
    core_stats.concat(plugin_stats)
  end

  def self.calculate(stats)
    stats.map { |stat| stat.calculate }.reduce(Hash.new, :merge)
  end

  def self.core_stats
    list = [
      Stat.new("topics", expose_via_api: true) { Statistics.topics },
      Stat.new("posts", expose_via_api: true) { Statistics.posts },
      Stat.new("users", expose_via_api: true) { Statistics.users },
      Stat.new("active_users", expose_via_api: true) { Statistics.active_users },
      Stat.new("likes", expose_via_api: true) { Statistics.likes },
      Stat.new("participating_users", expose_via_api: true) { Statistics.participating_users },
    ]

    if SiteSetting.display_eu_visitor_stats
      list.concat(
        [
          Stat.new("visitors", expose_via_api: true) { Statistics.visitors },
          Stat.new("eu_visitors", expose_via_api: true) { Statistics.eu_visitors },
        ],
      )
    end

    list
  end

  def self._api_stats
    _all_stats.select { |stat| stat.expose_via_api }
  end

  def self.plugin_stats
    DiscoursePluginRegistry.stats
  end

  private_class_method :_all_stats, :calculate, :core_stats, :_api_stats, :plugin_stats
end