total5/release.js

Total.js framework v5

require('./index');

// Variables
var WATCHER = process.connected === true;
var options;
var app;
var firstinit = true;
var pidname;
var unexpectedexit = false;
var restarting;

module.exports = function(opt) {

	options = opt || {};

	// options.ip = '127.0.0.1';
	// options.port = parseInt(process.argv[2]);
	// options.unixsocket = require('node:path').join(require('node:os').tmpdir(), 'app_name');
	// options.config = { name: 'Total.js' };
	// options.https = { key: Fs.readFileSync('keys/agent2-key.pem'), cert: Fs.readFileSync('keys/agent2-cert.pem')};
	// options.sleep = 3000;
	// options.inspector = 9229;
	// options.debugger = 40894;
	// options.watch = ['adminer'];
	// options.livereload = true;
	// options.watcher = false;
	// options.cluster = 'auto' || or NUMBER
	// options.limit = 10;
	// options.timeout = 5000;
	// options.edit = 'wss://.....com/?id=myprojectname'

	if (!WATCHER)
		WATCHER = process.argv.indexOf('--watcher') === -1 && !options.watcher;

};

function makestamp() {
	return '--- # --- [ ' + new Date().format('yyyy-MM-dd HH:mm:ss') + ' ] ';
}

function restart() {
	restarting && clearTimeout(restarting);
	restarting = setTimeout(run, 100);
}

function run() {

	restarting = null;

	var arr = F.TUtils.clone(process.argv).slice(2);
	var port = arr.pop();
	var directory = process.cwd();

	if (!firstinit)
		arr.push('--restart');

	port && arr.push(port);
	var filename = F.TUtils.getName(process.argv[1] || 'index.js');

	pidname = F.Path.join(directory, filename.replace(/\.js$/, '.pid'));
	app = F.Fork(F.Path.join(process.cwd(), filename), arr);
	app.on('message', function(msg) {
		switch (msg) {
			case 'total:eaddrinuse':
				process.exit(1);
				break;
			case 'total:ready':
				if (firstinit) {
					app.send('total:debug');
					firstinit = false;
				}
				break;
			case 'total:restart':
				console.log(makestamp().replace('#', 'RES'));
				unexpectedexit = true;
				setTimeout(restart, 1000);
				process.kill(app.pid);
				app = null;
				break;
		}
	});

	F.Fs.writeFileSync(pidname, process.pid + '');

	app.on('exit', function() {

		// checks unexpected exit
		if (unexpectedexit === true)
			app = null;
		else
			restart();

		unexpectedexit = false;
	});

	F.emit('watcher', app);
}

function init() {

	if (options.cluster) {
		options.release = true;
		options.count = options.cluster;
		F.TCluster.http(options);
		return;
	}

	if (WATCHER) {

		delete options.watcher;

		if (options.servicemode) {
			var types = options.servicemode === true || options.servicemode === 1 ? '' : options.servicemode.split(',').trim();
			global.DEBUG = false;
			F.load(types);
			if (!process.connected)
				F.console();
		} else {
			global.DEBUG = false;
			F.http(options);
		}
		return;
	}

	var end = function() {

		if (process.isending)
			return;

		process.isending = true;
		F.Fs.unlink(pidname, NOOP);

		if (app) {
			unexpectedexit = true;
			process.kill(app.pid);
			app = null;
		}

		process.exit(0);
	};

	process.env.NODE_TLS_REJECT_UNAUTHORIZED = '1';
	console.log('-------- RELEASE PID: ' + process.pid + ' (v' + F.version_header + ')');
	process.on('uncaughtException', e => e.toString().indexOf('ESRCH') == -1 && console.log(e));
	process.title = 'total: release';
	process.on('SIGTERM', end);
	process.on('SIGINT', end);
	process.on('exit', end);

	setInterval(function() {
		F.Fs.stat(pidname, function(err) {
			if (err) {
				unexpectedexit = true;
				process.kill(app.pid);
				setTimeout(() => process.exit(0), 100);
			}
		});
	}, 4000);

	if (!process.connected && options.edit) {
		require('./edit').init(options.edit.replace(/^http/, 'ws'));
		setTimeout(run, 1000);
	} else
		setImmediate(run);
}

setTimeout(init, 50);