yapi/server/models/group.js

const yapi = require('../yapi.js');
const baseModel = require('./base.js');

class groupModel extends baseModel {
    getName() {
        return 'group';
    }

    getSchema() {
        return {
            uid: Number,
            group_name: String,
            group_desc: String,
            add_time: Number,
            up_time: Number,
            type: {type:String,default: 'public', enum: ['public', 'private']},
            members: [
                {
                    uid: Number,
                    role: {type: String, enum:['owner', 'dev']}
                }
            ]
        };
    }

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

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

    getByPrivateUid(uid){
        return this.model.findOne({
            uid: uid,
            type: 'private'
        }).select('group_name _id group_desc add_time up_time type').exec();
    }

    getGroupById(id) {
        return this.model.findOne({
            _id: id
        }).select("uid group_name group_desc add_time up_time type").exec();
    }

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

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

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

    changeMemberRole(id, uid, role) {
        return this.model.update(
            {
                _id: id,
                 "members.uid": uid
            }, {
                "$set": { "members.$.role": role}
            }
        );
    }

    checkMemberRepeat(id, uid){
        return this.model.count({
            _id: id,
            "members.uid": uid
        });
    }

    list() {
        return this.model.find({
            type: 'public'
        }).select('group_name _id group_desc add_time up_time type uid').exec();
    }

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

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

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

module.exports = groupModel;