mirror of
https://github.com/YMFE/yapi.git
synced 2024-12-21 05:19:42 +08:00
534 lines
19 KiB
JavaScript
534 lines
19 KiB
JavaScript
import interfaceModel from '../models/interface.js';
|
||
import interfaceCatModel from '../models/interfaceCat.js';
|
||
import baseController from './base.js';
|
||
import yapi from '../yapi.js';
|
||
import userModel from '../models/user.js';
|
||
import projectModel from '../models/project.js';
|
||
|
||
class interfaceController extends baseController {
|
||
constructor(ctx) {
|
||
super(ctx);
|
||
this.Model = yapi.getInst(interfaceModel);
|
||
this.catModel = yapi.getInst(interfaceCatModel);
|
||
this.projectModel = yapi.getInst(projectModel);
|
||
}
|
||
|
||
/**
|
||
* 添加项目分组
|
||
* @interface /interface/add
|
||
* @method POST
|
||
* @category interface
|
||
* @foldnumber 10
|
||
* @param {Number} project_id 项目id，不能为空
|
||
* @param {String} title 接口标题，不能为空
|
||
* @param {String} path 接口请求路径，不能为空
|
||
* @param {String} method 请求方式
|
||
* @param {Array} [req_headers] 请求的header信息
|
||
* @param {String} [req_headers[].name] 请求的header信息名
|
||
* @param {String} [req_headers[].value] 请求的header信息值
|
||
* @param {Boolean} [req_headers[].required] 是否是必须，默认为否
|
||
* @param {String} [req_headers[].desc] header描述
|
||
* @param {String} [req_body_type] 请求参数方式，有["form", "json", "text", "xml"]四种
|
||
* @param {Array} [req_params] name, desc两个参数
|
||
* @param {Mixed} [req_body_form] 请求参数,如果请求方式是form，参数是Array数组，其他格式请求参数是字符串
|
||
* @param {String} [req_body_form[].name] 请求参数名
|
||
* @param {String} [req_body_form[].value] 请求参数值，可填写生成规则（mock）。如@email，随机生成一条email
|
||
* @param {String} [req_body_form[].type] 请求参数类型，有["text", "file"]两种
|
||
* @param {String} [req_body_other] 非form类型的请求参数可保存到此字段
|
||
* @param {String} [res_body_type] 相应信息的数据格式，有["json", "text", "xml"]三种
|
||
* @param {String} [res_body] 响应信息，可填写任意字符串，如果res_body_type是json,则会调用mock功能
|
||
* @param {String} [desc] 接口描述
|
||
* @returns {Object}
|
||
* @example ./api/interface/add.json
|
||
*/
|
||
async add(ctx) {
|
||
let params = ctx.request.body;
|
||
|
||
params = yapi.commons.handleParams(params, {
|
||
project_id: 'number',
|
||
title: 'string',
|
||
path: 'string',
|
||
method: 'string',
|
||
desc: 'string',
|
||
catid: 'number'
|
||
});
|
||
params.method = params.method || 'GET';
|
||
params.method = params.method.toUpperCase();
|
||
params.res_body_type = params.res_body_type ? params.res_body_type.toLowerCase() : 'json';
|
||
|
||
if (!params.project_id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '项目id不能为空');
|
||
}
|
||
|
||
if (!params.path) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口请求路径不能为空');
|
||
}
|
||
|
||
if (!yapi.commons.verifyPath(params.path)) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口path第一位必须是/，最后一位不能为/');
|
||
}
|
||
|
||
let checkRepeat = await this.Model.checkRepeat(params.project_id, params.path, params.method);
|
||
|
||
if (checkRepeat > 0) {
|
||
return ctx.body = yapi.commons.resReturn(null, 401, '已存在的接口:' + params.path + '[' + params.method + ']');
|
||
}
|
||
|
||
try {
|
||
let data = {
|
||
project_id: params.project_id,
|
||
catid: params.catid,
|
||
title: params.title,
|
||
path: params.path,
|
||
desc: params.desc,
|
||
method: params.method,
|
||
req_headers: params.req_headers,
|
||
req_body_type: params.req_body_type,
|
||
res_body: params.res_body,
|
||
res_body_type: params.res_body_type,
|
||
uid: this.getUid(),
|
||
add_time: yapi.commons.time(),
|
||
up_time: yapi.commons.time()
|
||
};
|
||
|
||
if (params.req_query) {
|
||
data.req_query = params.req_query;
|
||
}
|
||
|
||
if (params.req_body_form) {
|
||
data.req_body_form = params.req_body_form;
|
||
}
|
||
if (params.req_params && Array.isArray(params.req_params) && params.req_params.length > 0) {
|
||
data.type = 'var'
|
||
data.req_params = params.req_params;
|
||
} else {
|
||
data.type = 'static'
|
||
}
|
||
if (params.req_body_other) {
|
||
data.req_body_other = params.req_body_other;
|
||
}
|
||
|
||
let result = await this.Model.save(data);
|
||
let username = this.getUsername();
|
||
// let project = await this.projectModel.get(params.project_id);
|
||
let cate = await this.catModel.get(params.catid);
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 为分类 "${cate.name}" 添加了接口 "${data.title}"`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: params.project_id
|
||
});
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (e) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, e.message);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 添加项目分组
|
||
* @interface /interface/get
|
||
* @method GET
|
||
* @category interface
|
||
* @foldnumber 10
|
||
* @param {Number} id 接口id，不能为空
|
||
* @returns {Object}
|
||
* @example ./api/interface/get.json
|
||
*/
|
||
async get(ctx) {
|
||
let params = ctx.request.query;
|
||
|
||
if (!params.id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口id不能为空');
|
||
}
|
||
|
||
try {
|
||
let result = await this.Model.get(params.id);
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (e) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, e.message);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 接口列表
|
||
* @interface /interface/list
|
||
* @method GET
|
||
* @category interface
|
||
* @foldnumber 10
|
||
* @param {Number} project_id 项目id，不能为空
|
||
* @returns {Object}
|
||
* @example ./api/interface/list.json
|
||
*/
|
||
async list(ctx) {
|
||
let project_id = ctx.request.query.project_id;
|
||
|
||
if (!project_id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '项目id不能为空');
|
||
}
|
||
|
||
try {
|
||
let result = await this.Model.list(project_id);
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (err) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, err.message);
|
||
}
|
||
}
|
||
|
||
async listByCat(ctx) {
|
||
let catid = ctx.request.query.catid;
|
||
if (!catid) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, 'catid不能为空');
|
||
}
|
||
try {
|
||
let result = await this.Model.listByCatid(catid)
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (err) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, err.message);
|
||
}
|
||
|
||
}
|
||
|
||
async listByMenu(ctx) {
|
||
let project_id = ctx.request.query.project_id;
|
||
if (!project_id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '项目id不能为空');
|
||
}
|
||
try {
|
||
let result = await this.catModel.list(project_id), newResult = [];
|
||
for(let i=0, item, list;i< result.length; i++){
|
||
item = result[i].toObject()
|
||
list = await this.Model.listByCatid(item._id, '_id title method path')
|
||
for(let j=0; j< list.length; j++){
|
||
list[j] = list[j].toObject()
|
||
}
|
||
item.list = list;
|
||
newResult[i] = item
|
||
}
|
||
ctx.body = yapi.commons.resReturn(newResult);
|
||
} catch (err) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, err.message);
|
||
}
|
||
|
||
}
|
||
|
||
/**
|
||
* 编辑接口
|
||
* @interface /interface/up
|
||
* @method POST
|
||
* @category interface
|
||
* @foldnumber 10
|
||
* @param {Number} id 接口id，不能为空
|
||
* @param {String} [path] 接口请求路径
|
||
* @param {String} [method] 请求方式
|
||
* @param {Array} [req_headers] 请求的header信息
|
||
* @param {String} [req_headers[].name] 请求的header信息名
|
||
* @param {String} [req_headers[].value] 请求的header信息值
|
||
* @param {Boolean} [req_headers[].required] 是否是必须，默认为否
|
||
* @param {String} [req_headers[].desc] header描述
|
||
* @param {String} [req_body_type] 请求参数方式，有["form", "json", "text", "xml"]四种
|
||
* @param {Mixed} [req_body_form] 请求参数,如果请求方式是form，参数是Array数组，其他格式请求参数是字符串
|
||
* @param {String} [req_body_form[].name] 请求参数名
|
||
* @param {String} [req_body_form[].value] 请求参数值，可填写生成规则（mock）。如@email，随机生成一条email
|
||
* @param {String} [req_body_form[].type] 请求参数类型，有["text", "file"]两种
|
||
* @param {String} [req_body_other] 非form类型的请求参数可保存到此字段
|
||
* @param {String} [res_body_type] 相应信息的数据格式，有["json", "text", "xml"]三种
|
||
* @param {String} [res_body] 响应信息，可填写任意字符串，如果res_body_type是json,则会调用mock功能
|
||
* @param {String} [desc] 接口描述
|
||
* @returns {Object}
|
||
* @example ./api/interface/up.json
|
||
*/
|
||
|
||
async up(ctx) {
|
||
let params = ctx.request.body;
|
||
|
||
params = yapi.commons.handleParams(params, {
|
||
title: 'string',
|
||
path: 'string',
|
||
method: 'string',
|
||
desc: 'string',
|
||
catid: 'number'
|
||
});
|
||
params.method = params.method || 'GET';
|
||
params.method = params.method.toUpperCase();
|
||
|
||
let id = ctx.request.body.id;
|
||
|
||
if (!id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口id不能为空');
|
||
}
|
||
let interfaceData = await this.Model.get(id);
|
||
|
||
if (params.path && !yapi.commons.verifyPath(params.path)) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口path第一位必须是/，最后一位不能为/');
|
||
}
|
||
|
||
if (params.path && params.path !== interfaceData.path && params.method !== interfaceData.method) {
|
||
let checkRepeat = await this.Model.checkRepeat(interfaceData.project_id, params.path, params.method);
|
||
if (checkRepeat > 0) {
|
||
return ctx.body = yapi.commons.resReturn(null, 401, '已存在的接口:' + params.path + '[' + params.method + ']');
|
||
}
|
||
}
|
||
|
||
let data = {
|
||
up_time: yapi.commons.time()
|
||
};
|
||
|
||
if (params.path) {
|
||
data.path = params.path;
|
||
}
|
||
if (params.title) {
|
||
data.title = params.title;
|
||
}
|
||
if (params.desc) {
|
||
data.desc = params.desc;
|
||
}
|
||
if (params.method) {
|
||
data.method = params.method;
|
||
}
|
||
|
||
if(params.catid){
|
||
data.catid = params.catid;
|
||
}
|
||
|
||
if (params.req_headers) {
|
||
data.req_headers = params.req_headers;
|
||
}
|
||
|
||
if (params.req_body_form) {
|
||
data.req_body_form = params.req_body_form;
|
||
}
|
||
if (params.req_params && Array.isArray(params.req_params) && params.req_params.length > 0) {
|
||
data.type = 'var'
|
||
data.req_params = params.req_params;
|
||
} else {
|
||
data.type = 'static'
|
||
}
|
||
|
||
if (params.req_query) {
|
||
data.req_query = params.req_query;
|
||
}
|
||
if (params.req_body_other) {
|
||
data.req_body_other = params.req_body_other;
|
||
}
|
||
|
||
if (params.req_body_type) {
|
||
data.req_body_type = params.req_body_type;
|
||
}
|
||
|
||
if (params.res_body_type) {
|
||
data.res_body_type = params.res_body_type;
|
||
}
|
||
if (params.res_body) {
|
||
data.res_body = params.res_body;
|
||
}
|
||
|
||
if(params.status){
|
||
data.status = params.status;
|
||
}
|
||
|
||
try {
|
||
let result = await this.Model.up(id, data);
|
||
let username = this.getUsername();
|
||
let cate;
|
||
if(params.catid){
|
||
cate = await this.catModel.get(+params.catid);
|
||
}else{
|
||
let inter = await this.Model.get(id);
|
||
cate = await this.catModel.get(inter.catid);
|
||
}
|
||
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 更新了分类 "${cate.name}" 下的接口 "${data.title}"`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: cate.project_id
|
||
});
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (e) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, e.message);
|
||
}
|
||
|
||
}
|
||
|
||
/**
|
||
* 删除接口
|
||
* @interface /interface/del
|
||
* @method GET
|
||
* @category interface
|
||
* @foldnumber 10
|
||
* @param {Number} id 接口id，不能为空
|
||
* @returns {Object}
|
||
* @example ./api/interface/del.json
|
||
*/
|
||
|
||
async del(ctx) {
|
||
try {
|
||
let id = ctx.request.body.id;
|
||
|
||
if (!id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '接口id不能为空');
|
||
}
|
||
|
||
let data = await this.Model.get(ctx.request.body.id);
|
||
|
||
if (data.uid != this.getUid()) {
|
||
if (await this.jungeProjectAuth(data.project_id) !== true) {
|
||
return ctx.body = yapi.commons.resReturn(null, 405, '没有权限');
|
||
}
|
||
}
|
||
|
||
let inter = await this.Model.get(id);
|
||
let result = await this.Model.del(id);
|
||
|
||
let username = this.getUsername();
|
||
let cate = await this.catModel.get(inter.catid);
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 删除了分类 "${cate.name}" 下的接口 "${inter.title}"`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: cate.project_id
|
||
});
|
||
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
} catch (err) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, err.message);
|
||
}
|
||
}
|
||
|
||
async solveConflict(ctx) {
|
||
try {
|
||
let id = parseInt(ctx.query.id, 10), result, userInst, userinfo, data;
|
||
if (!id) return ctx.websocket.send("id 参数有误");
|
||
result = await this.Model.get(id), userinfo;
|
||
if (result.edit_uid !== 0 && result.edit_uid !== this.getUid()) {
|
||
userInst = yapi.getInst(userModel);
|
||
userinfo = await userInst.findById(result.edit_uid);
|
||
data = {
|
||
errno: result.edit_uid,
|
||
data: { uid: result.edit_uid, username: userinfo.username }
|
||
}
|
||
} else {
|
||
this.Model.upEditUid(id, this.getUid()).then()
|
||
data = {
|
||
errno: 0,
|
||
data: result
|
||
}
|
||
}
|
||
ctx.websocket.send(JSON.stringify(data));
|
||
ctx.websocket.on('close', () => {
|
||
this.Model.upEditUid(id, 0).then()
|
||
})
|
||
} catch (err) {
|
||
yapi.commons.log(err, 'error')
|
||
}
|
||
}
|
||
|
||
async addCat(ctx) {
|
||
try {
|
||
let params = ctx.request.body;
|
||
params = yapi.commons.handleParams(params, {
|
||
name: 'string',
|
||
project_id: 'number',
|
||
desc: 'string'
|
||
});
|
||
|
||
if (!params.project_id) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '项目id不能为空');
|
||
}
|
||
if (!params.name) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '名称不能为空');
|
||
}
|
||
|
||
let result = await this.catModel.save({
|
||
name: params.name,
|
||
project_id: params.project_id,
|
||
desc: params.desc,
|
||
uid: this.getUid(),
|
||
add_time: yapi.commons.time(),
|
||
up_time: yapi.commons.time()
|
||
})
|
||
|
||
let username = this.getUsername();
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 添加了分类 "${params.name}"`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: params.project_id
|
||
});
|
||
|
||
ctx.body = yapi.commons.resReturn(result);
|
||
|
||
} catch (e) {
|
||
ctx.body = yapi.commons.resReturn(null, 402, e.message);
|
||
}
|
||
}
|
||
|
||
async upCat(ctx) {
|
||
try {
|
||
let params = ctx.request.body;
|
||
let result = await this.catModel.up(params.catid, {
|
||
name: params.name,
|
||
desc: params.desc,
|
||
up_time: yapi.commons.time()
|
||
});
|
||
|
||
let username = this.getUsername();
|
||
let cate = await this.catModel.get(params.catid);
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 更新了分类 "${cate.name}"`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: cate.project_id
|
||
});
|
||
|
||
ctx.body = yapi.commons.resReturn(result)
|
||
} catch (e) {
|
||
ctx.body = yapi.commons.resReturn(null, 400, e.message)
|
||
}
|
||
}
|
||
|
||
async delCat(ctx) {
|
||
try {
|
||
let id = ctx.request.body.catid;
|
||
let catData = await this.catModel.get(id);
|
||
if (!catData) {
|
||
ctx.body = yapi.commons.resReturn(null, 400, "不存在的分类")
|
||
}
|
||
|
||
if (catData.uid !== this.getUid()) {
|
||
let auth = await this.checkAuth(catData.project_id, 'project', 'danger')
|
||
if (!auth) {
|
||
return ctx.body = yapi.commons.resReturn(null, 400, '没有权限');
|
||
}
|
||
}
|
||
|
||
let cate = await this.catModel.get(id);
|
||
let result = await this.catModel.del(id);
|
||
let r = await this.Model.delByCatid(id);
|
||
let username = this.getUsername();
|
||
|
||
yapi.commons.saveLog({
|
||
content: `用户 "${username}" 删除了分类 "${cate.name}" 及该分类下的接口`,
|
||
type: 'project',
|
||
uid: this.getUid(),
|
||
username: username,
|
||
typeid: cate.project_id
|
||
});
|
||
|
||
return ctx.body = yapi.commons.resReturn(r);
|
||
|
||
|
||
} catch (e) {
|
||
yapi.commons.resReturn(null, 400, e.message)
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
|
||
module.exports = interfaceController; |