Updated Store Delete Route

clever-developer-club · Nov 11, 2021 · cbdddea · cbdddea
1 parent 59b9e9a
commit cbdddea
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 3 deletions.
diff --git a/server/controllers/store.controller.js b/server/controllers/store.controller.js
@@ -2,7 +2,7 @@ const Store = require('../models/Store')
 const Product = require('../models/Product')
 const getError = require('../utils/dbErrorHandler')
 const {uploadImages, deleteImages} = require('../utils/multipleImageOperations')
-const { deleteStorefromProducts } = require('../utils/relationOperations')
+const { deleteStorefromProducts, deleteStorefromUser } = require('../utils/relationOperations')
 const {storePopulate} = require('../utils/populateObjects')
 const User = require('../models/User')
 
@@ -701,6 +701,7 @@ module.exports = {
             .then( async(store) => {
                 if(store){
                     await deleteStorefromProducts(store)
+					await deleteStorefromUser(req.user._id,store._id)
                     await deleteImages(store.images)
                     await store.deleteOne()
                     return res.status(200).json({

diff --git a/server/models/User.js b/server/models/User.js
@@ -120,6 +120,11 @@ UserSchema.methods = {
 		if(!this.hasStore(id)){
             this.stores.push(id)
         }
+	},
+	removeStore : function(id){
+		if(this.hasStore(id)){
+            this.stores = this.stores.filter(e => e.toString() !== id.toString())
+        }
 	}
 }
 

diff --git a/server/utils/populateObjects.js b/server/utils/populateObjects.js
@@ -68,7 +68,18 @@ module.exports = {
         },
         {
             path:'stores',
-            select:'name description addresses rating products '
+            select:'name description addresses rating products',
+			populate: {
+                path: 'products',
+                populate:{
+                    path:'productId',
+                    select: 'name description images',
+                    populate:{
+                        path:'images',
+                        select:'url'
+                    }
+                }
+            } 
         }
 
     ],

diff --git a/server/utils/relationOperations.js b/server/utils/relationOperations.js
@@ -1,5 +1,6 @@
 const Product = require('../models/Product')
 const Store = require('../models/Store')
+const User = require('../models/User')
 
 module.exports = {
 
@@ -30,6 +31,15 @@ module.exports = {
             }))
         }
         return Promise.all(promises)
-    }
+    },
+
+	deleteStorefromUser : async (userId,storeId) => {
+		User.findById(userId).then(async (user) => {
+			if(user){
+				user.removeStore(storeId)
+				await user.save()
+			}
+		})
+	}
 
 }