directives

Author: miorel

workspaces/leetcode-api/src/api/question-list/fetchGraphQL.generated.ts

@@ -7,10 +7,36 @@ import {
   GraphQLObjectType,
   GraphQLScalarType,
   Kind,
+  type ConstDirectiveNode,
   type GraphQLNamedType,
   type SelectionSetNode,
 } from "graphql";
 
+// TODO: get some of these directives added to the schema
+type DirectivesConfig = {
+  nonnegative: boolean;
+};
+
+function parseDirectives(
+  directives: readonly ConstDirectiveNode[],
+): DirectivesConfig {
+  const res = { nonnegative: false };
+  for (const directive of directives) {
+    switch (directive.name.value) {
+      case "nonnegative": {
+        invariant(!directive.arguments?.length, "No arguments allowed!");
+        res.nonnegative = true;
+        break;
+      }
+      default: {
+        throw new Error("Unsupported directive " + directive.name.value);
+      }
+    }
+  }
+
+  return res;
+}
+
 class ZodOutput {
   constructor(
     public text: string,
@@ -34,19 +60,29 @@ class ZodOutput {
 function generateZod(
   currentType: GraphQLNamedType,
   selectionSet: SelectionSetNode | undefined,
+  directives: readonly ConstDirectiveNode[] = [],
 ): ZodOutput {
   if (currentType instanceof GraphQLScalarType) {
     switch (currentType.name) {
       case "Boolean": {
+        invariant(directives.length === 0, "Directives not supported here.");
         return new ZodOutput("z.boolean()", true);
       }
       case "Int": {
-        return new ZodOutput("z.number().int()", true);
+        const directiveConfig = parseDirectives(directives);
+        return new ZodOutput(
+          [
+            "z.number().int()",
+            directiveConfig.nonnegative ? ".nonnegative()" : "",
+          ].join(""),
+          true,
+        );
       }
       case "Date":
       case "DateTime":
       case "ID":
       case "String": {
+        invariant(directives.length === 0, "Directives not supported here.");
         return new ZodOutput("z.string()", true);
       }
       default: {
@@ -56,37 +92,46 @@ function generateZod(
   }
 
   if (currentType instanceof GraphQLObjectType) {
+    invariant(directives.length === 0, "Directives not supported here.");
+
     const selections = nullthrows(selectionSet).selections.map((s) => {
       invariant(s.kind === Kind.FIELD, "Expected direct fields!");
 
       const processStack: ((prev: ZodOutput) => ZodOutput)[] = [];
 
-      let fieldType = currentType.getFields()[s.name.value].type;
-      while (
-        fieldType instanceof GraphQLList ||
-        fieldType instanceof GraphQLNonNull
-      ) {
+      const field = currentType.getFields()[s.name.value];
+      let fieldType = field.type;
+
+      while (true) {
         if (fieldType instanceof GraphQLList) {
           processStack.push(
             (prev) => new ZodOutput(`z.array(${prev.stringify(false)})`, true),
           );
-        } else {
+          fieldType = fieldType.ofType;
+          continue;
+        }
+
+        if (fieldType instanceof GraphQLNonNull) {
           processStack.push((prev) => prev.setIsNullable(false));
+          fieldType = fieldType.ofType;
+          continue;
         }
 
-        fieldType = fieldType.ofType;
+        return (
+          JSON.stringify(s.name.value) +
+          ": " +
+          processStack
+            .reduceRight(
+              (acc, fn) => fn(acc),
+              generateZod(
+                fieldType,
+                s.selectionSet,
+                field.astNode?.directives ?? [],
+              ),
+            )
+            .stringify()
+        );
       }
-
-      return (
-        JSON.stringify(s.name.value) +
-        ": " +
-        processStack
-          .reduceRight(
-            (acc, fn) => fn(acc),
-            generateZod(fieldType, s.selectionSet),
-          )
-          .stringify()
-      );
     });
 
     return new ZodOutput(`z.object({${selections.join(",")}})`, true);