Generate parametrized input class for interfaces #327 (#330)

Author: kobylynskyi

src/main/java/com/kobylynskyi/graphql/codegen/GraphQLCodegen.java

@@ -279,8 +279,7 @@ private List<File> processDefinitions(ExtendedDocument document) {
         for (ExtendedInterfaceTypeDefinition definition : document.getInterfaceDefinitions()) {
             generateFieldResolver(context, definition.getFieldDefinitions(), definition).ifPresent(generatedFiles::add);
         }
-        System.out.println(String.format("Generated %d definition classes in folder %s",
-                generatedFiles.size(), outputDir.getAbsolutePath()));
+        System.out.printf("Generated %d definition classes in folder %s%n", generatedFiles.size(), outputDir.getAbsolutePath());
         return generatedFiles;
     }
 
@@ -304,6 +303,13 @@ private List<File> generateInterface(MappingContext mappingContext, ExtendedInte
         if (Boolean.TRUE.equals(mappingConfig.getGenerateClient())) {
             Map<String, Object> responseProjDataModel = RequestResponseDefinitionToDataModelMapper.mapResponseProjection(mappingContext, definition);
             generatedFiles.add(GraphQLCodegenFileCreator.generateFile(FreeMarkerTemplatesRegistry.responseProjectionTemplate, responseProjDataModel, outputDir));
+
+            for (ExtendedFieldDefinition fieldDefinition : definition.getFieldDefinitions()) {
+                if (!Utils.isEmpty(fieldDefinition.getInputValueDefinitions())) {
+                    Map<String, Object> fieldProjDataModel = RequestResponseDefinitionToDataModelMapper.mapParametrizedInput(mappingContext, fieldDefinition, definition);
+                    generatedFiles.add(GraphQLCodegenFileCreator.generateFile(FreeMarkerTemplatesRegistry.parametrizedInputTemplate, fieldProjDataModel, outputDir));
+                }
+            }
         }
         return generatedFiles;
     }
