57 lines
1.4 KiB
JavaScript
57 lines
1.4 KiB
JavaScript
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.virtual('projectId').get(function projectId() {
|
|
return this.populated('project');
|
|
});
|
|
|
|
collectedProjectSchema.virtual('isDeleted').get(function isDeleted() {
|
|
return this.project == null;
|
|
});
|
|
|
|
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);
|