Added integration of root types

Author: michaelstaib

src/HotChocolate/Core/src/Execution/Configuration/ConfigurationContext.cs

@@ -1,10 +1,14 @@
+using HotChocolate.Types.Descriptors;
+
 namespace HotChocolate.Execution.Configuration;
 
 /// <summary>
 /// The configuration context is used during the setup of the schema and request executor.
 /// </summary>
 public sealed class ConfigurationContext : IHasContextData
 {
+    private IDescriptorContext? _descriptorContext;
+
     /// <summary>
     /// Initializes a new instance of <see cref="ConfigurationContext"/>.
     /// </summary>
@@ -52,4 +56,20 @@ public ConfigurationContext(
     /// Gets the configuration context data which can be used by hooks to store arbitrary state.
     /// </summary>
     public IDictionary<string, object?> ContextData { get; } = new Dictionary<string, object?>();
+
+    /// <summary>
+    /// Gets the descriptor context.
+    /// </summary>
+    public IDescriptorContext DescriptorContext
+    {
+        get
+        {
+            if(_descriptorContext is null)
+            {
+                _descriptorContext = SchemaBuilder.CreateContext();
+            }
+
+            return _descriptorContext;
+        }
+    }
 }

src/HotChocolate/Core/src/Execution/DependencyInjection/RequestExecutorBuilderExtensions.cs

@@ -2,6 +2,7 @@
 using HotChocolate.Execution;
 using HotChocolate.Execution.Configuration;
 using HotChocolate.Execution.Options;
+using HotChocolate.Types.Descriptors;
 using Microsoft.Extensions.Options;
 
 // ReSharper disable once CheckNamespace
@@ -118,6 +119,30 @@ public static IRequestExecutorBuilder ConfigureSchema(
                     (ctx, sp) => configureSchema(sp, ctx.SchemaBuilder))));
     }
 
+    /// <summary>
+    /// Adds a delegate that will be used to configure the descriptor context.
+    /// </summary>
+    public static IRequestExecutorBuilder ConfigureDescriptorContext(
+        this IRequestExecutorBuilder builder,
+        Action<IDescriptorContext> configure)
+    {
+        if (builder is null)
+        {
+            throw new ArgumentNullException(nameof(builder));
+        }
+
+        if (configure is null)
+        {
+            throw new ArgumentNullException(nameof(configure));
+        }
+
+        return Configure(
+            builder,
+            options => options.OnConfigureSchemaBuilderHooks.Add(
+                new OnConfigureSchemaBuilderAction(
+                    (ctx, _) => configure(ctx.DescriptorContext))));
+    }
+
     /// <summary>
     /// Adds a delegate that will be used to configure a named <see cref="ISchema"/>.
     /// </summary>

src/HotChocolate/Core/src/Execution/RequestExecutorResolver.cs

@@ -377,7 +377,7 @@ private static async ValueTask<ISchema> CreateSchemaAsync(
             .AddServices(schemaServices)
             .SetContextData(typeof(RequestExecutorOptions).FullName!, executorOptions);
 
-        var descriptorContext = context.SchemaBuilder.CreateContext();
+        var descriptorContext = context.DescriptorContext;
 
         await foreach (var member in
            typeModuleChangeMonitor.CreateTypesAsync(descriptorContext)

src/HotChocolate/Core/src/Types.Analyzers/FileBuilders/IOutputTypeFileBuilder.cs

@@ -7,9 +7,11 @@ public interface IOutputTypeFileBuilder
     void WriteHeader();
 
     void WriteBeginNamespace();
+
     void WriteEndNamespace();
 
     string WriteBeginClass(string typeName);
+
     void WriteEndClass();
 
     void WriteInitializeMethod(IOutputTypeInfo typeInfo);

src/HotChocolate/Core/src/Types.Analyzers/FileBuilders/InterfaceTypeExtensionFileBuilder.cs

@@ -47,7 +47,7 @@ public void WriteInitializeMethod(IOutputTypeInfo typeInfo)
     {
         _writer.WriteIndentedLine(
             "internal static void Initialize(global::HotChocolate.Types.IInterfaceTypeDescriptor<{0}> descriptor)",
-            typeInfo.RuntimeType.ToFullyQualified());
+            typeInfo.RuntimeType!.ToFullyQualified());
         _writer.WriteIndentedLine("{");
 
         using (_writer.IncreaseIndent())
@@ -127,6 +127,6 @@ public void WriteConfigureMethod(IOutputTypeInfo typeInfo)
     {
         _writer.WriteIndentedLine(
             "static partial void Configure(global::HotChocolate.Types.IInterfaceTypeDescriptor<{0}> descriptor);",
-            typeInfo.RuntimeType.ToFullyQualified());
+            typeInfo.RuntimeType!.ToFullyQualified());
     }
 }

src/HotChocolate/Core/src/Types.Analyzers/FileBuilders/ModuleFileBuilder.cs

@@ -1,6 +1,7 @@
 using System.Text;
 using HotChocolate.Types.Analyzers.Helpers;
 using HotChocolate.Types.Analyzers.Models;
+using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.Text;
 
 namespace HotChocolate.Types.Analyzers.FileBuilders;
@@ -85,23 +86,27 @@ public void WriteEnsureObjectTypeExtensionIsRegistered(string runtimeTypeName)
     public void WriteEnsureInterfaceTypeExtensionIsRegistered(string runtimeTypeName)
         => _writer.WriteIndentedLine("builder.AddType<InterfaceType<{0}>>();", runtimeTypeName);
 
-    public void WriteRegisterObjectTypeExtension(string runtimeTypeName, string extensionType)
+    public void WriteRegisterTypeExtension(string key, string runtimeTypeName, string extensionType)
     {
-        _writer.WriteIndentedLine("global::{0}.TryAdd<{1}>(", WellKnownTypes.TypeConfiguration, runtimeTypeName);
+        _writer.WriteIndentedLine(
+            "builder.ConfigureDescriptorContext(ctx => ctx.TypeConfiguration.TryAdd<{0}>(",
+            runtimeTypeName);
         using (_writer.IncreaseIndent())
         {
-            _writer.WriteIndentedLine("\"{0}\",", extensionType);
-            _writer.WriteIndentedLine("() => {0}.Initialize);", extensionType);
+            _writer.WriteIndentedLine("\"{0}\",", key);
+            _writer.WriteIndentedLine("() => {0}.Initialize));", extensionType);
         }
     }
 
-    public void WriteRegisterInterfaceTypeExtension(string runtimeTypeName, string extensionType)
+    public void WriteRegisterRootTypeExtension(string key, OperationType operation, string extensionType)
     {
-        _writer.WriteIndentedLine("global::{0}.TryAdd<{1}>(", WellKnownTypes.TypeConfiguration, runtimeTypeName);
+        _writer.WriteIndentedLine("builder.ConfigureDescriptorContext(ctx => ctx.TypeConfiguration.TryAdd(");
+
         using (_writer.IncreaseIndent())
         {
-            _writer.WriteIndentedLine("\"{0}\",", extensionType);
-            _writer.WriteIndentedLine("() => {0}.Initialize);", extensionType);
+            _writer.WriteIndentedLine("\"{0}\",", key);
+            _writer.WriteIndentedLine("global::HotChocolate.Types.OperationType.{0},", operation);
+            _writer.WriteIndentedLine("() => {0}.Initialize));", extensionType);
         }
     }