I maintain Orchid, a project for documenting code, especially Kotlin, and have just started having some discussions around adding something like that. Not currently supported out-of-the-box, but is definitely possible with some customization and is quickly becoming a highly desired feature. I would love to hear more about what specifically your’re looking for in these diagrams, if you wanted to share your thoughts in this issue
https://github.com/JavaEden/Orchid/issues/275