adding finish step

src/components/workout/WorkoutManager.tsx

@@ -232,9 +232,15 @@ export class WorkoutManager implements WorkoutManagerI {
   }
 
   async setStep(step: number) {
-    if (!this.workout || step < 0 || step > this.workout.steps.length) {
+    if (!this.workout || step < 0) {
       throw new Error("Invalid step set. Must be valid and session created.");
     }
+
+    if (this._currentStep >= this.workout.steps.length - 1) {
+      await this.endWorkout();
+      return;
+    }
+
     this._set = true;
     this._currentStep = step;
     if (this._timer && this._workout) {
@@ -254,14 +260,6 @@ export class WorkoutManager implements WorkoutManagerI {
     if (this.workout?.completed) {
       return;
     }
-    if (
-      this.workout &&
-      this.workout.steps &&
-      this._currentStep >= this.workout.steps.length - 1
-    ) {
-      await this.endWorkout();
-      return;
-    }
     await this.setStep(this._currentStep + 1);
   }
 

src/lib/workout.ts

@@ -51,13 +51,23 @@ export const createSteps = (config: RoutineConfiguration): Step[] => {
   steps.push({
     name: config.CoolDown.name,
     duration: config.CoolDown.duration,
-    color: "bg-blue-500",
+    color: "bg-teal-500",
     cycle: 0,
     totalSets: 1,
     totalCycles: config.Cycles.value,
     set: 0,
   });
 
+  steps.push({
+    name: "Finish",
+    duration: 0,
+    color: "bg-blue-500",
+    cycle: 0,
+    totalSets: 0,
+    totalCycles: 0,
+    set: 0,
+  });
+
   return steps;
 };
 

src/pages/PlayScreen/index.tsx

@@ -79,7 +79,12 @@ function StepView() {
   //   workoutState.manager.ready,
   // ]);
 
-  const whiteListedNames = ["Rest Between Sets", "Cool Down", "Preparation"];
+  const whiteListedNames = [
+    "Rest Between Sets",
+    "Cool Down",
+    "Preparation",
+    "Finish",
+  ];
 
   return (
     <div className="overflow-y-auto">