yapi-next/vendors/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'] },
          username: String,
          email: String
        }
      ],

      custom_field1: {
        name: String,
        enable: { type: Boolean, default: false }
      }
      // custom_field2: {
      //   name: String,
      //   enable: { type: Boolean, default: false }
      // },
      // custom_field3: {
      //   name: String,
      //   enable: { type: Boolean, default: false }
      // }
    };
  }

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

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

  updateMember(data) {
    return this.model.update(
      {
        'members.uid': data.uid
      },
      {
        $set: {
          'members.$.username': data.username,
          'members.$.email': data.email
        }
      },
      { multi: true }
    );
  }

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

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

  checkRepeat(name) {
    return this.model.countDocuments({
      group_name: name
    });
  }
  //  分组数量统计
  getGroupListCount() {
    return this.model.countDocuments({ type: 'public' });
  }

  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.countDocuments({
      _id: id,
      'members.uid': uid
    });
  }

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

  getAuthList(uid){
    return this.model.find({
      $or: [{
        'members.uid': uid,
        'type': 'public'
      }, {
        'type': 'public',
        uid
      }]
    }).select(' _id group_name group_desc add_time up_time type uid custom_field1')
    .exec();
    
  }

  findByGroups(ids = []){
    return this.model.find({
      _id: {
        $in: ids
      },
      type: 'public'
    })
  }

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

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

  getcustomFieldName(name) {
    return this.model
      .find({
        'custom_field1.name': name,
        'custom_field1.enable': true
      })
      .select('_id')
      .exec();
  }

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

module.exports = groupModel;