net.js

var http = require('http');
var https = require('https');
var url = require("url");
var pass = require('stream').PassThrough;

var getHeader = function(req) {
  var ret = {};
  for (var i in req.headers) {
    if (!/^(host|connection|Access-|origin|referer|user-agent|user-doclever|path-doclever|url-doclever|method-doclever|headers-doclever|X-Requested-With)/i.test(i)) {
      ret[i] = req.headers[i];
    }
  }
  ret["accept"] = "*/*";
  var headers = req.headers["headers-doclever"];
  if (headers) {
    headers = JSON.parse(headers);
    for (var key in headers) {
      ret[key] = headers[key];
    }
  }
  return ret
};

var filterResHeader = function(headers, res) {
  var ret = {};
  res.setHeader("Cache-Control", "no-cache,no-store");
  res.setHeader("Pragrma", "no-cache");
  res.setHeader("Expires", 0);
  var resHeaders = res.getHeader("Access-Control-Expose-Headers") ? res.getHeader("Access-Control-Expose-Headers").toLowerCase() : "";
  for (var i in headers) {
    if (!/Access-/i.test(i)) {
      if (/set-cookie/i.test(i)) {
        for (let index = 0; index < headers[i].length; index++) {
          headers[i][index] = headers[i][index].split(" ")[0];
        }
        ret[i] = headers[i];
      } else {
        ret[i] = headers[i];
      }
    }
    if (!resHeaders) {
      res.setHeader("Access-Control-Expose-Headers", i);
    } else if (resHeaders.indexOf(i.toLowerCase() + ",") == -1 && resHeaders.indexOf("," + i.toLowerCase()) == -1) {
      res.setHeader("Access-Control-Expose-Headers", res.getHeader("Access-Control-Expose-Headers") + "," + i);
    }
  }
  return ret;
};

var getPath = function(req) {
  var url = req.url;
  if (url.substr(0, 7).toLowerCase() === 'http://') {
    var i = url.indexOf('/', 7);
    if (i !== -1) {
      url = url.substr(i);
    }
  }
  return url;
};

function getHost(req) {
  var url = req.headers["url-doclever"];
  url = url.replace(/^(http:\/\/|https:\/\/)/i, "");
  var arr = url.split(":");
  return arr[0];
}

function getPort(req) {
  var url = req.headers["url-doclever"];
  var defaultPort;
  if (req.headers["url-doclever"].toLowerCase().startsWith("https://")) {
    defaultPort = 443;
  } else {
    defaultPort = 80;
  }
  url = url.replace(/^(http:\/\/|https:\/\/)/i, "");
  var arr = url.split(":");
  return arr.length > 1 ? arr[1] : defaultPort;
}

var onProxy = function(req, res) {
  handleHeader(req, res);
  if (req.method == "OPTIONS") {
    return;
  }
  if (!req.headers["url-doclever"]) {
    mock(req, res);
  } else {
    proxy(req, res);
  }
};

function mock(req, res) {
  var obj = url.parse(mockUrl);
  var path = req.headers.path || url.parse(req.url).path;
  var opt = {
    host: obj.hostname,
    path: (obj.pathname == "/" ? "" : obj.pathname) + path,
    method: req.method,
    headers: getHeader(req),
    port: obj.port ? obj.port : 80,
  }
  console.log("初次请求：method:" + opt.method + "host:" + opt.host + "port:" + opt.port + "path:" + opt.path)
  if (opt.headers["content-length"]) {
    delete opt.headers["content-length"]
  }
  var tmp = url.parse(realUrl).protocol == "http:" ? http : https;
  var req2 = tmp.request(opt, function(res2) {
    if (!realUrl || res2.headers["finish-doclever"] == "0") {
      console.log("接口开发中，返回mock数据");
      res.writeHead(res2.statusCode, filterResHeader(res2.headers, res));
      res2.pipe(res);
      res2.on('end', function() {

      });
    } else {
      if (res2.headers["finish-doclever"] == "1") {
        console.log("接口已完成，调用真实接口");
      } else {
        console.log("接口或者项目未找到，转调真实接口");
      }
      var headers = getHeader(req);
      var objUrl = url.parse(realUrl);
      var request1, opt1;
      if (objUrl.protocol == "http:") {
        opt1 = {
          host: objUrl.hostname,
          path: (objUrl.pathname == "/" ? "" : objUrl.pathname) + path,
          method: req.method,
          port: objUrl.port ? objUrl.port : 80,
          headers: headers
        }
        request1 = http.request;
      } else {
        opt1 = {
          host: objUrl.hostname,
          path: (objUrl.pathname == "/" ? "" : objUrl.pathname) + path,
          method: req.method,
          port: objUrl.port ? objUrl.port : 443,
          headers: headers,
          rejectUnauthorized: false,
          requestCert: true,
        }
        request1 = https.request;
      }
      console.log("调用真实接口：method:" + opt1.method + "host:" + opt1.host + "port:" + opt1.port + "path:" + opt1.path)
      var req3 = request1(opt1, function(res3) {
        console.log("真实接口调用完成。status:" + res3.statusCode)
        res.writeHead(res3.statusCode, filterResHeader(res3.headers, res));
        res3.pipe(res);
        res3.on('end', function() {

        });
      })
      if (/POST|PUT/i.test(req.method)) {
        stream.pipe(req3);
      } else {
        req3.end();
      }
      req3.on('error', function(err) {
        res.end(err.stack);
      });
    }
  });
  var stream;
  if (/POST|PUT|PATCH/i.test(req.method)) {
    stream = new pass();
    req.pipe(stream);
    req.pipe(req2);
  } else {
    req2.end();
  }
  req2.on('error', function(err) {
    res.end(err.stack);
  });
}

