Added WebSockets with funcionality to send messages from any module

Author: pawelmalak

Socket.js

@@ -0,0 +1,21 @@
+const WebSocket = require('ws');
+
+class Socket {
+  constructor(server) {
+    this.webSocketServer = new WebSocket.Server({ server })
+
+    this.webSocketServer.on('listening', () => {
+      console.log('socket listen');
+    })
+
+    this.webSocketServer.on('connection', (webSocketClient) => {
+      console.log('new connection');
+    })
+  }
+
+  send(msg) {
+    this.webSocketServer.clients.forEach(client => client.send(msg));
+  }
+}
+
+module.exports = Socket;

Sockets.js

@@ -0,0 +1,20 @@
+class Sockets {
+  constructor() {
+    this.sockets = [];
+  }
+
+  registerSocket(name, socket) {
+    this.sockets.push({ name, socket });
+  }
+
+  getAllSockets() {
+    return this.sockets;
+  }
+
+  getSocket(name) {
+    const socket = this.sockets.find(socket => socket.name === name);
+    return socket;
+  }
+}
+
+module.exports = new Sockets();

api.js

@@ -0,0 +1,21 @@
+const express = require('express');
+const errorHandler = require('./middleware/errorHandler');
+
+const api = express();
+
+api.get('/', (req, res) => {
+  res.send('Server is working');
+})
+
+// Body parser
+api.use(express.json());
+
+// Link controllers with routes
+api.use('/api/apps', require('./routes/apps'));
+api.use('/api/config', require('./routes/config'));
+api.use('/api/weather', require('./routes/weather'));
+
+// Custom error handler
+api.use(errorHandler);
+
+module.exports = api;

package-lock.json

@@ -20,7 +20,8 @@
     "express": "^4.17.1",
     "node-schedule": "^2.0.0",
     "sequelize": "^6.6.2",
-    "sqlite3": "^5.0.2"
+    "sqlite3": "^5.0.2",
+    "ws": "^7.4.5"
   },
   "devDependencies": {
     "nodemon": "^2.0.7"

package.json

@@ -1,30 +1,23 @@
-const express = require('express');
+const http = require('http');
 const { connectDB } = require('./db');
-const errorHandler = require('./middleware/errorHandler');
+const api = require('./api');
 const jobs = require('./utils/jobs');
-const colors = require('colors');
-require('dotenv').config();
-
-connectDB();
+const Socket = require('./Socket');
+const Sockets = require('./Sockets');
 
-const app = express();
+require('dotenv').config();
 const PORT = process.env.PORT || 5005;
 
-app.get('/', (req, res) => {
-  res.send('Server is working');
-})
-
-// Body parser
-app.use(express.json());
+connectDB();
 
-// Link controllers with routes
-app.use('/api/apps', require('./routes/apps'));
-app.use('/api/config', require('./routes/config'));
-app.use('/api/weather', require('./routes/weather'));
+// Create server for Express API and WebSockets
+const server = http.createServer();
+server.on('request', api);
 
-// Custom error handler
-app.use(errorHandler);
+// Register weatherSocket
+const weatherSocket = new Socket(server);
+Sockets.registerSocket('weather', weatherSocket);
 
-app.listen(PORT, () => {
+server.listen(PORT, () => {
   console.log(`Server is running on port ${PORT} in ${process.env.NODE_ENV} mode`.yellow.bold);
 })

server.js

@@ -1,10 +1,12 @@
 const schedule = require('node-schedule');
 const getExternalWeather = require('./getExternalWeather');
+const Sockets = require('../Sockets');
 
 const weatherJob = schedule.scheduleJob('updateWeather', '0 */15 * * * *', async () => {
   try {
     await getExternalWeather();
     console.log('weather updated');
+    Sockets.getSocket('weather').socket.send('weather updated');
   } catch (err) {
     console.log(err);
   }