Generates data fetcher interface for query types

graphql-dgs-codegen-core/src/main/kotlin/com/netflix/graphql/dgs/codegen/CodeGen.kt

@@ -262,14 +262,20 @@ class CodeGen(private val config: CodeGenConfig) {
             }
             .fold(CodeGenResult()) { t: CodeGenResult, u: CodeGenResult -> t.merge(u) }
 
+        val dataFetchersResult = definitions.asSequence()
+            .filterIsInstance<ObjectTypeDefinition>()
+            .filter { it.name == "Query" }
+            .map { KotlinDataFetcherGenerator(config).generate(it) }
+            .fold(CodeGenResult()) { t: CodeGenResult, u: CodeGenResult -> t.merge(u) }
+
         val constantsClass = KotlinConstantsGenerator(config, document).generate()
 
         val client = generateJavaClientApi(definitions)
         val entitiesClient = generateJavaClientEntitiesApi(definitions)
         val entitiesRepresentationsTypes = generateKotlinClientEntitiesRepresentations(definitions)
 
         return datatypesResult.merge(inputTypes).merge(interfacesResult).merge(unionResult).merge(enumsResult)
-            .merge(client).merge(entitiesClient).merge(entitiesRepresentationsTypes).merge(constantsClass)
+            .merge(client).merge(entitiesClient).merge(entitiesRepresentationsTypes).merge(dataFetchersResult).merge(constantsClass)
     }
 
     private fun generateKotlinClientEntitiesRepresentations(definitions: Collection<Definition<*>>): CodeGenResult {

graphql-dgs-codegen-core/src/main/kotlin/com/netflix/graphql/dgs/codegen/generators/kotlin/KotlinDataFetcherGenerator.kt

@@ -0,0 +1,53 @@
+package com.netflix.graphql.dgs.codegen.generators.kotlin
+
+import com.netflix.graphql.dgs.DgsData
+import com.netflix.graphql.dgs.codegen.CodeGenConfig
+import com.netflix.graphql.dgs.codegen.CodeGenResult
+import com.squareup.kotlinpoet.AnnotationSpec
+import com.squareup.kotlinpoet.FileSpec
+import com.squareup.kotlinpoet.FunSpec
+import com.squareup.kotlinpoet.KModifier
+import com.squareup.kotlinpoet.TypeName
+import com.squareup.kotlinpoet.TypeSpec
+import graphql.language.FieldDefinition
+import graphql.language.ObjectTypeDefinition
+import graphql.schema.DataFetchingEnvironment
+
+class KotlinDataFetcherGenerator(private val config: CodeGenConfig) {
+
+    private val packageName = config.packageNameDatafetchers
+    private val typeUtils = KotlinTypeUtils(config.packageNameTypes, config)
+
+    fun generate(query: ObjectTypeDefinition): CodeGenResult =
+        query.fieldDefinitions
+            .map { generateField(it) }
+            .fold(CodeGenResult()) { left, right -> left.merge(right) }
+
+    private fun generateField(field: FieldDefinition): CodeGenResult {
+        val fieldName = field.name.substring(0, 1).toUpperCase() + field.name.substring(1)
+        val className = fieldName + "Datafetcher"
+
+        val returnType = typeUtils.findReturnType(field.type)
+
+        val methodSpec = FunSpec.builder("get$fieldName")
+            .addAnnotation(AnnotationSpec.builder(DgsData::class).addMember("parentType", "\$S", "Query").addMember("field", "\$S", field.name).build())
+            .addModifiers(KModifier.ABSTRACT)
+            .also { builder ->
+                field.inputValueDefinitions.forEach {
+                    val inputType: TypeName = typeUtils.findReturnType(it.type)
+                    builder.addParameter(it.name, inputType)
+                }
+            }
+            .addParameter("dataFetchingEnvironment", DataFetchingEnvironment::class)
+            .returns(returnType)
+            .build()
+
+        val interfaceBuilder = TypeSpec.interfaceBuilder(className)
+            .addFunction(methodSpec)
+            .build()
+
+        val fileSpec = FileSpec.get(packageName, interfaceBuilder)
+
+        return CodeGenResult(kotlinDataFetchers = listOf(fileSpec))
+    }
+}

graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt

@@ -20,6 +20,7 @@ package com.netflix.graphql.dgs.codegen
 
 import com.squareup.kotlinpoet.*
 import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy
+import graphql.schema.DataFetchingEnvironment
 import org.assertj.core.api.Assertions.assertThat
 import org.junit.jupiter.api.Nested
 import org.junit.jupiter.api.Test
@@ -32,6 +33,10 @@ import java.util.stream.Stream
 
 class KotlinCodeGenTest {
 
+    val basePackageName = "com.netflix.graphql.dgs.codegen.tests.generated"
+    val typesPackageName = "$basePackageName.types"
+    val datafetchersPackageName = "$basePackageName.datafetchers"
+
     @Test
     fun generateDataClassWithStringProperties() {
 
@@ -66,6 +71,91 @@ class KotlinCodeGenTest {
         assertCompilesKotlin(dataTypes)
     }
 
+    @Test
+    fun generateDataFetcherInterfaceWithFunction() {
+
+        val schema = """
+            type Query {
+                people: [Person]
+            }
+            
+            type Person {
+                firstname: String
+                lastname: String
+            }
+        """.trimIndent()
+
+        val dataFetchers = CodeGen(
+            CodeGenConfig(
+                schemas = setOf(schema),
+                packageName = basePackageName,
+                language = Language.KOTLIN
+            )
+        ).generate().kotlinDataFetchers
+
+        assertThat(dataFetchers.size).isEqualTo(1)
+        assertThat(dataFetchers[0].name).isEqualTo("PeopleDatafetcher")
+        assertThat(dataFetchers[0].packageName).isEqualTo(datafetchersPackageName)
+        val type = dataFetchers[0].members[0] as TypeSpec
+
+        assertThat(type.kind).isEqualTo(TypeSpec.Kind.INTERFACE)
+        assertThat(type.funSpecs).hasSize(1)
+        val fn = type.funSpecs.single()
+        assertThat(fn.name).isEqualTo("getPeople")
+        val returnType = fn.returnType as ParameterizedTypeName
+        assertThat(fn.returnType)
+        assertThat(returnType.rawType.canonicalName).isEqualTo(List::class.qualifiedName)
+        assertThat(returnType.typeArguments).hasSize(1)
+        val arg0 = returnType.typeArguments.single() as ClassName
+        assertThat(arg0.canonicalName).isEqualTo("$typesPackageName.Person")
+        assertThat(fn.parameters).hasSize(1)
+        val param0 = fn.parameters.single()
+        assertThat(param0.name).isEqualTo("dataFetchingEnvironment")
+        assertThat((param0.type as ClassName).canonicalName).isEqualTo(DataFetchingEnvironment::class.qualifiedName)
+    }
+
+
+    @Test
+    fun generateDataFetcherInterfaceWithArgument() {
+
+        val schema = """
+            type Query {
+                person(name: String): Person
+            }
+            
+            type Person {
+                firstname: String
+                lastname: String
+            }
+        """.trimIndent()
+
+        val dataFetchers = (CodeGen(
+            CodeGenConfig(
+                schemas = setOf(schema),
+                packageName = basePackageName,
+                language = Language.KOTLIN
+            )
+        ).generate() as CodeGenResult).kotlinDataFetchers
+
+        assertThat(dataFetchers.size).isEqualTo(1)
+        assertThat(dataFetchers[0].name).isEqualTo("PersonDatafetcher")
+        assertThat(dataFetchers[0].packageName).isEqualTo(datafetchersPackageName)
+        val type = dataFetchers[0].members[0] as TypeSpec
+
+        assertThat(type.kind).isEqualTo(TypeSpec.Kind.INTERFACE)
+        assertThat(type.funSpecs).hasSize(1)
+        val fn = type.funSpecs.single()
+        assertThat(fn.name).isEqualTo("getPerson")
+        assertThat((fn.returnType as ClassName).canonicalName).isEqualTo("$typesPackageName.Person")
+        assertThat(fn.parameters).hasSize(2);
+        val arg0 = fn.parameters[0]
+        assertThat(arg0.name).isEqualTo("name")
+        assertThat((arg0.type as ClassName).canonicalName).isEqualTo(String::class.qualifiedName)
+        val arg1 = fn.parameters[1]
+        assertThat(arg1.name).isEqualTo("dataFetchingEnvironment")
+        assertThat((arg1.type as ClassName).canonicalName).isEqualTo(DataFetchingEnvironment::class.qualifiedName)
+    }
+
     @Test
     fun generateDataClassWithNullablePrimitive() {