After looking for good tools in this space, to create documentation diagrams, I stayed with Draw.io and yEd[1].

The latter has a pretty cool feature that re-arranges all objects according to some predefined layout. Quite neat.

[1]: https://www.yworks.com/products/yed