eea-searchserver/lib/Server.js

EEA Node.js Search Server module

function getSettings(settingsFile, callback) {
    var nconf = require('nconf');
    nconf.env({separator: '_',
               whitelist: ['elastic_host',
                           'elastic_path',
                           'elastic_port',
                           'elastic_index',
                           'elastic_type']})
         .file(settingsFile)
         .defaults({'elastic:path': '/',
                    'elastic:port': 9200});
    var elasticConf = nconf.get()['elastic'];
    if (elasticConf.host === undefined) {
        callback(new Error("Please set the 'elastic_host' env var or" +
                           " elastic:host in " + settingsFile));
        return null;
    }
    if (elasticConf.index === undefined) {
        callback(new Error("Please set the 'elastic_index' env var or" +
                           " elastic:index in " + settingsFile));
        return null;
    }
    if (elasticConf.type === undefined) {
        callback(new Error("Please set the 'elastic_type' env var or" +
                           " elastic:type in " + settingsFile));
        return null;
    }
    callback(null, nconf);
    return nconf;
}

function Server(app, settings, callback) {
    var _ = require('underscore');
    var getenv = require('getenv');
    var DEFAULT_PORT = 3000;
    var AUTO_INDEXING = getenv.bool('AUTO_INDEXING', true);

    /* Get config and fail if error */
    var nconf = getSettings(settings.settingsFile, function(err, res) {
        if (err) callback(err);
    });
    if (!nconf) {
        return null;
    }
    var isConstruct = settings.isConstruct;

    function exists_index(elastic){
        var request = require('sync-request');

        var indexed_url = 'http://' + elastic.host + ':' + elastic.port + elastic.path + elastic.index + '/_count';
        var count_current = 0;
        var res;
        try{
            res = request('GET', indexed_url)
            count_current = JSON.parse(res.getBody('utf8')).count;
            if (isNaN(count_current)){
                count_current = 0;
            }
        }catch(e){
            if(res.statusCode != 404){
                return exists_index(elastic);
            }else{
                return false
            }
        }
        return (count_current > 0);
    }
    /* Get server entrypoint commands */
    function runServer() {
        var http = require('http');
        var server = app.listen(nconf.get('http:port') || DEFAULT_PORT);
    }

    var commands = {
        'runserver': runServer,
    }
    if (app.get('managementCommands')) {
        commands = _.extend(commands, app.get('managementCommands'));
    }
    if (commands.help === undefined) {
        commands.help = function() {
            console.log("List of available commands: " + _.keys(commands));
        }
    }

    function run(cmd, app_dir, args, callback) {
        var cmdFunc = commands[cmd];
        if (cmdFunc === undefined) {
            callback(new Error("Command '" + cmd + "' unrecognized."));
            return;
        }
        cmdFunc(app_dir, args);
        callback(null, this);

        if (cmd == 'runserver' && AUTO_INDEXING === true){
            var waitForPort = require('wait-for-port');
            var elastic = nconf.get()['elastic'];
            if (!isConstruct){
              elastic.real_index = elastic.index + '_blue';
            }
            else {
              elastic.real_index = elastic.index;
            }
            waitForPort(elastic.host, elastic.port, { numRetries: 30 }, function(err) {
                if (err) throw new Error(err);
                var exists = exists_index(elastic);
                if (exists === false){
                    commands['create_index'](settings.indexing, args);
                }
            });
        }

    }

    var res = {
        run: run,
        nconf: nconf
    };

    callback(null, res);
    return res;
}

module.exports = Server