Pcc/support native enum type (#8)

Co-authored-by: pcc <[email protected]>
Co-authored-by: Pierre Cardona <[email protected]>

Author: aidansunbury

README.MD

@@ -169,6 +169,7 @@ The following are supported
 - Effects
 - Enum
 - Literal
+- NativeEnum
 - Nullable
 - Null
 - Nullish

packages/dev-app/src/router.ts

@@ -51,6 +51,11 @@ export const appRouter = createTRPCRouter({
       .query(() => {
         return "It's an input";
       }),
+    nativeEnumInput: procedure
+    .input(z.object({ aEnumInput: z.nativeEnum({ONE: "one", TWO: "two"}) }))
+    .query(() => {
+      return "It's an input";
+    }),
     stringArrayInput: procedure
       .input(z.object({ aStringArray: z.string().array() }))
       .query(() => {

packages/trpc-panel/src/parse/input-mappers/__tests__/zod/nativeEnum.test.ts

@@ -0,0 +1,26 @@
+import { defaultReferences } from "@src/parse/input-mappers/defaultReferences";
+import { parseZodNativeEnumDef } from "@src/parse/input-mappers/zod/parsers/parseZodNativeEnumDef";
+import { EnumNode } from "@src/parse/parseNodeTypes";
+import { z } from "zod";
+
+describe("Parse ZodNativeEnum", () => {
+  it("should parse a zod native enum", () => {
+    const expected: EnumNode = {
+      type: "enum",
+      enumValues: ["one", "two", "three"],
+      path: [],
+    };
+
+    enum ExampleEnum {
+      ONE = "one",
+      TWO = "two",
+      THREE = "three",
+    };
+
+    const parsed = parseZodNativeEnumDef(
+      z.nativeEnum(ExampleEnum)._def,
+      defaultReferences()
+    );
+    expect(expected).toStrictEqual(parsed);
+  });
+});

packages/trpc-panel/src/parse/input-mappers/zod/parsers/parseZodNativeEnumDef.ts

@@ -0,0 +1,12 @@
+import { nodePropertiesFromRef } from "@src/parse/utils";
+import { ZodNativeEnumDef } from "zod";
+import { EnumNode, ParseFunction } from "../../../parseNodeTypes";
+
+export const parseZodNativeEnumDef: ParseFunction<ZodNativeEnumDef, EnumNode> = (
+  def,
+  refs
+) => {
+  const values = Object.values(def.values) as string[];
+  refs.addDataFunctions.addDescriptionIfExists(def, refs);
+  return { type: "enum", enumValues: values, ...nodePropertiesFromRef(refs) };
+};

packages/trpc-panel/src/parse/input-mappers/zod/selector.ts

@@ -6,6 +6,7 @@ import {
   ZodDefaultDef,
   ZodEffectsDef,
   ZodEnumDef,
+  ZodNativeEnumDef,
   ZodFirstPartyTypeKind,
   ZodLiteralDef,
   ZodNullableDef,
@@ -29,6 +30,7 @@ import {
   ZodDiscriminatedUnionDefUnversioned,
 } from "./parsers/parseZodDiscriminatedUnionDef";
 import { parseZodEnumDef } from "./parsers/parseZodEnumDef";
+import { parseZodNativeEnumDef } from "./parsers/parseZodNativeEnumDef";
 import { parseZodLiteralDef } from "./parsers/parseZodLiteralDef";
 import { parseZodNumberDef } from "./parsers/parseZodNumberDef";
 import { parseZodObjectDef } from "./parsers/parseZodObjectDef";
@@ -65,6 +67,8 @@ export const zodSelectorFunction: ParserSelectorFunction<ZodDefWithType> = (
       );
     case ZodFirstPartyTypeKind.ZodEnum:
       return parseZodEnumDef(def as ZodEnumDef, references);
+    case ZodFirstPartyTypeKind.ZodNativeEnum:
+      return parseZodNativeEnumDef(def as ZodNativeEnumDef, references);
     case ZodFirstPartyTypeKind.ZodLiteral:
       return parseZodLiteralDef(def as ZodLiteralDef, references);
     case ZodFirstPartyTypeKind.ZodNumber: