From 805b554e8e00dac9a2dc4a72f559d5174aff6eee Mon Sep 17 00:00:00 2001
From: Tom Dudley <dudleyt88@gmail.com>
Date: Sun, 18 Nov 2018 23:15:46 +0000
Subject: [PATCH] Add support for custom workouts

---
 .../com.github.mgifos.workouts/model/Workout.scala   | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/scala/com.github.mgifos.workouts/model/Workout.scala b/src/main/scala/com.github.mgifos.workouts/model/Workout.scala
index 3df6d75..86f2104 100644
--- a/src/main/scala/com.github.mgifos.workouts/model/Workout.scala
+++ b/src/main/scala/com.github.mgifos.workouts/model/Workout.scala
@@ -13,7 +13,7 @@ case class WorkoutDef(sport: String, name: String, steps: Seq[Step] = Nil) exten
   def json: JsValue = Json.obj(
     "sportType" -> Json.obj(
       "sportTypeId" -> sportId(sport),
-      "sportTypeKey" -> sport),
+      "sportTypeKey" -> sportTypeKey(sport)),
     "workoutName" -> name,
     "workoutSegments" -> Json.arr(
       Json.obj(
@@ -34,7 +34,7 @@ case class WorkoutNote(note: String) extends Workout {
 
 object Workout {
 
-  private val WorkoutHeader = """^(running|cycling):\s([\u0020-\u007F]+)(([\r\n]+\s*\-\s[a-z]+:.*)*)$""".r
+  private val WorkoutHeader = """^(running|cycling|custom):\s([\u0020-\u007F]+)(([\r\n]+\s*\-\s[a-z]+:.*)*)$""".r
   private val NextStepRx = """^((-\s\w*:\s.*)(([\r\n]+\s{1,}-\s.*)*))(([\s].*)*)$""".r
 
   def parseDef(x: String)(implicit msys: MeasurementSystems.MeasurementSystem): Either[String, WorkoutDef] = {
@@ -59,6 +59,12 @@ object Workout {
   def sportId(sport: String) = sport match {
     case "running" => 1
     case "cycling" => 2
-    case _ => throw new IllegalArgumentException("Only running and cycling workouts are supported.")
+    case "custom" => 3
+    case _ => throw new IllegalArgumentException("Only running, cycling and 'custom' workouts are supported.")
+  }
+
+  def sportTypeKey(sport: String) = sport match {
+    case "custom" => "other"
+    case _ => sport
   }
 }