obs-portal/models/Track.js

66 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');
2020-11-20 10:02:30 +00:00
const TrackSchema = new mongoose.Schema(
{
slug: { type: String, lowercase: true, unique: true },
title: String,
description: String,
body: String,
visible: Boolean,
uploadedByUserAgent: String,
2020-11-20 10:02:30 +00:00
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 },
);
2020-11-20 10:02:30 +00:00
TrackSchema.plugin(uniqueValidator, { message: 'is already taken' });
2020-11-20 10:02:30 +00:00
TrackSchema.pre('validate', function (next) {
if (!this.slug) {
this.slugify();
}
next();
});
2020-11-20 10:02:30 +00:00
TrackSchema.methods.slugify = function () {
this.slug = slug(this.title) + '-' + ((Math.random() * Math.pow(36, 6)) | 0).toString(36);
};
TrackSchema.methods.isVisibleTo = function (user) {
if (this.visible) {
return true;
}
if (!user) {
return false;
}
if (user._id.toString() === this.author._id.toString()) {
return true;
}
return false;
};
TrackSchema.methods.toJSONFor = function (user, include) {
return {
slug: this.slug,
title: this.title,
description: this.description,
createdAt: this.createdAt,
updatedAt: this.updatedAt,
2020-08-16 21:23:18 +00:00
visibleForAll: this.author ? this.author.areTracksVisibleForAll : false,
2020-08-28 19:50:53 +00:00
visible: this.visible,
author: this.author.toProfileJSONFor(user),
...(include && include.body ? { body: this.body } : {}),
};
};
mongoose.model('Track', TrackSchema);