feat(mock): infer type from enum values for string and number (#1491)

* feat(mock): infer type from enum values for string and number

* cleanup

packages/mock/src/faker/getters/scalar.ts

@@ -108,7 +108,9 @@ export const getMockScalar = ({
     };
   }
 
-  switch (item.type) {
+  const type = getItemType(item);
+
+  switch (type) {
     case 'number':
     case 'integer': {
       let value = getNullable(
@@ -303,3 +305,15 @@ export const getMockScalar = ({
     }
   }
 };
+
+function getItemType(item: MockSchemaObject) {
+  if (item.type) return item.type;
+  if (!item.enum) return;
+
+  const uniqTypes = new Set(item.enum.map((value) => typeof value));
+  if (uniqTypes.size > 1) return;
+
+  const type = Array.from(uniqTypes.values()).at(0);
+  if (!type) return;
+  return ['string', 'number'].includes(type) ? type : undefined;
+}

tests/configs/mock.config.ts

@@ -126,4 +126,12 @@ export default defineConfig({
       target: '../specifications/enum-refs.yaml',
     },
   },
+  typelessEnum: {
+    input: '../specifications/typelessEnum.yaml',
+    output: {
+      schemas: '../generated/mock/typelessEnum/schemas',
+      target: '../generated/mock/typelessEnum',
+      mock: true,
+    },
+  },
 });

tests/specifications/typelessEnum.yaml

@@ -0,0 +1,29 @@
+openapi: 3.0.3
+info:
+  title: Typeless enums
+  version: 1.0.0
+paths:
+  /api/colors:
+    get:
+      summary: sample colors
+      responses:
+        '200':
+          description: OK
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/ColorObject'
+components:
+  schemas:
+    Colors1:
+      type: string
+      enum: [red, blue, yellow]
+    Colors2:
+      enum: [green, purple, orange]
+    ColorObject:
+      type: object
+      properties:
+        color:
+          oneOf:
+            - $ref: '#/components/schemas/Colors1'
+            - $ref: '#/components/schemas/Colors2'