fix: enhance union type / array handling in input fields

getcronit · Dec 4, 2024 · 75398e6 · 75398e6 · github-actions · Dec 4, 2024
1 parent 871c6d0
commit 75398e6
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 10 deletions.
diff --git a/packages/pylon-builder/src/schema/schema-parser.ts b/packages/pylon-builder/src/schema/schema-parser.ts
@@ -823,18 +823,19 @@ export class SchemaParser {
               ])
             })
           } else {
-            const firstType = type.types[0]
+            let firstType = type.getNonNullableType()
 
-            consola.warn(
-              `Warning: Union types in input fields are not supported yet. Defaulting to the first type (${this.checker.typeToString(
-                firstType
-              )}) at path: ${path.join(' > ')}`
-            )
+            if (firstType.isUnion() && !isPrimitive(firstType)) {
+              consola.warn(
+                `Warning: Union types in input fields are not supported yet. Defaulting to the first type (${this.checker.typeToString(
+                  firstType
+                )}) at path: ${path.join(' > ')}`
+              )
 
-            recLoop(firstType.getNonNullableType(), info, processing, [
-              ...path,
-              'NON_NULLABLE'
-            ])
+              firstType = firstType.types[0]
+            }
+
+            recLoop(firstType, info, processing, [...path, 'NON_NULLABLE'])
           }
         }
       } else if (isFunction(type)) {

diff --git a/packages/pylon-builder/src/schema/types-helper.ts b/packages/pylon-builder/src/schema/types-helper.ts
@@ -17,6 +17,15 @@ export const isList = (checker: ts.TypeChecker, type: ts.Type) => {
       typeNode.kind === ts.SyntaxKind.TupleType)
   )
 
+  if (!is) {
+    // Check if type references an array
+    const isArray = type.getSymbol()?.getName() === 'Array'
+
+    if (isArray) {
+      return true
+    }
+  }
+
   return is
 }