const { MongoClient } = require("mongodb");
const express = require('express');
const minio = require('minio');

/*============== VARIABLE DECLARATION ==============*/
// Mongo set-up variables
const mongoCollection = process.env.MONGO_COLLECTION || 'eventlog';
const mongoUri = process.env.MONGO_URI || 'mongodb://127.0.0.1:27017/default?replicaSet=rs0';

// Minio set-up variables
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);
console.info("Using bucket:", minioBucket);
const historyNumber = parseInt(process.env.HISTORY_AMOUNT) || 10;

// Stream set-up variables
let changeStream;
const options = { fullDocument: "updateLookup" };
const pipeline = [];
const PORT = process.env.PORT || 3002;

// Create Mongo client
const mongoClient = new MongoClient(mongoUri);


/*============== CODE ==============*/
async function run() {

  console.log('server.js has been launched');
  const app = express();

  await mongoClient.connect();
  const collection = mongoClient.db().collection(mongoCollection);

  changeStream = collection.watch(pipeline, options);
  console.log("Started watching changes in database");

  // Triggers on GET at /event route
  app.get('/events', async function (request, response) {
    let minioClient = new minio.Client({
      endPoint: minioURI.hostname,
      port: parseInt(minioURI.port) || (minioURI.protocol == 'https:' ? 443 : 80),
      useSSL: minioURI.protocol == 'https:',
      accessKey: minioURI.username,
      secretKey: minioURI.password
    });

    async function wrapEvent(doc) {
      let arr = [];
      let blob;
      if (doc && doc.screenshot_count) {
        for (let i = 1; i <= doc.screenshot_count ; i++) {
          arr.push({
            url: await minioClient.presignedUrl('GET', minioBucket,
              `${doc.camera}/${doc._id}/${i}.jpg`, 60 * 60)
          });
        }
      } else if (doc && doc.screenshots) {
        for (let j = 0; j < doc.screenshots.length; j++) {
          let path = doc.screenshots[j];
          arr.push({
            url: await minioClient.presignedUrl('GET', minioBucket,
              `thumb/${path}`, 60 * 60),
            orig: await minioClient.presignedUrl('GET', minioBucket,
              `${path}`, 60 * 60)
          });
        }
      };

      if (arr.length > 0) {
        blob = JSON.stringify({...doc, screenshots: [...arr]});
      } else {
        blob = JSON.stringify({...doc});
      }
      return `event: log-entry\ndata: ${blob}\n\n`
    }

    // Notify SSE to React
    const header = { 'Content-Type': 'text/event-stream', 'Connection': 'keep-alive' };
    response.writeHead(200, "OK", header);

    const historyCursor = collection.find().sort({$natural:-1}).limit(historyNumber);

    historyCursor.forEach(async (document) => {
      response.write(await wrapEvent(document));
    })

    changeStream.on("change", async(data) => {
      response.write(await wrapEvent(data.fullDocument));
    });
  });

  app.listen(PORT);
  console.log(`Server listening at 127.0.0.1:${PORT}`);
}

run().catch(console.dir);