pass test

Author: danadajian

src/definitions/input.ts

@@ -27,6 +27,12 @@ export function buildInputObjectDefinition(
     return "";
   }
 
+  const typeNameWithoutInput = node.name.value.replace("Input", "");
+  const correspondingType = schema.getType(typeNameWithoutInput);
+  if (correspondingType) {
+    return "";
+  }
+
   const classMembers = (node.fields ?? [])
     .map((arg) => {
       const typeToUse = buildTypeMetadata(arg.type, schema, config);

src/definitions/object.ts

@@ -34,6 +34,12 @@ export function buildObjectTypeDefinition(
     return "";
   }
 
+  const typeNameWithInput = `${node.name.value}Input`;
+  const correspondingInput = schema.getType(typeNameWithInput);
+  const inputOutputAnnotation = correspondingInput
+    ? "@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.INPUT_OBJECT, GraphQLValidObjectLocations.Locations.OBJECT])\n"
+    : "";
+
   const annotations = buildAnnotations({
     config,
     definitionNode: node,
@@ -57,7 +63,7 @@ ${getDataClassMembers({ node, schema, config, completableFuture: true })}
 }`;
   }
 
-  return `${annotations}data class ${name}(
+  return `${annotations}${inputOutputAnnotation}data class ${name}(
 ${getDataClassMembers({ node, schema, config })}
 )${interfaceInheritance}`;
 }

test/unit/should_consolidate_input_and_output_types/expected.kt

@@ -2,20 +2,24 @@ package com.kotlin.generated
 
 import com.expediagroup.graphql.generator.annotations.*
 
-data class MyTypeToConsolidate1(
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.INPUT_OBJECT, GraphQLValidObjectLocations.Locations.OBJECT])
+data class MyTypeToConsolidate(
     val field: String? = null
 )
 
 @GraphQLDescription("A description for MyTypeToConsolidate2")
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.INPUT_OBJECT, GraphQLValidObjectLocations.Locations.OBJECT])
 data class MyTypeToConsolidate2(
     val field: String? = null
 )
 
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.INPUT_OBJECT, GraphQLValidObjectLocations.Locations.OBJECT])
 data class MyTypeToConsolidate3(
     val field: String? = null
 )
 
 @GraphQLDescription("It always uses the type description when consolidating")
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.INPUT_OBJECT, GraphQLValidObjectLocations.Locations.OBJECT])
 data class MyTypeToConsolidate4(
     val field: String? = null
 )
@@ -24,6 +28,7 @@ data class MyTypeNotToConsolidate(
     val field: String? = null
 )
 
+@GraphQLDescription("The type name must exactly match in order to consolidate")
 data class MyTypeToNotConsolidateInput(
     val field: String? = null
 )