Add resolver argument annotation on input value

Signed-off-by: Clément BUCHART <[email protected]>

Author: clement-buchart

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

@@ -20,6 +20,7 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 import static com.kobylynskyi.graphql.codegen.mapper.GraphQLTypeMapper.getDirectives;
 import static com.kobylynskyi.graphql.codegen.mapper.GraphQLTypeMapper.getMandatoryType;
@@ -123,9 +124,12 @@ public List<String> getAnnotations(MappingContext mappingContext, String graphQL
             }
         }
         // 6. Add annotations for resolver arguments
-        if (!Utils.isEmpty(mappingContext.getResolverArgumentAnnotations())
-                && mappingContext.getOperationsName().contains(parentTypeName)) {
-            annotations.addAll(mappingContext.getResolverArgumentAnnotations());
+        if (!Utils.isEmpty(mappingContext.getResolverArgumentAnnotations())) {
+            if (mappingContext.getOperationsName().contains(parentTypeName)
+                || (def instanceof InputValueDefinition
+                && !mappingContext.getInputsName().contains(parentTypeName))) {
+                annotations.addAll(mappingContext.getResolverArgumentAnnotations());
+            }
         }
         // 7. Add annotations for parametrized resolvers
         if (!Utils.isEmpty(mappingContext.getParametrizedResolverAnnotations())

src/main/java/com/kobylynskyi/graphql/codegen/model/MappingContext.java

@@ -31,6 +31,7 @@ public class MappingContext implements GraphQLCodegenConfiguration {
     private final Set<String> interfacesName;
     private final Set<String> unionsName;
     private final Set<String> operationsName;
+    private final Set<String> inputsName;
     private final Map<String, Set<String>> interfaceChildren;
     private final GeneratedInformation generatedInformation;
     private final DataModelMapperFactory dataModelMapperFactory;
@@ -49,6 +50,8 @@ private MappingContext(File outputDirectory,
         this.typesUnionsInterfacesNames = document.getTypesUnionsInterfacesNames();
         this.interfacesName = document.getInterfacesNames();
         this.unionsName = document.getUnionsNames();
+        this.inputsName = document.getInputDefinitions().stream().map(ExtendedDefinition::getName)
+                .collect(Collectors.toSet());
         this.interfaceChildren = document.getInterfaceChildren();
         this.generatedInformation = generatedInformation;
         this.operationsName = document.getOperationsNames();
@@ -366,6 +369,10 @@ public Set<String> getOperationsName() {
         return operationsName;
     }
 
+    public Set<String> getInputsName() {
+        return inputsName;
+    }
+
     public Map<String, Set<String>> getInterfaceChildren() {
         return interfaceChildren;
     }

src/test/resources/expected-classes/annotation/EventPropertyResolver_ArgumentAnnotations.java.txt

@@ -13,11 +13,11 @@ public interface EventPropertyResolver {
     /**
      * Properties
      */
-    java.util.List<EventProperty> child(EventProperty eventProperty, Integer first, Integer last, graphql.schema.DataFetchingEnvironment env) throws Exception;
+    java.util.List<EventProperty> child(EventProperty eventProperty, @org.springframework.graphql.data.method.annotation.Argument Integer first, @org.springframework.graphql.data.method.annotation.Argument Integer last, graphql.schema.DataFetchingEnvironment env) throws Exception;
 
     /**
      * Parent event of the property
      */
-    Event parent(EventProperty eventProperty, EventStatus withStatus, String createdAfter, graphql.schema.DataFetchingEnvironment env) throws Exception;
+    Event parent(EventProperty eventProperty, @org.springframework.graphql.data.method.annotation.Argument EventStatus withStatus, @org.springframework.graphql.data.method.annotation.Argument String createdAfter, graphql.schema.DataFetchingEnvironment env) throws Exception;
 
 }