Pass services to definition transformer

Author: gnawf

lib/src/main/java/graphql/nadel/NadelSchemas.kt

@@ -234,9 +234,8 @@ data class NadelSchemas(
 
         private fun createEngineSchema(services: List<Service>): GraphQLSchema {
             val overallSchemaGenerator = OverallSchemaGenerator()
-            val serviceRegistries = services.map(Service::definitionRegistry)
             val schema = overallSchemaGenerator.buildOverallSchema(
-                serviceRegistries,
+                services,
                 builder.overallWiringFactory,
                 builder.schemaDefinitionTransformationHook,
             )

lib/src/main/java/graphql/nadel/schema/NadelSchemaDefinitionTransformationHook.kt

@@ -1,11 +1,15 @@
 package graphql.nadel.schema
 
+import graphql.nadel.Service
 import graphql.nadel.util.AnySDLDefinition
 
 fun interface NadelSchemaDefinitionTransformationHook {
-    operator fun invoke(definitions: List<AnySDLDefinition>): List<AnySDLDefinition>
+    operator fun invoke(
+        services: List<Service>,
+        definitions: List<AnySDLDefinition>,
+    ): List<AnySDLDefinition>
 
     companion object {
-        val Identity = NadelSchemaDefinitionTransformationHook { originalSchema -> originalSchema }
+        val Identity = NadelSchemaDefinitionTransformationHook { _, definitions -> definitions }
     }
 }

lib/src/main/java/graphql/nadel/schema/OverallSchemaGenerator.kt

@@ -9,6 +9,7 @@ import graphql.language.SchemaDefinition
 import graphql.language.SourceLocation
 import graphql.nadel.NadelOperationKind
 import graphql.nadel.NadelTypeDefinitionRegistry
+import graphql.nadel.Service
 import graphql.nadel.util.AnyNamedNode
 import graphql.nadel.util.AnySDLDefinition
 import graphql.nadel.util.AnySDLNamedDefinition
@@ -22,10 +23,11 @@ import graphql.schema.idl.WiringFactory
 
 internal class OverallSchemaGenerator {
     fun buildOverallSchema(
-        serviceRegistries: List<NadelTypeDefinitionRegistry>,
+        services: List<Service>,
         wiringFactory: WiringFactory,
         schemaDefinitionTransformationHook: NadelSchemaDefinitionTransformationHook,
     ): GraphQLSchema {
+        val serviceRegistries = services.map(Service::definitionRegistry)
         val schemaGenerator = SchemaGenerator()
         val runtimeWiring = RuntimeWiring.newRuntimeWiring()
             .wiringFactory(wiringFactory)
@@ -34,7 +36,7 @@ internal class OverallSchemaGenerator {
 
         val definitions = getDefinitions(serviceRegistries)
             .let {
-                schemaDefinitionTransformationHook.invoke(it)
+                schemaDefinitionTransformationHook.invoke(services, it)
             }
 
         val typeRegistry = createTypeRegistry(definitions)