debian-mirror-gitlab/lib/api/entities/project_with_access.rb

# frozen_string_literal: true

module API
  module Entities
    class ProjectWithAccess < Project
      expose :permissions do
        expose :project_access, using: Entities::ProjectAccess do |project, options|
          if options[:project_members]
            options[:project_members].find { |member| member.source_id == project.id }
          else
            project.member(options[:current_user])
          end
        end

        expose :group_access, using: Entities::GroupAccess do |project, options|
          if project.group
            if options[:group_members]
              options[:group_members].find { |member| member.source_id == project.namespace_id }
            else
              project.group.highest_group_member(options[:current_user])
            end
          end
        end
      end

      # rubocop: disable CodeReuse/ActiveRecord
      def self.preload_relation(projects_relation, options = {})
        if ::Feature.enabled?(:projects_preloader_fix)
          super(projects_relation, options)
        else
          relation = super(projects_relation, options)
          # use reselect to override the existing select and
          # prevent an error `subquery has too many columns`
          project_ids = relation.reselect('projects.id')
          namespace_ids = relation.reselect(:namespace_id)

          options[:project_members] = options[:current_user]
            .project_members
            .where(source_id: project_ids)
            .preload(:source, user: [notification_settings: :source])

          options[:group_members] = options[:current_user]
            .group_members
            .where(source_id: namespace_ids)
            .preload(:source, user: [notification_settings: :source])

          relation
        end
      end

      def self.postload_relation(projects_relation, options = {})
        return unless ::Feature.enabled?(:projects_preloader_fix)

        options[:project_members] = options[:current_user]
          .project_members
          .where(source_id: projects_relation.subquery(:id))
          .preload(:source, user: [notification_settings: :source])

        options[:group_members] = options[:current_user]
          .group_members
          .where(source_id: projects_relation.subquery(:namespace_id))
          .preload(:source, user: [notification_settings: :source])
      end
      # rubocop: enable CodeReuse/ActiveRecord
    end
  end
end