I also need a WebSocket solution and was planning to do the same thing you did so I am really interested in how this turns out.
I have just been through a similar exercise updating the Webduino libraries to use the DigiFi (as opposed to EthernetServer+EthernetClient from the standard Arduino distribution). It has been a bit of hit or miss, with very unreliable WiFi connections, it seems the connection is not closed or is lost and the browsers get stuck waiting forever for my page to slowly load (first the html, then a js file, then a css file sometimes, then a jpeg or two, then a few more, etc.).