update referral CA

Author: Varun-Kolanu

app.js

@@ -45,8 +45,9 @@ passport.use(
 app.get(
   "/api/v1/Oauth2/google",
   (req, res, next) => {
+    const { referralCode } = req.query;
     const frontendUrl = req.headers.referer;
-    const state = JSON.stringify({ frontendUrl });
+    const state = JSON.stringify({ frontendUrl, referralCode });
     const authUrl = passport.authenticate("google", {
       scope: ["profile", "email"],
       state,

controllers/auth.js

@@ -7,10 +7,11 @@ import jwt from "jsonwebtoken";
 import { generateOTP, sendVerification } from "../utils/sendVerification.js";
 import { Verification } from "../models/verification.js";
 import validator from "validator";
+import { CARequest } from "../models/ca_request.js";
 
 export const signup = async (req, res, next) => {
   try {
-    const { name, email, password } = req.body;
+    const { name, email, password, referralCode } = req.body;
 
     if (!validator.isLength(name, { min: 2, max: 50 })) {
       return next(
@@ -30,16 +31,12 @@ export const signup = async (req, res, next) => {
 
     let user = await User.findOne({ email });
     if (user) {
-      return next(new ErrorHandler("User Already Exist", 400));
+      return next(new ErrorHandler("User Already Exists", 400));
     }
 
     const hashedpswd = await bcrypt.hash(password, 10);
 
-    await sendVerification(email, generateOTP(), name, hashedpswd);
-    return res.status(200).json({
-      status: "success",
-      message: "Email sent for verification",
-    });
+    sendVerification(email, generateOTP(), name, hashedpswd, referralCode, res);
   } catch (error) {
     next(error);
   }
@@ -65,12 +62,17 @@ export const createGoogleUser = async (
 
 export const googleCallback = async (user, req, res, next) => {
   try {
-    const frontendUrl = req.frontendUrl;
+    const { frontendUrl, referralCode } = req;
     const email = user.emails[0].value;
     let googleUser = await User.findOne({ email }).select("+password");
 
-    if (!googleUser.password)
+    if (!googleUser.password) {
+      if (referralCode) {
+        googleUser.referredBy = referralCode
+        await googleUser.save();
+      }
       res.redirect(`${frontendUrl}setPassword?email=${email}`);
+    }
     else {
       const refreshToken = await generateRefreshToken(googleUser);
       const token = jwt.sign({ _id: googleUser._id }, process.env.JWT_SECRET, {
@@ -138,6 +140,16 @@ export const passwordSetter = async (req, res, next) => {
 
     const hashedpswd = await bcrypt.hash(password, 10);
 
+    const referralCode = user.referredBy;
+    if (referralCode) {
+      const ca = await CARequest.findOne({ referralCode });
+      if (ca && ca.status === 'approved') {
+        const points = 10;
+        ca.points += points;
+        await ca.save();
+      }
+    }
+
     user.password = hashedpswd;
     await user.save();
 
@@ -203,10 +215,20 @@ export const verifyEmail = async (req, res, next) => {
     const verification = await Verification.findOne({ email });
     if (!verification || verification.code != otp)
       return next(new ErrorHandler("OTP Invalid or Expired", 400));
+
+    if (verification.referralCode) {
+      const ca = await CARequest.findOne({ referralCode: verification.referralCode });
+      if (ca && ca.status === "approved") {
+        const points = 10;
+        ca.points += points;
+        await ca.save();
+      }
+    }
     const user = await User.create({
       name: verification.name,
       email: verification.email,
       password: verification.password,
+      referredBy: verification.referralCode
     });
     const refreshToken = await generateRefreshToken(user);
     await Verification.deleteOne({ email });

controllers/ca.js

@@ -1,13 +1,16 @@
 import ErrorHandler from "../middlewares/error.js";
 import { CARequest } from "../models/ca_request.js";
 import { User } from "../models/user.js";
+import { generateCAReferral } from "../utils/features.js";
 
 export const register = async (req, res, next) => {
   try {
     const request = await CARequest.findOne({ user: req.user._id });
     if (request)
       return next(new ErrorHandler("CA Request already exists", 400));
     const { institute, userDescription, ca_brought_by, branch, graduation_year, contact_number, whatsapp_number } = req.body;
+
+    const referralCode = await generateCAReferral();
     const newRequest = await CARequest.create({
       user: req.user._id,
       institute,
@@ -16,7 +19,8 @@ export const register = async (req, res, next) => {
       branch,
       graduation_year,
       contact_number,
-      whatsapp_number
+      whatsapp_number,
+      referralCode
     });
     res.status(201).json(newRequest);
   } catch (error) {
@@ -61,13 +65,6 @@ export const updateRequest = async (req, res, next) => {
       );
     }
     if (status) {
-      if (req.user.role === "admin" && status === "approved") {
-        await User.findByIdAndUpdate(request.user, { role: "ca" });
-      }
-
-      if (req.user.role === "admin" && status !== "approved" && request.status === "approved") {
-        await User.findByIdAndUpdate(request.user, { role: "user" });
-      }
       request.status = status;
     }
 
@@ -91,3 +88,19 @@ export const updateRequest = async (req, res, next) => {
     next(error);
   }
 };
+
+export const getCALeaderboard = async (req, res, next) => {
+  try {
+    const ca_requests = await CARequest.find({ status: "approved" }).populate("user", "name").sort({ points: -1 });
+    const ca_leaderboard = ca_requests.map(request => {
+      return {
+        name: request.user.name,
+        institute: request.institute,
+        points: request.points
+      }
+    })
+    res.status(200).json(ca_leaderboard)
+  } catch (error) {
+    next(error)
+  }
+}

controllers/user.js

@@ -24,7 +24,7 @@ export const updateUser = async (req, res, next) => {
         )
       );
     const { role, institute, phone_num } = req.body;
-    if (role && role !== "user" && role !== "ca" && role !== "admin")
+    if (role && role !== "user" && role !== "admin")
       return next(new ErrorHandler("Invalid role", 400));
 
     let user = await User.findById(id);

middlewares/auth.js

@@ -27,10 +27,12 @@ export const isAuthenticated = async (req, res, next) => {
 export const cbMiddleware = (req, res, next) => {
   const state = JSON.parse(req.query.state || "{}");
   const frontend = state.frontendUrl || frontendUrl;
+  const referralCode = state.referralCode || "";
   const callBack = passport.authenticate("google", {
     failureRedirect: `${frontend}backend_redirect?error=unknown`,
     session: false,
   });
   req.frontendUrl = frontend;
+  req.referralCode = referralCode;
   callBack(req, res, next);
 }

models/ca_request.js

@@ -26,6 +26,14 @@ const schema = new mongoose.Schema({
   adminMessage: {
     type: String,
   },
+  referralCode: {
+    type: String,
+    unique: true
+  },
+  points: {
+    type: Number,
+    default: 0,
+  }
 });
 
 export const CARequest = mongoose.model("CARequest", schema);

models/user.js

@@ -16,7 +16,7 @@ const schema = new mongoose.Schema({
   },
   role: {
     type: String,
-    enum: ["user", "ca", "admin"],
+    enum: ["user", "admin"],
     default: "user",
   },
   institute: {
@@ -25,6 +25,7 @@ const schema = new mongoose.Schema({
   phone_num: {
     type: String,
   },
+  referredBy: String,
   createdAt: {
     type: Date,
     default: Date.now,

models/verification.js

@@ -21,6 +21,7 @@ const schema = new mongoose.Schema({
     type: Date,
     required: true,
   },
+  referralCode: String
 });
 
 schema.index({ expiry: 1 }, { expireAfterSeconds: 0 });

routes/ca.js

@@ -5,6 +5,7 @@ import {
   getMyRequest,
   register,
   updateRequest,
+  getCALeaderboard
 } from "../controllers/ca.js";
 
 const router = express.Router();
@@ -13,5 +14,6 @@ router.post("/", isAuthenticated, register);
 router.get("/my", isAuthenticated, getMyRequest);
 router.patch("/:id", isAuthenticated, updateRequest);
 router.get("/all", isAuthenticated, getAllRequests);
+router.get("/leaderboard", getCALeaderboard);
 
 export default router;

utils/features.js

@@ -3,6 +3,7 @@ import { Session } from "../models/session.js";
 import { Team } from "../models/team.js";
 import fs from "fs"
 import YAML from "yamljs";
+import { CARequest } from "../models/ca_request.js";
 
 export const sendJwt = async (
   user,
@@ -62,6 +63,22 @@ export const generateRandomCode = async () => {
   return result;
 };
 
+export const generateCAReferral = async () => {
+  const characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+  let result = "";
+
+  do {
+    result = "";
+    for (let i = 0; i < 8; i++) {
+      result += characters.charAt(
+        Math.floor(Math.random() * characters.length)
+      );
+    }
+  } while (await CARequest.findOne({ referralCode: result }));
+
+  return result;
+};
+
 export const loadSwaggerWithDynamicUrl = (filePath) => {
   let swaggerYaml = fs.readFileSync(filePath, 'utf8');
 

utils/sendVerification.js

@@ -6,17 +6,17 @@ export function generateOTP() {
 	return otp;
 }
 
-export const sendVerification = async (to, otp, name, password) => {
+export const sendVerification = async (to, otp, name, password, referralCode, res) => {
 	const transporter = nodemailer.createTransport({
 		service: "gmail",
 		auth: {
-			user: "[email protected]",
+			user: process.env.SMTP_EMAIL_ID,
 			pass: process.env.SMTP_APP_PASS,
 		},
 	});
 
-	const image_url = "https://iili.io/2cwlabs.png";
-	const from = "[email protected]";
+	const image_url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTP63mBXmRPHkCh88H6n2upFPU-8ibISHho3A&s";
+	const from = process.env.SMTP_EMAIL_ID;
 	const mailOptions = {
 		from,
 		to,
@@ -69,7 +69,7 @@ export const sendVerification = async (to, otp, name, password) => {
         `,
 	};
 
-	await transporter.sendMail(mailOptions, async (error, info) => {
+	transporter.sendMail(mailOptions, async (error, info) => {
 		if (error) {
 			console.error("Error sending email:", error);
 			throw new Error("Error sending email:", error);
@@ -82,9 +82,13 @@ export const sendVerification = async (to, otp, name, password) => {
 				code: otp,
 				password,
 				expiry: new Date(Date.now() + 60 * 60 * 1000),
+				referralCode
 			},
 			{ upsert: true }
 		);
-
+		res.status(200).json({
+			status: "success",
+			message: "Email sent for verification",
+		});
 	});
 };