socketio-jwt/test/fixture/secret_function.js

'use strict' // Node 4.x workaround

const express = require('express')
const http = require('http')

const socketIo = require('socket.io')
const socketio_jwt = require('../../lib')

const jwt = require('jsonwebtoken')
const xtend = require('xtend')
const enableDestroy = require('server-destroy')

let sio

exports.start = (options, callback) => {
  const SECRETS = {
    123: 'aaafoo super sercret',
    555: 'other'
  }

  if (typeof options == 'function') {
    callback = options
    options = {}
  }

  options = xtend(
    {
      secret: (request, decodedToken, callback) => {
        callback(null, SECRETS[decodedToken.id])
      },
      timeout: 1000,
      handshake: true
    },
    options
  )

  const app = express()
  const server = http.createServer(app)
  sio = socketIo.listen(server)

  app.use(express.json())
  app.post('/login', (req, res) => {
    const profile = {
      first_name: 'John',
      last_name: 'Doe',
      email: 'john@doe.com',
      id: req.body.username === 'valid_signature' ? 123 : 555
    }

    // We are sending the profile inside the token
    const token = jwt.sign(profile, SECRETS[123], { expiresIn: 60 * 60 * 5 })
    res.json({ token: token })
  })

  if (options.handshake) {
    sio.use(socketio_jwt.authorize(options))

    sio.sockets.on('echo', (m) => {
      sio.sockets.emit('echo-response', m)
    })
  } else {
    sio.sockets
      .on('connection', socketio_jwt.authorize(options))
      .on('authenticated', (socket) => {
        socket.on('echo', (m) => {
          socket.emit('echo-response', m)
        })
      })
  }

  server.__sockets = []
  server.on('connection', (c) => {
    server.__sockets.push(c)
  })

  server.listen(9000, callback)
  enableDestroy(server)
}

exports.stop = (callback) => {
  sio.close()
  try {
    server.destroy()
  } catch (er) {}

  callback()
}