apps/Notes/models/Note.js


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

const { encrypt, decrypt } = require('helpers/crypt')
const mongoose = require('mongoose')

const noteSchema = new mongoose.Schema({
  content: { type: String, required: true, minlength: 1 }
})

noteSchema.statics.getNote = async (id) => {
  const note = await Note.findById(id)
  if (!note) throw new Error('Could not fetch note')

  const content = decrypt(note.content)

  return { ...note, content }
}

noteSchema.statics.updateNote = async (id, content) => {
  const note = await Note.findByIdAndUpdate(id, { content: encrypt(content) }, { new: true })

  if (!note) throw new Error('Could not update note')

  return { ...note, content }
}

noteSchema.pre('save', async function (next) {
  const note = this

  if (note.isModified('content')) {
    note.content = encrypt(note.content)
  }

  next()
})

const Note = mongoose.models.Note || mongoose.model('Note', noteSchema)

export default Note