server.js

var _ = require('lodash');
var logger = require('./lib/utils/logger');
var chalk = require('chalk');
var http = require('http');

// Init WS SECRET
var WS_SECRET;

if( !_.isUndefined(process.env.WS_SECRET) && !_.isNull(process.env.WS_SECRET) )
{
    if( process.env.WS_SECRET.indexOf('|') > 0 )
    {
        WS_SECRET = process.env.WS_SECRET.split('|');
    }
    else
    {
        WS_SECRET = [process.env.WS_SECRET];
    }
}
else
{
    try {
        var tmp_secret_json = require('./ws_secret.json');
        WS_SECRET = _.values(tmp_secret_json);
    }
    catch (e)
    {
        console.error("WS_SECRET NOT SET!!!");
    }
}

var banned = require('./lib/utils/config').banned;

// Init http server
if( process.env.NODE_ENV !== 'production' )
{
    var app = require('./lib/express');
    server = http.createServer();
}
else
    server = http.createServer();

// Init socket vars
var Primus = require('primus');
var api;
var client;
var server;


// Init API Socket connection
api = new Primus(server, {
    transformer: 'websockets',
    pathname: '/api',
    parser: 'JSON'
});

api.plugin('emit', require('primus-emit'));
api.plugin('spark-latency', require('primus-spark-latency'));


// Init Client Socket connection
client = new Primus(server, {
    transformer: 'websockets',
    pathname: '/primus',
    parser: 'JSON'
});

client.plugin('emit', require('primus-emit'));


// Init external API
external = new Primus(server, {
    transformer: 'websockets',
    pathname: '/external',
    parser: 'JSON'
});

external.plugin('emit', require('primus-emit'));

// Init collections
var Collection = require('./lib/collection');
var Nodes = new Collection(external);

Nodes.setChartsCallback(function (err, charts)
{
    if(err !== null)
    {
        console.error('COL', 'CHR', 'Charts error:', err);
    }
    else
    {
        client.write({
            action: 'charts',
            data: charts
        });
        //console.log(charts['miners']);
    }
});


