2022-02-09 11:56:34 +00:00
|
|
|
const { MongoClient } = require("mongodb");
|
|
|
|
const express = require('express');
|
|
|
|
const minio = require('minio');
|
|
|
|
|
2022-02-15 08:55:59 +00:00
|
|
|
/*============== VARIABLE DECLARATION ==============*/
|
2022-02-09 11:56:34 +00:00
|
|
|
// Mongo set-up variables
|
2022-02-11 09:36:53 +00:00
|
|
|
const mongoCollection = process.env.MONGO_COLLECTION || 'eventlog';
|
|
|
|
const mongoUri = process.env.MONGO_URI || 'mongodb://127.0.0.1:27017/default?replicaSet=rs0';
|
2022-02-09 11:56:34 +00:00
|
|
|
|
|
|
|
// Minio set-up variables
|
2022-02-11 09:36:53 +00:00
|
|
|
const minioURI = new URL(process.env.MINIO_URI || 'http://kspace-mugshot:2mSI6HdbJ8@127.0.0.1:9000/kspace-mugshot');
|
|
|
|
const minioBucket = minioURI.pathname.substring(1);
|
2022-02-15 16:34:10 +00:00
|
|
|
const historyNumber = parseInt(process.env.HISTORY_AMOUNT) || 1000;
|
2022-02-09 11:56:34 +00:00
|
|
|
|
|
|
|
// Stream set-up variables
|
|
|
|
let changeStream;
|
|
|
|
const options = { fullDocument: "updateLookup" };
|
|
|
|
const pipeline = [];
|
|
|
|
const PORT = process.env.PORT || 3002;
|
|
|
|
|
|
|
|
// Create Mongo client
|
2022-02-11 09:36:53 +00:00
|
|
|
const mongoClient = new MongoClient(mongoUri);
|
2022-02-09 11:56:34 +00:00
|
|
|
|
|
|
|
|
|
|
|
/*============== CODE ==============*/
|
|
|
|
async function run() {
|
|
|
|
|
2022-02-11 09:36:53 +00:00
|
|
|
console.log('server.js has been launched');
|
2022-02-09 11:56:34 +00:00
|
|
|
const app = express();
|
|
|
|
|
|
|
|
await mongoClient.connect();
|
2022-02-11 09:36:53 +00:00
|
|
|
const collection = mongoClient.db().collection(mongoCollection);
|
2022-02-09 11:56:34 +00:00
|
|
|
|
|
|
|
changeStream = collection.watch(pipeline, options);
|
|
|
|
console.log("Started watching changes in database");
|
|
|
|
|
2022-02-15 16:34:10 +00:00
|
|
|
// Triggers on GET at /event route
|
2022-02-09 11:56:34 +00:00
|
|
|
app.get('/events', async function (request, response) {
|
|
|
|
let minioClient = new minio.Client({
|
2022-02-11 09:36:53 +00:00
|
|
|
endPoint: minioURI.hostname,
|
|
|
|
port: parseInt(minioURI.port) || (minioURI.protocol == 'https' ? 443 : 80),
|
|
|
|
useSSL: minioURI.protocol == 'https',
|
|
|
|
accessKey: minioURI.username,
|
|
|
|
secretKey: minioURI.password
|
2022-02-09 11:56:34 +00:00
|
|
|
});
|
|
|
|
|
2022-02-15 19:34:59 +00:00
|
|
|
function wrapEvent(doc) {
|
2022-02-15 20:14:27 +00:00
|
|
|
if (doc && doc.screenshot_count) {
|
2022-02-15 19:34:59 +00:00
|
|
|
doc.screenshots = [];
|
|
|
|
for (let i = 1; i <= doc.screenshot_count ; i++) {
|
2022-02-15 20:14:27 +00:00
|
|
|
minioClient.presignedUrl('GET', minioBucket, `${doc.camera}/${doc._id}/${i}.jpg`, 60 * 60, (err, presignedUrl) => {
|
2022-02-15 19:34:59 +00:00
|
|
|
if (err) { return console.log(err) };
|
|
|
|
doc.screenshots.push({ url: presignedUrl });
|
|
|
|
})
|
|
|
|
}
|
|
|
|
};
|
|
|
|
let blob = JSON.stringify(doc);
|
|
|
|
return `event: log-entry\ndata: ${blob}\n\n`
|
|
|
|
}
|
|
|
|
|
2022-02-15 16:34:10 +00:00
|
|
|
// Notify SSE to React
|
2022-02-11 09:36:53 +00:00
|
|
|
const header = { 'Content-Type': 'text/event-stream', 'Connection': 'keep-alive' };
|
2022-02-09 11:56:34 +00:00
|
|
|
response.writeHead(200, "OK", header);
|
|
|
|
|
2022-02-15 16:34:10 +00:00
|
|
|
const historyCursor = collection.find({}).sort({$natural : -1}).limit(historyNumber);
|
|
|
|
|
|
|
|
historyCursor.forEach((document) => {
|
2022-02-15 19:34:59 +00:00
|
|
|
response.write(wrapEvent(document));
|
2022-02-15 16:34:10 +00:00
|
|
|
})
|
|
|
|
|
2022-02-09 11:56:34 +00:00
|
|
|
changeStream.on("change", data => {
|
2022-02-15 19:34:59 +00:00
|
|
|
response.write(wrapEvent(data.fullDocument));
|
2022-02-09 11:56:34 +00:00
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
app.listen(PORT);
|
|
|
|
console.log(`Server listening at 127.0.0.1:${PORT}`);
|
|
|
|
}
|
|
|
|
|
|
|
|
run().catch(console.dir);
|