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 req2 = http.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的接口测试页面进行内网测试！");