fix: missing enum type reference

plantain-00 · Nov 12, 2018 · d79d540 · d79d540
1 parent 472b401
commit d79d540
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/demo/cases.json b/demo/cases.json
@@ -829,6 +829,24 @@
       ],
       "additionalProperties": false
     },
+    "StringEnum": {
+      "enum": [
+        "enum member 1",
+        "enum member 2"
+      ]
+    },
+    "NumberEnum": {
+      "enum": [
+        0,
+        1
+      ]
+    },
+    "NumberEnum2": {
+      "enum": [
+        3,
+        4
+      ]
+    },
     "StringType": {
       "type": "object",
       "properties": {

diff --git a/src/json-schema-generator.ts b/src/json-schema-generator.ts
@@ -23,6 +23,18 @@ export function generateJsonSchemas(typeDeclarations: TypeDeclaration[]) {
         type: undefined,
         $ref: `#/definitions/${typeDeclaration.name}`
       }
+    } else if (typeDeclaration.kind === 'enum') {
+      if (typeDeclaration.members.length === 1) {
+        definitions[typeDeclaration.name] = {
+          type: undefined,
+          const: typeDeclaration.members[0]
+        }
+      } else {
+        definitions[typeDeclaration.name] = {
+          type: undefined,
+          enum: typeDeclaration.members.map((m) => m.value)
+        }
+      }
     }
   }
   return typeDeclarations.filter(m => (m.kind === 'object' || m.kind === 'array' || m.kind === 'union') && m.entry)