obs-portal/models/Track.js

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;