I'm not done with this yet and it's not typechecking but feel free to look at https://github.com/pantsbuild/pants/pull/9158