feat: 解析枚举值

Author: geekact

src/lib/parse-schema.ts

@@ -9,6 +9,14 @@ export const parseSchema = (
   const parsed = refToObject(docs, schema);
   const nullable = parsed.nullable ? ' | null' : '';
 
+  if (parsed.enum?.length) {
+    return `${parsed.enum.map((item) => (typeof item === 'number' ? item : `"${item}"`)).join(' | ')}${nullable}`;
+  }
+
+  if (parsed.oneOf) {
+    return parsed.oneOf.map((schema) => parseSchema(docs, schema)).join(' | ');
+  }
+
   switch (parsed.type) {
     case 'array':
       return `(${parseSchema(docs, parsed.items)})[]${nullable}`;
@@ -29,9 +37,5 @@ export const parseSchema = (
       return `string${nullable}`;
   }
 
-  if (parsed.oneOf) {
-    return parsed.oneOf.map((schema) => parseSchema(docs, schema)).join(' | ');
-  }
-
   return 'unknown';
 };

test/lib/parse-schema.test.ts

@@ -75,6 +75,12 @@ describe('常规', () => {
     });
     expect(type).toMatchInlineSnapshot(`"string | boolean | number"`);
   });
+
+  test('枚举', () => {
+    expect(
+      parseSchema(docs, { enum: ['foo', 'bar', 'baz', 1, 2] }),
+    ).toMatchInlineSnapshot(`""foo" | "bar" | "baz" | 1 | 2"`);
+  });
 });
 
 describe('nullable', () => {
@@ -128,6 +134,12 @@ describe('nullable', () => {
       parseSchema(docs, { type: 'string', format: 'binary', nullable: true }),
     ).toMatchInlineSnapshot(`"Blob | null"`);
   });
+
+  test('枚举', () => {
+    expect(
+      parseSchema(docs, { enum: ['foo', 'bar', 'baz'], nullable: true }),
+    ).toMatchInlineSnapshot(`""foo" | "bar" | "baz" | null"`);
+  });
 });
 
 test('未知类型', () => {