pass test

Author: danadajian

bun.lockb

@@ -22,6 +22,7 @@
     "@graphql-codegen/java-common": "3.0.0",
     "@graphql-codegen/plugin-helpers": "5.0.4",
     "@graphql-codegen/visitor-plugin-common": "5.2.0",
+    "title-case": "4.3.1",
     "ts-deepmerge": "7.0.0",
     "valibot": "0.30.0"
   },

package.json

@@ -28,6 +28,7 @@ import { buildFieldDefinition } from "../helpers/build-field-definition";
 import { CodegenConfigWithDefaults } from "../helpers/build-config-with-defaults";
 import { inputTypeHasMatchingOutputType } from "../helpers/input-type-has-matching-output-type";
 import { findTypeInResolverInterfacesConfig } from "../helpers/findTypeInResolverInterfacesConfig";
+import { titleCase } from "title-case";
 
 export function buildObjectTypeDefinition(
   node: ObjectTypeDefinitionNode,
@@ -64,10 +65,32 @@ export function buildObjectTypeDefinition(
     node,
     config,
   );
+  const fieldsWithArguments = node.fields?.filter(
+    (fieldNode) => fieldNode.arguments?.length,
+  );
+  const fieldNodes = typeInResolverInterfacesConfig
+    ? node.fields
+    : fieldsWithArguments;
   const shouldGenerateFunctions = Boolean(
     typeInResolverInterfacesConfig ||
       node.fields?.some((fieldNode) => fieldNode.arguments?.length),
   );
+
+  if (node.name.value === "Query" || node.name.value === "Mutation") {
+    const individualQueryClasses = node.fields?.map((fieldNode) => {
+      const className = `${titleCase(fieldNode.name.value)}Query`;
+      return `${annotations}${outputRestrictionAnnotation}open class ${className}${interfaceInheritance} {
+${getDataClassMembers({ node, fieldNodes: [fieldNode], schema, config, shouldGenerateFunctions })}
+}`;
+    });
+    const consolidatedQueryClass = `${annotations}${outputRestrictionAnnotation}open class ${name}${interfaceInheritance} {
+${getDataClassMembers({ node, fieldNodes, schema, config, shouldGenerateFunctions })}
+}`;
+    return [consolidatedQueryClass, ...(individualQueryClasses ?? [])].join(
+      "\n\n",
+    );
+  }
+
   if (shouldGenerateFunctions) {
     const atLeastOneFieldHasNoArguments = node.fields?.some(
       (fieldNode) => !fieldNode.arguments?.length,
@@ -94,12 +117,6 @@ export function buildObjectTypeDefinition(
             .join(",\n")}\n)`
         : "";
 
-    const fieldsWithArguments = node.fields?.filter(
-      (fieldNode) => fieldNode.arguments?.length,
-    );
-    const fieldNodes = typeInResolverInterfacesConfig
-      ? node.fields
-      : fieldsWithArguments;
     return `${annotations}${outputRestrictionAnnotation}open class ${name}${constructor}${interfaceInheritance} {
 ${getDataClassMembers({ node, fieldNodes, schema, config, shouldGenerateFunctions })}
 }`;

src/definitions/object.ts

@@ -10,7 +10,7 @@ open class Query {
 
 @GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.OBJECT])
 open class GetStuffQuery {
-    open fun getStuff(): String = throw NotImplementedError("Query.getStuff must be implemented.")
+    open fun getStuff(dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Query.getStuff must be implemented.")
 }
 
 @GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.OBJECT])