wip

Author: michaelstaib

.build/Helpers.cs

@@ -25,7 +25,7 @@ static class Helpers
         Path.Combine("HotChocolate", "Data"),
         Path.Combine("HotChocolate", "Marten"),
         Path.Combine("HotChocolate", "MongoDb"),
-        Path.Combine("HotChocolate", "OpenApi"),
+        // Path.Combine("HotChocolate", "OpenApi"),
         Path.Combine("HotChocolate", "Primitives"),
         Path.Combine("HotChocolate", "Raven"),
         Path.Combine("HotChocolate", "Mutable"),

src/HotChocolate/Fusion-vnext/src/Fusion.Composition/SchemaVisitors/DiscoverLookupsSchemaVisitor.cs

@@ -7,7 +7,7 @@
 namespace HotChocolate.Fusion.SchemaVisitors;
 
 internal sealed class DiscoverLookupsSchemaVisitor(MutableSchemaDefinition schema)
-    : SchemaVisitor<DiscoverLookupsContext>
+    : MutableSchemaDefinitionVisitor<DiscoverLookupsContext>
 {
     private readonly MultiValueDictionary<string, MutableObjectTypeDefinition>
         _implementingTypesByInterfaceName = [];

src/HotChocolate/Mutable/src/Types.Mutable/MutableSchemaDefinition.cs

@@ -10,8 +10,8 @@ namespace HotChocolate.Types.Mutable;
 /// </summary>
 public class MutableSchemaDefinition
     : INamedTypeSystemMemberDefinition<MutableSchemaDefinition>
-    , ISchemaDefinition
-    , IFeatureProvider
+        , ISchemaDefinition
+        , IFeatureProvider
 {
     private readonly List<SchemaCoordinate> _allDefinitionCoordinates = [];
     private MutableObjectTypeDefinition? _queryType;
@@ -45,7 +45,8 @@ public MutableObjectTypeDefinition? QueryType
         }
     }
 
-    IObjectTypeDefinition? ISchemaDefinition.QueryType => QueryType;
+    IObjectTypeDefinition ISchemaDefinition.QueryType
+        => QueryType ?? throw new InvalidOperationException("The query type is not defined.");
 
     /// <summary>
     /// Gets or sets the mutation type.
@@ -251,6 +252,48 @@ public MutableObjectTypeDefinition GetOperationType(OperationType operationType)
     IObjectTypeDefinition ISchemaDefinition.GetOperationType(OperationType operationType)
         => GetOperationType(operationType);
 
+    public IEnumerable<MutableObjectTypeDefinition> GetPossibleTypes(ITypeDefinition abstractType)
+    {
+        if (abstractType.Kind is not TypeKind.Union and not TypeKind.Interface and not TypeKind.Object)
+        {
+            throw new ArgumentException(
+                "The specified type is not an abstract type.",
+                nameof(abstractType));
+        }
+
+        if (abstractType is MutableUnionTypeDefinition unionType)
+        {
+            foreach (var possibleType in unionType.Types.AsEnumerable())
+            {
+                yield return possibleType;
+            }
+
+            yield break;
+        }
+
+        if (abstractType is MutableInterfaceTypeDefinition interfaceType)
+        {
+            foreach (var type in Types)
+            {
+                if (type is MutableObjectTypeDefinition obj
+                    && obj.Implements.ContainsName(interfaceType.Name))
+                {
+                    yield return obj;
+                }
+            }
+
+            yield break;
+        }
+
+        if (abstractType is MutableObjectTypeDefinition objectType)
+        {
+            yield return objectType;
+        }
+    }
+
+    IEnumerable<IObjectTypeDefinition> ISchemaDefinition.GetPossibleTypes(ITypeDefinition abstractType)
+        => GetPossibleTypes(abstractType);
+
     /// <summary>
     /// Gets the type and directive definitions that are defined in this schema in insert order.
     /// </summary>