feat: add purpose field and make discord invite API be able to generate invites n times by superusers

controllers/discordactions.js

@@ -8,6 +8,7 @@
 const { generateCloudFlareHeaders } = require("../utils/discord-actions");
 const discordDeveloperRoleId = config.get("discordDeveloperRoleId");
 const discordMavenRoleId = config.get("discordMavenRoleId");
+const crypto = require("crypto");
 
 const { setUserDiscordNickname, getDiscordMembers } = discordServices;
 
@@ -286,7 +287,7 @@
     const nickNameUpdatedUsers = [];
     let counter = 0;
     for (let i = 0; i < usersToBeEffected.length; i++) {
       const { discordId, username, first_name: firstName } = usersToBeEffected[i];
       try {
         if (counter % 10 === 0 && counter !== 0) {
           await new Promise((resolve) => setTimeout(resolve, 5500));
@@ -302,7 +303,7 @@
           if (message) {
             counter++;
             totalNicknamesUpdated.count++;
             nickNameUpdatedUsers.push(usersToBeEffected[i].id);
           }
         }
       } catch (error) {
@@ -407,17 +408,30 @@
   }
 };
 
+const generateRandomId = () => {
+  return crypto.randomBytes(16).toString("hex");
+};
+
 const generateInviteForUser = async (req, res) => {
   try {
-    const { userId } = req.query;
+    const { userId, dev } = req.query;
     const userIdForInvite = userId || req.userData.id;
-
-    const modelResponse = await discordRolesModel.getUserDiscordInvite(userIdForInvite);
-
-    if (!modelResponse.notFound) {
-      return res.status(409).json({
-        message: "User invite is already present!",
-      });
+    let inviteLink = "";
+    let randomId = "";
+
+    if (!dev) {
+      const modelResponse = await discordRolesModel.getUserDiscordInvite(userIdForInvite);
+      if (!modelResponse.notFound) {
+        return res.status(409).json({
+          message: "User invite is already present!",
+        });
+      }
+    } else {
+      const modelResponse = await discordRolesModel.getUserDiscordInvite(userIdForInvite);
+      if (!modelResponse.notFound) {
+        // Generate a random ID
+        randomId = generateRandomId();
+      }
     }
 
     const channelId = config.get("discordNewComersChannelId");
@@ -437,14 +451,27 @@
     const discordInviteResponse = await response.json();
 
     const inviteCode = discordInviteResponse.data.code;
-    const inviteLink = `discord.gg/${inviteCode}`;
+    inviteLink = `discord.gg/${inviteCode}`;
 
-    await discordRolesModel.addInviteToInviteModel({ userId: userIdForInvite, inviteLink });
+    if (dev) {
+      const purpose = req.body.purpose;
+      await discordRolesModel.addInviteToInviteModel({ userId: randomId || userIdForInvite, inviteLink, purpose });
 
-    return res.status(201).json({
-      message: "invite generated successfully",
-      inviteLink,
-    });
+      return res.status(201).json({
+        message: "invite generated successfully",
+        inviteLink,
+        purpose,
+        userId,
+      });
+    } else {
+      await discordRolesModel.addInviteToInviteModel({ userId: userIdForInvite, inviteLink });
+
+      return res.status(201).json({
+        message: "invite generated successfully",
+        inviteLink,
+        userId,
+      });
+    }
   } catch (err) {
     logger.error(`Error in generating invite for user: ${err}`);
     return res.boom.badImplementation(INTERNAL_SERVER_ERROR);

middlewares/validators/discordactions.js

@@ -40,8 +40,24 @@ const validateUpdateUsersNicknameStatusBody = async (req, res, next) => {
   }
 };
 
+export const validateGenerateInviteForUserBody = async (req, res, next) => {
+  const schema = Joi.object().strict().keys({
+    purpose: Joi.string().trim(),
+  });
+
+  try {
+    await schema.validateAsync(req.body, { abortEarly: false });
+    next();
+  } catch (error) {
+    const errorMessages = error.details.map((detail) => detail.message);
+    logger.error(`Error while validating invite creation payload: ${errorMessages}`);
+    res.boom.badRequest(errorMessages);
+  }
+};
+
 module.exports = {
   validateGroupRoleBody,
   validateMemberRoleBody,
   validateUpdateUsersNicknameStatusBody,
+  validateGenerateInviteForUserBody,
 };

routes/discordactions.js

@@ -20,6 +20,7 @@
   validateGroupRoleBody,
   validateMemberRoleBody,
   validateUpdateUsersNicknameStatusBody,
+  validateGenerateInviteForUserBody,
 } = require("../middlewares/validators/discordactions");
 const checkIsVerifiedDiscord = require("../middlewares/verifydiscord");
 const checkCanGenerateDiscordLink = require("../middlewares/checkCanGenerateDiscordLink");
@@ -36,7 +37,13 @@
 router.get("/groups", authenticate, checkIsVerifiedDiscord, getAllGroupRoles);
 router.post("/roles", authenticate, checkIsVerifiedDiscord, validateMemberRoleBody, addGroupRoleToMember);
 router.get("/invite", authenticate, getUserDiscordInvite);
-router.post("/invite", authenticate, checkCanGenerateDiscordLink, generateInviteForUser);
+router.post(
+  "/invite",
+  authenticate,
+  checkCanGenerateDiscordLink,
+  generateInviteForUser,
+  validateGenerateInviteForUserBody
+);
 router.delete("/roles", authenticate, checkIsVerifiedDiscord, deleteRole);
 router.get("/roles", authenticate, checkIsVerifiedDiscord, getGroupsRoleId);
 router.patch(

test/unit/middlewares/discordactions-validators.test.js

@@ -3,6 +3,7 @@ const {
   validateGroupRoleBody,
   validateMemberRoleBody,
   validateUpdateUsersNicknameStatusBody,
+  validateGenerateInviteForUserBody,
 } = require("../../../middlewares/validators/discordactions");
 const { expect } = require("chai");
 
@@ -127,4 +128,34 @@ describe("Middleware | Validators | discord actions", function () {
       expect(nextSpy.callCount).to.be.equal(0);
     });
   });
