freaker2k7 / dockerserver

lib/network.js

var IP = null;
var args = null;
var machine = null;
var request = null;

var get_protocol = function() {
	return 'http' + (args.https && 's' || '') + '://';
};

var get_req = function(req, host) {
	return new Promise(function(resolve) {
		return request({
			url: host + req.originalUrl,
			headers: Object.assign(req.headers, {'docker-server': 'force', 'content-type': 'application/json'}),
			form: req.body,
			method: req.method
		}, function (err, res, body) {
			// console.log('request:', host, err, typeof body, body);
			if (err) {
				console.error('err:', host, err);
				return resolve({'body': [err], 'host': host});
			}
			
			try {
				return resolve({'body': typeof body === 'string' && JSON.parse(body) || body, 'host': host});
			} catch(e) {
				if (body.startsWith('error')) {
					body = {'error': body};
				} else {
					console.error('Error in get_req on', host, ':', e, typeof body, body);
				}
				
				return resolve({'body': [body], 'host': host});
			}
		});
	});
};

var handle_cluster_request = function(req, res, next) {
	return function(ip) {
		if (ip === IP) {
			next();
		} else {
			get_req(req, ip).then(function(ip_obj) {
				res.send(ip_obj.body || null);
			});
		}
	};
};

var handle_cluster = function(req, res, next) {
	if (req.method === 'PUT') {
		machine.next(IP).then(handle_cluster_request(req, res, next));
	} else if(~['GET', 'DELETE', 'POST'].indexOf(req.method)) {
		machine.all(IP).then(function(ips) {
			var promises = [];
			for (var i in ips) {

A for in loop automatically includes the property of any prototype object, consider checking the key using hasOwnProperty.

				promises.push(get_req(req, ips[i].host));
			}
			
			return Promise.all(promises).then(function(responses) {
				var response = [];
				for (var i = 0; i < responses.length; ++i) {
					// Add the host to each result for more clearance for the client.
					for (var j in responses[i].body) {

A for in loop automatically includes the property of any prototype object, consider checking the key using hasOwnProperty.

						responses[i].body[j].host = responses[i].host;
					}
					response = response.concat(responses[i].body);
				}
				res.send(response);
			});
		});
	} else {
		res.sendStatus(400);
	}
};


module.exports = function(cli_args) {
	args = cli_args;
	
	return {
		'get_protocol': get_protocol,
		'check': function(token) {
			return function (req, res, next) {
				if (req.get('Authorization') === token) {
					next();
				} else {
					res.sendStatus(401);
				}
			};
		},
		'protocol': function(app) {
			if (args.https) {
				const fs = require('fs');
				
				if (fs.existsSync('/certs/privkey.pem') && fs.existsSync('/certs/cert.pem')) {
					try {
						return require('https').createServer({
							key: fs.readFileSync('/certs/privkey.pem', 'utf8'),
							cert: fs.readFileSync('/certs/cert.pem', 'utf8'),
							ca: fs.existsSync('/certs/chain.pem') && fs.readFileSync('/certs/chain.pem', 'utf8') || undefined
						}, app);
					} catch (e) {
						console.error('Could not start secure server:', e);
						return require('http').createServer(app);
					}
				}
			}
			
			return require('http').createServer(app);
		},
		'balance': function(port) {
			if (args.cluster) {
				IP = get_protocol() + require('ip').address() + ':' + port;
				machine = require('./machine.js')(args);
				request = require('request');
				
				machine.cpu(IP);
				setInterval(function() { machine.cpu(IP); }, 30000);
			}
			
			return function (req, res, next) {
				// console.log('000 ??? body', JSON.stringify(req.body));
				if (args.cluster && req.get('docker-server') !== 'force') {
					handle_cluster(req, res, next);
				} else {
					next();
				}
			};
		}
	};
};