preloading of fields

app/graphql/types/base_object.rb

@@ -62,14 +62,29 @@ def self.expose_abilities(abilities, entity_name: graphql_name, subject_resolver
       define_method(:user_abilities) { object || {} }
     end
 
-    def self.lookahead_field(field, base_scope:, lookaheads: [], conditional_lookaheads: {})
+    def self.lookahead_preload(field, preload, profile: nil)
+      @lookahead_preloads ||= Hash.new { |hash, key| hash[key] = {} }
+      @lookahead_preloads[profile][field] = preload
+    end
+
+    def self.lookahead_preloads(profile: nil)
+      own_preloads = @lookahead_preloads || {}
+      base_preloads = superclass.respond_to?(:lookahead_preloads) ? superclass.lookahead_preloads : {}
+
+      base_preloads.merge(own_preloads.fetch(nil, {})).merge(own_preloads.fetch(profile, {}))
+    end
+
+    def self.lookahead_field(field, base_scope:, lookaheads: [], conditional_lookaheads: {}, preload_type: nil,
+                             preload_profile: nil)
       define_method(field) do |*_args, lookahead:, **_kwargs|
         field_selected = lambda do |f|
           lookahead.selects?(f) ||
             lookahead.selection(:nodes).selects?(f) ||
             lookahead.selection(:edges).selection(:node).selects?(f)
         end
 
+        type_lookaheads = preload_type&.lookahead_preloads(profile: preload_profile) || {}
+        conditional_lookaheads = conditional_lookaheads.merge(type_lookaheads)
         scope = lookaheads.reduce(base_scope.call(object)) { |acc, f| acc.preload(f) }
 
         conditional_lookaheads.reduce(scope) do |acc, (f, preload_field)|

app/graphql/types/namespace_member_type.rb

@@ -12,6 +12,10 @@ class NamespaceMemberType < Types::BaseObject
     field :member_roles, NamespaceMemberRoleType.connection_type, null: false, description: 'Memberroles of the member'
     field :roles, NamespaceRoleType.connection_type, null: false, description: 'Roles of the member'
 
+    lookahead_preload :namespace, :namespace
+    lookahead_preload :user, :user
+    lookahead_preload :namespace, { namespace: :namespace_members }, profile: :namespace_memberships
+
     expose_abilities %i[
       assign_member_roles
       delete_member

app/graphql/types/namespace_type.rb

@@ -24,7 +24,7 @@ class NamespaceType < Types::BaseObject
     field :runtimes, Types::RuntimeType.connection_type, null: false, description: 'Runtime of the namespace'
 
     lookahead_field :members, base_scope: ->(object) { object.namespace_members },
-                              conditional_lookaheads: { user: :user, namespace: :namespace }
+                              preload_type: Types::NamespaceMemberType
 
     expose_abilities %i[
       invite_member

app/graphql/types/user_type.rb

@@ -48,7 +48,8 @@ class UserType < Types::BaseObject
 
     lookahead_field :namespace_memberships,
                     base_scope: ->(object) { object.namespace_memberships },
-                    conditional_lookaheads: { user: :user, namespace: { namespace: :namespace_members } }
+                    preload_type: Types::NamespaceMemberType,
+                    preload_profile: :namespace_memberships
 
     expose_abilities %i[
       manage_mfa