# frozen_string_literal: true

module Gitlab::UsageDataCounters
  class NoteCounter < BaseCounter
    KNOWN_EVENTS = %w[create].freeze
    PREFIX = 'note'
    COUNTABLE_TYPES = %w[Snippet Commit MergeRequest].freeze

    class << self
      def redis_key(event, noteable_type)
        "#{super(event)}_#{noteable_type}".upcase
      end

      def count(event, noteable_type)
        return unless countable?(noteable_type)

        increment(redis_key(event, noteable_type))
      end

      def read(event, noteable_type)
        return 0 unless countable?(noteable_type)

        total_count(redis_key(event, noteable_type))
      end

      def totals
        COUNTABLE_TYPES.map do |countable_type|
          [:"#{countable_type.underscore}_comment", read(:create, countable_type)]
        end.to_h
      end

      private

      def countable?(noteable_type)
        COUNTABLE_TYPES.include?(noteable_type.to_s)
      end
    end
  end
end