Resolve enums correctly

Urigo · Mar 14, 2024 · 5c3528e · 5c3528e
1 parent 435e6f6
commit 5c3528e
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 8 deletions.
diff --git a/.changeset/shaggy-berries-compare.md b/.changeset/shaggy-berries-compare.md
@@ -0,0 +1,5 @@
+---
+"sofa-api": patch
+---
+
+Resolve enums correctly
diff --git a/example/types.ts b/example/types.ts
@@ -1,6 +1,11 @@
 export const typeDefs = /* GraphQL */ `
+  enum Dough {
+    THIN
+    THICK
+  }
+  
   type Pizza {
-    dough: String!
+    dough: Dough!
     toppings: [String!]
   }
 

diff --git a/src/router.ts b/src/router.ts
@@ -6,6 +6,7 @@ import {
   OperationTypeNode,
   isIntrospectionType,
   isInputObjectType,
+  isEnumType,
 } from 'graphql';
 import { buildOperationNodeForField, createGraphQLError } from '@graphql-tools/utils';
 import { getOperationInfo, OperationInfo } from './ast.js';
@@ -95,13 +96,20 @@ export function createRouter(sofa: Sofa) {
   for (const typeName in types) {
     const type = types[typeName];
 
-    if (
-      (isObjectType(type) || isInputObjectType(type)) &&
-      !isIntrospectionType(type)
-    ) {
-      sofa.openAPI!.components!.schemas![typeName] = buildSchemaObjectFromType(type, {
-        customScalars: sofa.customScalars,
-      });
+    if (!isIntrospectionType(type)) {
+      if (
+        (isObjectType(type) || isInputObjectType(type))
+      ) {
+        sofa.openAPI!.components!.schemas![typeName] = buildSchemaObjectFromType(type, {
+          customScalars: sofa.customScalars,
+        });
+      } else if (isEnumType(type)) {
+        sofa.openAPI!.components!.schemas![typeName] = {
+          title: type.name,
+          type: 'string',
+          enum: type.getValues().map((value) => value.name),
+        }
+      }
     }
   }