add support to other roles update in staging (#1354)

* add support to other roles

* modify condition to forbid in production

Author: RitikJaiswal75

controllers/staging.js

@@ -3,33 +3,21 @@ const { addOrUpdate } = require("../models/users");
 const updateRoles = async (req, res) => {
   try {
     const userData = await req.userData;
-    if (process.env.NODE_ENV !== "production") {
+    if (process.env.NODE_ENV === "production") {
       return res.status(403).json({
         message: "FORBIDDEN | To be used only in staging and development",
       });
     }
     const userId = req.userData.id;
-    if (req.body.super_user) {
-      await addOrUpdate(
-        {
-          roles: {
-            ...userData.roles,
-            super_user: req.body.super_user,
-          },
+    await addOrUpdate(
+      {
+        roles: {
+          ...userData.roles,
+          ...req.body,
         },
-        userId
-      );
-    } else if (req.body.member) {
-      await addOrUpdate(
-        {
-          roles: {
-            ...userData.roles,
-            member: req.body.member,
-          },
-        },
-        userId
-      );
-    }
+      },
+      userId
+    );
     return res.status(200).json({
       message: "Roles Updated successfully",
     });

middlewares/validators/staging.js

@@ -1,11 +1,12 @@
 const joi = require("joi");
 
 const validateUserRoles = async (req, res, next) => {
-  const config = {
+  const schema = joi.object().strict().keys({
     super_user: joi.boolean().optional(),
     member: joi.boolean().optional(),
-  };
-  const schema = joi.object(config).xor("super_user", "member");
+    archive: joi.boolean().optional(),
+    in_discord: joi.boolean().optional(),
+  });
 
   try {
     await schema.validateAsync(req.body);