feat: add properties and functions to cient and server

vasco-santos · Sep 20, 2019 · 0c7e43b · 0c7e43b
1 parent 845986b
commit 0c7e43b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/client.js b/client.js
@@ -13,6 +13,7 @@ module.exports = function (addr, opts) {
 
   const stream = duplex(socket, opts)
   stream.remoteAddress = url
+  stream.socket = socket
   stream.close = () => new Promise((resolve, reject) => {
     socket.addEventListener('close', resolve)
     socket.close()

diff --git a/server.js b/server.js
@@ -41,6 +41,9 @@ module.exports = !WebSocket.Server ? null : function (opts, onConnection) {
   wsServer.on('connection', function (socket, req) {
     var stream = ws(socket)
     stream.remoteAddress = req.socket.remoteAddress
+    stream.remotePort = req.socket.remotePort
+    stream.close = () => emitter.close()
+
     emitter.emit('connection', stream, req)
   })