feat: add properties and functions to client and server

README.md

@@ -50,6 +50,8 @@ await server.listen(PORT)
 
 Create a websocket client connection. Set `binary: true` to get a stream of arrayBuffers (on the browser). Defaults to true on node, but to strings on the browser. This may cause a problems if your application assumes binary.
 
+For adding options to the Websocket instance, as [websockets/ws/blob/master/doc/ws.md#new-websocketaddress-protocols-options](https://github.com/websockets/ws/blob/master/doc/ws.md#new-websocketaddress-protocols-options), you can provide an object with the `websocket` property into the connect options.
+
 ```js
 const stream = connect(url)
 // stream is duplex and is both a `source` and `sink`.

client.js

@@ -7,12 +7,14 @@ var wsurl = require('./ws-url')
 
 module.exports = function (addr, opts) {
   const location = typeof window === 'undefined' ? {} : window.location
+  opts = opts || {}
 
   const url = wsurl(addr, location)
-  const socket = new WebSocket(url)
+  const socket = new WebSocket(url, opts.websocket)
 
   const stream = duplex(socket, opts)
   stream.remoteAddress = url
+  stream.socket = socket
   stream.close = () => new Promise((resolve, reject) => {
     socket.addEventListener('close', resolve)
     socket.close()