catproxy/_book/src/tools.js

a node proxy or host change tools

import log from './log';
import childProcess from 'child_process';
import Promise from 'promise';
import net from 'net';
import uuid from 'node-uuid';
export const hostReg = /:(\/\/([^\/]+))/;
export let getUrl = ({port, path: pathname, protocol, hostname, host})=> {
	if (protocol && (hostname || host)) {
		hostname = hostname || host;
		hostname = hostname.split(':')[0];
		protocol === "https" ? "http" : "https";
		if (+port === 80 && protocol === "http") {
			port = "";
		}
		if (+port === 443 && protocol === "https") {
			port = "";
		}
		if (port) {
			port  = ":" + port;
		}
		pathname = pathname || "";
		return `${protocol}://${hostname}${port}${pathname}`;
	}
};

export let writeErr = (err) => {
	err = err || "系统内部错误";
	if (err.stack && err.message) {
		err = err.message + "<br>" + err.stack;
	}
	return err;
};

let portReg = /EADDRINUSE\s*[^0-9]*([0-9]+)/i;

export let error = err => {
	if (err.code === 'EACCES' || err.message && err.message.indexOf("EACCES") > -1) {
		log.error("请用sudo管理员权限打开");
		process.exit(1);
	} else if (err.code === "EADDRINUSE" || err.message.indexOf("EADDRINUSE") > -1) {
		let port = err.port || (err.message.match(portReg) || ['', ''])[1];
		log.error(`端口${port}被占用，请检查端口占用情况`);
		process.exit(1);
	} else {
		process.exit(1);
		log.error("出现错误：" + (err && err.stack ? err.stack : err));
	}
};

export let openCmd = uri => {
	var cmd;
	if (process.platform === 'win32') {
		cmd = 'start';
	} else if (process.platform === 'linux') {
		cmd = 'xdg-open';
	} else if (process.platform === 'darwin') {
		cmd = 'open';
	}
	childProcess.exec([cmd, uri].join(' '));
};

export let sendErr = (res, err, uri) => {
	if (!res) {
		return;
	}
	err = err || "";

	if (res.finished) {
		return;
	}
	var message = "";
	var t = typeof err;
	if (t === 'string') {
		message = err;
	} else if(t === 'object') {
		message = (err.message || "") + (err.msg || "") + (err.stack || ""); 
	}
	res.headers = res.headers || {};
	if (!res.headers['content-type']|| !res.headers['Content-Type']) {
		res.headers['Content-Type'] = "text/html; charset=utf-8";
	}
	if (res.headers['content-length']) {
		delete res.headers['content-length'];
	}
	if (res.headers['content-encoding']) {
		delete res.headers['content-encoding'];
	}
	let statusCode = '500';
	if (message.indexOf('ETIMEDOUT') > -1) {
		statusCode = '408';
	} else if (message.indexOf('ENOTFOUND') > -1){
		statusCode = '504';
	}
	if (statusCode === '500') {
		log.error(`url:${uri || ``}`);
		log.error(`${message}`);
	}
	res.writeHead(statusCode, res.headers);
	res.end(message);
};

export let getPort = () => {
	return new Promise(function (resolve, reject) {
		var server = net.createServer();
		server.unref();
		server.on('error', reject);
		server.listen(0, function () {
			var port = server.address().port;
			server.close(function () {
				resolve(port);
			});
		});
	});
};

let getGuids = (start) => {
	start = +start || 1;
	return () => {
		return start++;
	};
};
// 监控要用的id
export let getMonitorId = getGuids(+new Date());

// weinre要用得id
export let weinreId = uuid.v4();