storage limit check

Author: maxchistt

server/middleware/storage.middleware.js

@@ -0,0 +1,36 @@
+/** @file storage.middleware.js */
+const Note = require('../models/Note')
+const Media = require('../models/Media')
+require('dotenv').config()
+const STORAGE_LIMIT = process.env.STORAGE_LIMIT || 5024000
+
+/**
+ * Функция-Middleware для проверки занятого пространства
+ * @param {*} req 
+ * @param {*} res 
+ * @param {*} next 
+ */
+module.exports = async (req, res, next) => {
+    //проверка работы сервера
+    if (req.method === 'OPTIONS') {
+        return next()
+    }
+
+    try {
+        /**Нахождение пользовательских заметок в бд */
+        const notes = await Note.find({ owner: req.user.userId })
+        /**Нахождение пользовательских media в бд */
+        const media = await Media.find({ owner: req.user.userId })
+        /**Вычисление обьема данных */
+        const dataLength = JSON.stringify([...notes, ...media]).length
+
+        const leftPercent = ~~(((STORAGE_LIMIT - dataLength) / STORAGE_LIMIT) * 100)
+        //console.log("data left: ", leftPercent, "%")
+        if (!(leftPercent > 0)) return res.status(409).json({ message: "Привышен лимит данных" })
+
+        next()
+    } catch (e) {
+        res.status(500).json({ message: 'Что-то пошло не так, попробуйте снова' })
+    }
+
+}

server/routes/media.routes.js

@@ -6,13 +6,14 @@ const { Router } = require('express')
 const Media = require('../models/Media')
 const Note = require('../models/Note')
 const auth = require('../middleware/auth.middleware')
+const storage = require('../middleware/storage.middleware')
 const router = Router()
 
 /**
  * Добавление и редактирование media
  * /api/media/set
  */
-router.post('/set', auth, async (req, res) => {
+router.post('/set', auth, storage, async (req, res) => {
     try {
         /**получение данных о media и запись в бд */
         const media = tryParce(req.body.media)

server/routes/notes.routes.js

@@ -6,6 +6,7 @@ const { Router } = require('express')
 const Note = require('../models/Note')
 const Media = require('../models/Media')
 const auth = require('../middleware/auth.middleware')
+const storage = require('../middleware/storage.middleware')
 const router = Router()
 
 const { checkNote } = require('../validation/NoteCheck')
@@ -14,7 +15,7 @@ const { checkNote } = require('../validation/NoteCheck')
  * Добавление и редактирование заметки
  * /api/notes/set
  */
-router.post('/set', auth, async (req, res) => {
+router.post('/set', auth, storage, async (req, res) => {
     try {
         /**получение данных о заметке и запись в бд */
         const note = tryParce(req.body.note)