validation based on other field in schema

[Hooman-studio]

Hi everyone!

I am just starting with zod, and I can't seem to figure out a way to do the following:

1. Have an enum type lets say: ["one", "two", "three"]
2. when that enum type is one don't add any field
3. when that enum type is two I want the schema to have a required field extra of type string.
4. when that enum type is three I want the schema to have a required field another_extra of type string

I have read about descriminative unions, but ideally I don't want the enum type to be inside a union (i want it to be accessible as schema.myEnum not schema.myUnion.myEnum

Wondering if there is a way to accomplish this!
Thank you!