A websocket middleware for express

README.md

A websocket middleware for express

A convenient way to expose a websocket connections in route handles.

Usage

const express = require('express')
    , ws = require('./ws')

const app = express()

app.get('/', (req, res) => 
  req.ws
    ? res.ws(socket => /* do something with the socket */ )
    : res.send('Hello you non websocket')
)

app.listen(PORT)

server.on('upgrade', ws(app))

ws.js

const uws = require('uws')
    , http = require('http')

const wss = new uws.Server({ noServer: true })

module.exports = app => (req, socket, head) => {
  const res = new http.ServerResponse(req)
  res.assignSocket(socket)
  res.on('finish', () => res.socket.destroy())
  req.ws = true
  res.ws = fn => wss.handleUpgrade(req, socket, head, fn)
  app(req, res)
}