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);