diff options
Diffstat (limited to 'lib/gitlab/graphql/project/dast_profile_connection_extension.rb')
-rw-r--r-- | lib/gitlab/graphql/project/dast_profile_connection_extension.rb | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/lib/gitlab/graphql/project/dast_profile_connection_extension.rb b/lib/gitlab/graphql/project/dast_profile_connection_extension.rb new file mode 100644 index 00000000000..a3c3f2f2b7e --- /dev/null +++ b/lib/gitlab/graphql/project/dast_profile_connection_extension.rb @@ -0,0 +1,22 @@ +# frozen_string_literal: true +module Gitlab + module Graphql + module Project + class DastProfileConnectionExtension < GraphQL::Schema::Field::ConnectionExtension + def after_resolve(value:, object:, context:, **rest) + preload_authorizations(context[:project_dast_profiles]) + context[:project_dast_profiles] = nil + value + end + + def preload_authorizations(dast_profiles) + return unless dast_profiles + + projects = dast_profiles.map(&:project) + users = dast_profiles.filter_map { |dast_profile| dast_profile.dast_profile_schedule&.owner } + Preloaders::UsersMaxAccessLevelInProjectsPreloader.new(projects: projects, users: users).execute + end + end + end + end +end |