pass test but broke others

Author: danadajian

src/annotations/build-directive-annotations.ts

@@ -17,6 +17,7 @@ import { ConstDirectiveNode } from "graphql/language";
 import { GraphQLSchema, isInputObjectType, Kind } from "graphql";
 import { shouldConsolidateTypes } from "../utils/should-consolidate-types";
 import { sanitizeName } from "../utils/sanitize-name";
+import { titleCase } from "../definitions/object";
 
 export function buildDirectiveAnnotations(
   definitionNode: DefinitionNode,
@@ -50,17 +51,22 @@ export function buildDirectiveAnnotations(
           return !typeWillBeConsolidated;
         },
       );
-      if (!directiveReplacementFromConfig) return "";
-      const kotlinAnnotations = buildKotlinAnnotations(
+      if (!directiveReplacementFromConfig)
+        return buildDefaultKotlinAnnotations(directive);
+      const kotlinAnnotationsFromConfig = buildKotlinAnnotationsFromConfig(
         directive,
         directiveReplacementFromConfig.kotlinAnnotations,
       );
-      return kotlinAnnotations.join("\n") + "\n";
+      return kotlinAnnotationsFromConfig.join("\n") + "\n";
     })
     .join("");
 }
 
-function buildKotlinAnnotations(
+function buildDefaultKotlinAnnotations(directive: ConstDirectiveNode) {
+  return `@${titleCase(directive.name.value)}\n`;
+}
+
+function buildKotlinAnnotationsFromConfig(
   directive: ConstDirectiveNode,
   kotlinAnnotations: NonNullable<
     CodegenConfigWithDefaults["directiveReplacements"]

src/definitions/object.ts

@@ -162,6 +162,6 @@ export function shouldGenerateFunctionsInClass(
   );
 }
 
-function titleCase(str: string) {
+export function titleCase(str: string) {
   return str.charAt(0).toUpperCase() + str.slice(1);
 }

src/visitor.ts

@@ -13,6 +13,7 @@ limitations under the License.
 
 import { BaseVisitor, RawConfig } from "@graphql-codegen/visitor-plugin-common";
 import {
+  DirectiveDefinitionNode,
   EnumTypeDefinitionNode,
   GraphQLSchema,
   InputObjectTypeDefinitionNode,
@@ -24,7 +25,7 @@ import { CodegenConfigWithDefaults } from "./config/build-config-with-defaults";
 import { buildEnumTypeDefinition } from "./definitions/enum";
 import { buildInterfaceDefinition } from "./definitions/interface";
 import { buildInputObjectDefinition } from "./definitions/input";
-import { buildObjectTypeDefinition } from "./definitions/object";
+import { buildObjectTypeDefinition, titleCase } from "./definitions/object";
 import { buildUnionTypeDefinition } from "./definitions/union";
 import { ParsedConfig } from "@graphql-codegen/visitor-plugin-common/typings/base-visitor";
 
@@ -39,6 +40,10 @@ export class KotlinVisitor extends BaseVisitor<
     super(rawConfig, rawConfig);
   }
 
+  DirectiveDefinition(node: DirectiveDefinitionNode): string {
+    return buildDirectiveDefinition(node);
+  }
+
   EnumTypeDefinition(node: EnumTypeDefinitionNode): string {
     return buildEnumTypeDefinition(node, this._schema, this.config);
   }
@@ -59,3 +64,12 @@ export class KotlinVisitor extends BaseVisitor<
     return buildUnionTypeDefinition(node, this._schema, this.config);
   }
 }
+
+function buildDirectiveDefinition(node: DirectiveDefinitionNode): string {
+  return `@GraphQLDirective(
+    name = "${titleCase(node.name.value)}",
+    description = "${node.description?.value ?? ""}",
+    locations = [${node.locations.map((location) => `graphql.introspection.Introspection.DirectiveLocation.${location.value}`).join(", ")}]
+)
+annotation class MyCustomDirective`;
+}

test/unit/should_generate_default_directives/expected.kt

@@ -5,7 +5,7 @@ import com.expediagroup.graphql.generator.annotations.*
 @GraphQLDirective(
     name = "MyCustomDirective",
     description = "A description for MyCustomDirective",
-    locations = [graphql.introspection.Introspection.DirectiveLocation.FIELD_DEFINITION]
+    locations = [graphql.introspection.Introspection.DirectiveLocation.OBJECT, graphql.introspection.Introspection.DirectiveLocation.FIELD_DEFINITION, graphql.introspection.Introspection.DirectiveLocation.INPUT_OBJECT, graphql.introspection.Introspection.DirectiveLocation.INPUT_FIELD_DEFINITION]
 )
 annotation class MyCustomDirective
 

test/unit/should_generate_default_directives/schema.graphql

@@ -1,3 +1,4 @@
+"A description for MyCustomDirective"
 directive @myCustomDirective on OBJECT | FIELD_DEFINITION | INPUT_OBJECT | INPUT_FIELD_DEFINITION
 
 type MyTypeWithCustomDirectiveOnObject @myCustomDirective {