weseek__growi/lib/models/bookmark.js

module.exports = function(crowi) {
  var debug = require('debug')('crowi:models:bookmark')
    , mongoose = require('mongoose')
    , ObjectId = mongoose.Schema.Types.ObjectId
    , bookmarkSchema;


  bookmarkSchema = new mongoose.Schema({
    page: { type: ObjectId, ref: 'Page', index: true },
    user: { type: ObjectId, ref: 'User', index: true },
    createdAt: { type: Date, default: Date.now() }
  });
  bookmarkSchema.index({page: 1, user: 1}, {unique: true});

  bookmarkSchema.statics.populatePage = function(bookmarks) {
    var Bookmark = this;
    var User = crowi.model('User');
    var Page = crowi.model('Page');

    return new Promise(function(resolve, reject) {
      Bookmark.populate(bookmarks, {path: 'page'}, function(err, bookmarks) {
        if (err) {
          return reject(err);
        }

        Bookmark.populate(bookmarks, {path: 'page.revision', model: 'Revision'}, function(err, bookmarks) {
          if (err) {
            return reject(err);
          }

          Bookmark.populate(bookmarks, {path: 'page.revision.author', model: 'User', select: User.USER_PUBLIC_FIELDS}, function(err, bookmarks) {
            if (err) {
              return reject(err);
            }

            return resolve(bookmarks);
          });
        });
      });
    });
  };

  // bookmark チェック用
  bookmarkSchema.statics.findByPageIdAndUserId = function(pageId, userId) {
    var Bookmark = this;

    return new Promise(function(resolve, reject) {
      return Bookmark.findOne({ page: pageId, user: userId }, function(err, doc) {
        if (err) {
          return reject(err);
        }

        return resolve(doc);
      });
    });
  };

  bookmarkSchema.statics.findByUser = function(user, option) {
    var User = crowi.model('User');
    var Bookmark = this;

    var limit = option.limit || 50;
    var offset = option.offset || 0;
    var populatePage = option.populatePage || false;

    return new Promise(function(resolve, reject) {
      Bookmark
        .find({ user: user._id })
        .sort({createdAt: -1})
        .skip(offset)
        .limit(limit)
        .exec(function(err, bookmarks) {
          if (err) {
            return reject(err);
          }

          if (!populatePage) {
            return resolve(bookmarks);
          }

          return Bookmark.populatePage(bookmarks).then(resolve).catch(reject);
        });
    });
  };

  bookmarkSchema.statics.add = function(page, user) {
    var Bookmark = this;

    return new Promise(function(resolve, reject) {
      var newBookmark = new Bookmark;

      newBookmark.page = page;
      newBookmark.user = user;
      newBookmark.createdAt = Date.now();
      newBookmark.save(function(err, bookmark) {
        debug('Bookmark.save', err, bookmark);
        if (err) {
          if (err.code === 11000) { // duplicate key (dummy reesponse of new object)
            return resolve(newBookmark);
          }
          return reject(err);
        }

        resolve(bookmark);
      });
    });
  };

  bookmarkSchema.statics.remove = function(page, user) {
    var Bookmark = this;

    return new Promise(function(resolve, reject) {
      Bookmark.findOneAndRemove({page: page, user: user}, function(err, data) {
        if (err) {
          debug('Bookmark.findOneAndRemove failed', err);
          return reject(err);
        }

        return resolve(data);
      });
    });
  };

  return mongoose.model('Bookmark', bookmarkSchema);
};