import mongoose from 'mongoose'; import shortid from 'shortid'; import slugify from 'slugify'; const { Schema } = mongoose; const collectedProjectSchema = new Schema( { project: { type: Schema.Types.ObjectId, ref: 'Project' }, }, { timestamps: true, _id: true, usePushEach: true } ); collectedProjectSchema.virtual('id').get(function getId() { return this._id.toHexString(); }); collectedProjectSchema.set('toJSON', { virtuals: true }); const collectionSchema = new Schema( { _id: { type: String, default: shortid.generate }, name: { type: String, default: 'My collection' }, description: { type: String }, slug: { type: String }, owner: { type: Schema.Types.ObjectId, ref: 'User' }, items: { type: [collectedProjectSchema] } }, { timestamps: true, usePushEach: true } ); collectionSchema.virtual('id').get(function getId() { return this._id; }); collectionSchema.set('toJSON', { virtuals: true }); collectionSchema.pre('save', function generateSlug(next) { const collection = this; collection.slug = slugify(collection.name, '_'); return next(); }); export default mongoose.model('Collection', collectionSchema);