Add types to ZodFirstPartySchemaTypes (#3247)

* Add `ZodSymbol` and `ZodReadonly` to `ZodFirstPartySchemaTypes` * Test `ZodFirstPartySchemaTypes` in sync with `ZodFirstPartyTypeKind`
colinhacks · Mar 14, 2024 · cb94d30 · cb94d30
1 parent 7fa37bb
commit cb94d30
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/src/__tests__/firstpartyschematypes.test.ts b/src/__tests__/firstpartyschematypes.test.ts
@@ -0,0 +1,20 @@
+// @ts-ignore TS6133
+import { test } from "@jest/globals";
+import { util } from "../helpers/util";
+
+import { 
+  ZodFirstPartySchemaTypes,
+  ZodFirstPartyTypeKind,
+} from "..";
+
+test("Identify missing [ZodFirstPartySchemaTypes]", () => {
+  type ZodFirstPartySchemaForType<T extends ZodFirstPartyTypeKind> = ZodFirstPartySchemaTypes extends infer Schema
+    ? Schema extends { _def: { typeName: T } }
+      ? Schema
+      : never
+    : never;
+  type ZodMappedTypes = { [key in ZodFirstPartyTypeKind]: ZodFirstPartySchemaForType<key> };
+  type ZodFirstPartySchemaTypesMissingFromUnion = keyof { [key in keyof ZodMappedTypes as ZodMappedTypes[key] extends { _def: never } ? key : never]: unknown };
+
+  util.assertNever({} as ZodFirstPartySchemaTypesMissingFromUnion);
+});
diff --git a/src/types.ts b/src/types.ts
@@ -4972,7 +4972,9 @@ export type ZodFirstPartySchemaTypes =
   | ZodCatch<any>
   | ZodPromise<any>
   | ZodBranded<any, any>
-  | ZodPipeline<any, any>;
+  | ZodPipeline<any, any>
+  | ZodReadonly<any>
+  | ZodSymbol;
 
 // requires TS 4.4+
 abstract class Class {