All files / lib/models watcher.ts

88% Statements 22/25
100% Branches 0/0
62.5% Functions 5/8
88% Lines 22/25

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99          16x 16x 16x                                                       16x                                                 16x       16x       16x       16x 3x 3x 3x 3x     16x 2x     16x 9x     16x 9x     16x 16x   16x   16x    
import Crowi from 'server/crowi'
import { Types, Document, Model, Schema, model } from 'mongoose'
import ActivityDefine from '../util/activityDefine'
// import Debug from 'debug'
 
const STATUS_WATCH = 'WATCH'
const STATUS_IGNORE = 'IGNORE'
const STATUSES = [STATUS_WATCH, STATUS_IGNORE]
 
export interface WatcherDocument extends Document {
  _id: Types.ObjectId
  user: Types.ObjectId
  targetModel: string
  target: Types.ObjectId
  status: string
  createdAt: Date
 
  isWatching(): boolean
  isIgnoring(): boolean
}
 
export interface WatcherModel extends Model<WatcherDocument> {
  findByUserIdAndTargetId(userId: Types.ObjectId, targetId: Types.ObjectId): any
  upsertWatcher(user: Types.ObjectId, targetModel: string, target: Types.ObjectId, status: string): any
  watchByPageId(user: Types.ObjectId, pageId: Types.ObjectId, status: string): any
  getWatchers(target: Types.ObjectId): Promise<Types.ObjectId[]>
  getIgnorers(target: Types.ObjectId): Promise<Types.ObjectId[]>
 
  STATUS_WATCH: string
  STATUS_IGNORE: string
}
 
export default (crowi: Crowi) => {
  // const debug = Debug('crowi:models:watcher')
 
  const watcherSchema = new Schema<WatcherDocument, WatcherModel>({
    user: {
      type: Schema.Types.ObjectId,
      ref: 'User',
      index: true,
      required: true,
    },
    targetModel: {
      type: String,
      require: true,
      enum: ActivityDefine.getSupportTargetModelNames(),
    },
    target: {
      type: Schema.Types.ObjectId,
      refPath: 'targetModel',
      require: true,
    },
    status: {
      type: String,
      require: true,
      enum: STATUSES,
    },
    createdAt: { type: Date, default: Date.now },
  })
 
  watcherSchema.methods.isWatching = function() {
    return this.status === STATUS_WATCH
  }
 
  watcherSchema.methods.isIgnoring = function() {
    return this.status === STATUS_IGNORE
  }
 
  watcherSchema.statics.findByUserIdAndTargetId = function(userId, targetId) {
    return this.findOne({ user: userId, target: targetId })
  }
 
  watcherSchema.statics.upsertWatcher = function(user, targetModel, target, status) {
    const query = { user, targetModel, target }
    const doc = { ...query, status }
    const options = { upsert: true, new: true, setDefaultsOnInsert: true, runValidators: true }
    return Watcher.findOneAndUpdate(query, doc, options)
  }
 
  watcherSchema.statics.watchByPageId = function(user, pageId, status) {
    return this.upsertWatcher(user, 'Page', pageId, status)
  }
 
  watcherSchema.statics.getWatchers = async function(target) {
    return Watcher.find({ target, status: STATUS_WATCH }).distinct('user')
  }
 
  watcherSchema.statics.getIgnorers = async function(target) {
    return Watcher.find({ target, status: STATUS_IGNORE }).distinct('user')
  }
 
  watcherSchema.statics.STATUS_WATCH = STATUS_WATCH
  watcherSchema.statics.STATUS_IGNORE = STATUS_IGNORE
 
  const Watcher = model<WatcherDocument, WatcherModel>('Watcher', watcherSchema)
 
  return Watcher
}