feat: exercises seeders

package-lock.json

@@ -14,6 +14,7 @@
         "bcrypt": "^5.1.1",
         "class-transformer": "^0.5.1",
         "cors": "^2.8.5",
+        "csv-parser": "^3.0.0",
         "dotenv": "^16.3.1",
         "express": "^4.18.2",
         "express-joi-validation": "^5.0.1",
@@ -799,6 +800,20 @@
         "node": ">= 8"
       }
     },
+    "node_modules/csv-parser": {
+      "version": "3.0.0",
+      "resolved": "https://registry.npmjs.org/csv-parser/-/csv-parser-3.0.0.tgz",
+      "integrity": "sha512-s6OYSXAK3IdKqYO33y09jhypG/bSDHPuyCme/IdEHfWpLf/jKcpitVFyOC6UemgGk8v7Q5u2XE0vvwmanxhGlQ==",
+      "dependencies": {
+        "minimist": "^1.2.0"
+      },
+      "bin": {
+        "csv-parser": "bin/csv-parser"
+      },
+      "engines": {
+        "node": ">= 10"
+      }
+    },
     "node_modules/debug": {
       "version": "4.3.4",
       "resolved": "https://registry.npmjs.org/debug/-/debug-4.3.4.tgz",
@@ -1800,7 +1815,6 @@
       "version": "1.2.8",
       "resolved": "https://registry.npmjs.org/minimist/-/minimist-1.2.8.tgz",
       "integrity": "sha512-2yyAR8qBkN3YuheJanUpWC5U3bb5osDywNB8RzDVlDwDHbocAJveqqj1u8+SVD7jkWT4yvsHCpWqqWqAxb0zCA==",
-      "dev": true,
       "funding": {
         "url": "https://github.com/sponsors/ljharb"
       }

package.json

@@ -28,6 +28,7 @@
     "bcrypt": "^5.1.1",
     "class-transformer": "^0.5.1",
     "cors": "^2.8.5",
+    "csv-parser": "^3.0.0",
     "dotenv": "^16.3.1",
     "express": "^4.18.2",
     "express-joi-validation": "^5.0.1",

src/common/enums/exercise-type.enum.ts

@@ -0,0 +1,4 @@
+export enum ExerciseType {
+  DURATION = 'duration',
+  WEIGHT = 'weight',
+}

src/common/models/exercise.model.ts

@@ -1,3 +1,4 @@
+import { ExerciseType } from "@common/enums/exercise-type.enum";
 import mongoose, { ObjectId } from "mongoose";
 const { Schema } = mongoose;
 
@@ -5,12 +6,13 @@ export interface IExercise {
   name: string;
   category: string;
   duration?: number | null;
+  exerciseType: ExerciseType;
   expectedDurationRange: {
     min: number;
     max: number;
   };
-  reps: number;
-  sets: number;
+  reps?: number;
+  sets?: number;
   instructions: string;
   benefits: string;
   targetMuscles: {
@@ -25,16 +27,21 @@ export interface IExercise {
   };
 }
 
-const exerciseSchema = new Schema({
+const exerciseSchema = new Schema<IExercise>({
   name: { type: String, required: true, unique: true, dropDups: true },
   category: { type: String, required: true },
   duration: { type: Number, required: false },
+  exerciseType: {
+    type: String,
+    required: false,
+    enum: ExerciseType
+  },
   expectedDurationRange: {
     min: { type: Number, required: true },
     max: { type: Number, required: true },
   },
-  reps: { type: Number, required: true },
-  sets: { type: Number, required: true },
+  reps: { type: Number, required: false },
+  sets: { type: Number, required: false },
   instructions: { type: String, required: true },
   benefits: { type: String, required: true },
   targetMuscles: {

src/common/serializers/exercise.serialization.ts

@@ -45,6 +45,10 @@ export class ExerciseSerialization {
   @SwaggerResponseProperty({ type: "string" })
   category: string;
 
+  @Expose()
+  @SwaggerResponseProperty({ type: "string" })
+  exerciseType: string;
+
   @Expose()
   @SwaggerResponseProperty({ type: "number" })
   duration: number | null;

src/common/serializers/exercisePopulate.serialization.ts

@@ -47,6 +47,10 @@ export class ExercisePopulateSerialization {
   @SwaggerResponseProperty({ type: "string" })
   category: string;
 
+  @Expose()
+  @SwaggerResponseProperty({ type: "string" })
+  exerciseType: string;
+
   @Expose()
   @SwaggerResponseProperty({ type: "number" })
   duration: number | null;

src/seeder/helpers/db-store.ts

@@ -1,3 +1,13 @@
-export const dbStore = {
+import { IExerciseCSV } from './load-exercises-dataset';
+
+export const dbStore: {
+  dbConnected: boolean;
+  excerisesDataset: IExerciseCSV[];
+  musclesDataset: IExerciseCSV['target'][];
+  equipmentsDataset: IExerciseCSV['equipment'][];
+} = {
   dbConnected: false,
-};
+  excerisesDataset: [],
+  musclesDataset: [],
+  equipmentsDataset: [],
+}

src/seeder/helpers/load-exercises-dataset.ts

@@ -0,0 +1,68 @@
+import { ExerciseType } from "@common/enums/exercise-type.enum";
+import * as path from "path";
+const csv = require('csv-parser')
+const fs = require('fs')
+
+type EBodyPart = "waist" | "upper legs" | "upper arms" | "lower legs" | "chest" | "lower arms" | "back" | "neck" | "shoulders" | "cardio";
+type EType = ExerciseType;
+type EGoal = "Gain Muscle" | "Get Fitter & Lose Weight";
+type ELevel = "Beginner" | "Intermediate" | "Advanced";
+type EEquipment = "band" | "barbell" | "kettlebell" | "dumbbell" | "weighted" | "cable" | "Machine" | "body weight" | "medicine ball" | "Exercise Ball" | "Foam Roll" | "ez barbell" | "leverage machine" | "assisted" | "rope" | "sled machine" | "upper body ergometer" | "olympic barbell" | "bosu ball" | "skierg machine" | "hammer" | "smith machine" | "wheel roller" | "stationary bike" | "tire" | "trap bar" | "elliptical machine" | "stepmill machine";
+type ETarget = "abs" | "adductors" | "abductors" | "biceps" | "calves" | "pectorals" | "forearms" | "glutes" | "hamstrings" | "lats" | "Lower Back" | "Middle Back" | "traps" | "levator scapulae" | "quads" | "delts" | "triceps" | "cardiovascular system" | "spine" | "upper back" | "serratus anterior";
+
+export interface IExerciseCSV {
+    name: string;
+    target: ETarget;
+    equipment: EEquipment;
+    level: ELevel;
+    gym: boolean;
+    home: boolean;
+    goal: EGoal;
+    bodyPart: EBodyPart;
+    type: EType;
+    sets: number;
+}
+
+
+const filePath = path.join(__dirname, '../../resources/exercises.csv');
+
+export const loadExercisesDataset = async (): Promise<IExerciseCSV[]> => {
+  let results: IExerciseCSV[] = [];
+
+  await new Promise((resolve) => {
+    fs
+    .createReadStream(filePath)
+    .pipe(csv())
+    .on('data', (data: any) => {
+      // skip empty rows
+      if (Object.values(data).some((v: any) => !v)) {
+        return;
+      }
+      if(!data.type){
+        console.log("type is null", data);
+      }
+
+      results.push({
+        ...data,
+        gym: data.gym == '1',
+        home: data.home == '1',
+        sets: parseInt(data.sets)
+      })
+    })
+    .on('end', () => {
+      resolve(results);
+    });
+  });
+
+  // remove duplicates by name
+  const uniqueNames = new Set(results.map(e => e.name));
+  results = results.filter(e => {
+    const found = uniqueNames.has(e.name)
+    uniqueNames.delete(e.name);
+    return found;
+  });
+
+  console.log(`Loaded ${results.length} exercises from dataset`);
+
+  return results;
+}

src/seeder/seed.ts

@@ -9,6 +9,18 @@
 
 import * as glob from "glob";
 import path from "path";
+import { loadExercisesDataset } from "./helpers/load-exercises-dataset";
+import { dbStore } from "./helpers/db-store";
+
+const loadDatasets = async() => {
+  const exercisesDataset = await loadExercisesDataset();
+  const musclesDataset = Array.from(new Set(exercisesDataset.map((exercise) => exercise.target)));
+  const equipmentsDataset = Array.from(new Set(exercisesDataset.map((exercise) => exercise.equipment)))
+  
+  dbStore.excerisesDataset = exercisesDataset;
+  dbStore.musclesDataset = musclesDataset;
+  dbStore.equipmentsDataset = equipmentsDataset;
+}
 
 const main = async () => {
   // get cli arguments
@@ -33,7 +45,11 @@ const main = async () => {
       }
 
       return seederNames.includes(path.basename(file));
-    });
+    })
+    .sort();
+
+  // load datasets
+  await loadDatasets();
 
   // run all seeders
   let count = 0;

src/seeder/seeders/3-muscles.seeder.ts

@@ -0,0 +1,9 @@
+import { Muscle } from "@common/models/muscle.model";
+import { dbStore } from "seeder/helpers/db-store";
+import { seederWrapper } from "seeder/helpers/seeder-wrapper";
+
+export default seederWrapper(Muscle, async () => {
+  await Promise.all(dbStore.musclesDataset.map(async function (m) {
+    return Muscle.create({ name: m, image: `https://placehold.co/600x400`})
+  }))
+})

src/seeder/seeders/4-equipments.seeder.ts

@@ -0,0 +1,8 @@
+import { Equipment } from "@common/models/equipment.model";
+import { dbStore } from "seeder/helpers/db-store";
+import { seederWrapper } from "seeder/helpers/seeder-wrapper";
+
+export default seederWrapper(Equipment, async () => { await Promise.all(dbStore.equipmentsDataset.map(async function (e) {
+  return Equipment.create({ name: e, image: `https://placehold.co/600x400`})
+}))
+})

src/seeder/seeders/5-exercises.seeder.ts

@@ -0,0 +1,47 @@
+import { ExerciseType } from "@common/enums/exercise-type.enum";
+import { Equipment } from "@common/models/equipment.model";
+import { Exercise, IExercise } from "@common/models/exercise.model";
+import { Muscle } from "@common/models/muscle.model";
+import { dbStore } from "seeder/helpers/db-store";
+import { IExerciseCSV } from "seeder/helpers/load-exercises-dataset";
+import { seederWrapper } from "seeder/helpers/seeder-wrapper";
+
+export default seederWrapper(Exercise, async () => {
+  console.log('preparing exercises data... (this may take a while)');
+  const data = await Promise.all(dbStore.excerisesDataset.map(async function (e: IExerciseCSV) {
+    return {
+      name: e.name,
+      category: "unknown",
+      exerciseType: e.type,
+      ...(
+        e.type === ExerciseType.WEIGHT && 
+        {
+          reps: Math.floor(Math.random() * 10),
+          sets: e.sets,
+        } 
+        ||
+        {
+          duration: Math.floor(Math.random() * 100),
+        }
+      ),
+      expectedDurationRange: {
+        min: Math.floor(Math.random() * 10),
+        max: 10 + Math.floor(Math.random() * 10),
+      },
+      instructions: "Do this exercise",
+      benefits: "You will get stronger",
+      targetMuscles: {
+        primary: (await Muscle.findOne({ name: e.target }).exec())._id,
+        secondary: (await Muscle.findOne({ name: e.target }).exec())._id,
+      },
+      equipments: [(await Equipment.findOne({name: e.equipment}).exec())._id],
+      coverImage: "https://placehold.co/600x400",
+      media: {
+        type: 'image',
+        url: "https://placehold.co/600x400",
+      }
+    } satisfies IExercise;
+  }))
+  console.log('inserting exercises...');
+  await Exercise.insertMany(data);
+})