oneuptime/init-script/server.js

process.on('exit', () => {
    // eslint-disable-next-line no-console
    console.log('Shutting Shutdown');
});

process.on('unhandledRejection', err => {
    // eslint-disable-next-line no-console
    console.error('Unhandled rejection in process occurred');
    // eslint-disable-next-line no-console
    console.error(err);
});

process.on('uncaughtException', err => {
    // eslint-disable-next-line no-console
    console.error('Uncaught exception in process occurred');
    // eslint-disable-next-line no-console
    console.error(err);
});

const { NODE_ENV } = process.env;
if (!NODE_ENV || NODE_ENV === 'development') {
    // Load env vars from /backend/.env
    require('custom-env').env();
}

const fs = require('fs');
const util = require('./util/db');
const scripts = require('./scripts');
const express = require('express');
const app = express();
const { find, save, update, removeMany } = require('./util/db');
const bodyParser = require('body-parser');
const cors = require('cors');

async function interactWithDB(req, res) {
    if (req.params.dbFunction === 'find') {
        res.send(await find(req.body.collection, req.body.query));
    }
    if (req.params.dbFunction === 'save') {
        res.send(await save(req.body.collection, req.body.docs));
    }
    if (req.params.dbFunction === 'update') {
        res.send(
            await update(req.body.collection, req.body.query, req.body.value)
        );
    }
    if (req.params.dbFunction === 'removeMany') {
        res.send(await removeMany(req.body.collection, req.body.query));
    }
}

// IMPORTANT: only attach this server in development.
if (process.env['NODE_ENV'] === 'development') {
    app.use(cors());

    app.use(bodyParser.urlencoded({ limit: '10mb', extended: true }));
    app.use(bodyParser.json({ limit: '10mb' }));

    app.listen(1447, function() {
        // eslint-disable-next-line no-console
        console.log('Server running on: 1447');
    });

    app.get('/:dbFunction', async function(req, res) {
        return await interactWithDB(req, res);
    });

    app.post('/:dbFunction', async function(req, res) {
        return await interactWithDB(req, res);
    });
}

async function run() {
    const excludedScripts = ['index.js', 'start.js', 'end.js'];

    // eslint-disable-next-line no-console
    console.log('Connecting to MongoDB.');

    const connection = await util.connectToDb();
    global.db = connection.db();

    // eslint-disable-next-line no-console
    console.log('Connected to MongoDB.');

    let currentVersion = await util.getVersion();

    // eslint-disable-next-line no-console
    console.log('Current Version: ' + currentVersion);

    if (currentVersion) {
        currentVersion = currentVersion.split('.')[2];
    }

    // eslint-disable-next-line no-console
    console.log('START SCRIPT: Running script.');

    await scripts.start();

    // eslint-disable-next-line no-console
    console.log('START SCRIPT: Completed');

    const files = fs
        .readdirSync('./scripts')
        .filter(file => excludedScripts.indexOf(file) < 0) // Exclude index, start and end scripts
        .sort((a, b) =>
            parseInt(a.split('.')[2]) > parseInt(b.split('.')[2]) ? 1 : 0
        );

    // Switched to for loop, forEach does not await the callback
    for (let i = 0; i < files.length; i++) {
        const file = files[i];

        // eslint-disable-next-line no-console
        console.log(file + ': Running script.');

        await require(`./scripts/${file}`)();

        // eslint-disable-next-line no-console
        console.log(file + ': Completed. ');
    }

    // eslint-disable-next-line no-console
    console.log('END SCRIPT: Running script.');

    await scripts.end();

    // eslint-disable-next-line no-console
    console.log('END SCRIPT: Completed');
    // keep connection open in dev
    if (process.env['NODE_ENV'] !== 'development') {
        connection.close();
        // eslint-disable-next-line no-console
        console.log('Mongo connection closed.');
    } else {
        // eslint-disable-next-line no-console
        console.log('Mongo connection open in development mode.');
    }
}

module.exports = run;

run();