Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

deny unexpected keys @ ZodObject's .omit(mask),.pick(mask),.required(mask) & .partial(mask) at compile time. #1564

Merged
merged 9 commits into from Feb 8, 2023
Merged

deny unexpected keys @ ZodObject's .omit(mask),.pick(mask),.required(mask) & .partial(mask) at compile time. #1564

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
ee7f362
deny unexpected keys @ `ZodObject.omit(...)` & `ZodObject.pick(...)`.
igalklebanov Nov 16, 2022
95104bf
make runtime unit tests ignore unexpected key ts error.
igalklebanov Nov 16, 2022
bd0c300
forgot to run yarn build:deno.
igalklebanov Nov 16, 2022
c6e7b08
apply same restrictions for `required(...)` & `partial(...)` masks.
igalklebanov Nov 16, 2022
796dd6b
add "non existent key" tests @ partials & pickomit.
igalklebanov Dec 16, 2022
6c79e02
Naming tweak
Dec 24, 2022
ca8bd05
Merge branch 'master' into fix-pick-omit-allow-unexpected-keys
igalklebanov Dec 24, 2022
84fc595
Merge branch 'master' into fix-pick-omit-allow-unexpected-keys
igalklebanov Jan 29, 2023
1102a5d
Merge branch 'master' into fix-pick-omit-allow-unexpected-keys
colinhacks Feb 8, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
5 changes: 3 additions & 2 deletions deno/lib/__tests__/pickomit.test.ts
View file
Open in desktop
Expand Up @@ -66,13 +66,13 @@ test("nonstrict inference", () => {
});

test("nonstrict parsing - pass", () => {
const laxfish = fish.nonstrict().pick({ name: true });
const laxfish = fish.passthrough().pick({ name: true });
laxfish.parse({ name: "asdf", whatever: "asdf" });
laxfish.parse({ name: "asdf", age: 12, nested: {} });
});

test("nonstrict parsing - fail", () => {
const laxfish = fish.nonstrict().pick({ name: true });
const laxfish = fish.passthrough().pick({ name: true });
const bad = () => laxfish.parse({ whatever: "asdf" } as any);
expect(bad).toThrow();
});
Expand All @@ -85,6 +85,7 @@ test("pick a nonexistent key", () => {

const pickedSchema = schema.pick({
a: true,
// @ts-expect-error
doesntExist: true,
igalklebanov marked this conversation as resolved.
Show resolved Hide resolved
});

Expand Down
14 changes: 10 additions & 4 deletions deno/lib/types.ts
View file
Open in desktop
Expand Up @@ -1772,6 +1772,12 @@ export type SomeZodObject = ZodObject<
any
>;

export type objectKeyMask<Obj> = { [k in keyof Obj]?: true };

export type optionalPickWith<Obj, Shape> = {
[k in keyof Obj]?: k extends keyof Shape ? Obj[k] : never;
};

function deepPartialify(schema: ZodTypeAny): any {
if (schema instanceof ZodObject) {
const newShape: any = {};
Expand Down Expand Up @@ -2015,8 +2021,8 @@ export class ZodObject<
}) as any;
}

pick<Mask extends { [k in keyof T]?: true }>(
mask: Mask
pick<Mask extends objectKeyMask<T>>(
mask: optionalPickWith<Mask, objectKeyMask<T>>
): ZodObject<Pick<T, Extract<keyof T, keyof Mask>>, UnknownKeys, Catchall> {
const shape: any = {};
util.objectKeys(mask).map((key) => {
Expand All @@ -2029,8 +2035,8 @@ export class ZodObject<
}) as any;
}

omit<Mask extends { [k in keyof T]?: true }>(
mask: Mask
omit<Mask extends objectKeyMask<T>>(
mask: optionalPickWith<Mask, objectKeyMask<T>>
): ZodObject<Omit<T, keyof Mask>, UnknownKeys, Catchall> {
const shape: any = {};
util.objectKeys(this.shape).map((key) => {
Expand Down
5 changes: 3 additions & 2 deletions src/__tests__/pickomit.test.ts
View file
Open in desktop
Expand Up @@ -65,13 +65,13 @@ test("nonstrict inference", () => {
});

test("nonstrict parsing - pass", () => {
const laxfish = fish.nonstrict().pick({ name: true });
const laxfish = fish.passthrough().pick({ name: true });
igalklebanov marked this conversation as resolved.
Show resolved Hide resolved
laxfish.parse({ name: "asdf", whatever: "asdf" });
laxfish.parse({ name: "asdf", age: 12, nested: {} });
});

test("nonstrict parsing - fail", () => {
const laxfish = fish.nonstrict().pick({ name: true });
const laxfish = fish.passthrough().pick({ name: true });
const bad = () => laxfish.parse({ whatever: "asdf" } as any);
expect(bad).toThrow();
});
Expand All @@ -84,6 +84,7 @@ test("pick a nonexistent key", () => {

const pickedSchema = schema.pick({
a: true,
// @ts-expect-error
doesntExist: true,
});

Expand Down
14 changes: 10 additions & 4 deletions src/types.ts
View file
Open in desktop
Expand Up @@ -1772,6 +1772,12 @@ export type SomeZodObject = ZodObject<
any
>;

export type objectKeyMask<Obj> = { [k in keyof Obj]?: true };

export type optionalPickWith<Obj, Shape> = {
[k in keyof Obj]?: k extends keyof Shape ? Obj[k] : never;
};

function deepPartialify(schema: ZodTypeAny): any {
if (schema instanceof ZodObject) {
const newShape: any = {};
Expand Down Expand Up @@ -2015,8 +2021,8 @@ export class ZodObject<
}) as any;
}

pick<Mask extends { [k in keyof T]?: true }>(
mask: Mask
pick<Mask extends objectKeyMask<T>>(
mask: optionalPickWith<Mask, objectKeyMask<T>>
): ZodObject<Pick<T, Extract<keyof T, keyof Mask>>, UnknownKeys, Catchall> {
const shape: any = {};
util.objectKeys(mask).map((key) => {
Expand All @@ -2029,8 +2035,8 @@ export class ZodObject<
}) as any;
}

omit<Mask extends { [k in keyof T]?: true }>(
mask: Mask
omit<Mask extends objectKeyMask<T>>(
mask: optionalPickWith<Mask, objectKeyMask<T>>
): ZodObject<Omit<T, keyof Mask>, UnknownKeys, Catchall> {
const shape: any = {};
util.objectKeys(this.shape).map((key) => {
Expand Down