88 lines
1.8 KiB
JavaScript
88 lines
1.8 KiB
JavaScript
'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()
|
|
}
|