function handleHeader(req, res) {
  if (!req.headers.origin) {
    return;
  }
  res.setHeader("Access-Control-Allow-Origin", req.headers.origin);
  res.setHeader("Access-Control-Allow-Headers", "X-Requested-With");
  res.setHeader("Access-Control-Allow-Methods", "PUT,POST,GET,DELETE");
  res.setHeader("Access-Control-Allow-Credentials", "true");
  if (req.headers["access-control-request-headers"]) {
    res.setHeader("Access-Control-Allow-Headers", req.headers["access-control-request-headers"])
  }
  res.setHeader("Access-Control-Expose-Headers", "connection,content-length,date,x-powered-by,content-encoding,server,etag,accept-ranges,allow,content-language,set-cookie,doclever-request");
  if (req.method == "OPTIONS") {
    res.writeHead(200, { 'Content-Type': 'text/plain' });
    res.end();
    return;
  }
}

function redirect(res, bHttps, opt, location) {
  var urlRedirect = location;
  if (urlRedirect.startsWith("/")) {
    urlRedirect = (bHttps ? "https://" : "http://") + opt.host + ":" + opt.port + urlRedirect;
  }
  var objUrl = url.parse(urlRedirect);
  var request1, opt1;
  if (objUrl.protocol == "http:") {
    opt1 = {
      host: objUrl.hostname,
      path: objUrl.path,
      method: "GET",
      port: objUrl.port ? objUrl.port : 80,
    }
    request1 = http.request;
    bHttps = false;
  } else {
    opt1 = {
      host: objUrl.hostname,
      path: objUrl.path,
      method: "GET",
      port: objUrl.port ? objUrl.port : 443,
      rejectUnauthorized: false,
      requestCert: true,
    }
    request1 = https.request;
    bHttps = true;
  }
  var req3 = request1(opt1, function(res3) {
    if (res3.statusCode == 302) {
      redirect(res, bHttps, opt, res3.headers.location)
    } else {
      var resHeader = filterResHeader(res3.headers, res)
      resHeader["doclever-request"] = JSON.stringify(handleSelfCookie(req3));
      res.writeHead(res3.statusCode, resHeader);
      res3.pipe(res);
      res3.on('end', function() {

      });
    }
  })
  req3.end();
  req3.on('error', function(err) {
    res.end(err.stack);
  });
}

function handleSelfCookie(req) {
  var arr = req._headers;
  arr["url"] = req.method + " " + req.path;
  var cookie = arr["cookie"];
  if (!cookie) {
    return arr;
  }
  var arrCookie = cookie.split(";");
  var keys = ["id", "name", "photo", "qq", "sex", "company", "phone", "loginCount", "age", "email"];
  arrCookie = arrCookie.filter(function(obj) {
    obj = obj.trim();
    for (let key of keys) {
      if (obj.startsWith(key + "=")) {
        return false;
      }
    }
    return true;
  })
  arr["cookie"] = arrCookie.join(";");
  return arr;
}

function proxy(req, res) {
  var bHttps = false;
  if (req.headers["url-doclever"].toLowerCase().startsWith("https://")) {
    bHttps = true;
  }
  var opt, request;
  if (bHttps) {
    opt = {
      host: getHost(req),
      path: req.headers["path-doclever"],
      method: req.headers["method-doclever"],
      headers: getHeader(req),
      port: getPort(req),
      rejectUnauthorized: false,
      requestCert: true,
    };
    request = https.request;
  } else {
    opt = {
      host: getHost(req),
      path: req.headers["path-doclever"],
      method: req.headers["method-doclever"],
      headers: getHeader(req),
      port: getPort(req)
    };
    request = http.request;
  }
  var req2 = request(opt, function(res2) {
    if (res2.statusCode == 302) {
      redirect(res, bHttps, opt, res2.headers.location)
    } else {
      var resHeader = filterResHeader(res2.headers, res)
      resHeader["doclever-request"] = JSON.stringify(handleSelfCookie(req2));
      res.writeHead(res2.statusCode, resHeader);
      res2.pipe(res);
      res2.on('end', function() {

      });
    }
  });
  if (/POST|PUT|PATCH/i.test(req.method)) {
    req.pipe(req2);
  } else {
    req2.end();
  }
  req2.on('error', function(err) {
    res.end(err.stack);
  });
}

var arguments = process.argv.splice(2);
var mockUrl = arguments[0];
var realUrl = arguments[1];
var port = arguments[2] ? arguments[2] : 36742;
var server = http.createServer(onProxy);
server.listen(port);
console.log(arguments.length > 0 ? ("内网测试，Mock数据正监听端口：" + port) : "正在运行中，请用DOClever的接口测试页面进行内网测试！");