49 lines
1.1 KiB
JavaScript
49 lines
1.1 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.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);
|