mirror of
https://github.com/YMFE/yapi.git
synced 2025-01-30 13:20:24 +08:00
258 lines
7.5 KiB
JavaScript
Executable File
258 lines
7.5 KiB
JavaScript
Executable File
const yapi = require('../yapi.js');
|
||
const projectModel = require('../models/project.js');
|
||
const interfaceModel = require('../models/interface.js');
|
||
const mockExtra = require('../../common/mock-extra.js');
|
||
const _ = require('underscore');
|
||
const Mock = require('mockjs');
|
||
|
||
/**
|
||
*
|
||
* @param {*} apiPath /user/tom
|
||
* @param {*} apiRule /user/:username
|
||
*/
|
||
function matchApi(apiPath, apiRule) {
|
||
|
||
let apiRules = apiRule.split("/");
|
||
let apiPaths = apiPath.split("/");
|
||
let pathRules = {}
|
||
if (apiPaths.length !== apiRules.length) {
|
||
return false;
|
||
}
|
||
for (let i = 0; i < apiRules.length; i++) {
|
||
if (apiRules[i]) {
|
||
apiRules[i] = apiRules[i].trim();
|
||
} else {
|
||
continue;
|
||
}
|
||
if (apiRules[i].length > 2 && apiRules[i][0] === '{' && apiRules[i][apiRules[i].length - 1] === '}') {
|
||
|
||
pathRules[apiRules[i].substr(1, apiRules[i].length - 2)] = apiPaths[i];
|
||
} else if (apiRules[i].indexOf(":") === 0) {
|
||
pathRules[apiRules[i].substr(1)] = apiPaths[i]
|
||
} else if (apiRules[i].length > 2 && apiRules[i].indexOf('{') > -1 && apiRules[i].indexOf('}') > -1) {
|
||
let params = [];
|
||
apiRules[i] = apiRules[i].replace(/\{(.+?)\}/g, function (src, match) {
|
||
|
||
params.push(match);
|
||
return '(.+)';
|
||
})
|
||
apiRules[i] = new RegExp(apiRules[i]);
|
||
if (!apiRules[i].test(apiPaths[i])) {
|
||
return false;
|
||
}
|
||
|
||
let matchs = apiPaths[i].match(apiRules[i]);
|
||
|
||
params.forEach((item, index) => {
|
||
pathRules[item] = matchs[index + 1];
|
||
})
|
||
} else {
|
||
if (apiRules[i] !== apiPaths[i]) {
|
||
return false;
|
||
}
|
||
}
|
||
}
|
||
return pathRules;
|
||
}
|
||
|
||
function parseCookie(str) {
|
||
if (!str || typeof str !== 'string') {
|
||
return str;
|
||
}
|
||
if (str.split(';')[0]) {
|
||
let c = str.split(';')[0].split('=');
|
||
return { name: c[0], value: c[1] || '' }
|
||
}
|
||
return null;
|
||
}
|
||
|
||
function handleCorsRequest(ctx) {
|
||
let header = ctx.request.header;
|
||
ctx.set('Access-Control-Allow-Origin', header.origin);
|
||
ctx.set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, HEADER, PATCH, OPTIONS");
|
||
ctx.set('Access-Control-Allow-Headers', header['access-control-request-headers']);
|
||
ctx.set('Access-Control-Allow-Credentials', true);
|
||
ctx.set('Access-Control-Max-Age', 1728000);
|
||
ctx.body = 'ok';
|
||
|
||
}
|
||
|
||
module.exports = async (ctx, next) => {
|
||
// no used variable 'hostname' & 'config'
|
||
// let hostname = ctx.hostname;
|
||
// let config = yapi.WEBCONFIG;
|
||
let path = ctx.path;
|
||
|
||
|
||
if (path.indexOf('/mock/') !== 0) {
|
||
if (next) await next();
|
||
return true;
|
||
}
|
||
|
||
let paths = path.split("/");
|
||
let projectId = paths[2];
|
||
paths.splice(0, 3);
|
||
path = "/" + paths.join("/");
|
||
if (!projectId) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, 'projectId不能为空');
|
||
}
|
||
|
||
let projectInst = yapi.getInst(projectModel), project;
|
||
try {
|
||
project = await projectInst.get(projectId);
|
||
} catch (e) {
|
||
return ctx.body = yapi.commons.resReturn(null, 403, e.message);
|
||
}
|
||
|
||
if (!project) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '不存在的项目');
|
||
}
|
||
|
||
let interfaceData, newpath;
|
||
let interfaceInst = yapi.getInst(interfaceModel);
|
||
|
||
try {
|
||
newpath = path.substr(project.basepath.length);
|
||
interfaceData = await interfaceInst.getByPath(project._id, newpath, ctx.method);
|
||
|
||
|
||
//处理query_path情况
|
||
if (!interfaceData || interfaceData.length === 0) {
|
||
interfaceData = await interfaceInst.getByQueryPath(project._id, newpath, ctx.method);
|
||
|
||
let i, l, j, len, curQuery, match = false;
|
||
for (i = 0, l = interfaceData.length; i < l; i++) {
|
||
match = false;
|
||
let currentInterfaceData = interfaceData[i];
|
||
curQuery = currentInterfaceData.query_path;
|
||
if (!curQuery || typeof curQuery !== 'object' || !curQuery.path) {
|
||
continue;
|
||
}
|
||
for (j = 0, len = curQuery.params.length; j < len; j++) {
|
||
if (ctx.query[curQuery.params[j].name] !== curQuery.params[j].value) {
|
||
continue;
|
||
}
|
||
if (j === len - 1) {
|
||
match = true;
|
||
}
|
||
}
|
||
if (match) {
|
||
interfaceData = [currentInterfaceData];
|
||
break;
|
||
}
|
||
if (i === l - 1) {
|
||
interfaceData = [];
|
||
}
|
||
}
|
||
}
|
||
|
||
//处理动态路由
|
||
if (!interfaceData || interfaceData.length === 0) {
|
||
let newData = await interfaceInst.getVar(project._id, ctx.method);
|
||
|
||
let findInterface = _.find(newData, (item) => {
|
||
let m = matchApi(newpath, item.path)
|
||
if (m !== false) {
|
||
ctx.request.query = Object.assign(m, ctx.request.query);
|
||
return true;
|
||
}
|
||
return false;
|
||
});
|
||
|
||
if (!findInterface) {
|
||
//非正常跨域预检请求回应
|
||
if (ctx.method === 'OPTIONS' && ctx.request.header['access-control-request-method']) {
|
||
return handleCorsRequest(ctx);
|
||
}
|
||
return ctx.body = yapi.commons.resReturn(null, 404, `不存在的api, 当前请求path为 ${newpath}， 请求方法为 ${ctx.method} ，请确认是否定义此请求。`);
|
||
}
|
||
interfaceData = [
|
||
await interfaceInst.get(findInterface._id)
|
||
]
|
||
|
||
}
|
||
|
||
if (interfaceData.length > 1) {
|
||
return ctx.body = yapi.commons.resReturn(null, 405, '存在多个api，请检查数据库');
|
||
} else {
|
||
interfaceData = interfaceData[0];
|
||
}
|
||
|
||
ctx.set("Access-Control-Allow-Origin", "*")
|
||
let res;
|
||
|
||
res = interfaceData.res_body;
|
||
try {
|
||
if (interfaceData.res_body_type === 'json') {
|
||
res = mockExtra(
|
||
yapi.commons.json_parse(interfaceData.res_body),
|
||
{
|
||
query: ctx.request.query,
|
||
body: ctx.request.body,
|
||
params: Object.assign({}, ctx.request.query, ctx.request.body)
|
||
}
|
||
);
|
||
try {
|
||
res = Mock.mock(res);
|
||
} catch (e) {
|
||
yapi.commons.log(e, 'error')
|
||
}
|
||
}
|
||
|
||
let context = {
|
||
projectData: project,
|
||
interfaceData: interfaceData,
|
||
ctx: ctx,
|
||
mockJson: res,
|
||
resHeader: {},
|
||
httpCode: 200,
|
||
delay: 0
|
||
}
|
||
await yapi.emitHook('mock_after', context);
|
||
let handleMock = new Promise(resolve => {
|
||
setTimeout(() => {
|
||
resolve(true)
|
||
}, context.delay)
|
||
})
|
||
await handleMock;
|
||
if (context.resHeader && typeof context.resHeader === 'object') {
|
||
for (let i in context.resHeader) {
|
||
let cookie;
|
||
if (i === 'Set-Cookie') {
|
||
if (context.resHeader[i] && typeof context.resHeader[i] === 'string') {
|
||
cookie = parseCookie(context.resHeader[i]);
|
||
if (cookie && typeof cookie === 'object') {
|
||
ctx.cookies.set(cookie.name, cookie.value, {
|
||
maxAge: 864000000
|
||
});
|
||
}
|
||
} else if (context.resHeader[i] && Array.isArray(context.resHeader[i])) {
|
||
context.resHeader[i].forEach(item => {
|
||
cookie = parseCookie(item);
|
||
if (cookie && typeof cookie === 'object') {
|
||
ctx.cookies.set(cookie.name, cookie.value, {
|
||
maxAge: 864000000
|
||
});
|
||
}
|
||
})
|
||
}
|
||
} else ctx.set(i, context.resHeader[i]);
|
||
}
|
||
}
|
||
|
||
ctx.status = context.httpCode;
|
||
return ctx.body = context.mockJson;
|
||
} catch (e) {
|
||
yapi.commons.log(e, 'error')
|
||
return ctx.body = {
|
||
errcode: 400,
|
||
errmsg: '解析出错，请检查。Error: ' + e.message,
|
||
data: null
|
||
}
|
||
}
|
||
} catch (e) {
|
||
console.error(e)
|
||
return ctx.body = yapi.commons.resReturn(null, 409, e.message);
|
||
}
|
||
};
|