Populate field annotation for GraphQL input type #7

Author: kobylynskyi

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

@@ -51,7 +51,7 @@ static OperationDefinition mapFieldDefinition(MappingConfig mappingConfig, Field
         operation.setName(fieldDef.getName());
         operation.setType(GraphqlTypeToJavaTypeMapper.getJavaType(mappingConfig, fieldDef.getType(), fieldDef.getName(), parentTypeName));
         operation.setAnnotations(GraphqlTypeToJavaTypeMapper.getAnnotations(mappingConfig, fieldDef.getType(), fieldDef.getName(), parentTypeName));
-        operation.setParameters(InputValueDefinitionToParameterMapper.map(mappingConfig, fieldDef.getInputValueDefinitions()));
+        operation.setParameters(InputValueDefinitionToParameterMapper.map(mappingConfig, fieldDef.getInputValueDefinitions(), fieldDef.getName()));
         return operation;
     }
 

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

@@ -23,10 +23,11 @@ public static ParameterDefinition map(MappingConfig mappingConfig, FieldDefiniti
         return parameter;
     }
 
-    public static ParameterDefinition map(MappingConfig mappingConfig, InputValueDefinition inputValueDefinition) {
+    public static ParameterDefinition map(MappingConfig mappingConfig, InputValueDefinition inputValueDefinition, String parentTypeName) {
         ParameterDefinition parameter = new ParameterDefinition();
         parameter.setName(MapperUtils.capitalizeIfRestricted(inputValueDefinition.getName()));
         parameter.setType(getJavaType(mappingConfig, inputValueDefinition.getType()));
+        parameter.setAnnotations(getAnnotations(mappingConfig, inputValueDefinition.getType(), inputValueDefinition.getName(), parentTypeName));
         return parameter;
     }
 
@@ -69,7 +70,7 @@ private static String getJavaType(MappingConfig mappingConfig, String graphlType
     }
 
     static List<String> getAnnotations(MappingConfig mappingConfig, Type type, String name, String parentTypeName) {
-        return getAnnotations(mappingConfig, type, name, parentTypeName, true);
+        return getAnnotations(mappingConfig, type, name, parentTypeName, false);
     }
 
     private static List<String> getAnnotations(MappingConfig mappingConfig, Type type, String name, String parentTypeName,

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

@@ -29,7 +29,7 @@ public static Map<String, Object> map(MappingConfig mappingConfig, InputObjectTy
         dataModel.put(IMPORTS, MapperUtils.getImports(mappingConfig, packageName));
         dataModel.put(CLASS_NAME, MapperUtils.getClassNameWithPrefixAndSuffix(mappingConfig, typeDefinition));
         dataModel.put(NAME, typeDefinition.getName());
-        dataModel.put(FIELDS, InputValueDefinitionToParameterMapper.map(mappingConfig, typeDefinition.getInputValueDefinitions()));
+        dataModel.put(FIELDS, InputValueDefinitionToParameterMapper.map(mappingConfig, typeDefinition.getInputValueDefinitions(), typeDefinition.getName()));
         return dataModel;
     }
 

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

@@ -15,12 +15,12 @@
  */
 public class InputValueDefinitionToParameterMapper {
 
-    public static List<ParameterDefinition> map(MappingConfig mappingConfig, List<InputValueDefinition> valueDefinitions) {
+    public static List<ParameterDefinition> map(MappingConfig mappingConfig, List<InputValueDefinition> valueDefinitions, String parentTypeName) {
         if (valueDefinitions == null) {
             return Collections.emptyList();
         }
         return valueDefinitions.stream()
-                .map(inputValueDefinition -> GraphqlTypeToJavaTypeMapper.map(mappingConfig, inputValueDefinition))
+                .map(inputValueDefinition -> GraphqlTypeToJavaTypeMapper.map(mappingConfig, inputValueDefinition, parentTypeName))
                 .collect(Collectors.toList());
     }
 

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

@@ -75,8 +75,8 @@ void generate_ClassNameWithSuffix_Prefix() throws Exception {
                 StringContains.containsString("public interface GithubAssigneeTO "));
 
         // verify proper class name for GraphQL input
-        assertThat(getGeneratedFileContent(files, "GithubChangeUserStatusInputTO.java"),
-                StringContains.containsString("public class GithubChangeUserStatusInputTO "));
+        assertEquals(Utils.getFileContent(new File("src/test/resources/expected-classes/GithubAcceptTopicSuggestionInputTO.java.txt").getPath()),
+                getGeneratedFileContent(files, "GithubAcceptTopicSuggestionInputTO.java"));
 
         // verify proper class name for GraphQL type and references to interfaces/types/unions for GraphQL type
         assertEquals(Utils.getFileContent(new File("src/test/resources/expected-classes/GithubCommitTO.java.txt").getPath()),

src/test/resources/expected-classes/GithubAcceptTopicSuggestionInputTO.java.txt

@@ -0,0 +1,37 @@
+package com.github.graphql;
+
+import java.util.*;
+
+public class GithubAcceptTopicSuggestionInputTO {
+
+    private String clientMutationId;
+    @javax.validation.constraints.NotNull
+    private String name;
+    @javax.validation.constraints.NotNull
+    private String repositoryId;
+
+    public GithubAcceptTopicSuggestionInputTO() {
+    }
+
+    public String getClientMutationId() {
+        return clientMutationId;
+    }
+    public void setClientMutationId(String clientMutationId) {
+        this.clientMutationId = clientMutationId;
+    }
+
+    public String getName() {
+        return name;
+    }
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public String getRepositoryId() {
+        return repositoryId;
+    }
+    public void setRepositoryId(String repositoryId) {
+        this.repositoryId = repositoryId;
+    }
+
+}

src/test/resources/schemas/test.graphqls

@@ -9,19 +9,19 @@ schema {
 
 type Query {
     # Version of the application.
-    version: String
+    version: String!
 
     # List of events of a specified category.
-    eventsByCategoryAndStatus(categoryId: ID!, status: EventStatus): [Event]
+    eventsByCategoryAndStatus(categoryId: ID!, status: EventStatus): [Event]!
 
     # Single event by ID.
-    eventById(id: ID!): Event
+    eventById(id: ID!): Event!
 }
 
 type Mutation {
     # Create a new event.
     createEvent(categoryId: String!,
-                createdBy: String!): Event
+                createdBy: String!): Event!
 }