How to define Map with correlation between a key type and a value type, while they both are unions

Question

Here is the example which shows what I want to achieve. It almost works except 2 problems:

1. Set does not show an error on incorrect code
2. Immer Draft type (o