Pass the coderegistry builder to custom directives fix #268

Author: oliemansm

src/main/kotlin/com/coxautodev/graphql/tools/SchemaParser.kt

@@ -172,7 +172,7 @@ class SchemaParser internal constructor(scanResult: ScannedSchemaObjects, privat
 
         val objectType = builder.build()
 
-        return directiveGenerator.onObject(objectType, DirectiveBehavior.Params(runtimeWiring))
+        return directiveGenerator.onObject(objectType, DirectiveBehavior.Params(runtimeWiring, codeRegistryBuilder))
     }
 
     private fun buildDirectives(directives: List<Directive>, directiveDefinitions: Set<GraphQLDirective>, directiveLocation: Introspection.DirectiveLocation): Array<GraphQLDirective> {
@@ -207,7 +207,7 @@ class SchemaParser internal constructor(scanResult: ScannedSchemaObjects, privat
             builder.field(fieldBuilder.build())
         }
 
-        return directiveGenerator.onInputObject(builder.build(), DirectiveBehavior.Params(runtimeWiring))
+        return directiveGenerator.onInputObject(builder.build(), DirectiveBehavior.Params(runtimeWiring, codeRegistryBuilder))
     }
 
     private fun createEnumObject(definition: EnumTypeDefinition): GraphQLEnumType {
@@ -241,7 +241,7 @@ class SchemaParser internal constructor(scanResult: ScannedSchemaObjects, privat
             }
         }
 
-        return directiveGenerator.onEnum(builder.build(), DirectiveBehavior.Params(runtimeWiring))
+        return directiveGenerator.onEnum(builder.build(), DirectiveBehavior.Params(runtimeWiring, codeRegistryBuilder))
     }
 
     private fun createInterfaceObject(interfaceDefinition: InterfaceTypeDefinition): GraphQLInterfaceType {
@@ -257,7 +257,7 @@ class SchemaParser internal constructor(scanResult: ScannedSchemaObjects, privat
             builder.field { field -> createField(field, fieldDefinition) }
         }
 
-        return directiveGenerator.onInterface(builder.build(), DirectiveBehavior.Params(runtimeWiring))
+        return directiveGenerator.onInterface(builder.build(), DirectiveBehavior.Params(runtimeWiring, codeRegistryBuilder))
     }
 
     private fun createUnionObject(definition: UnionTypeDefinition, types: List<GraphQLObjectType>): GraphQLUnionType {
@@ -270,7 +270,7 @@ class SchemaParser internal constructor(scanResult: ScannedSchemaObjects, privat
         builder.withDirectives(*buildDirectives(definition.directives, setOf(), Introspection.DirectiveLocation.UNION))
 
         getLeafUnionObjects(definition, types).forEach { builder.possibleType(it) }
-        return directiveGenerator.onUnion(builder.build(), DirectiveBehavior.Params(runtimeWiring))
+        return directiveGenerator.onUnion(builder.build(), DirectiveBehavior.Params(runtimeWiring, codeRegistryBuilder))
     }
 
     private fun getLeafUnionObjects(definition: UnionTypeDefinition, types: List<GraphQLObjectType>): List<GraphQLObjectType> {

src/main/kotlin/graphql/schema/idl/DirectiveBehavior.kt

@@ -1,5 +1,6 @@
 package graphql.schema.idl
 
+import graphql.schema.GraphQLCodeRegistry
 import graphql.schema.GraphQLEnumType
 import graphql.schema.GraphQLInputObjectType
 import graphql.schema.GraphQLInterfaceType
@@ -30,8 +31,8 @@ class DirectiveBehavior {
     fun onInputObject(element: GraphQLInputObjectType, params: Params): GraphQLInputObjectType =
             directiveHelper.onInputObjectType(element, params.toParameters())
 
-    data class Params(val runtimeWiring: RuntimeWiring) {
-        internal fun toParameters() = SchemaGeneratorDirectiveHelper.Parameters(null, runtimeWiring, null, null)
+    data class Params(val runtimeWiring: RuntimeWiring, val codeRegistryBuilder: GraphQLCodeRegistry.Builder) {
+        internal fun toParameters() = SchemaGeneratorDirectiveHelper.Parameters(null, runtimeWiring, null, codeRegistryBuilder)
     }
 
 }