feat: support union types

R-unic · Oct 1, 2023 · 2905501 · 2905501
1 parent 45f1e6a
commit 2905501
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/code-generator.ts b/src/code-generator.ts
@@ -10,8 +10,9 @@ import {
   LiteralLikeNode,
   StringLiteral,
   TypeNode,
-  ArrayLiteralExpression,
+  UnionTypeNode,
   ArrayTypeNode,
+  ArrayLiteralExpression,
   TypeAliasDeclaration,
   ModifierLike,
   TypeReferenceNode,
@@ -1282,6 +1283,15 @@ export default class CodeGenerator extends StringBuilder {
 
   private walkType(type: TypeNode): void {
     switch(type.kind) {
+      case SyntaxKind.UnionType: {
+        const union = <UnionTypeNode>type;
+        for (const type of union.types) {
+          this.walk(type);
+          if (Util.isNotLast(type, union.types))
+            this.append(" | ");
+        }
+        break;
+      }
       case SyntaxKind.NumberKeyword: {
         this.append("Num");
         break;