From a822ae325421a15e983fb03a1fb04c248b51169f Mon Sep 17 00:00:00 2001
From: Alex Holliday <ajhollid@gmail.com>
Date: Tue, 23 Jul 2024 11:24:20 -0700
Subject: [PATCH] Add role to invite token

---
 Server/db/MongoDB.js         | 2 ++
 Server/models/InviteToken.js | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/Server/db/MongoDB.js b/Server/db/MongoDB.js
index 104e49f38..438c7c306 100644
--- a/Server/db/MongoDB.js
+++ b/Server/db/MongoDB.js
@@ -46,6 +46,7 @@ const insertUser = async (req, res) => {
       const avatar = await GenerateAvatarImage(req.file);
       userData.avatarImage = avatar;
     }
+    console.log(userData);
     const newUser = new UserModel(userData);
     await newUser.save();
     return await UserModel.findOne({ _id: newUser._id })
@@ -174,6 +175,7 @@ const requestInviteToken = async (req, res) => {
     await InviteToken.deleteMany({ email: req.body.email });
     let inviteToken = new InviteToken({
       email: req.body.email,
+      role: req.body.role,
       token: crypto.randomBytes(32).toString("hex"),
     });
     await inviteToken.save();
diff --git a/Server/models/InviteToken.js b/Server/models/InviteToken.js
index 6abc3d30b..e1e3413e9 100644
--- a/Server/models/InviteToken.js
+++ b/Server/models/InviteToken.js
@@ -7,6 +7,10 @@ const InviteTokenSchema = mongoose.Schema(
       required: true,
       unique: true,
     },
+    role: {
+      type: String,
+      required: true,
+    },
     token: {
       type: String,
       required: true,