lib/gitlab/graphql/todos_project_permission_preloader/field_extension.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

# frozen_string_literal: true

module Gitlab
  module Graphql
    module TodosProjectPermissionPreloader
      class FieldExtension < ::GraphQL::Schema::FieldExtension
        def after_resolve(value:, memo:, **rest)
          todos = value.to_a

          Preloaders::UserMaxAccessLevelInProjectsPreloader.new(
            todos.map(&:project).compact,
            current_user(rest)
          ).execute

          value
        end

        private

        def current_user(options)
          options.dig(:context, :current_user)
        end
      end
    end
  end
end