yapi/server/controllers/interface.js

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;