Merge pull request #417 from 0xOlias/kjs/fix-enum-type-infer

Fix bug with inferring enum types
ponder-sh · Nov 10, 2023 · 390c0f6 · 390c0f6 · vercel · Nov 10, 2023
2 parents 49fa33c + 583e066
commit 390c0f6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/packages/core/src/schema/schema.test-d.ts b/packages/core/src/schema/schema.test-d.ts
@@ -96,5 +96,5 @@ test("schema", () => {
   type t = p.Infer<typeof s>;
   //   ^?
 
-  assertType<t>({} as { t: { id: string; e: ["ONE", "TWO"] } });
+  assertType<t>({} as { t: { id: string; e: "ONE" | "TWO" } });
 });
diff --git a/packages/core/src/schema/types.ts b/packages/core/src/schema/types.ts
@@ -262,7 +262,12 @@ export type RecoverEnumType<
     | ReferenceColumn
     | EnumColumn
     | VirtualColumn
-> = TColumn extends EnumColumn ? TEnums[TColumn["type"] & keyof TEnums] : never;
+> = TColumn extends EnumColumn
+  ? TEnums[TColumn["type"] &
+      keyof TEnums] extends infer _enum extends readonly string[]
+    ? _enum[number]
+    : never
+  : never;
 
 export type RecoverTableType<
   TEnums extends Record<string, Enum>,