yapi/server/models/project.js

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

class projectModel extends baseModel {
    getName() {
        return 'project';
    }

    getSchema() {
        return {
            uid: { type: Number, required: true },
            name: { type: String, required: true },
            basepath: {
                type: String, required: true, validate: {
                    validator: (v) => {
                        return v && v[0] === '/';
                    },
                    message: 'basepath必须是/开头'
                }
            },
            desc: String,
            group_id: { type: Number, required: true },
            members: Array,
            protocol: { type: String, required: true },
            prd_host: { type: String, required: true },
            env: [
                { name: String, domain: String }
            ],
            add_time: Number,
            up_time: Number
        };
    }

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

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

    getByDomain(domain) {
        return this.model.find({
            prd_host: domain
        }).exec();
    }

    checkNameRepeat(name) {
        return this.model.count({
            name: name
        });
    }

    checkDomainRepeat(domain, basepath) {
        return this.model.count({
            prd_host: domain,
            basepath: basepath
        });
    }

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

    listWithPaging(group_id, page, limit) {
        page = parseInt(page);
        limit = parseInt(limit);
        return this.model.find({
            group_id: group_id
        }).sort({ _id: -1 }).skip((page - 1) * limit).limit(limit).exec();
    }

    listCount(group_id) {
        return this.model.count({
            group_id: group_id
        });
    }

    countByGroupId(group_id) {
        return this.model.count({
            group_id: group_id
        });
    }

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

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

    addMember(id, uid) {
        return this.model.update(
            {
                _id: id
            }, {
                $push: { members: uid }
            }
        );
    }

    delMember(id, uid) {
        return this.model.update(
            {
                _id: id
            }, {
                $pull: { members: uid }
            }
        );
    }

    checkMemberRepeat(id, uid) {
        return this.model.count({
            _id: id,
            members: {$in: [uid]}
        });
    }

    search(keyword) {
        return this.model.find({
            name: new RegExp(keyword, 'ig')
        })
            .limit(10);
    }
}

module.exports = projectModel;