@moquyun/proxy/index.js

Multi-user & multi-env web debugging proxy based on whistle

// 避免第三方模块没处理好异常导致程序crash
require('whistle/lib/util/patch');
const fse = require('fs-extra');
const fs = require('fs');
const path = require('path');
const os = require('os');
const qs = require('querystring');
const { getWhistlePath } = require('whistle/lib/config');
const pkg = require('./package.json');
const initConfig = require('./lib/config');

const PURE_URL_RE = /^((?:https?:)?\/\/[\w.-]+[^?#]*)/;

// 设置存储路径
process.env.WHISTLE_PATH = process.env.NOHOST_PATH || getWhistlePath();
fse.ensureDirSync(process.env.WHISTLE_PATH); // eslint-disable-line


const getPureUrl = (url) => {
  if (!url || !PURE_URL_RE.test(url)) {
    return;
  }
  return RegExp.$1.replace(/\/+$/, '');
};

const getErrorStack = (err) => {
  if (!err) {
    return '';
  }
  let stack;
  try {
    stack = err.stack;
  } catch (e) {}
  stack = stack || err.message || err;
  const result = [
    `From: nohost@${pkg.version}`,
    `Node: ${process.version}`,
    `Date: ${new Date().toLocaleString()}`,
    stack];
  return result.join('\r\n');
};

const handleUncaughtException = (err) => {
  if (!err || err.code !== 'ERR_IPC_CHANNEL_CLOSED') {
    const stack = getErrorStack(err);
    fs.writeFileSync(path.join(process.cwd(), 'nohost.log'), `\r\n${stack}\r\n`, { flag: 'a' }); // eslint-disable-line
    console.error(stack); // eslint-disable-line
  }
  process.exit(1);
};

process.on('unhandledRejection', handleUncaughtException);
process.on('uncaughtException', handleUncaughtException);

const parseCluster = ({ cluster }) => {
  if (!cluster) {
    return;
  }
  if (/^\d+$/.test(cluster)) {
    return { workers: parseInt(cluster, 10) };
  }
  if (typeof cluster !== 'string') {
    return { workers: os.cpus().length };
  }
  const { workers, plugins } = qs.parse(cluster);
  const result = { workers: workers > 0 ? parseInt(workers, 10) : os.cpus().length };
  if (plugins && typeof plugins === 'string') {
    result.plugins = plugins.split(',').map((plugin) => path.resolve(plugin));
  }
  return result;
};

module.exports = (options, cb) => {
  if (typeof options === 'function') {
    cb = options;
    options = {};
  } else if (!options) {
    options = {};
  }
  if (options.__maxHttpHeaderSize > 0) {
    process.env.PFORK_MAX_HTTP_HEADER_SIZE = options.__maxHttpHeaderSize;
  }
  if (options.debugMode) {
    const mode = typeof options.mode === 'string' ? options.mode.trim().split(/\s*[|,&]\s*/) : [];
    if (mode.includes('prod') || mode.includes('production')) {
      options.debugMode = false;
    } else {
      process.env.PFORK_MODE = 'bind';
    }
  }
  const clusterOpts = parseCluster(options);
  if (clusterOpts) {
    options.cluster = clusterOpts.workers;
    options.pluginPaths = clusterOpts.plugins;
  }
  options.redirect = getPureUrl(options.redirect);
  initConfig(options);
  require('./lib')(options, cb); // eslint-disable-line
};