debian-mirror-gitlab/lib/tasks/gitlab/packages/events.rake

require 'logger'

desc "GitLab | Packages | Events | Generate hll counter events file for packages"
namespace :gitlab do
  namespace :packages do
    namespace :events do
      task generate: :environment do
        Rake::Task["gitlab:packages:events:generate_guest"].invoke
        Rake::Task["gitlab:packages:events:generate_unique"].invoke
      rescue => e
        logger.error("Error building events list: #{e}")
      end

      task generate_guest: :environment do
        logger = Logger.new(STDOUT)
        logger.info('Building list of package events...')

        path = Gitlab::UsageDataCounters::GuestPackageEventCounter::KNOWN_EVENTS_PATH
        File.open(path, "w") { |file| file << guest_events_list.to_yaml }

        logger.info("Events file `#{path}` generated successfully")
      rescue => e
        logger.error("Error building events list: #{e}")
      end

      task generate_unique: :environment do
        logger = Logger.new(STDOUT)
        logger.info('Building list of package events...')

        path = File.join(File.dirname(Gitlab::UsageDataCounters::HLLRedisCounter::KNOWN_EVENTS_PATH), 'package_events.yml')
        File.open(path, "w") { |file| file << generate_unique_events_list.to_yaml }

        logger.info("Events file `#{path}` generated successfully")
      rescue => e
        logger.error("Error building events list: #{e}")
      end

      private

      def event_pairs
        Packages::Event.event_types.keys.product(Packages::Event::EVENT_SCOPES.keys)
      end

      def generate_unique_events_list
        events = event_pairs.each_with_object([]) do |(event_type, event_scope), events|
          Packages::Event.originator_types.keys.excluding('guest').each do |originator|
            if name = Packages::Event.allowed_event_name(event_scope, event_type, originator)
              events << {
                "name" => name,
                "category" => "#{event_scope}_packages",
                "aggregation" => "weekly",
                "redis_slot" => "package",
                "feature_flag" => "collect_package_events_redis"
              }
            end
          end
        end

        events.sort_by { |event| event["name"] }
      end

      def guest_events_list
        event_pairs.map do |event_type, event_scope|
          Packages::Event.allowed_event_name(event_scope, event_type, "guest")
        end.compact.sort
      end
    end
  end
end