// Init API Socket events
api.on('connection', function (spark)
{
    console.info('API', 'CON', 'Open:', spark.address.ip);

    spark.on('hello', function (data)
    {
        console.info('API', 'CON', 'Hello', data['id']);

        if( _.isUndefined(data.secret) || WS_SECRET.indexOf(data.secret) === -1 || banned.indexOf(spark.address.ip) >= 0 )
        {
            spark.end(undefined, { reconnect: false });
            console.error('API', 'CON', 'Closed - wrong auth', data);

            return false;
        }

        if( !_.isUndefined(data.id) && !_.isUndefined(data.info) )
        {
            data.ip = spark.address.ip;
            data.spark = spark.id;
            data.latency = spark.latency || 0;

            Nodes.add( data, function (err, info)
            {
                if(err !== null)
                {
                    console.error('API', 'CON', 'Connection error:', err);
                    return false;
                }

                if(info !== null)
                {
                    spark.emit('ready');

                    console.success('API', 'CON', 'Connected', data.id);

                    client.write({
                        action: 'add',
                        data: info
                    });
                }
            });
        }
    });


    spark.on('update', function (data)
    {
        if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
        {
            Nodes.update(data.id, data.stats, function (err, stats)
            {
                if(err !== null)
                {
                    console.error('API', 'UPD', 'Update error:', err);
                }
                else
                {
                    if(stats !== null)
                    {
                        client.write({
                            action: 'update',
                            data: stats
                        });

                        console.info('API', 'UPD', 'Update from:', data.id, 'for:', stats);

                        Nodes.getCharts();
                    }
                }
            });
        }
        else
        {
            console.error('API', 'UPD', 'Update error:', data);
        }
    });


    spark.on('block', function (data)
    {
        if( !_.isUndefined(data.id) && !_.isUndefined(data.block) )
        {
            Nodes.addBlock(data.id, data.block, function (err, stats)
            {
                if(err !== null)
                {
                    console.error('API', 'BLK', 'Block error:', err);
                }
                else
                {
                    if(stats !== null)
                    {
                        client.write({
                            action: 'block',
                            data: stats
                        });

                        console.success('API', 'BLK', 'Block:', data.block['number'], 'from:', data.id);

                        Nodes.getCharts();
                    }
                    //console.log(stats);
                }
            });
        }
        else
        {
            console.error('API', 'BLK', 'Block error:', data);
        }
    });


    spark.on('pending', function (data)
    {
        if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
        {
            Nodes.updatePending(data.id, data.stats, function (err, stats) {
                if(err !== null)
                {
                    console.error('API', 'TXS', 'Pending error:', err);
                }

                if(stats !== null)
                {
                    client.write({
                        action: 'pending',
                        data: stats
                    });

                    console.success('API', 'TXS', 'Pending:', data.stats['pending'], 'from:', data.id);
                }
            });
        }
        else
        {
            console.error('API', 'TXS', 'Pending error:', data);
        }
    });


    spark.on('stats', function (data)
    {
        if( !_.isUndefined(data.id) && !_.isUndefined(data.stats) )
        {

            Nodes.updateStats(data.id, data.stats, function (err, stats)
            {
                if(err !== null)
                {
                    console.error('API', 'STA', 'Stats error:', err);
                }
                else
                {
                    if(stats !== null)
                    {
                        client.write({
                            action: 'stats',
                            data: stats
                        });

                        console.success('API', 'STA', 'Stats from:', data.id);
                    }
                    //console.log(stats);
                }
            });
        }
        else
        {
            console.error('API', 'STA', 'Stats error:', data);
        }
    });


    spark.on('history', function (data)
    {
        console.success('API', 'HIS', 'Got history from:', data.id);
        
        //console.log(data);

        var time = chalk.reset.cyan((new Date()).toJSON()) + " ";
        console.time(time, 'COL', 'CHR', 'Got charts in');

        Nodes.addHistory(data.id, data.history, function (err, history)
        {
            console.timeEnd(time, 'COL', 'CHR', 'Got charts in');

            if(err !== null)
            {
                console.error('COL', 'CHR', 'History error:', err);
            }
            else
            {
                client.write({
                    action: 'charts',
                    data: history
                });
            }
        });
    });


    spark.on('node-ping', function (data)
    {
        var start = (!_.isUndefined(data) && !_.isUndefined(data.clientTime) ? data.clientTime : null);

        spark.emit('node-pong', {
            clientTime: start,
            serverTime: _.now()
        });

        console.info('API', 'PIN', 'Ping from:', data['id']);
    });


    spark.on('latency', function (data)
    {
        if( !_.isUndefined(data.id) )
        {
            Nodes.updateLatency(data.id, data.latency, function (err, latency)
            {
                if(err !== null)
                {
                    console.error('API', 'PIN', 'Latency error:', err);
                }

                if(latency !== null)
                {
                    // client.write({
                    // 	action: 'latency',
                    // 	data: latency
                    // });

                    console.info('API', 'PIN', 'Latency:', latency, 'from:', data.id);
                }
            });

            if( Nodes.requiresUpdate(data.id) )
            {
                var range = Nodes.getHistory().getHistoryRequestRange();

                spark.emit('history', range);
                console.info('API', 'HIS', 'Asked:', data.id, 'for history:', range.min, '-', range.max);

                Nodes.askedForHistory(true);
            }
        }
    });


    spark.on('end', function (data)
    {
        Nodes.inactive(spark.id, function (err, stats)
        {
            if(err !== null)
            {
                console.error('API', 'CON', 'Connection end error:', err);
            }
            else
            {
                client.write({
                    action: 'inactive',
                    data: stats
                });

                console.warn('API', 'CON', 'Connection with:', spark.id, 'ended:', data);
            }
        });
    });
});



client.on('connection', function (clientSpark)
{
    clientSpark.on('ready', function (data)
    {
        clientSpark.emit('init', { nodes: Nodes.all() });

        Nodes.getCharts();
    });

    clientSpark.on('client-pong', function (data)
    {
        var serverTime = _.get(data, "serverTime", 0);
        var latency = Math.ceil( (_.now() - serverTime) / 2 );

        clientSpark.emit('client-latency', { latency: latency });
    });
});

var latencyTimeout = setInterval( function ()
{
    // client.write({
    //     action: 'client-ping',
    //     data: {
    //         serverTime: _.now()
    //     }
    // });
}, 5000);


// Cleanup old inactive nodes
var nodeCleanupTimeout = setInterval( function ()
{
    client.write({
        action: 'init',
        data: Nodes.all()
    });

    Nodes.getCharts();

}, 1000*60*60);


//console.log('SERVER = ',server);
server.listen(process.env.PORT || 3000);

module.exports = server;

/*
let WebSocketClient = require('websocket').w3cwebsocket;

const wsclient = new WebSocketClient('ws://127.0.0.1:3000/primus');

wsclient.onopen = () => {
    console.log('WebSocket Client Connected');
};

wsclient.onerror = function() {
    console.log('Connection Error');
};

wsclient.onmessage = function(data) {
    //console.log("Received: " + data.data.length);
};

wsclient.onclose = function() {
    console.log('Client Closed');
};
*/