Add support for overriding assigned/generated types

Signed-off-by: Brendan Hay <[email protected]>

Author: brendanhay