index.js

/*
 * @Author: Eduardo Policarpo
 * @contact: +55 43996611437
 * @Date: 2021-05-10 18:09:49
 * @LastEditTime: 2021-06-07 03:18:01
 */
'use strict';
const cors = require('cors');
const express = require('express');
const app = express();
const path = require('path')
const server = require('http').Server(app);
const serveIndex = require('serve-index');
const motor = require('./engines');
const config = require('./config');
const { yo } = require('yoo-hoo');
const router = motor.engines[process.env.ENGINE].router
require('events').EventEmitter.prototype._maxListeners = 999;
const { startAllSessions } = require('./startup');

const io = require('socket.io')(server, {
    cors: {
        origins: ["*"],
        methods: ["GET", "POST"],
        transports: ['websocket', 'polling'],
        credentials: true
    },
    allowEIO3: true
});

app.use(cors());
app.use(express.json({ limit: '50mb' }));
app.set('view engine', 'ejs');
app.set('views', path.join(__dirname, 'views'));
app.set('json spaces', 2);
app.use(express.static('public'));
express.static(path.join(__dirname, '/public'));
app.use('/files', express.static('files-received'), serveIndex('files-received', { icons: true }));

app.use((req, res, next) => {
    req.io = io;
    next();
});

app.use(router);

io.on('connection', sock => {
    console.log(`ID: ${sock.id} socket in`)

    sock.on('event', data => {
        console.log(data)
    });

    sock.on('disconnect', () => {
        console.log(`ID: ${sock.id} socket out`)
    });
});

app.get('/start', function (req, res) {
    res.render('index', { port: config.port, host: config.host })
});

if (config.https == 1) {
    https.createServer(
        {
            key: fs.readFileSync(config.ssl_key_path),
            cert: fs.readFileSync(config.ssl_cert_path)
        },
        server).listen(config.port, async (error) => {
            if (error) {
                console.log(error)
            }
            else {
                console.log('\n\nWelcome to')
                yo('MyZAP', {
                    color: 'rainbow',
                    spacing: 1,
                });
                console.log(`Http server running on ${config.host}:${config.port}\n\n`);
                if (config.start_all_sessions === 'true') {
                    let result = await startAllSessions()
                    if (result != undefined) {
                        console.log(result)
                    }
                }
            }
        });
} else {
    server.listen(config.port, async (error) => {
        if (error) {
            console.log(error)
        }
        else {
            console.log('\n\nWelcome to')
            yo('MyZAP', {
                color: 'rainbow',
                spacing: 1,
            });
            console.log(`Http server running on ${config.host}:${config.port}\n\n`);
            if (config.start_all_sessions === 'true') {
                let result = await startAllSessions()
                if (result != undefined) {
                    console.log(result)
                }
            }
        }
    });
}

process.stdin.resume();

async function exitHandler(options, exitCode) {
    if (options.cleanup) {
        console.log('cleanup');
        await Sessions.getSessions().forEach(async session => {
            await Sessions.closeSession(session);
        });
    }
    if (exitCode || exitCode === 0) {
        console.log(exitCode);
    }

    if (options.exit) {
        process.exit();
    }
}

process.on('exit', exitHandler.bind(null, { cleanup: true }));
process.on('SIGINT', exitHandler.bind(null, { exit: true }));
process.on('SIGUSR1', exitHandler.bind(null, { exit: true }));
process.on('SIGUSR2', exitHandler.bind(null, { exit: true }));
process.on('uncaughtException', exitHandler.bind(null, { exit: true }));