obs-portal/models/Article.js

60 lines
1.6 KiB
JavaScript
Raw Normal View History

2020-11-20 10:02:30 +00:00
const mongoose = require('mongoose');
const uniqueValidator = require('mongoose-unique-validator');
const slug = require('slug');
const User = mongoose.model('User');
const ArticleSchema = new mongoose.Schema(
{
slug: { type: String, lowercase: true, unique: true },
title: String,
description: String,
body: String,
favoritesCount: { type: Number, default: 0 },
comments: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Comment' }],
tagList: [{ type: String }],
author: { type: mongoose.Schema.Types.ObjectId, ref: 'User' },
},
{ timestamps: true },
);
ArticleSchema.plugin(uniqueValidator, { message: 'is already taken' });
ArticleSchema.pre('validate', function (next) {
if (!this.slug) {
this.slugify();
}
next();
});
2020-11-20 10:02:30 +00:00
ArticleSchema.methods.slugify = function () {
this.slug = slug(this.title) + '-' + ((Math.random() * Math.pow(36, 6)) | 0).toString(36);
};
2020-11-20 10:02:30 +00:00
ArticleSchema.methods.updateFavoriteCount = function () {
const article = this;
2020-11-20 10:02:30 +00:00
return User.count({ favorites: { $in: [article._id] } }).then(function (count) {
article.favoritesCount = count;
return article.save();
});
};
2020-11-20 10:02:30 +00:00
ArticleSchema.methods.toJSONFor = function (user) {
return {
slug: this.slug,
title: this.title,
description: this.description,
body: this.body,
createdAt: this.createdAt,
updatedAt: this.updatedAt,
tagList: this.tagList,
favorited: user ? user.isFavorite(this._id) : false,
favoritesCount: this.favoritesCount,
2020-11-20 10:02:30 +00:00
author: this.author.toProfileJSONFor(user),
};
};
mongoose.model('Article', ArticleSchema);