You should take a look at mobx-state-tree. It provides this capability in a straightforward way and is more approachable in general.