From a863c77ea12a4c7060b9c496e5df81bbdcca2374 Mon Sep 17 00:00:00 2001
From: piotrruss <mail@pruss.it>
Date: Thu, 9 Sep 2021 23:23:53 +0200
Subject: add db transactions, move db models

---
 apps/Notes/models/NoteList.js | 68 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 apps/Notes/models/NoteList.js

(limited to 'apps/Notes/models/NoteList.js')

diff --git a/apps/Notes/models/NoteList.js b/apps/Notes/models/NoteList.js
new file mode 100644
index 0000000..0af648a
--- /dev/null
+++ b/apps/Notes/models/NoteList.js
@@ -0,0 +1,68 @@
+const { encrypt, decrypt } = require('helpers/crypt')
+const mongoose = require('mongoose')
+
+const decryptTitles = l => ({ notes: l.notes.map(n => ({ ...n, title: decrypt(n.title) })) })
+
+const noteListSchema = new mongoose.Schema({
+  notes: [{
+    title: {
+      type: String,
+      maxlength: 1000,
+      required: true
+    },
+    noteId: {
+      type: mongoose.Schema.Types.ObjectId,
+      ref: 'Note',
+      required: true
+    },
+    created_at: { type: Date, default: Date.now },
+    updated_at: { type: Date, default: Date.now }
+  }]
+})
+
+noteListSchema.statics.getList = async (id) => {
+  const newList = await NoteList.findById(id).lean()
+
+  return { notes: newList.notes.map(n => ({ ...n, title: decrypt(n.title) })) }
+}
+
+noteListSchema.statics.getNoteId = async (_id, id) => {
+  const noteList = await NoteList.findById(_id).lean()
+  return noteList.notes
+    .find(n => n._id.toString() === id.toString()).noteId
+}
+
+noteListSchema.statics.addNote = async (_id, noteId, title) => {
+  const noteList = await NoteList.findOneAndUpdate(
+    { _id }, { $push: { notes: { title: encrypt(title || 'no title'), noteId } } }, { new: true }
+  ).lean()
+
+  return decryptTitles(noteList)
+}
+
+noteListSchema.statics.removeNote = async (_id, id) => {
+  const noteList = await NoteList.findOneAndUpdate(
+    { _id }, { $pull: { notes: { _id: id } } }
+  ).lean()
+
+  return decryptTitles(noteList)
+}
+
+noteListSchema.statics.updateList = async (_id, noteId, title) => {
+  const noteList = await NoteList.findOneAndUpdate(
+    { _id, 'notes.noteId': noteId },
+    {
+      $set: {
+        'notes.$.title': encrypt(title || 'No title'),
+        'notes.$.updated_at': Date.now()
+      }
+    },
+    { new: true }
+  ).lean()
+
+  return decryptTitles(noteList)
+}
+
+const NoteList = mongoose.models.NoteList || mongoose.model('NoteList', noteListSchema)
+
+export default NoteList
-- 
cgit v1.2.3