Clean up augmented schema to contain only types used in the resulting schema (#234)

resolves #233

Author: Andy2003

core/src/main/kotlin/org/neo4j/graphql/SchemaBuilder.kt

@@ -119,8 +119,29 @@ class SchemaBuilder(
             .filter { it.name == queryTypeName || it.name == mutationTypeName || it.name == subscriptionTypeName }
             .forEach { type -> handler.forEach { h -> h.augmentType(type) } }
 
-        // TODO copy over only the types used in the source schema
-        typeDefinitionRegistry.merge(neo4jTypeDefinitionRegistry)
+        val types = mutableListOf<Type<*>>()
+        neo4jTypeDefinitionRegistry.directiveDefinitions.values
+            .filterNot { typeDefinitionRegistry.getDirectiveDefinition(it.name).isPresent }
+            .forEach { directiveDefinition ->
+                typeDefinitionRegistry.add(directiveDefinition)
+                directiveDefinition.inputValueDefinitions.forEach { types.add(it.type) }
+            }
+        typeDefinitionRegistry.types()
+            .values
+            .flatMap { typeDefinition ->
+                when (typeDefinition) {
+                    is ImplementingTypeDefinition -> typeDefinition.fieldDefinitions
+                        .flatMap { fieldDefinition -> fieldDefinition.inputValueDefinitions.map { it.type } + fieldDefinition.type }
+                    is InputObjectTypeDefinition -> typeDefinition.inputValueDefinitions.map { it.type }
+                    else -> emptyList()
+                }
+            }
+            .forEach { types.add(it) }
+        types
+            .map { TypeName(it.name()) }
+            .filterNot { typeDefinitionRegistry.hasType(it) }
+            .mapNotNull { neo4jTypeDefinitionRegistry.getType(it).unwrap() }
+            .forEach { typeDefinitionRegistry.add(it) }
     }
 
     /**

core/src/test/kotlin/org/neo4j/graphql/utils/GraphQLSchemaTestSuite.kt

@@ -57,6 +57,7 @@ class GraphQLSchemaTestSuite(fileName: String) : AsciiDocTestSuite(
 
                 diff(expectedSchema, augmentedSchema)
                 diff(augmentedSchema, expectedSchema)
+                targetSchemaBlock.adjustedCode = SCHEMA_PRINTER.print(augmentedSchema)
             } catch (e: Throwable) {
                 if (ignore) {
                     Assumptions.assumeFalse(true, e.message)
@@ -65,9 +66,7 @@ class GraphQLSchemaTestSuite(fileName: String) : AsciiDocTestSuite(
                         Assertions.fail<Throwable>(e)
                     }
                     val actualSchema = SCHEMA_PRINTER.print(augmentedSchema)
-                    targetSchemaBlock.adjustedCode = actualSchema + "\n" +
-                            // this is added since the SCHEMA_PRINTER is not able to print global directives
-                            javaClass.getResource("/lib_directives.graphql").readText()
+                    targetSchemaBlock.adjustedCode = actualSchema
                     throw AssertionFailedError("augmented schema differs for '$title'",
                             expectedSchema?.let { SCHEMA_PRINTER.print(it) } ?: targetSchema,
                             actualSchema,
@@ -84,7 +83,7 @@ class GraphQLSchemaTestSuite(fileName: String) : AsciiDocTestSuite(
         private val METHOD_PATTERN = Pattern.compile("(add|delete|update|merge|create)(.*)")
 
         private val SCHEMA_PRINTER = SchemaPrinter(SchemaPrinter.Options.defaultOptions()
-            .includeDirectives(true)
+            .includeDirectives(false)
             .includeScalarTypes(true)
             .includeSchemaDefinition(true)
             .includeIntrospectionTypes(false)