Further improve Java compatibility

After merging #411 new opportunities for strenghtening and fixing Java API ergonomics arose, particularly around the AST Transformers API.

This is an umbrella issue to cover those improvements (that don't break or significantly change the APIs).