70 lines
1.6 KiB
JavaScript
70 lines
1.6 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const uniqueValidator = require('mongoose-unique-validator');
|
|
const slug = require('slug');
|
|
|
|
const schema = new mongoose.Schema(
|
|
{
|
|
slug: { type: String, lowercase: true, unique: true },
|
|
title: String,
|
|
description: String,
|
|
body: String,
|
|
visible: Boolean,
|
|
uploadedByUserAgent: String,
|
|
numEvents: { type: Number, default: 0 },
|
|
comments: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Comment' }],
|
|
author: { type: mongoose.Schema.Types.ObjectId, ref: 'User' },
|
|
trackData: { type: mongoose.Schema.Types.ObjectId, ref: 'TrackData' },
|
|
},
|
|
{ timestamps: true },
|
|
);
|
|
|
|
schema.plugin(uniqueValidator, { message: 'is already taken' });
|
|
|
|
schema.pre('validate', function (next) {
|
|
if (!this.slug) {
|
|
this.slugify();
|
|
}
|
|
|
|
next();
|
|
});
|
|
|
|
class Track extends mongoose.Model {
|
|
slugify() {
|
|
this.slug = slug(this.title) + '-' + ((Math.random() * Math.pow(36, 6)) | 0).toString(36);
|
|
}
|
|
|
|
isVisibleTo(user) {
|
|
if (this.visible) {
|
|
return true;
|
|
}
|
|
|
|
if (!user) {
|
|
return false;
|
|
}
|
|
|
|
if (user._id.toString() === this.author._id.toString()) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
toJSONFor(user, include) {
|
|
return {
|
|
slug: this.slug,
|
|
title: this.title,
|
|
description: this.description,
|
|
createdAt: this.createdAt,
|
|
updatedAt: this.updatedAt,
|
|
visibleForAll: this.author ? this.author.areTracksVisibleForAll : false,
|
|
visible: this.visible,
|
|
author: this.author.toProfileJSONFor(user),
|
|
...(include && include.body ? { body: this.body } : {}),
|
|
};
|
|
}
|
|
}
|
|
|
|
mongoose.model(Track, schema);
|
|
|
|
module.exports = Track;
|