Add websocket handler

Author: simonwep

package-lock.json

@@ -28,7 +28,8 @@
  "mongodb": "^3.1.10",
  "mongoose": "^5.4.0",
  "multer": "^1.4.1",
- "pm2": "^3.2.9"
+ "pm2": "^3.2.9",
+ "ws": "^6.1.3"
  },
  "devDependencies": {
  "eslint": "^5.10.0",

package.json

@@ -1,5 +1,6 @@
 const compression = require('compression');
 const express = require('express');
+const server = require('http').createServer();
 const fs = require('fs');
 const cors = require('cors');
 const path = require('path');
@@ -31,5 +32,9 @@ app.use(cors());
 // GraphQL API Module
 app.use('/api', require('./api/api'));
 
-// Start
-app.listen(config.server.port);
+// Spawn Websocket
+require('./websocket')(server);
+
+// Add express http-server
+server.on('request', app);
+server.listen(config.server.port);

src/app.js

@@ -0,0 +1,72 @@
+const userModel = require('./models/user');
+const WebSocket = require('ws');
+
+module.exports = server => {
+ const wss = new WebSocket.Server({server: server});
+ const userMap = {};
+
+ wss.on('connection', ws => {
+ let user;
+
+ ws.on('message', async message => {
+
+ // Try to parse message
+ try {
+ message = JSON.parse(message);
+ } catch (ignored) {
+ return;
+ }
+
+ const {type, value} = message;
+ switch (type) {
+ case 'register': {
+ if (typeof value === 'string' && value) {
+ user = await userModel.findOne({apikeys: {$elemMatch: {key: value}}});
+
+ if (!user) {
+ return;
+ }
+
+ const userid = user.id;
+ if (!(userid in userMap)) {
+ userMap[userid] = [];
+ }
+
+ userMap[userid].push(ws);
+ }
+
+ break;
+ }
+ case 'broadcast': {
+ if (user) {
+ const socketList = userMap[user.id];
+
+ // Broadcast message
+ for (let i = 0, l = socketList.length; i < l; i++) {
+ const socket = socketList[i];
+
+ if (socket !== ws) {
+ socket.send(JSON.stringify(value));
+ }
+ }
+ }
+ break;
+ }
+ }
+ });
+
+ ws.on('close', () => {
+
+ // Check if socket was registered
+ if (user) {
+ const socketList = userMap[user.id];
+ const idx = socketList.indexOf(ws);
+
+ // Remove socket
+ if (~idx) {
+ socketList.splice(idx, 1);
+ }
+ }
+ });
+ });
+};