feat: support model creating workouts

src/common/models/user.model.ts

@@ -17,7 +17,7 @@ export interface IUser {
   password: string;
   image: string;
   role: AuthenticatableType;
-  gender: string;
+  gender: Gender;
   dob: Date;
   height: number;
   weight: number;

src/common/models/workout.model.ts

@@ -30,7 +30,8 @@ export interface IWorkout {
                 },
             ],
         },
-    ]
+    ],
+    aiGenerated: boolean;
 }
 
 const workoutSchema = new Schema({
@@ -60,10 +61,11 @@ const workoutSchema = new Schema({
                 },
             ],
         },
-    ]
+    ],
+    aiGenerated: { type: Boolean, required: true, default: false },
 });
 
 
 export type WorkoutDocument = IWorkout & mongoose.Document;
 
-export const Workout = mongoose.model<WorkoutDocument>("workouts", workoutSchema);
+export const Workout = mongoose.model<WorkoutDocument>("workouts", workoutSchema);

src/lib/services/crud.service.ts

@@ -3,7 +3,10 @@ import { AnyKeys, Document, FilterQuery, Model } from "mongoose";
 
 
 export const CrudService = <ModelDoc extends Document>(
-  model: Model<ModelDoc>
+  model: Model<ModelDoc>,
+  crudOptions?: {
+    defaultFilter?: FilterQuery<ModelDoc>;
+  }
 ) => {
   return class CrudServiceClass {
     protected model: Model<ModelDoc> = model;
@@ -16,6 +19,7 @@ export const CrudService = <ModelDoc extends Document>(
       filter: FilterQuery<ModelDoc>,
       data: AnyKeys<ModelDoc>
     ): Promise<ModelDoc> {
+      filter = { ...crudOptions?.defaultFilter, ...filter };
       await this.existsOrThrow(filter);
       await this.model.updateOne(filter, data);
       return this.findOneOrFail(filter);
@@ -25,12 +29,14 @@ export const CrudService = <ModelDoc extends Document>(
       filter: FilterQuery<ModelDoc>,
       data: AnyKeys<ModelDoc>
     ): Promise<ModelDoc[]> {
+      filter = { ...crudOptions?.defaultFilter, ...filter };
       await this.existsOrThrow(filter);
       await this.model.updateMany(filter, data);
       return this.model.find(filter);
     }
 
     async deleteOne(filter: FilterQuery<ModelDoc>): Promise<ModelDoc> {
+      filter = { ...crudOptions?.defaultFilter, ...filter };
       await this.existsOrThrow(filter);
       return this.model.findOneAndDelete(filter);
     }
@@ -57,6 +63,8 @@ export const CrudService = <ModelDoc extends Document>(
       };
     }> {
       if (options?.filterOptions) filter = { ...filter, ...options.filterOptions };
+      filter = { ...crudOptions?.defaultFilter, ...filter };
+      
       const queryInstruction = this.model
         .find(filter)
         .limit(paginationOptions.limit)
@@ -74,6 +82,18 @@ export const CrudService = <ModelDoc extends Document>(
       return { docs, paginationData };
     }
 
+    async listAll(
+      filter: FilterQuery<ModelDoc>,
+      options?: {
+        populateArray: any
+      },
+    ): Promise<ModelDoc[]> {
+      filter = { ...crudOptions?.defaultFilter, ...filter };
+      const queryInstruction = this.model.find(filter);
+      if (options?.populateArray) queryInstruction.populate(options.populateArray);
+      return queryInstruction;
+    }
+
     async search(
       filter: FilterQuery<ModelDoc>,
       paginationOptions: {
@@ -94,6 +114,7 @@ export const CrudService = <ModelDoc extends Document>(
         perPage: number;
       };
     }> {
+      filter = { ...crudOptions?.defaultFilter, ...filter };
       const queryInstruction = this.model
         .find(filter)
         .limit(paginationOptions.limit)
@@ -116,7 +137,7 @@ export const CrudService = <ModelDoc extends Document>(
       options?: {
         populateArray: any
       }): Promise<ModelDoc | null> {
-      const queryInstruction = this.model.findOne(filter);
+      const queryInstruction = this.model.findOne();
       if (options?.populateArray) queryInstruction.populate(options.populateArray);
       const document = await queryInstruction
       return document;

src/lib/utils/age.ts

@@ -0,0 +1,5 @@
+export const calcAge = (dob: Date): number => {
+  const diff = Date.now() - dob.getTime();
+  const ageDate = new Date(diff);
+  return Math.abs(ageDate.getUTCFullYear() - 1970);
+}

src/modules/users/modules/user-registered-workouts/controllers/user-registered-workouts.controller.ts

@@ -1,5 +1,5 @@
 import { UserRegisteredWorkoutsService } from "../services/user-registered-workouts.service";
-import { Request, Response } from "express";
+import { Response } from "express";
 import { JsonResponse } from "@lib/responses/json-response";
 import { parsePaginationQuery } from "@helpers/pagination";
 import { bodyValidator } from "@helpers/validation.helper";
@@ -21,13 +21,10 @@ import { SwaggerSummary } from "@lib/decorators/swagger-summary.decorator";
 import { SwaggerDescription } from "@lib/decorators/swagger-description.decorator";
 import { SwaggerResponse } from "@lib/decorators/swagger-response.decorator";
 import { SwaggerRequest } from "@lib/decorators/swagger-request.decorator";import { updateUserRegisteredWorkoutsSchema } from "../validations/update-user-registered-workouts.validation";
+import { IUserRequest } from "@common/interfaces/user-request.interface";
  4
 
 
-interface userRequest extends Request {
-  jwtPayload?: any;
-}
-
 @Controller("/user/myWorkouts")
 @ControllerMiddleware(UsersGuardMiddleware())
 export class userRegisteredWorkoutsController extends BaseController {
@@ -52,7 +49,7 @@ export class userRegisteredWorkoutsController extends BaseController {
   @SwaggerResponse([UserRegisteredWorkoutsPopulateSerialization])
   @SwaggerSummary("List my workouts")
   @SwaggerDescription("List all user registered workouts (workouts that the user had started)")
-  list = async (req: userRequest, res: Response) => {
+  list = async (req: IUserRequest, res: Response) => {
     const paginationQuery = parsePaginationQuery(req.query);
     const { docs, paginationData } =
       await this.userRegisteredWorkoutsService.list(
@@ -79,7 +76,7 @@ export class userRegisteredWorkoutsController extends BaseController {
   @SwaggerResponse(UserRegisteredWorkoutsPopulateSerialization)
   @SwaggerSummary("today's workout && my trainer --> my plan && weekly")
   @SwaggerDescription("Get a single workout from user registered workouts (workouts that the user had started)")
-  get = async (req: userRequest, res: Response) => {
+  get = async (req: IUserRequest, res: Response) => {
     const data = await this.userRegisteredWorkoutsService.findOneOrFail(
       { _id: req.params.id },
       {
@@ -107,7 +104,7 @@ export class userRegisteredWorkoutsController extends BaseController {
   @SwaggerRequest(createUserRegisteredWorkoutsSchema)
   @SwaggerSummary("Create workout")
   @SwaggerDescription("Create a new workout for the user")
-  create = async (req: userRequest, res: Response) => {
+  create = async (req: IUserRequest, res: Response) => {
     const data = await this.userRegisteredWorkoutsService.createForUser(req.body, req.jwtPayload.id);
     return JsonResponse.success(
       {
@@ -123,7 +120,7 @@ export class userRegisteredWorkoutsController extends BaseController {
   @SwaggerRequest(updateUserRegisteredWorkoutsSchema)
   @SwaggerSummary("Update Workout Progress")
   @SwaggerDescription("Update the progress of a workout")
-  updateProgress = async (req: userRequest, res: Response) => {
+  updateProgress = async (req: IUserRequest, res: Response) => {
     const urwId: string = req.params.id;
     const weekNumber: number = Number(req.params.week);
     const dayNumber: number = Number(req.params.day);

src/modules/users/modules/user-registered-workouts/services/user-registered-workouts.service.ts

@@ -1,17 +1,33 @@
 import { UserRegisteredWorkout } from "@common/models/user-registered-workout.model";
 import { CrudService } from "@lib/services/crud.service";
 import { ICreateUserRegisteredWorkouts } from "../validations/create-user-registered-workouts.validation";
-import { WorkoutService } from "../../workouts/services/workouts.service";
 import { IUpdateUserRegisteredWorkouts } from "../validations/update-user-registered-workouts.validation";
 import { HttpError } from "@lib/error-handling/http-error";
+import { Workout } from "@common/models/workout.model";
 
-export class UserRegisteredWorkoutsService extends CrudService(UserRegisteredWorkout) {
-  private workoutsService: WorkoutService = new WorkoutService();
+export class UserRegisteredWorkoutsService extends CrudService(UserRegisteredWorkout, {
+  defaultFilter: {
+    is_active: true,
+  },
+}) {
+  private workoutsService = new (CrudService(Workout))()
+
+  async unregisterCurrentWorkout(userId: string) {
+    return await this.updateMany({
+      user: userId,
+      is_active: true,
+    }, {
+      is_active: false,
+    });
+  }
   
   async createForUser(data: ICreateUserRegisteredWorkouts, userId: string) {
     const workout = await this.workoutsService.findOneOrFail({
       _id: data.workout,
     });
+
+    await this.unregisterCurrentWorkout(userId); 
+
     return await this.create({
       ...data,
       user: userId,

src/modules/users/modules/workouts/controllers/workouts.controller.ts

@@ -15,6 +15,7 @@ import { SwaggerResponse } from "@lib/decorators/swagger-response.decorator";
 import { SwaggerSummary } from "@lib/decorators/swagger-summary.decorator";
 import { SwaggerDescription } from "@lib/decorators/swagger-description.decorator";
 import { SwaggerQuery } from "@lib/decorators/swagger-query.decorator";
+import { IUserRequest } from "@common/interfaces/user-request.interface";
 
 
 @Controller("/user/workouts")
@@ -37,7 +38,7 @@ export class UsersWorkoutController extends BaseController {
     filterName: "string",
     filterVal: "string",
   })
-  list = async (req: Request, res: Response): Promise<Response> => {
+  list = async (req: IUserRequest, res: Response): Promise<Response> => {
     const paginationQuery = parsePaginationQuery(req.query);
     
     let filterName = req.query.filterName, filterVal = req.query.filterVal;
@@ -48,7 +49,13 @@ export class UsersWorkoutController extends BaseController {
     }
 
     const { docs, paginationData } = await this.workoutsService.list(
-      filter,
+      {
+        ...filter,
+        $or: [
+          { aiGenerated: true, created_by: req.jwtPayload.id },
+          { aiGenerated: false },
+        ],
+      },
       paginationQuery
     );
 

src/modules/users/modules/workouts/services/workouts.service.ts

@@ -1,4 +1,71 @@
+import { WorkoutPlace } from "@common/enums/workout-place.enum";
+import { UserDocument } from "@common/models/user.model";
 import { Workout } from "@common/models/workout.model";
+import { FitnessModel, IFWParams, IFitnessPredictionItem } from "@lib/models/fitness-model";
 import { CrudService } from "@lib/services/crud.service";
+import { calcAge } from "@lib/utils/age";
+import { ExerciseService } from "../../exercises/services/exercises.service";
+import { UserRegisteredWorkoutsService } from "../../user-registered-workouts/services/user-registered-workouts.service";
 
-export class WorkoutService extends CrudService(Workout) {}
+export class WorkoutService extends CrudService(Workout, {
+  defaultFilter: {
+    aiGenerated: false,
+  },
+}) {
+  private exerciseService = new ExerciseService();
+  private userRegisteredWorkoutsService = new UserRegisteredWorkoutsService();
+
+  public async createModelWorkout(user: UserDocument) {
+    const params: IFWParams = {
+      home_or_gym: user.preferences.workout_place === WorkoutPlace.GYM ? 1 : 0,
+      level: user.fitness_level,
+      goal: user.preferences.fitness_goal,
+      gender: user.gender,
+      age: calcAge(user.dob),
+      feedback: false,
+      old_weight: user.weight,
+      equipments: user.preferences.preferred_equipment,
+    };
+
+    const pworkout = await FitnessModel.predictWorkout(params);
+
+    // partition the workout days into weeks
+    // each week has 7 days
+    const weeks: IFitnessPredictionItem[][][] = [];
+    for (let i = 0; i < pworkout.length; i += 7) {
+      weeks.push(pworkout.slice(i, i + 7));
+    }
+
+    const exercisesNames = pworkout.flat().map((e) => e.name);
+    const exercises = await this.exerciseService.listAll({ name: { $in: exercisesNames } });
+
+    const workout = await this.create({
+      aiGenerated: true,
+      name: `AI Generated Workout for ${user.name} (${new Date().toLocaleDateString()})`,
+      description: `AI Generated Workout for ${user.name} (${new Date().toLocaleDateString()})`,
+      type: "AI Generated",
+      created_by: user._id,
+      image: "https://placehold.co/300x400",
+      fitness_level: user.fitness_level,
+      fitness_goal: user.preferences.fitness_goal,
+      place: [user.preferences.workout_place],
+      min_per_day: 30,
+      total_number_days: pworkout.flat().length,
+      template_weeks: weeks.map((week, i) => ({
+        week_number: i + 1,
+        week_name: `Week ${i + 1}`,
+        week_description: `Week ${i + 1}`,
+        days: week.map((day, j) => ({
+          day_number: j + 1,
+          total_number_exercises: day.length,
+          day_type: "full_body", // #TODO: Change this
+          exercises: day.map((e) => exercises.find((ex) => ex.name === e.name)?._id),
+        })),
+      })),
+    });
+
+    await this.userRegisteredWorkoutsService.createForUser({
+      workout: workout._id,
+    }, user._id);
+  }
+}