feat: gain loyalty points upon booking

- fix itinerary tests
- refactor redeem points

backend/src/constants/index.ts

@@ -0,0 +1 @@
+export const LOYALTY_POINT_GAIN = 600000;

backend/src/controllers/booking.controller.ts

@@ -4,17 +4,20 @@ import { logger } from '../middlewares/logger.middleware';
 import Validator from '../utils/Validator.utils';
 import bookingRepo from '../database/repositories/booking.repo';
 import { checkUserLegalAge } from '../utils/AgeVerification.utils';
+import userRepo from '../database/repositories/user.repo';
+import { LOYALTY_POINT_GAIN } from '../constants';
 
 class BookingController {
   async bookItinerary(req: Request, res: Response) {
     try {
       Validator.validateId(req.body.itinerary_id, 'incorrect itinerary id');
 
       if (!(await checkUserLegalAge(req.user.userId))) {
-        res.status(403).json({ message: 'Cannot book as user is under 18' });
+        res.status(ResponseStatusCodes.FORBIDDEN).json({ message: 'Cannot book as user is under 18' });
         return;
       }
       const booking = await bookingRepo.bookItinerary(req.user.userId, req.body.itinerary_id);
+      await userRepo.updateUserLoyaltyPoints(req.user.userId, LOYALTY_POINT_GAIN);
 
       const response = {
         message: 'Booking successful',
@@ -35,11 +38,12 @@ class BookingController {
       Validator.validateId(req.body.activity_id, 'incorrect activity id');
 
       if (!(await checkUserLegalAge(req.user.userId))) {
-        res.status(403).json({ message: 'Cannot book as user is under 18' });
+        res.status(ResponseStatusCodes.FORBIDDEN).json({ message: 'Cannot book as user is under 18' });
         return;
       }
 
       const booking = await bookingRepo.bookActivity(req.user.userId, req.body.activity_id);
+      await userRepo.updateUserLoyaltyPoints(req.user.userId, LOYALTY_POINT_GAIN);
 
       const response = {
         message: 'Booking successful',

backend/src/controllers/loyalty.controller.ts

@@ -14,16 +14,11 @@ async function redeemPoints(req: Request, res: Response) {
 
     if (!user) throw new Error('User not found');
 
-    if (user.loyalty_points < points) {
-      throw new Error('Insufficient points');
-    }
-
-    user.loyalty_points -= points;
-    user.loyalty_level = getLoyaltyLevel(user.loyalty_points);
     user.wallet += points * 0.01;
 
     // Update user
     await userRepo.updateUser(userId, user as UserType);
+    await userRepo.updateUserLoyaltyPoints(userId, -points);
 
     res.status(ResponseStatusCodes.OK).json({ message: 'Points redeemed successfully' });
   } catch (error: any) {
@@ -32,16 +27,4 @@ async function redeemPoints(req: Request, res: Response) {
   }
 }
 
-function getLoyaltyLevel(points: number): number {
-  let level: number = 1;
-
-  if (points >= 1000000) {
-    level = 3;
-  } else if (points >= 500000) {
-    level = 2;
-  }
-
-  return level;
-}
-
 export { redeemPoints };

backend/src/database/repositories/user.repo.ts

@@ -2,6 +2,7 @@ import { ObjectId } from 'mongodb';
 import { User } from '../models/user.model';
 import { UserType } from '../../types/User.types';
 import { accountType } from '../../types/User.types';
+import getUserLevel from '../../utils/UserLevel.util';
 
 class UserRepository {
   async getUsers() {
@@ -36,6 +37,23 @@ class UserRepository {
     return await User.updateOne({ _id: new ObjectId(id) }, user);
   }
 
+  async updateUserLoyaltyPoints(id: string, points: number) {
+    const user = await this.findUserById(id);
+    const oldPoints = user?.loyalty_points ?? 0;
+    const newPoints = oldPoints + points;
+
+    if (newPoints < 0) {
+      throw new Error('Insufficient points');
+    }
+
+    return await User.findByIdAndUpdate(id, {
+      $set: {
+        loyalty_points: newPoints,
+        loyalty_level: getUserLevel(newPoints),
+      },
+    });
+  }
+
   async acceptUser(id: string) {
     return await User.updateOne({ _id: new ObjectId(id) }, { accepted: true });
   }

backend/tests/itinerary.test.ts

@@ -40,7 +40,7 @@ let activityRef = {
 
 let newItinerary: ItineraryType = {
   name: 'Test Itinerary',
-  category: 'Test Category',
+  category: '',
   tags: [],
   activities: [],
   locations: [location],
@@ -92,6 +92,8 @@ describe('Itinerary tests', () => {
       activityRef.activity = newActivity.body.data.activityId;
       newItinerary.tags.push(newTag.body.data.tagId);
       newItinerary.activities.push(activityRef);
+      newItinerary.category = newCategory.body.data.categoryId;
+
       const response = await requestWithAuth('post', '/api/itineraries').send(newItinerary);
 
       itineraryId = response.body.data.itineraryId;