Hello!
I think you should consider making a diagram in the style of Akka Streams blocks like the one in your code. It helped me think about the constructed flow and how to do composition in a websocket flow.
Thanks for your work! Feel free to close if you are not interested.