feat: sockets for live reloading

jakowenko · Sep 20, 2021 · 50fef76 · 50fef76
1 parent 6d7ad32
commit 50fef76
Show file tree

Hide file tree

Showing 11 changed files with 406 additions and 135 deletions.
diff --git a/api/package-lock.json b/api/package-lock.json
diff --git a/api/package.json b/api/package.json
@@ -36,6 +36,7 @@
     "multer": "^1.4.3",
     "node-schedule": "^2.0.0",
     "sharp": "^0.28.3",
+    "socket.io": "^4.2.0",
     "traverse": "^0.6.6",
     "uuid": "^8.3.2",
     "winston": "^3.3.3"

diff --git a/api/server.js b/api/server.js
@@ -1,5 +1,6 @@
 require('./src/util/logger.util').init();
 const http = require('http');
+const socket = require('./src/util/socket.util');
 const { SERVER } = require('./src/constants');
 const { version } = require('./package.json');
 const mqtt = require('./src/util/mqtt.util');
@@ -14,9 +15,10 @@ module.exports.start = async () => {
   console.log(`Double Take v${version}`);
   console.log(config());
   await database.init();
-  http.Server(app).listen(SERVER.PORT);
+  const server = http.Server(app).listen(SERVER.PORT);
   mqtt.connect();
   storage.purge();
+  socket.connect(server);
 };
 
 try {

diff --git a/api/src/controllers/recognize.controller.js b/api/src/controllers/recognize.controller.js
@@ -8,6 +8,7 @@ const recognize = require('../util/recognize.util');
 const frigate = require('../util/frigate.util');
 const { jwt } = require('../util/auth.util');
 const mqtt = require('../util/mqtt.util');
+const { emit } = require('../util/socket.util');
 const { BAD_REQUEST } = require('../constants/http-status');
 const DETECTORS = require('../constants/config').detectors();
 const { AUTH, FRIGATE, TOKEN } = require('../constants');
@@ -167,12 +168,9 @@ module.exports.start = async (req, res) => {
     res.send(output);
 
     mqtt.recognize(output);
-
     notify.publish(output, camera, results);
-
-    if (output.matches.length) {
-      IDS.push(id);
-    }
+    if (output.matches.length) IDS.push(id);
+    if (results.length) emit('recognize', true);
   } catch (error) {
     PROCESSING = false;
     res.send(error);

diff --git a/api/src/util/socket.util.js b/api/src/util/socket.util.js
@@ -0,0 +1,13 @@
+const socket = require('socket.io');
+
+let io = false;
+
+module.exports.connect = (server) => {
+  io = socket(server, {
+    cors: {
+      origin: true,
+    },
+  });
+};
+
+module.exports.emit = (event, message) => (io ? io.emit(event, message) : false);