trainjs/bin/server.js

/*

	This file is a part of node-on-train project.

	Copyright (C) Thanh D. Dang <thanhdd.it@gmail.com>

	node-on-train is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	node-on-train is distributed in the hope that it will be useful, but
	WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
	General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.

	*/


	var fs = require('fs');
	var trainjs = require('../lib/train.js');
	var http = require('http');
	var url = require('url');
	var exec = require('child_process').exec;
	var Fiber = require('fibers');
	var path = require('path');


	var global_v_init = trainjs.initServer();

	var pre_run_path = ROOT_APP + '/config/pre_run.js';
	var pre_run = fs.existsSync( pre_run_path );

	var app_config = require( ROOT_APP + '/config/application.js' );
	var ApplicationController = require( ROOT_APP + '/app/controllers/application_controller.js' );


	var app = require( ROOT_APP + '/app.js' );

	var beforeAction = function(req, res, next, class_object, action, before_action_i) {
		if (class_object['before_action']) {
			var before_action = class_object['before_action'][before_action_i];
			if (!before_action.only || before_action.only && before_action.only.indexOf(action) > -1) {
			if (class_object[before_action.action](req, res, next)) { //if res.end()
				return;
			}

			if (before_action_i < class_object['before_action'].length - 1) {
				before_action_i = before_action_i + 1;
				beforeAction(req, res, next, class_object, action, before_action_i);
			} else {
				class_object[action](req, res, next);
			}
		} else {
			class_object[action](req, res, next);
		}
	} else {
		class_object[action](req, res, next);
	}
};

function runServer() {
	app.use(function (req, res, next) {
		Fiber(function() {
			var url_parts = url.parse(req.url, true);
			req.query = url_parts.query;

			if ( app_config.cors ) {
				res.setHeader("Access-Control-Allow-Origin", "*");
				res.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE, PUT");
				if (req.headers['access-control-request-headers'])
					res.setHeader("Access-Control-Allow-Headers", req.headers['access-control-request-headers']);
			}

			if ( req.method === 'OPTIONS' ) {
				res.end();
			} else {
				res.setHeader("Content-Type", "application/json");
				next();
			}
		}).run();
	});

	var application = new ApplicationController();
	app.use(application.before);
	trainjs.newServer(app, global_v_init);

	app.use(function (req, res, next) {
		var assets = ['image/', 'text/html', 'application/javascript', 'text/css', 'application/font'];
		var content_type = 'text/plain';

		if (req.headers.accept) {
			var headers_accept = req.headers.accept.split(',');
			content_type = headers_accept[0];

			for (var i in headers_accept) {
				if ( headers_accept[i].indexOf('text/plain') > -1 ) {
					content_type = 'text/html';
				}
			}

			if (req.headers['content-type'])
				content_type = req.headers['content-type'];

			if (content_type == '*/*' && /\.js$/.test(req.url)) {
				content_type = 'application/javascript';
			}

			if (/\.png$/.test(req.url)) {
				content_type = 'image/png';
			}

			if (content_type == '*/*' && /\.jpg$/.test(req.url)) {
				content_type = 'image/jpeg';
			}
		}

		var url_path = req.url.split('?')[0];

		var is_asset = false;
		for (var k in assets) {
			if ( content_type.indexOf(assets[k]) > -1 ) {
				is_asset = true;
				break;
			}
		}

		if ( url_path.indexOf('/node_modules/') > -1 && url_path.indexOf('../') < 0 ) {
			url_path = '..' + url_path;
		}

		var file_path = path.join(ROOT_APP, '/public', url_path);
		if (is_asset && fs.existsSync(file_path)) {
			if (url_path == '/') {
				fs.readFile(ROOT_APP + '/public/index.html', function(err, page) {
					res.writeHead(200, {'Content-Type': 'text/html'});
					res.write(page);
					res.end();
				});
			} else {
				fs.readFile(file_path, function(err, page) {
					res.writeHead(200, {'Content-Type': content_type});
					res.write(page);
					res.end();
				});
			}
		} else {
			file_path = path.join(ROOT_APP, '', url_path);
			if (fs.existsSync(file_path)) {
				try {
					var stats = fs.lstatSync(file_path);
					if (stats.isDirectory()) {
						var index_file = file_path + '/index.html';
						if (fs.existsSync(index_file)) {
							if (url_path[url_path.length - 1] != '/') {
								res.writeHead(301, {'Location': url_path + '/'});
								res.end();
							} else {
								fs.readFile(index_file, function(err, page) {
									res.writeHead(200, {'Content-Type': 'text/html'});
									res.write(page);
									res.end();
								});
							}
						} else {
							res.end();
						}
					} else {
						if (is_asset) {
							fs.readFile(file_path, function(err, page) {
								res.writeHead(200, {'Content-Type': content_type});
								res.write(page);
								res.end();
							});
						} else {
							res.end();
						}
					}
				}
				catch (e) {
					console.log(e);
				}
			} else {
				var routes = global_v_init.routes;
				var controllers = global_v_init.controllers;
				var is_route = false;

				for (var k in ROUTE_PATTERNS) {
					var pattern = ROUTE_PATTERNS[k].pattern;
					var method = ROUTE_PATTERNS[k].method;
					var resource = ROUTE_PATTERNS[k].resource;
					var action = ROUTE_PATTERNS[k].action;

					if (pattern.matches(req.url) && method.toLowerCase() == req.method.toLowerCase()) {
						var params = pattern.match(req.url);
						req.params = params.namedParams;
						try {
							var class_object = new controllers[resource]();
							beforeAction(req, res, next, class_object, action, 0);
							is_route = true;
							break;
						} catch(e) {
							console.log( 'Error: '.bold.red + e );
							res.end();
						}
					}
				}

				if (!is_route)
					next();
			}
		}
	});

	var port = '1337';
	if (process.env.PORT) {
		port = process.env.PORT;
	} else {
		var port_param_index = process.argv.indexOf("-p");
		var isNormalInteger = function (str) {
			var n = ~~Number(str);
			return String(n) === str && n >= 0;
		};
		if ( port_param_index > -1 && process.argv[ port_param_index + 1 ] && isNormalInteger( process.argv[ port_param_index + 1 ] ) ) {
			port = process.argv[ port_param_index + 1 ];
		}
	}

	http.createServer(app).listen(port);
	console.log('=> Server running at http://0.0.0.0:' + port + '\n=> Ctrl-C to shutdown server');
}

if ( pre_run ) {
	var cmd = require( pre_run_path );
	console.log('=> Run config/pre_run.js');
	exec(cmd, function (error, stdout, stderr) {
		if (error) console.log(error);
		else {
			PRE_RUN_STDOUT = stdout;
			runServer();
		}
	});
} else {
	runServer();
}