+
+  describe("validateGenerateInviteForUserBody", function () {
+    it("lets the request pass to the next function", async function () {
+      const res = {};
+      const req = {
+        body: {
+          purpose: "testing",
+        },
+      };
+      const nextSpy = Sinon.spy();
+      await validateGenerateInviteForUserBody(req, res, nextSpy);
+      expect(nextSpy.calledOnce).to.be.equal(true);
+    });
+
+    it("stops the propogation to the next function", async function () {
+      const res = {
+        boom: {
+          badRequest: () => {},
+        },
+      };
+      const nextSpy = Sinon.spy();
+      const req = {
+        body: {},
+      };
+      await validateMemberRoleBody(req, res, nextSpy).catch((err) => {
+        expect(err).to.be.an.instanceOf(Error);
+      });
+      expect(nextSpy.callCount).to.be.equal(0);
+    });
+  });
 });

test/unit/models/discordactions.test.js

@@ -1,3 +1,4 @@
+import { generateDiscordInviteLink } from "../../../utils/discord-actions";
 const chai = require("chai");
 const expect = chai.expect;
 const sinon = require("sinon");
@@ -802,6 +803,54 @@ describe("discordactions", function () {
     });
   });
 
+  describe("generateInviteForUser", function () {
+    let fetchStub;
+
+    beforeEach(async function () {
+      fetchStub = sinon.stub(global, "fetch");
+    });
+
+    afterEach(function () {
+      fetchStub.restore();
+    });
+
+    it("should return the invite link and purpose", async function () {
+      const inviteLink = "discord.gg/xyz";
+      const discordInviteLink = {
+        data: {
+          code: "xyz",
+        },
+      };
+      fetchStub.resolves({
+        ok: true,
+        json: () => discordInviteLink,
+      });
+
+      const result = await generateDiscordInviteLink();
+      expect(result).to.be.equal(inviteLink);
+
+      const inviteObject = { userId: "123456", inviteLink: "discord.gg/xyz", purpose: "testing" };
+      await addInviteToInviteModel(inviteObject);
+
+      const invite = await getUserDiscordInvite("123456");
+      expect(invite).to.have.property("id");
+      expect(invite.notFound).to.be.equal(false);
+      expect(invite.userId).to.be.equal("123456");
+      expect(invite.inviteLink).to.be.equal("discord.gg/xyz");
+      expect(invite.purpose).to.be.equal("testing");
+    });
+
+    it("should resolve with an error", async function () {
+      const error = new Error("Error");
+      fetchStub.rejects(error);
+      try {
+        await generateDiscordInviteLink();
+      } catch (err) {
+        expect(err).to.be.equal(error);
+      }
+    });
+  });
+
   describe("groupUpdateLastJoinDate", function () {
     beforeEach(function () {
       sinon.stub(discordRoleModel, "doc").returns({