@@ -398,12 +404,14 @@ private List<File> generateType(MappingContext mappingContext, ExtendedObjectTyp
     private Optional<File> generateFieldResolver(MappingContext mappingContext,
                                                  List<ExtendedFieldDefinition> fieldDefinitions,
                                                  ExtendedDefinition<?, ?> parentDefinition) {
-        List<ExtendedFieldDefinition> fieldDefsWithResolvers = fieldDefinitions.stream()
-                .filter(fieldDef -> FieldDefinitionToParameterMapper.generateResolversForField(mappingContext, fieldDef, parentDefinition))
-                .collect(toList());
-        if (!fieldDefsWithResolvers.isEmpty()) {
-            Map<String, Object> dataModel = FieldDefinitionsToResolverDataModelMapper.mapToTypeResolver(mappingContext, fieldDefsWithResolvers, parentDefinition.getName());
-            return Optional.of(GraphQLCodegenFileCreator.generateFile(FreeMarkerTemplatesRegistry.operationsTemplate, dataModel, outputDir));
+        if (Boolean.TRUE.equals(mappingConfig.getGenerateApis())) {
+            List<ExtendedFieldDefinition> fieldDefsWithResolvers = fieldDefinitions.stream()
+                    .filter(fieldDef -> FieldDefinitionToParameterMapper.generateResolversForField(mappingContext, fieldDef, parentDefinition))
+                    .collect(toList());
+            if (!fieldDefsWithResolvers.isEmpty()) {
+                Map<String, Object> dataModel = FieldDefinitionsToResolverDataModelMapper.mapToTypeResolver(mappingContext, fieldDefsWithResolvers, parentDefinition.getName());
+                return Optional.of(GraphQLCodegenFileCreator.generateFile(FreeMarkerTemplatesRegistry.operationsTemplate, dataModel, outputDir));
+            }
         }
         return Optional.empty();
     }

src/main/java/com/kobylynskyi/graphql/codegen/mapper/RequestResponseDefinitionToDataModelMapper.java

@@ -64,7 +64,7 @@ public static Map<String, Object> mapResponseProjection(MappingContext mappingCo
      */
     public static Map<String, Object> mapParametrizedInput(MappingContext mappingContext,
                                                            ExtendedFieldDefinition fieldDefinition,
-                                                           ExtendedObjectTypeDefinition parentTypeDefinition) {
+                                                           ExtendedDefinition<?, ?> parentTypeDefinition) {
         String className = MapperUtils.getParametrizedInputClassName(mappingContext, fieldDefinition, parentTypeDefinition);
         Map<String, Object> dataModel = new HashMap<>();
         // ParametrizedInput classes are sharing the package with the model classes, so no imports are needed

src/test/java/com/kobylynskyi/graphql/codegen/GraphQLCodegenFieldsResolversTest.java

@@ -50,6 +50,19 @@ void generate_ParametrizedFields() throws Exception {
                 getFileByName(files, "CommitResolver.java"));
     }
 
+    @Test
+    void generate_ParametrizedFieldsInInterface() throws Exception {
+        mappingConfig.setGenerateClient(true);
+        mappingConfig.setGenerateApis(false);
+
+        new GraphQLCodegen(singletonList("src/test/resources/schemas/parametrized-input-client.graphqls"),
+                outputBuildDir, mappingConfig, TestUtils.getStaticGeneratedInfo()).generate();
+
+        File[] files = Objects.requireNonNull(outputJavaClassesDir.listFiles());
+        assertSameTrimmedContent(new File("src/test/resources/expected-classes/request/ProductLinkCodeParametrizedInput.java.txt"),
+                getFileByName(files, "ProductLinkCodeParametrizedInput.java"));
+    }
+
     @Test
     void generate_CustomFieldsResolvers() throws Exception {
         mappingConfig.setModelNamePrefix("Github");

src/test/java/com/kobylynskyi/graphql/codegen/GraphQLCodegenModelsForRootTypesTest.java

@@ -70,8 +70,8 @@ void generate_CheckFiles_generateApisFalse() throws Exception {
 
         File[] files = Objects.requireNonNull(outputJavaClassesDir.listFiles());
         List<String> generatedFileNames = Arrays.stream(files).map(File::getName).sorted().collect(toList());
-        assertEquals(Arrays.asList("Event.java", "EventProperty.java", "EventPropertyResolver.java", "EventStatus.java",
-                "Mutation.java", "MutationResolver.java", "Query.java", "QueryResolver.java", "Subscription.java"),
+        assertEquals(Arrays.asList("Event.java", "EventProperty.java", "EventStatus.java",
+                "Mutation.java", "Query.java", "Subscription.java"),
                 generatedFileNames);
     }
 

src/test/resources/expected-classes/request/ProductLinkCodeParametrizedInput.java.txt

@@ -0,0 +1,41 @@
+package com.github.graphql;
+
+import com.kobylynskyi.graphql.codegen.model.graphql.GraphQLParametrizedInput;
+import com.kobylynskyi.graphql.codegen.model.graphql.GraphQLRequestSerializer;
+import java.util.StringJoiner;
+
+/**
+ * Parametrized input for field linkCode in type Product
+ */
+@javax.annotation.Generated(
+    value = "com.kobylynskyi.graphql.codegen.GraphQLCodegen",
+    date = "2020-12-31T23:59:59-0500"
+)
+public class ProductLinkCodeParametrizedInput implements GraphQLParametrizedInput {
+
+    @javax.validation.constraints.NotNull
+    private String pid;
+
+    public ProductLinkCodeParametrizedInput() {
+    }
+
+    public ProductLinkCodeParametrizedInput(String pid) {
+        this.pid = pid;
+    }
+
+    public ProductLinkCodeParametrizedInput pid(String pid) {
+        this.pid = pid;
+        return this;
+    }
+
+
+    @Override
+    public String toString() {
+        StringJoiner joiner = new StringJoiner(", ", "(", ")");
+        if (pid != null) {
+            joiner.add("pid: " + GraphQLRequestSerializer.getEntry(pid));
+        }
+        return joiner.toString();
+    }
+
+}

src/test/resources/schemas/parametrized-input-client.graphqls

@@ -0,0 +1,15 @@
+type Query {
+    product(
+        companyId: ID!
+    ): Product!
+}
+
+interface Product {
+    linkCode(
+        pid: ID!
+    ): LinkCode
+}
+
+type LinkCode {
+    html: String
+}