yapi/server/models/interface.js

import yapi from '../yapi.js';
import baseModel from './base.js';

class interfaceModel extends baseModel {
    getName() {
        return 'interface';
    }

    getSchema() {
        return {
            title: { type: String, required: true },
            uid: { type: Number, required: true },
            path: { type: String, required: true },
            method: { type: String, required: true },
            project_id: { type: Number, required: true },
            status: {type: String, enum: ['undone', 'done'], default: 'undone'},
            desc: String,
            add_time: Number,
            up_time: Number,
            type: {type: String, enum: ['static', 'var'], default:'static'},
            req_query:[{
                name: String, value: String, desc: String, required: {
                    type:String,
                    enum: ["1", "0"],
                    default: "1"
                }
            }],
            req_headers: [{
                name: String, value: String, desc: String, required: {
                    type:String,
                    enum: ["1", "0"],
                    default: "1"
                }
            }],
            req_params:[{
                name: String,
                desc: String
            }],
            req_body_type: {
                type: String,
                enum: ['form', 'json', 'text', 'file']
            },
            req_body_form: [{
                name: String,  type: { type: String, enum: ['text', 'file'] }, desc: String, required: {
                    type:String,
                    enum: ["1", "0"],
                    default: "1"
                }
            }],
            req_body_other: String,
            res_body_type: {
                type: String,
                enum: ['json', 'text', 'xml']
            },
            res_body: String
        };
    }

    save(data) {
        let m = new this.model(data);
        return m.save();
    }

    get(id) {
        return this.model.findOne({
            _id: id
        })
            .exec();
    }

    getVar(project_id, method){
        return this.model.find({
            type: 'var',
            method: method
        }).select('_id path').exec()
    }

    getByPath(project_id, path, method) {
        return this.model.find({
            project_id: project_id,
            path: path,
            method: method
        })
            .exec();
    }

    checkRepeat(id, path, method) {
        return this.model.count({
            project_id: id,
            path: path,
            method: method
        });
    }

    countByProjectId(id) {
        return this.model.count({
            project_id: id
        });
    }

    list(project_id) {
        return this.model.find({
            project_id: project_id
        })
            .sort({ _id: -1 })
            .exec();
    }

    del(id) {
        return this.model.deleteOne({
            _id: id
        });
    }

    delByProjectId(id){
        return this.model.delete({
            project_id: id
        })
    }

    up(id, data) {
        data.up_time = yapi.commons.time();
        return this.model.update({
            _id: id
        }, data, { runValidators: true });
    }
}

module.exports = interfaceModel;