debian-mirror-gitlab/lib/api/helpers/badges_helpers.rb

# frozen_string_literal: true

module API
  module Helpers
    module BadgesHelpers
      include ::API::Helpers::MembersHelpers

      def find_badge(source)
        badge_id = params[:badge_id]

        if source.is_a?(Project)
          source.project_badges.find(badge_id)
        else
          source.badges.find(badge_id)
        end
      end

      def present_badges(source, records, options = {})
        entity_type = options[:with] || Entities::Badge
        badge_params = badge_source_params(source).merge(with: entity_type)

        present records, badge_params
      end

      def badge_source_params(source)
        project = if source.is_a?(Project)
                    source
                  else
                    GroupProjectsFinder.new(group: source, current_user: current_user).execute.first
                  end

        { project: project }
      end
    end
  end
end