put the websocket route in the map containing all routes

Instead of handling the websocket differently just handle it as a normal
route and upgrade it to a websocket.