shelloid/src/main.js

Simple and secure Node.js web application server.

#! /usr/bin/env node

/*
 Copyright (c) Shelloid Systems LLP. All rights reserved.
 The use and distribution terms for this software are covered by the
 GNU Lesser General Public License 3.0 (https://www.gnu.org/licenses/lgpl.html)
 which can be found in the file LICENSE at the root of this distribution.
 By using this software in any fashion, you are agreeing to be bound by
 the terms of this license.
 You must not remove this notice, or any other, from this software.
 */
global.lib_require = require("./lib/lib_require.js");

var http = require('http'),
	path = require("path"),
	cluster = require("cluster"),
	os = require("os");
	winston = require('winston');

/*require/init order is important*/
var	init = require("./init.js");

init.installGlobals();

var log = lib_require("log");

var loader = lib_require("loader"),
	utils = lib_require("utils"),
	route = lib_require("route"),
	app   = lib_require("app"),
	auth = lib_require("auth"),
	app_pkg = lib_require("app_pkg"),
	validate_globals = lib_require("validate_globals"),
	dbconfig = lib_require("dbconfig");
	
if(process.argv.length <= 2){
    console.log("Please provide app directory as the argument");
	process.exit(0);
}

var args = process.argv.slice(2);
var serverCtx = init.serverCtx.apply(null, args);

shelloid.serverCtx = serverCtx;

init.loadAppConfig(serverCtx.appCtx);

var numCPUs = os.cpus().length;

if(cluster.isMaster){
	winston.add(winston.transports.File, { filename: serverCtx.appCtx.config.log.file});
	winston.transports.Console.level = serverCtx.appCtx.config.log.level;
	winston.transports.File.level = serverCtx.appCtx.config.log.level;	
}

if (cluster.isMaster && serverCtx.appCtx.config.enableCluster) {
	if(serverCtx.appCtx.env == "sim"){
		sh.error("Cannot enable cluster in simulator mode. Please set config.enableCluster to false");
		process.exit(0);
	}
	console.log("Enabling Cluster: Starting workers");
	cluster.setupMaster({ silent: false });
	// Fork workers.
	for (var i = 0; i < numCPUs; i++) {
		cluster.fork();
	}
	cluster.on('exit', function(worker, code, signal) {
		console.log('Worker: ' + worker.process.pid + ' died. Exit code: ' + code);
	});
	    
	for(var k in cluster.workers){
		var worker = cluster.workers[k];
		worker.on("message", processWorkerMsg);
	}
}else{
	app_pkg.init(serverCtx.appCtx, app_pkg_initDone);
}

function processWorkerMsg(msg){
	if(msg.isLog){
		winston.log(msg.logLevel, msg.logMsg);
	}else{
		console.log("Don't know how to process the worker message", msg);
	}
}

function app_pkg_initDone(err){
	if(err){
		console.log("Server initialization error: " + err + "Exiting.");
		process.exit(0);
	}else{
		sh.info("Application package manager initialization done.");	
		serverCtx.appCtx.app = app.newInstance(serverCtx.appCtx);
		dbInit();
	}
}

function dbInit(){
	dbconfig.init(serverCtx, appInit);
}

function appInit(){
	init.appInit(loadAuthMods);
}

function loadAuthMods(){
	sh.info("Database initialization done");
	loader.loadAuthMods(serverCtx, authModsLoaded);
}

function authModsLoaded(){		
	if(serverCtx.appCtx.authMods.length == 0){
		sh.info("No authentication modules found.");
		authModsAdded();
	}else{
		sh.info("Authentication modules loaded.");
		auth.addAll(serverCtx.appCtx, authModsAdded);
	}
}

function authModsAdded(){
	loader.loadInterfaces(serverCtx, interfacesLoaded);
}

function interfacesLoaded(){
	if(serverCtx.appCtx.env == "sim"){
		var sim = lib_require("sim");
		sim.init();
		loader.loadRoutes(serverCtx, sim.enterSimMode);
	}else{
		loader.loadRoutes(serverCtx, routesLoaded);
	}
}

function routesLoaded(){

	if(serverCtx.appCtx.routes.length == 0){
		console.log("No routes configured! Exiting.");
		process.exit();
	}

	route.addAll(serverCtx.appCtx);
	
	if(serverCtx.appCtx.hasErrors){
		console.log("Application context has errors. Exiting.");
		process.exit(0);
	}

	var server = http.createServer(serverCtx.appCtx.app);
	server.listen(serverCtx.appCtx.config.port, function(){
	  shelloid.info('Shelloid server version: ' + serverCtx.packageJson.version + ' listening on ' + 
		serverCtx.appCtx.config.port);
	});
		
}