mirror of
https://github.com/usatiuk/ustk-todolist.git
synced 2025-10-28 15:47:48 +01:00
38 lines
991 B
JavaScript
38 lines
991 B
JavaScript
const mongoose = require('mongoose');
|
|
|
|
const { Schema } = mongoose;
|
|
|
|
const TodoListSchema = Schema({
|
|
name: {
|
|
type: String,
|
|
required: true,
|
|
},
|
|
todos: [{ type: Schema.Types.ObjectId, ref: 'Todo' }],
|
|
user: { type: Schema.Types.ObjectId, ref: 'User', required: true },
|
|
});
|
|
|
|
TodoListSchema.pre('save', async function () {
|
|
const user = await this.model('User').findById(this.user);
|
|
user.lists.push(this._id);
|
|
await user.save();
|
|
});
|
|
|
|
TodoListSchema.pre('remove', async function () {
|
|
const user = await this.model('User').findById(this.user);
|
|
user.lists.splice(user.todos.indexOf(this._id), 1);
|
|
await user.save();
|
|
await this.model('Todo').remove({ list: this._id });
|
|
});
|
|
|
|
TodoListSchema.methods.toJson = function () {
|
|
const todos = this.populated('todos') ? this.todos.map(todo => todo.toJson()) : this.todos;
|
|
return {
|
|
id: this._id.toString(),
|
|
user: this.user.toString(),
|
|
name: this.name,
|
|
todos,
|
|
};
|
|
};
|
|
|
|
mongoose.model('TodoList', TodoListSchema);
|