covid.js

if(process.argv.length < 4){
    console.log('Required command: node covid.js <ip_binding> <port_binding>');
    return false;
}

var express = require('express');
var app = express();
var tracker = require('./tracker');
var db = require('./database');
const fsys = require('fs');

if(process.argv[2] != 'proxy') {
    var http = require('http').createServer(app);
    app.use('/static', express.static('static'));
}

app.set('view engine', 'ejs');
app.use((req, res, next) => {
    res.removeHeader('X-Powered-By');
    res.removeHeader('Server');
    next();
});

app.get('/', async (req, res) => {
    let data = fsys.readFileSync(tracker.COUNTRIES_FILE);
    res.render('website/index', {
        title: 'Estadísticas COVID-19',
        countries: JSON.parse(data)
    })
});

app.get('/country', async (req, res) => {
    let data = fsys.readFileSync(tracker.COUNTRIES_FILE);
    res.render('website/overview', {
        title: 'Estadísticas por países',
        countries: JSON.parse(data)
    });
});

app.get('/country/day', async (req, res) => {
    let data = fsys.readFileSync(tracker.COUNTRIES_FILE);
    res.render('website/daily', {
        title: 'Estadística diaria',
        countries: JSON.parse(data)
    });
});

app.get('/country/week', async (req, res) => {
    let data = fsys.readFileSync(tracker.COUNTRIES_FILE);
    res.render('website/weekly', {
        title: 'Estadística semanal',
        countries: JSON.parse(data)
    });
});

app.get('/country/month', async (req, res) => {
    let data = fsys.readFileSync(tracker.COUNTRIES_FILE);
    res.render('website/monthly', {
        title: 'Estadística mensual',
        countries: JSON.parse(data)
    });
});

app.get('/list/countries', async (req, res) => {
    res.append('Content-Type', 'application/json');
    res.send(fsys.readFileSync(tracker.COUNTRIES_FILE));
});

app.get('/realtime/country/:country', async (req, res) => {
    tracker.downloadData(req.params.country).then((data) => {
        res.append('Content-Type', 'application/json');
        res.send(JSON.stringify(data));
    }).catch((err) => {
        console.log(err);
        res.send('Real-time data is not available');
    });
});

app.get('/history/country/:period/:country', async (req, res) => {
    let minDate = new Date(Date.now());
    switch(req.params.period) {
        case 'day':
            minDate.setUTCDate(minDate.getUTCDate() - 2);
            break;
        case 'week':
            minDate.setUTCDate(minDate.getUTCDate() - 8);
            break;
        case 'month':
            minDate.setUTCMonth(minDate.getUTCMonth() - 1);
            break;
        default:
            res.send('Not a valid time');
            return;
    }
    try {
        let response = await db.queryData(req.params.country, minDate.toISOString());
        res.append('Content-Type', 'application/json');
        res.send(JSON.stringify(response));
    } catch(err) {
        console.log(err);
        res.send('Unable to retrieve the history');
    }
});

if(process.argv[2] != 'proxy') {
    http.listen(process.argv[3], process.argv[2], () => {
        console.log('External server started');
        tracker.updateDatabase();
        setInterval(() => {
            tracker.updateDatabase();
        }, 1.2E6);
    });
} else {
    app.listen(process.argv[3], () => {
        console.log('Internal server started (no static file is being served)');
        tracker.updateDatabase();
        setInterval(() => {
            tracker.updateDatabase();
        }, 1.2E6);
    });
}