knomp/libs/website.js

var https = require('https');
var fs = require('fs');
var path = require('path');

var async = require('async');
var watch = require('node-watch');
//var redis = require('redis');

var dot = require('dot');
var express = require('express');
var bodyParser = require('body-parser');
var compress = require('compression');

//var Stratum = require('stratum-pool');
//var util = require('stratum-pool/lib/util.js');

var filterIterate = require('./filterIterate.js');

var api = require('./api.js');

module.exports = function(logger){

    dot.templateSettings.strip = false;

    var portalConfig = JSON.parse(process.env.portalConfig);
    var poolConfigs = JSON.parse(process.env.pools);

    var websiteConfig = portalConfig.website;
    var websiteTemplate = 'website/' + (typeof websiteConfig.template !== 'undefined' && websiteConfig.template ? websiteConfig.template : 'default');

    var portalApi = new api(logger, portalConfig, poolConfigs);
    var portalStats = portalApi.stats;

    var logSystem = 'Website';

    var pageFiles = {
        'index.html': 'index',
        'home.html': '',
        'getting_started.html': 'getting_started',
        'stats.html': 'stats',
        'tbs.html': 'tbs',
        'workers.html': 'workers',
        'api.html': 'api',
//        'admin.html': 'admin',
//        'mining_key.html': 'mining_key',
        'miner_stats.html': 'miner_stats',
        'payments.html': 'payments'
    };

    var pageTemplates = {};

    var pageProcessed = {};
    var indexesProcessed = {};

    var keyScriptTemplate = '';
    var keyScriptProcessed = '';

    var processTemplates = function(){

        for (var pageName in pageTemplates){
            if (pageName === 'index') continue;
            pageProcessed[pageName] = pageTemplates[pageName]({
                poolsConfigs: poolConfigs,
                stats: portalStats.stats,
                portalConfig: portalConfig
            });
            indexesProcessed[pageName] = pageTemplates.index({
                page: pageProcessed[pageName],
                selected: pageName,
                stats: portalStats.stats,
                poolConfigs: poolConfigs,
                portalConfig: portalConfig
            });
        }

        //logger.debug(logSystem, 'Stats', 'Website updated to latest stats');
    };

    var readPageFiles = function(files){
        async.each(files, function(fileName, callback){
            var filePath = websiteTemplate + (fileName === 'index.html' ? '/' : '/pages/') + fileName;
            fs.readFile(filePath, 'utf8', function(err, data){
                var pTemp = dot.template(data);
                pageTemplates[pageFiles[fileName]] = pTemp
                callback();
            });
        }, function(err){
            if (err){
                console.log('error reading files for creating dot templates: '+ JSON.stringify(err));
                return;
            }
            processTemplates();
        });
    };

    // if an html file was changed reload it
    /* requires node-watch 0.5.0 or newer */
    watch(['./'+websiteTemplate, './'+websiteTemplate+'/pages'], function(evt, filename){
        var basename;
        // support older versions of node-watch automatically
        if (!filename && evt)
            basename = path.basename(evt);
        else
            basename = path.basename(filename);

        if (basename in pageFiles){
            readPageFiles([basename]);
            logger.special(logSystem, 'Server', 'Reloaded file ' + basename);
        }
    });

    portalStats.getGlobalStats(function(){
        readPageFiles(Object.keys(pageFiles));
    });

    var buildUpdatedWebsite = function(){
        portalStats.getGlobalStats(function(){
            processTemplates();

            //Anonymize
            let anonPortalStats = JSON.parse(JSON.stringify(portalStats.stats));
            for (pool in anonPortalStats.pools) {
                filterIterate(anonPortalStats.pools[pool].confirmed.blocks, {split:{by:':', index:3}}, 'miner-');
                filterIterate(anonPortalStats.pools[pool].currentRoundShares, {key: true}, 'miner-');
                filterIterate(anonPortalStats.pools[pool].currentRoundTimes, {key: true}, 'miner-');
                filterIterate(anonPortalStats.pools[pool].miners, {key: true, prop: ['name']}, 'miner-', );
                filterIterate(anonPortalStats.pools[pool].pending.blocks, {split:{by:':', index:3}}, 'miner-');
                filterIterate(anonPortalStats.pools[pool].workers, {key: true, prop: ['name']}, 'worker-', );

                for (payment in anonPortalStats.pools[pool].payments) {
                    filterIterate(anonPortalStats.pools[pool].payments[payment].amounts, {key: true}, 'miner-', );
                    filterIterate(anonPortalStats.pools[pool].payments[payment].balances, {key: true}, 'miner-', );
                    filterIterate(anonPortalStats.pools[pool].payments[payment].work, {key: true}, 'miner-', );
                }
            }

            var statData = 'data: ' + JSON.stringify(anonPortalStats) + '\n\n';
            for (var uid in portalApi.liveStatConnections) {
                var res = portalApi.liveStatConnections[uid];
                res.write(statData);
            }
        });
    };

    setInterval(buildUpdatedWebsite, websiteConfig.stats.updateInterval * 1000);

    var getPage = function(pageId){
        if (pageId in pageProcessed){
            var requestedPage = pageProcessed[pageId];
            return requestedPage;
        }
    };

    var minerpage = function(req, res, next){
        var address = req.params.address || null;
        if (address != null) {
			address = address.split(".")[0];
            portalStats.getBalanceByAddress(address, function(){
                processTemplates();
		        res.header('Content-Type', 'text/html');
                res.end(indexesProcessed['miner_stats']);
            });
        }
        else
            next();
    };

    var payout = function(req, res, next){
        var address = req.params.address || null;
        if (address != null){
            portalStats.getPayout(address, function(data){
                res.write(data.toString());
                res.end();
            });
        }
        else
            next();
    };

    var shares = function(req, res, next){
        portalStats.getCoins(function(){
            processTemplates();
            res.end(indexesProcessed['user_shares']);
        });
    };

    var usershares = function(req, res, next){
        var coin = req.params.coin || null;
        if(coin != null){
            portalStats.getCoinTotals(coin, null, function(){
                processTemplates();
                res.end(indexesProcessed['user_shares']);
            });
        }
        else
            next();
    };

    var route = function(req, res, next){
        var pageId = req.params.page || '';
        if (pageId in indexesProcessed){
            res.header('Content-Type', 'text/html');
            res.end(indexesProcessed[pageId]);
        }
        else
            next();

    };

    var app = express();

    app.use('/robots.txt', function (req, res, next) {
        res.type('text/plain')
        res.send("User-agent: *\nDisallow: /workers/*");
    });

    app.use(bodyParser.json());

    app.get('/get_page', function(req, res, next){
        var requestedPage = getPage(req.query.id);
        if (requestedPage){
            res.end(requestedPage);
            return;
        }
        next();
    });

    //Don't think these were used at all
    // app.get('/key.html', function(req, res, next){
    //     res.end(keyScriptProcessed);
    // });
    //app.get('/stats/shares/:coin', usershares);
    //app.get('/stats/shares', shares);
	//app.get('/payout/:address', payout);

    app.use(compress());
    app.get('/workers/:address', minerpage);
    app.get('/:page', route);
    app.get('/', route);

    app.get('/api/:method', function(req, res, next){
        portalApi.handleApiRequest(req, res, next);
    });

    /* Don't see what this was really doing
    app.post('/api/admin/:method', function(req, res, next){
        if (portalConfig.website
            && portalConfig.website.adminCenter
            && portalConfig.website.adminCenter.enabled){
            if (portalConfig.website.adminCenter.password === req.body.password)
                portalApi.handleAdminApiRequest(req, res, next);
            else
                res.send(401, JSON.stringify({error: 'Incorrect Password'}));

        }
        else
            next();

    });
    */

    app.use(compress());
    app.use('/static', express.static(websiteTemplate + '/static'));

    app.use(function(err, req, res, next){
        console.error(err.stack);
        res.send(500, 'Something broke!');
    });

    try {
        if (portalConfig.website.tlsOptions && portalConfig.website.tlsOptions.enabled === true) {
            var TLSoptions = {
              key: fs.readFileSync(portalConfig.website.tlsOptions.key),
              cert: fs.readFileSync(portalConfig.website.tlsOptions.cert)
            };

            https.createServer(TLSoptions, app).listen(portalConfig.website.port, portalConfig.website.host, function() {
                logger.info(logSystem, 'Server', 'TLS Website started on ' + portalConfig.website.host + ':' + portalConfig.website.port);
            });
        } else {
          app.listen(portalConfig.website.port, portalConfig.website.host, function () {
            logger.info(logSystem, 'Server', 'Website started on ' + portalConfig.website.host + ':' + portalConfig.website.port);
          });
        }
    }
    catch(e){
        console.log(e)
        logger.error(logSystem, 'Server', 'Could not start website on ' + portalConfig.website.host + ':' + portalConfig.website.port
            +  ' - its either in use or you do not have permission');
    }
};